update unit test to support both HEAT and COOL mode

jmcollin78 · Oct 30, 2023 · 9c62bfb · 9c62bfb
1 parent a62a3d1
commit 9c62bfb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_switch_ac.py b/tests/test_switch_ac.py
@@ -56,7 +56,7 @@ def find_my_entity(entity_id) -> ClimateEntity:
         assert entity.ac_mode is True
         assert entity.hvac_action is HVACAction.OFF
         assert entity.hvac_mode is HVACMode.OFF
-        assert entity.hvac_modes == [HVACMode.COOL, HVACMode.OFF]
+        assert entity.hvac_modes == [HVACMode.HEAT, HVACMode.COOL, HVACMode.OFF]
         assert entity.target_temperature == entity.max_temp
         assert entity.preset_modes == [
             PRESET_NONE,