evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use DeviceClass Enums in arlo tests (#62095)

Browse source
This commit is contained in:
Dave T 2021-12-16 21:11:08 +00:00 • committed by GitHub
parent 773ac289dd
commit dab1a786a5
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
tests/components/arlo/test_sensor.py
View file

@ -6,12 +6,8 @@ import pytest
from homeassistant.components.arlo import DATA_ARLO, sensor as arlo
from homeassistant.components.arlo.sensor import SENSOR_TYPES
from homeassistant.const import (
DEVICE_CLASS_BATTERY,
DEVICE_CLASS_HUMIDITY,
DEVICE_CLASS_TEMPERATURE,
PERCENTAGE,
)
from homeassistant.components.sensor import SensorDeviceClass
from homeassistant.const import PERCENTAGE
def _get_named_tuple(input_dict):
@ -157,12 +153,12 @@ def test_sensor_state_default(default_sensor):
def test_sensor_device_class__battery(battery_sensor):
"""Test the battery device_class."""
assert battery_sensor.device_class == DEVICE_CLASS_BATTERY
assert battery_sensor.device_class == SensorDeviceClass.BATTERY
def test_sensor_device_class(temperature_sensor):
"""Test the device_class property."""
assert temperature_sensor.device_class == DEVICE_CLASS_TEMPERATURE
assert temperature_sensor.device_class == SensorDeviceClass.TEMPERATURE
def test_unit_of_measure(default_sensor, battery_sensor):
@ -174,8 +170,8 @@ def test_unit_of_measure(default_sensor, battery_sensor):
def test_device_class(default_sensor, temperature_sensor, humidity_sensor):
"""Test the device_class property."""
assert default_sensor.device_class is None
assert temperature_sensor.device_class == DEVICE_CLASS_TEMPERATURE
assert humidity_sensor.device_class == DEVICE_CLASS_HUMIDITY
assert temperature_sensor.device_class == SensorDeviceClass.TEMPERATURE
assert humidity_sensor.device_class == SensorDeviceClass.HUMIDITY
def test_attribution(default_sensor, temperature_sensor, humidity_sensor):