fi

basnijholt · Apr 6, 2024 · 97a5b8f · 97a5b8f
1 parent 85c23d2
commit 97a5b8f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_switch.py b/tests/test_switch.py
@@ -1351,9 +1351,9 @@ def mock_area_registry(
     year, month = (int(x) for x in ha_version.split(".")[:2])
     if year <= 2022 and month <= 12:
         area_kwargs["aliases"] = {}
-    elif year >= 2024 and month >= 2:
+    elif year >= 2024 and month == 2:
         area_kwargs["icon"] = None
-    elif year >= 2024 and month >= 3:
+    elif year >= 2024 and month >= 3 and month:
         area_kwargs["icon"] = None
         area_kwargs["floor_id"] = "test-floor"
         registry.areas = ar.AreaRegistryItems()