jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-19 11:17:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

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 import DATA_ARLO, sensor as arlo
from homeassistant.components.arlo.sensor import SENSOR_TYPES from homeassistant.components.arlo.sensor import SENSOR_TYPES
from homeassistant.const import ( from homeassistant.components.sensor import SensorDeviceClass
DEVICE_CLASS_BATTERY, from homeassistant.const import PERCENTAGE
DEVICE_CLASS_HUMIDITY,
DEVICE_CLASS_TEMPERATURE,
PERCENTAGE,
)
def _get_named_tuple(input_dict): 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): def test_sensor_device_class__battery(battery_sensor):
"""Test the battery device_class.""" """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): def test_sensor_device_class(temperature_sensor):
"""Test the device_class property.""" """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): 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): def test_device_class(default_sensor, temperature_sensor, humidity_sensor):
"""Test the device_class property.""" """Test the device_class property."""
assert default_sensor.device_class is None assert default_sensor.device_class is None
assert temperature_sensor.device_class == DEVICE_CLASS_TEMPERATURE assert temperature_sensor.device_class == SensorDeviceClass.TEMPERATURE
assert humidity_sensor.device_class == DEVICE_CLASS_HUMIDITY assert humidity_sensor.device_class == SensorDeviceClass.HUMIDITY
def test_attribution(default_sensor, temperature_sensor, humidity_sensor): def test_attribution(default_sensor, temperature_sensor, humidity_sensor):