Use DeviceClass Enums in canary tests (#62113)

This commit is contained in:
Dave T 2021-12-16 20:47:59 +00:00 committed by GitHub
parent 373790cb15
commit cabd6375d1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -9,12 +9,9 @@ from homeassistant.components.canary.sensor import (
STATE_AIR_QUALITY_NORMAL, STATE_AIR_QUALITY_NORMAL,
STATE_AIR_QUALITY_VERY_ABNORMAL, STATE_AIR_QUALITY_VERY_ABNORMAL,
) )
from homeassistant.components.sensor import SensorDeviceClass
from homeassistant.const import ( from homeassistant.const import (
ATTR_UNIT_OF_MEASUREMENT, ATTR_UNIT_OF_MEASUREMENT,
DEVICE_CLASS_BATTERY,
DEVICE_CLASS_HUMIDITY,
DEVICE_CLASS_SIGNAL_STRENGTH,
DEVICE_CLASS_TEMPERATURE,
PERCENTAGE, PERCENTAGE,
SIGNAL_STRENGTH_DECIBELS_MILLIWATT, SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
TEMP_CELSIUS, TEMP_CELSIUS,
@ -56,14 +53,14 @@ async def test_sensors_pro(hass, canary) -> None:
"20_temperature", "20_temperature",
"21.12", "21.12",
TEMP_CELSIUS, TEMP_CELSIUS,
DEVICE_CLASS_TEMPERATURE, SensorDeviceClass.TEMPERATURE,
None, None,
), ),
"home_dining_room_humidity": ( "home_dining_room_humidity": (
"20_humidity", "20_humidity",
"50.46", "50.46",
PERCENTAGE, PERCENTAGE,
DEVICE_CLASS_HUMIDITY, SensorDeviceClass.HUMIDITY,
None, None,
), ),
"home_dining_room_air_quality": ( "home_dining_room_air_quality": (
@ -182,14 +179,14 @@ async def test_sensors_flex(hass, canary) -> None:
"20_battery", "20_battery",
"70.46", "70.46",
PERCENTAGE, PERCENTAGE,
DEVICE_CLASS_BATTERY, SensorDeviceClass.BATTERY,
None, None,
), ),
"home_dining_room_wifi": ( "home_dining_room_wifi": (
"20_wifi", "20_wifi",
"-57.0", "-57.0",
SIGNAL_STRENGTH_DECIBELS_MILLIWATT, SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
DEVICE_CLASS_SIGNAL_STRENGTH, SensorDeviceClass.SIGNAL_STRENGTH,
None, None,
), ),
} }