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

Black

Browse source
This commit is contained in:
Paulus Schoutsen 2019-07-31 12:25:30 -07:00
parent da05dfe708
commit 4de97abc3a
2676 changed files with 163166 additions and 140084 deletions
Download patch file Download diff file Expand all files Collapse all files

32
tests/components/air_quality/test_air_quality.py
View file

@ -1,42 +1,36 @@
"""The tests for the Air Quality component."""
from homeassistant.components.air_quality import (
ATTR_ATTRIBUTION, ATTR_N2O,
ATTR_OZONE, ATTR_PM_10)
ATTR_ATTRIBUTION,
ATTR_N2O,
ATTR_OZONE,
ATTR_PM_10,
)
from homeassistant.setup import async_setup_component
async def test_state(hass):
"""Test Air Quality state."""
config = {
'air_quality': {
'platform': 'demo',
}
}
config = {"air_quality": {"platform": "demo"}}
assert await async_setup_component(hass, 'air_quality', config)
assert await async_setup_component(hass, "air_quality", config)
state = hass.states.get('air_quality.demo_air_quality_home')
state = hass.states.get("air_quality.demo_air_quality_home")
assert state is not None
assert state.state == '14'
assert state.state == "14"
async def test_attributes(hass):
"""Test Air Quality attributes."""
config = {
'air_quality': {
'platform': 'demo',
}
}
config = {"air_quality": {"platform": "demo"}}
assert await async_setup_component(hass, 'air_quality', config)
assert await async_setup_component(hass, "air_quality", config)
state = hass.states.get('air_quality.demo_air_quality_office')
state = hass.states.get("air_quality.demo_air_quality_office")
assert state is not None
data = state.attributes
assert data.get(ATTR_PM_10) == 16
assert data.get(ATTR_N2O) is None
assert data.get(ATTR_OZONE) is None
assert data.get(ATTR_ATTRIBUTION) == \
'Powered by Home Assistant'
assert data.get(ATTR_ATTRIBUTION) == "Powered by Home Assistant"