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

Use new enums in litterrobot tests (#62515)

Browse Source 
* 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
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/components/litterrobot/test_switch.py
View File

@@ -10,14 +10,10 @@ from homeassistant.components.switch import (
SERVICE_TURN_OFF,
SERVICE_TURN_ON,
)
from homeassistant.const import (
ATTR_ENTITY_ID,
ENTITY_CATEGORY_CONFIG,
STATE_OFF,
STATE_ON,
)
from homeassistant.const import ATTR_ENTITY_ID, STATE_OFF, STATE_ON
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry
from homeassistant.helpers.entity import EntityCategory
from homeassistant.util.dt import utcnow
from .conftest import setup_integration
@@ -39,7 +35,7 @@ async def test_switch(hass: HomeAssistant, mock_account: MagicMock):
ent_reg = entity_registry.async_get(hass)
entity_entry = ent_reg.async_get(NIGHT_LIGHT_MODE_ENTITY_ID)
assert entity_entry
assert entity_entry.entity_category == ENTITY_CATEGORY_CONFIG
assert entity_entry.entity_category is EntityCategory.CONFIG
@pytest.mark.parametrize(