Correct return type of climate entity feature

PeteRager · Feb 7, 2025 · 4a7a8a8 · 4a7a8a8
1 parent ea71cdb
commit 4a7a8a8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/custom_components/lennoxs30/climate.py b/custom_components/lennoxs30/climate.py
@@ -195,10 +195,10 @@ def is_single_setpoint_active(self) -> bool:
         return True
 
     @property
-    def supported_features(self):
+    def supported_features(self) -> ClimateEntityFeature:
         """Return the list of supported features."""
         if self.is_zone_disabled:
-            return 0
+            return ClimateEntityFeature(0)
 
         mask = SUPPORT_FLAGS
 

diff --git a/tests/test_climate.py b/tests/test_climate.py
@@ -752,7 +752,7 @@ async def test_climate_supported_features(hass, manager_mz: Manager):
     assert feat != 0
     system.zoningMode = LENNOX_ZONING_MODE_CENTRAL
     feat = c1.supported_features
-    assert feat == 0
+    assert feat == ClimateEntityFeature(0)
     feat = c.supported_features
     assert feat != 0