Merge remote-tracking branch 'upstream/master' into feat/light_module_feats

tests/smart/modules/test_autooff.py

@@ -1,6 +1,5 @@
 from __future__ import annotations
 
-import sys
 from datetime import datetime
 
 import pytest
@@ -25,10 +24,6 @@ autooff = parametrize(
         ("auto_off_at", "auto_off_at", datetime | None),
     ],
 )
-@pytest.mark.skipif(
-    sys.version_info < (3, 10),
-    reason="Subscripted generics cannot be used with class and instance checks",
-)
 async def test_autooff_features(
     dev: SmartDevice, feature: str, prop_name: str, type: type
 ):

tests/smart/modules/test_energy.py

@@ -0,0 +1,21 @@
+import pytest
+
+from kasa import Module, SmartDevice
+from kasa.interfaces.energy import Energy
+from kasa.smart.modules import Energy as SmartEnergyModule
+from tests.conftest import has_emeter_smart
+
+
+@has_emeter_smart
+async def test_supported(dev: SmartDevice):
+    energy_module = dev.modules.get(Module.Energy)
+    if not energy_module:
+        pytest.skip(f"Energy module not supported for {dev}.")
+
+    assert isinstance(energy_module, SmartEnergyModule)
+    assert energy_module.supports(Energy.ModuleFeature.CONSUMPTION_TOTAL) is False
+    assert energy_module.supports(Energy.ModuleFeature.PERIODIC_STATS) is False
+    if energy_module.supported_version < 2:
+        assert energy_module.supports(Energy.ModuleFeature.VOLTAGE_CURRENT) is False
+    else:
+        assert energy_module.supports(Energy.ModuleFeature.VOLTAGE_CURRENT) is True