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

Rewrite bayesian unittest tests to pytest style test functions. (#41740)

Browse Source
This commit is contained in:
thaohtp 2020-10-16 13:10:48 +02:00 committed by GitHub
parent 6f113e981b
commit aabf26bbc8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 380 additions and 384 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

266
tests/components/bayesian/test_binary_sensor.py
View File

@ -1,7 +1,6 @@
"""The test for the bayesian sensor platform.""" """The test for the bayesian sensor platform."""
import json import json
from os import path from os import path
import unittest
from homeassistant import config as hass_config from homeassistant import config as hass_config
from homeassistant.components.bayesian import DOMAIN, binary_sensor as bayesian from homeassistant.components.bayesian import DOMAIN, binary_sensor as bayesian
@ -17,24 +16,12 @@ from homeassistant.const import (
STATE_UNKNOWN, STATE_UNKNOWN,
) )
from homeassistant.core import Context, callback from homeassistant.core import Context, callback
from homeassistant.setup import async_setup_component, setup_component from homeassistant.setup import async_setup_component
from tests.async_mock import patch from tests.async_mock import patch
from tests.common import get_test_home_assistant
class TestBayesianBinarySensor(unittest.TestCase): async def test_load_values_when_added_to_hass(hass):
"""Test the threshold sensor."""
def setup_method(self, method):
"""Set up things to be run when tests are started."""
self.hass = get_test_home_assistant()
def teardown_method(self, method):
"""Stop everything that was started."""
self.hass.stop()
def test_load_values_when_added_to_hass(self):
"""Test that sensor initializes with observations of relevant entities.""" """Test that sensor initializes with observations of relevant entities."""
config = { config = {
@ -55,17 +42,18 @@ class TestBayesianBinarySensor(unittest.TestCase):
} }
} }
self.hass.states.set("sensor.test_monitored", "off") hass.states.async_set("sensor.test_monitored", "off")
self.hass.block_till_done() await hass.async_block_till_done()
assert setup_component(self.hass, "binary_sensor", config) assert await async_setup_component(hass, "binary_sensor", config)
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("binary_sensor.test_binary") state = hass.states.get("binary_sensor.test_binary")
assert state.attributes.get("observations")[0]["prob_given_true"] == 0.8 assert state.attributes.get("observations")[0]["prob_given_true"] == 0.8
assert state.attributes.get("observations")[0]["prob_given_false"] == 0.4 assert state.attributes.get("observations")[0]["prob_given_false"] == 0.4
def test_unknown_state_does_not_influence_probability(self):
async def test_unknown_state_does_not_influence_probability(hass):
"""Test that an unknown state does not change the output probability.""" """Test that an unknown state does not change the output probability."""
config = { config = {
@ -86,16 +74,17 @@ class TestBayesianBinarySensor(unittest.TestCase):
} }
} }
self.hass.states.set("sensor.test_monitored", STATE_UNKNOWN) hass.states.async_set("sensor.test_monitored", STATE_UNKNOWN)
self.hass.block_till_done() await hass.async_block_till_done()
assert setup_component(self.hass, "binary_sensor", config) assert await async_setup_component(hass, "binary_sensor", config)
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("binary_sensor.test_binary") state = hass.states.get("binary_sensor.test_binary")
assert state.attributes.get("observations") == [] assert state.attributes.get("observations") == []
def test_sensor_numeric_state(self):
async def test_sensor_numeric_state(hass):
"""Test sensor on numeric state platform observations.""" """Test sensor on numeric state platform observations."""
config = { config = {
"binary_sensor": { "binary_sensor": {
@ -122,28 +111,28 @@ class TestBayesianBinarySensor(unittest.TestCase):
} }
} }
assert setup_component(self.hass, "binary_sensor", config) assert await async_setup_component(hass, "binary_sensor", config)
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set("sensor.test_monitored", 4) hass.states.async_set("sensor.test_monitored", 4)
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("binary_sensor.test_binary") state = hass.states.get("binary_sensor.test_binary")
assert [] == state.attributes.get("observations") assert [] == state.attributes.get("observations")
assert 0.2 == state.attributes.get("probability") assert 0.2 == state.attributes.get("probability")
assert state.state == "off" assert state.state == "off"
self.hass.states.set("sensor.test_monitored", 6) hass.states.async_set("sensor.test_monitored", 6)
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set("sensor.test_monitored", 4) hass.states.async_set("sensor.test_monitored", 4)
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set("sensor.test_monitored", 6) hass.states.async_set("sensor.test_monitored", 6)
self.hass.states.set("sensor.test_monitored1", 6) hass.states.async_set("sensor.test_monitored1", 6)
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("binary_sensor.test_binary") state = hass.states.get("binary_sensor.test_binary")
assert state.attributes.get("observations")[0]["prob_given_true"] == 0.6 assert state.attributes.get("observations")[0]["prob_given_true"] == 0.6
assert state.attributes.get("observations")[1]["prob_given_true"] == 0.9 assert state.attributes.get("observations")[1]["prob_given_true"] == 0.9
assert state.attributes.get("observations")[1]["prob_given_false"] == 0.1 assert state.attributes.get("observations")[1]["prob_given_false"] == 0.1
@ -151,25 +140,26 @@ class TestBayesianBinarySensor(unittest.TestCase):
assert state.state == "on" assert state.state == "on"
self.hass.states.set("sensor.test_monitored", 6) hass.states.async_set("sensor.test_monitored", 6)
self.hass.states.set("sensor.test_monitored1", 0) hass.states.async_set("sensor.test_monitored1", 0)
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set("sensor.test_monitored", 4) hass.states.async_set("sensor.test_monitored", 4)
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("binary_sensor.test_binary") state = hass.states.get("binary_sensor.test_binary")
assert 0.2 == state.attributes.get("probability") assert 0.2 == state.attributes.get("probability")
assert state.state == "off" assert state.state == "off"
self.hass.states.set("sensor.test_monitored", 15) hass.states.async_set("sensor.test_monitored", 15)
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("binary_sensor.test_binary") state = hass.states.get("binary_sensor.test_binary")
assert state.state == "off" assert state.state == "off"
def test_sensor_state(self):
async def test_sensor_state(hass):
"""Test sensor on state platform observations.""" """Test sensor on state platform observations."""
config = { config = {
"binary_sensor": { "binary_sensor": {
@ -189,43 +179,44 @@ class TestBayesianBinarySensor(unittest.TestCase):
} }
} }
assert setup_component(self.hass, "binary_sensor", config) assert await async_setup_component(hass, "binary_sensor", config)
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set("sensor.test_monitored", "on") hass.states.async_set("sensor.test_monitored", "on")
state = self.hass.states.get("binary_sensor.test_binary") state = hass.states.get("binary_sensor.test_binary")
assert [] == state.attributes.get("observations") assert [] == state.attributes.get("observations")
assert 0.2 == state.attributes.get("probability") assert 0.2 == state.attributes.get("probability")
assert state.state == "off" assert state.state == "off"
self.hass.states.set("sensor.test_monitored", "off") hass.states.async_set("sensor.test_monitored", "off")
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set("sensor.test_monitored", "on") hass.states.async_set("sensor.test_monitored", "on")
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set("sensor.test_monitored", "off") hass.states.async_set("sensor.test_monitored", "off")
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("binary_sensor.test_binary") state = hass.states.get("binary_sensor.test_binary")
assert state.attributes.get("observations")[0]["prob_given_true"] == 0.8 assert state.attributes.get("observations")[0]["prob_given_true"] == 0.8
assert state.attributes.get("observations")[0]["prob_given_false"] == 0.4 assert state.attributes.get("observations")[0]["prob_given_false"] == 0.4
assert round(abs(0.33 - state.attributes.get("probability")), 7) == 0 assert round(abs(0.33 - state.attributes.get("probability")), 7) == 0
assert state.state == "on" assert state.state == "on"
self.hass.states.set("sensor.test_monitored", "off") hass.states.async_set("sensor.test_monitored", "off")
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set("sensor.test_monitored", "on") hass.states.async_set("sensor.test_monitored", "on")
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("binary_sensor.test_binary") state = hass.states.get("binary_sensor.test_binary")
assert round(abs(0.2 - state.attributes.get("probability")), 7) == 0 assert round(abs(0.2 - state.attributes.get("probability")), 7) == 0
assert state.state == "off" assert state.state == "off"
def test_sensor_value_template(self):
async def test_sensor_value_template(hass):
"""Test sensor on template platform observations.""" """Test sensor on template platform observations."""
config = { config = {
"binary_sensor": { "binary_sensor": {
@ -244,43 +235,44 @@ class TestBayesianBinarySensor(unittest.TestCase):
} }
} }
assert setup_component(self.hass, "binary_sensor", config) assert await async_setup_component(hass, "binary_sensor", config)
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set("sensor.test_monitored", "on") hass.states.async_set("sensor.test_monitored", "on")
state = self.hass.states.get("binary_sensor.test_binary") state = hass.states.get("binary_sensor.test_binary")
assert [] == state.attributes.get("observations") assert [] == state.attributes.get("observations")
assert 0.2 == state.attributes.get("probability") assert 0.2 == state.attributes.get("probability")
assert state.state == "off" assert state.state == "off"
self.hass.states.set("sensor.test_monitored", "off") hass.states.async_set("sensor.test_monitored", "off")
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set("sensor.test_monitored", "on") hass.states.async_set("sensor.test_monitored", "on")
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set("sensor.test_monitored", "off") hass.states.async_set("sensor.test_monitored", "off")
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("binary_sensor.test_binary") state = hass.states.get("binary_sensor.test_binary")
assert state.attributes.get("observations")[0]["prob_given_true"] == 0.8 assert state.attributes.get("observations")[0]["prob_given_true"] == 0.8
assert state.attributes.get("observations")[0]["prob_given_false"] == 0.4 assert state.attributes.get("observations")[0]["prob_given_false"] == 0.4
assert round(abs(0.33 - state.attributes.get("probability")), 7) == 0 assert round(abs(0.33 - state.attributes.get("probability")), 7) == 0
assert state.state == "on" assert state.state == "on"
self.hass.states.set("sensor.test_monitored", "off") hass.states.async_set("sensor.test_monitored", "off")
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set("sensor.test_monitored", "on") hass.states.async_set("sensor.test_monitored", "on")
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("binary_sensor.test_binary") state = hass.states.get("binary_sensor.test_binary")
assert round(abs(0.2 - state.attributes.get("probability")), 7) == 0 assert round(abs(0.2 - state.attributes.get("probability")), 7) == 0
assert state.state == "off" assert state.state == "off"
def test_threshold(self):
async def test_threshold(hass):
"""Test sensor on probability threshold limits.""" """Test sensor on probability threshold limits."""
config = { config = {
"binary_sensor": { "binary_sensor": {
@ -299,18 +291,19 @@ class TestBayesianBinarySensor(unittest.TestCase):
} }
} }
assert setup_component(self.hass, "binary_sensor", config) assert await async_setup_component(hass, "binary_sensor", config)
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set("sensor.test_monitored", "on") hass.states.async_set("sensor.test_monitored", "on")
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("binary_sensor.test_binary") state = hass.states.get("binary_sensor.test_binary")
assert round(abs(1.0 - state.attributes.get("probability")), 7) == 0 assert round(abs(1.0 - state.attributes.get("probability")), 7) == 0
assert state.state == "on" assert state.state == "on"
def test_multiple_observations(self):
async def test_multiple_observations(hass):
"""Test sensor with multiple observations of same entity.""" """Test sensor with multiple observations of same entity."""
config = { config = {
"binary_sensor": { "binary_sensor": {
@ -337,12 +330,12 @@ class TestBayesianBinarySensor(unittest.TestCase):
} }
} }
assert setup_component(self.hass, "binary_sensor", config) assert await async_setup_component(hass, "binary_sensor", config)
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set("sensor.test_monitored", "off") hass.states.async_set("sensor.test_monitored", "off")
state = self.hass.states.get("binary_sensor.test_binary") state = hass.states.get("binary_sensor.test_binary")
for key, attrs in state.attributes.items(): for key, attrs in state.attributes.items():
json.dumps(attrs) json.dumps(attrs)
@ -351,14 +344,14 @@ class TestBayesianBinarySensor(unittest.TestCase):
assert state.state == "off" assert state.state == "off"
self.hass.states.set("sensor.test_monitored", "blue") hass.states.async_set("sensor.test_monitored", "blue")
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set("sensor.test_monitored", "off") hass.states.async_set("sensor.test_monitored", "off")
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set("sensor.test_monitored", "blue") hass.states.async_set("sensor.test_monitored", "blue")
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("binary_sensor.test_binary") state = hass.states.get("binary_sensor.test_binary")
assert state.attributes.get("observations")[0]["prob_given_true"] == 0.8 assert state.attributes.get("observations")[0]["prob_given_true"] == 0.8
assert state.attributes.get("observations")[0]["prob_given_false"] == 0.4 assert state.attributes.get("observations")[0]["prob_given_false"] == 0.4
@ -366,17 +359,18 @@ class TestBayesianBinarySensor(unittest.TestCase):
assert state.state == "on" assert state.state == "on"
self.hass.states.set("sensor.test_monitored", "blue") hass.states.async_set("sensor.test_monitored", "blue")
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set("sensor.test_monitored", "red") hass.states.async_set("sensor.test_monitored", "red")
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("binary_sensor.test_binary") state = hass.states.get("binary_sensor.test_binary")
assert round(abs(0.11 - state.attributes.get("probability")), 7) == 0 assert round(abs(0.11 - state.attributes.get("probability")), 7) == 0
assert state.state == "off" assert state.state == "off"
def test_probability_updates(self):
async def test_probability_updates(hass):
"""Test probability update function.""" """Test probability update function."""
prob_given_true = [0.3, 0.6, 0.8] prob_given_true = [0.3, 0.6, 0.8]
prob_given_false = [0.7, 0.4, 0.2] prob_given_false = [0.7, 0.4, 0.2]
@ -396,7 +390,8 @@ class TestBayesianBinarySensor(unittest.TestCase):
assert round(abs(0.9130434782608695 - prior), 7) == 0 assert round(abs(0.9130434782608695 - prior), 7) == 0
def test_observed_entities(self):
async def test_observed_entities(hass):
"""Test sensor on observed entities.""" """Test sensor on observed entities."""
config = { config = {
"binary_sensor": { "binary_sensor": {
@ -421,34 +416,35 @@ class TestBayesianBinarySensor(unittest.TestCase):
} }
} }
assert setup_component(self.hass, "binary_sensor", config) assert await async_setup_component(hass, "binary_sensor", config)
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set("sensor.test_monitored", "on") hass.states.async_set("sensor.test_monitored", "on")
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set("sensor.test_monitored1", "off") hass.states.async_set("sensor.test_monitored1", "off")
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("binary_sensor.test_binary") state = hass.states.get("binary_sensor.test_binary")
assert [] == state.attributes.get("occurred_observation_entities") assert [] == state.attributes.get("occurred_observation_entities")
self.hass.states.set("sensor.test_monitored", "off") hass.states.async_set("sensor.test_monitored", "off")
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("binary_sensor.test_binary") state = hass.states.get("binary_sensor.test_binary")
assert ["sensor.test_monitored"] == state.attributes.get( assert ["sensor.test_monitored"] == state.attributes.get(
"occurred_observation_entities" "occurred_observation_entities"
) )
self.hass.states.set("sensor.test_monitored1", "on") hass.states.async_set("sensor.test_monitored1", "on")
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("binary_sensor.test_binary") state = hass.states.get("binary_sensor.test_binary")
assert ["sensor.test_monitored", "sensor.test_monitored1"] == sorted( assert ["sensor.test_monitored", "sensor.test_monitored1"] == sorted(
state.attributes.get("occurred_observation_entities") state.attributes.get("occurred_observation_entities")
) )
def test_state_attributes_are_serializable(self):
async def test_state_attributes_are_serializable(hass):
"""Test sensor on observed entities.""" """Test sensor on observed entities."""
config = { config = {
"binary_sensor": { "binary_sensor": {
@ -473,29 +469,29 @@ class TestBayesianBinarySensor(unittest.TestCase):
} }
} }
assert setup_component(self.hass, "binary_sensor", config) assert await async_setup_component(hass, "binary_sensor", config)
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set("sensor.test_monitored", "on") hass.states.async_set("sensor.test_monitored", "on")
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set("sensor.test_monitored1", "off") hass.states.async_set("sensor.test_monitored1", "off")
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("binary_sensor.test_binary") state = hass.states.get("binary_sensor.test_binary")
assert [] == state.attributes.get("occurred_observation_entities") assert [] == state.attributes.get("occurred_observation_entities")
self.hass.states.set("sensor.test_monitored", "off") hass.states.async_set("sensor.test_monitored", "off")
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("binary_sensor.test_binary") state = hass.states.get("binary_sensor.test_binary")
assert ["sensor.test_monitored"] == state.attributes.get( assert ["sensor.test_monitored"] == state.attributes.get(
"occurred_observation_entities" "occurred_observation_entities"
) )
self.hass.states.set("sensor.test_monitored1", "on") hass.states.async_set("sensor.test_monitored1", "on")
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("binary_sensor.test_binary") state = hass.states.get("binary_sensor.test_binary")
assert ["sensor.test_monitored", "sensor.test_monitored1"] == sorted( assert ["sensor.test_monitored", "sensor.test_monitored1"] == sorted(
state.attributes.get("occurred_observation_entities") state.attributes.get("occurred_observation_entities")
) )