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
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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():