From d9dada4fb3d6f61cf84e0aaea9d79f759df64961 Mon Sep 17 00:00:00 2001 From: Jan Bouwhuis Date: Mon, 8 Apr 2024 14:33:24 +0200 Subject: [PATCH] Remove condition from ecobee humidifier attribute test (#115197) Remove conditions from ecobee humidifier attribute test --- tests/components/ecobee/test_humidifier.py | 19 ++++++++----------- 1 file changed, 8 insertions(+), 11 deletions(-) diff --git a/tests/components/ecobee/test_humidifier.py b/tests/components/ecobee/test_humidifier.py index f35a7dc9237..696ca3d6c0d 100644 --- a/tests/components/ecobee/test_humidifier.py +++ b/tests/components/ecobee/test_humidifier.py @@ -44,21 +44,18 @@ async def test_attributes(hass: HomeAssistant) -> None: state = hass.states.get(DEVICE_ID) assert state.state == STATE_ON - if state.attributes.get(ATTR_CURRENT_HUMIDITY): - assert state.attributes.get(ATTR_CURRENT_HUMIDITY) == 15 - assert state.attributes.get(ATTR_MIN_HUMIDITY) == DEFAULT_MIN_HUMIDITY - assert state.attributes.get(ATTR_MAX_HUMIDITY) == DEFAULT_MAX_HUMIDITY - assert state.attributes.get(ATTR_HUMIDITY) == 40 - assert state.attributes.get(ATTR_AVAILABLE_MODES) == [ + assert state.attributes[ATTR_CURRENT_HUMIDITY] == 15 + assert state.attributes[ATTR_MIN_HUMIDITY] == DEFAULT_MIN_HUMIDITY + assert state.attributes[ATTR_MAX_HUMIDITY] == DEFAULT_MAX_HUMIDITY + assert state.attributes[ATTR_HUMIDITY] == 40 + assert state.attributes[ATTR_AVAILABLE_MODES] == [ MODE_OFF, MODE_AUTO, MODE_MANUAL, ] - assert state.attributes.get(ATTR_FRIENDLY_NAME) == "ecobee" - assert state.attributes.get(ATTR_DEVICE_CLASS) == HumidifierDeviceClass.HUMIDIFIER - assert ( - state.attributes.get(ATTR_SUPPORTED_FEATURES) == HumidifierEntityFeature.MODES - ) + assert state.attributes[ATTR_FRIENDLY_NAME] == "ecobee" + assert state.attributes[ATTR_DEVICE_CLASS] == HumidifierDeviceClass.HUMIDIFIER + assert state.attributes[ATTR_SUPPORTED_FEATURES] == HumidifierEntityFeature.MODES async def test_turn_on(hass: HomeAssistant) -> None: