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

Update template/test_weather.py to use pytest (#56223)

Browse Source
This commit is contained in:
jan iversen 2021-09-14 22:06:55 +02:00 committed by GitHub
parent 2c348dd2d7
commit 96a9af8cc4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 30 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
tests/components/template/test_weather.py
View File

@ -1,4 +1,6 @@
"""The tests for the Template Weather platform.""" """The tests for the Template Weather platform."""
import pytest
from homeassistant.components.weather import ( from homeassistant.components.weather import (
ATTR_WEATHER_ATTRIBUTION, ATTR_WEATHER_ATTRIBUTION,
ATTR_WEATHER_HUMIDITY, ATTR_WEATHER_HUMIDITY,
@ -10,14 +12,12 @@ from homeassistant.components.weather import (
ATTR_WEATHER_WIND_SPEED, ATTR_WEATHER_WIND_SPEED,
DOMAIN, DOMAIN,
) )
from homeassistant.setup import async_setup_component
async def test_template_state_text(hass): @pytest.mark.parametrize("count,domain", [(1, DOMAIN)])
"""Test the state text of a template.""" @pytest.mark.parametrize(
await async_setup_component( "config",
hass, [
DOMAIN,
{ {
"weather": [ "weather": [
{"weather": {"platform": "demo"}}, {"weather": {"platform": "demo"}},
@ -37,40 +37,27 @@ async def test_template_state_text(hass):
}, },
] ]
}, },
],
) )
async def test_template_state_text(hass, start_ha):
"""Test the state text of a template."""
for attr, v_attr, value in [
(
"sensor.attribution",
ATTR_WEATHER_ATTRIBUTION,
"The custom attribution",
),
("sensor.temperature", ATTR_WEATHER_TEMPERATURE, 22.3),
("sensor.humidity", ATTR_WEATHER_HUMIDITY, 60),
("sensor.pressure", ATTR_WEATHER_PRESSURE, 1000),
("sensor.windspeed", ATTR_WEATHER_WIND_SPEED, 20),
("sensor.windbearing", ATTR_WEATHER_WIND_BEARING, 180),
("sensor.ozone", ATTR_WEATHER_OZONE, 25),
("sensor.visibility", ATTR_WEATHER_VISIBILITY, 4.6),
]:
hass.states.async_set(attr, value)
await hass.async_block_till_done() await hass.async_block_till_done()
await hass.async_start()
await hass.async_block_till_done()
hass.states.async_set("sensor.attribution", "The custom attribution")
await hass.async_block_till_done()
hass.states.async_set("sensor.temperature", 22.3)
await hass.async_block_till_done()
hass.states.async_set("sensor.humidity", 60)
await hass.async_block_till_done()
hass.states.async_set("sensor.pressure", 1000)
await hass.async_block_till_done()
hass.states.async_set("sensor.windspeed", 20)
await hass.async_block_till_done()
hass.states.async_set("sensor.windbearing", 180)
await hass.async_block_till_done()
hass.states.async_set("sensor.ozone", 25)
await hass.async_block_till_done()
hass.states.async_set("sensor.visibility", 4.6)
await hass.async_block_till_done()
state = hass.states.get("weather.test") state = hass.states.get("weather.test")
assert state is not None assert state is not None
assert state.state == "sunny" assert state.state == "sunny"
assert state.attributes.get(v_attr) == value
data = state.attributes
assert data.get(ATTR_WEATHER_ATTRIBUTION) == "The custom attribution"
assert data.get(ATTR_WEATHER_TEMPERATURE) == 22.3
assert data.get(ATTR_WEATHER_HUMIDITY) == 60
assert data.get(ATTR_WEATHER_PRESSURE) == 1000
assert data.get(ATTR_WEATHER_WIND_SPEED) == 20
assert data.get(ATTR_WEATHER_WIND_BEARING) == 180
assert data.get(ATTR_WEATHER_OZONE) == 25
assert data.get(ATTR_WEATHER_VISIBILITY) == 4.6