test: fix bad mode preset_test

zabuldon · Apr 6, 2020 · e33d2df · e33d2df
1 parent 8144a3a
commit e33d2df
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/tests/unit_tests/test_climate.py b/tests/unit_tests/test_climate.py
@@ -231,9 +231,8 @@ async def test_set_preset_mode_invalid_modes(monkeypatch):
 
     await _climate.async_update()
 
-    preset_modes = _climate.preset_modes
-    with pytest.raises(UnknownPresetMode):
-        bad_modes = ["UKNOWN_MODE", "home", "auto", "away", "hot"]
-        for mode in bad_modes:
-            assert mode not in preset_modes
+    bad_modes = ["UKNOWN_MODE", "home", "auto", "away", "hot"]
+    for mode in bad_modes:
+        assert mode not in _climate.preset_modes
+        with pytest.raises(UnknownPresetMode):
             await _climate.set_preset_mode(mode)