From 1a594d2a8c0009c0358868bdd2eb48300fa0b47f Mon Sep 17 00:00:00 2001
From: Dave T <17680170+davet2001@users.noreply.github.com>
Date: Thu, 16 Dec 2021 19:59:55 +0000
Subject: [PATCH] Use DeviceClass Enums in abode tests (#61980)

---
 tests/components/abode/test_binary_sensor.py | 4 ++--
 tests/components/abode/test_sensor.py        | 5 ++---
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/tests/components/abode/test_binary_sensor.py b/tests/components/abode/test_binary_sensor.py
index e4aa08c7f5f..d8f19a19b77 100644
--- a/tests/components/abode/test_binary_sensor.py
+++ b/tests/components/abode/test_binary_sensor.py
@@ -2,8 +2,8 @@
 from homeassistant.components.abode import ATTR_DEVICE_ID
 from homeassistant.components.abode.const import ATTRIBUTION
 from homeassistant.components.binary_sensor import (
-    DEVICE_CLASS_WINDOW,
     DOMAIN as BINARY_SENSOR_DOMAIN,
+    BinarySensorDeviceClass,
 )
 from homeassistant.const import (
     ATTR_ATTRIBUTION,
@@ -37,4 +37,4 @@ async def test_attributes(hass):
     assert not state.attributes.get("no_response")
     assert state.attributes.get("device_type") == "Door Contact"
     assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Front Door"
-    assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_WINDOW
+    assert state.attributes.get(ATTR_DEVICE_CLASS) == BinarySensorDeviceClass.WINDOW
diff --git a/tests/components/abode/test_sensor.py b/tests/components/abode/test_sensor.py
index 5e3195430ab..3d68e4d3d8a 100644
--- a/tests/components/abode/test_sensor.py
+++ b/tests/components/abode/test_sensor.py
@@ -1,11 +1,10 @@
 """Tests for the Abode sensor device."""
 from homeassistant.components.abode import ATTR_DEVICE_ID
-from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN
+from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN, SensorDeviceClass
 from homeassistant.const import (
     ATTR_DEVICE_CLASS,
     ATTR_FRIENDLY_NAME,
     ATTR_UNIT_OF_MEASUREMENT,
-    DEVICE_CLASS_HUMIDITY,
     PERCENTAGE,
     TEMP_CELSIUS,
 )
@@ -35,7 +34,7 @@ async def test_attributes(hass):
     assert state.attributes.get("device_type") == "LM"
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
     assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Environment Sensor Humidity"
-    assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_HUMIDITY
+    assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.HUMIDITY
 
     state = hass.states.get("sensor.environment_sensor_lux")
     assert state.state == "1.0"