1
0
mirror of https://github.com/home-assistant/core.git synced 2025-12-23 20:39:01 +00:00

Use new enums in litterrobot tests (#62515)

* Use new enums in litterrobot tests

* Code review: swap == for is
This commit is contained in:
Dave T
2021-12-22 09:00:24 +00:00
committed by GitHub
parent d3d6965ba0
commit a7ef983a31
4 changed files with 12 additions and 19 deletions

View File

@@ -2,8 +2,8 @@
from unittest.mock import Mock
from homeassistant.components.litterrobot.sensor import LitterRobotSleepTimeSensor
from homeassistant.components.sensor import DOMAIN as PLATFORM_DOMAIN
from homeassistant.const import DEVICE_CLASS_TIMESTAMP, PERCENTAGE
from homeassistant.components.sensor import DOMAIN as PLATFORM_DOMAIN, SensorDeviceClass
from homeassistant.const import PERCENTAGE
from .conftest import create_mock_robot, setup_integration
@@ -30,7 +30,7 @@ async def test_sleep_time_sensor_with_none_state(hass):
assert sensor
assert sensor.state is None
assert sensor.device_class == DEVICE_CLASS_TIMESTAMP
assert sensor.device_class is SensorDeviceClass.TIMESTAMP
async def test_gauge_icon():