jeena/home-assistant
1
0
Fork 0
Code Issues Pull requests Projects 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

294
tests/components/bayesian/test_binary_sensor.py
View file

@ -21,234 +21,224 @@ class TestBayesianBinarySensor(unittest.TestCase):
def test_sensor_numeric_state(self):
"""Test sensor on numeric state platform observations."""
config = {
'binary_sensor': {
'platform':
'bayesian',
'name':
'Test_Binary',
'observations': [{
'platform': 'numeric_state',
'entity_id': 'sensor.test_monitored',
'below': 10,
'above': 5,
'prob_given_true': 0.6
}, {
'platform': 'numeric_state',
'entity_id': 'sensor.test_monitored1',
'below': 7,
'above': 5,
'prob_given_true': 0.9,
'prob_given_false': 0.1
}],
'prior':
0.2,
"binary_sensor": {
"platform": "bayesian",
"name": "Test_Binary",
"observations": [
{
"platform": "numeric_state",
"entity_id": "sensor.test_monitored",
"below": 10,
"above": 5,
"prob_given_true": 0.6,
},
{
"platform": "numeric_state",
"entity_id": "sensor.test_monitored1",
"below": 7,
"above": 5,
"prob_given_true": 0.9,
"prob_given_false": 0.1,
},
],
"prior": 0.2,
}
}
assert setup_component(self.hass, 'binary_sensor', config)
assert setup_component(self.hass, "binary_sensor", config)
self.hass.states.set('sensor.test_monitored', 4)
self.hass.states.set("sensor.test_monitored", 4)
self.hass.block_till_done()
state = self.hass.states.get('binary_sensor.test_binary')
state = self.hass.states.get("binary_sensor.test_binary")
assert [] == state.attributes.get('observations')
assert 0.2 == state.attributes.get('probability')
assert [] == state.attributes.get("observations")
assert 0.2 == state.attributes.get("probability")
assert state.state == 'off'
assert state.state == "off"
self.hass.states.set('sensor.test_monitored', 6)
self.hass.states.set("sensor.test_monitored", 6)
self.hass.block_till_done()
self.hass.states.set('sensor.test_monitored', 4)
self.hass.states.set("sensor.test_monitored", 4)
self.hass.block_till_done()
self.hass.states.set('sensor.test_monitored', 6)
self.hass.states.set('sensor.test_monitored1', 6)
self.hass.states.set("sensor.test_monitored", 6)
self.hass.states.set("sensor.test_monitored1", 6)
self.hass.block_till_done()
state = self.hass.states.get('binary_sensor.test_binary')
assert [{
'prob_false': 0.4,
'prob_true': 0.6
}, {
'prob_false': 0.1,
'prob_true': 0.9
}] == state.attributes.get('observations')
assert round(abs(0.77-state.attributes.get('probability')), 7) == 0
state = self.hass.states.get("binary_sensor.test_binary")
assert [
{"prob_false": 0.4, "prob_true": 0.6},
{"prob_false": 0.1, "prob_true": 0.9},
] == state.attributes.get("observations")
assert round(abs(0.77 - state.attributes.get("probability")), 7) == 0
assert state.state == 'on'
assert state.state == "on"
self.hass.states.set('sensor.test_monitored', 6)
self.hass.states.set('sensor.test_monitored1', 0)
self.hass.states.set("sensor.test_monitored", 6)
self.hass.states.set("sensor.test_monitored1", 0)
self.hass.block_till_done()
self.hass.states.set('sensor.test_monitored', 4)
self.hass.states.set("sensor.test_monitored", 4)
self.hass.block_till_done()
state = self.hass.states.get('binary_sensor.test_binary')
assert 0.2 == state.attributes.get('probability')
state = self.hass.states.get("binary_sensor.test_binary")
assert 0.2 == state.attributes.get("probability")
assert state.state == 'off'
assert state.state == "off"
self.hass.states.set('sensor.test_monitored', 15)
self.hass.states.set("sensor.test_monitored", 15)
self.hass.block_till_done()
state = self.hass.states.get('binary_sensor.test_binary')
state = self.hass.states.get("binary_sensor.test_binary")
assert state.state == 'off'
assert state.state == "off"
def test_sensor_state(self):
"""Test sensor on state platform observations."""
config = {
'binary_sensor': {
'name':
'Test_Binary',
'platform':
'bayesian',
'observations': [{
'platform': 'state',
'entity_id': 'sensor.test_monitored',
'to_state': 'off',
'prob_given_true': 0.8,
'prob_given_false': 0.4
}],
'prior':
0.2,
'probability_threshold':
0.32,
"binary_sensor": {
"name": "Test_Binary",
"platform": "bayesian",
"observations": [
{
"platform": "state",
"entity_id": "sensor.test_monitored",
"to_state": "off",
"prob_given_true": 0.8,
"prob_given_false": 0.4,
}
],
"prior": 0.2,
"probability_threshold": 0.32,
}
}
assert setup_component(self.hass, 'binary_sensor', config)
assert setup_component(self.hass, "binary_sensor", config)
self.hass.states.set('sensor.test_monitored', 'on')
self.hass.states.set("sensor.test_monitored", "on")
state = self.hass.states.get('binary_sensor.test_binary')
state = self.hass.states.get("binary_sensor.test_binary")
assert [] == state.attributes.get('observations')
assert 0.2 == state.attributes.get('probability')
assert [] == state.attributes.get("observations")
assert 0.2 == state.attributes.get("probability")
assert state.state == 'off'
assert state.state == "off"
self.hass.states.set('sensor.test_monitored', 'off')
self.hass.states.set("sensor.test_monitored", "off")
self.hass.block_till_done()
self.hass.states.set('sensor.test_monitored', 'on')
self.hass.states.set("sensor.test_monitored", "on")
self.hass.block_till_done()
self.hass.states.set('sensor.test_monitored', 'off')
self.hass.states.set("sensor.test_monitored", "off")
self.hass.block_till_done()
state = self.hass.states.get('binary_sensor.test_binary')
assert [{
'prob_true': 0.8,
'prob_false': 0.4
}] == state.attributes.get('observations')
assert round(abs(0.33-state.attributes.get('probability')), 7) == 0
state = self.hass.states.get("binary_sensor.test_binary")
assert [{"prob_true": 0.8, "prob_false": 0.4}] == state.attributes.get(
"observations"
)
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')
self.hass.states.set("sensor.test_monitored", "off")
self.hass.block_till_done()
self.hass.states.set('sensor.test_monitored', 'on')
self.hass.states.set("sensor.test_monitored", "on")
self.hass.block_till_done()
state = self.hass.states.get('binary_sensor.test_binary')
assert round(abs(0.2-state.attributes.get('probability')), 7) == 0
state = self.hass.states.get("binary_sensor.test_binary")
assert round(abs(0.2 - state.attributes.get("probability")), 7) == 0
assert state.state == 'off'
assert state.state == "off"
def test_threshold(self):
"""Test sensor on probabilty threshold limits."""
config = {
'binary_sensor': {
'name':
'Test_Binary',
'platform':
'bayesian',
'observations': [{
'platform': 'state',
'entity_id': 'sensor.test_monitored',
'to_state': 'on',
'prob_given_true': 1.0,
}],
'prior':
0.5,
'probability_threshold':
1.0,
"binary_sensor": {
"name": "Test_Binary",
"platform": "bayesian",
"observations": [
{
"platform": "state",
"entity_id": "sensor.test_monitored",
"to_state": "on",
"prob_given_true": 1.0,
}
],
"prior": 0.5,
"probability_threshold": 1.0,
}
}
assert setup_component(self.hass, 'binary_sensor', config)
assert setup_component(self.hass, "binary_sensor", config)
self.hass.states.set('sensor.test_monitored', 'on')
self.hass.states.set("sensor.test_monitored", "on")
self.hass.block_till_done()
state = self.hass.states.get('binary_sensor.test_binary')
assert round(abs(1.0-state.attributes.get('probability')), 7) == 0
state = self.hass.states.get("binary_sensor.test_binary")
assert round(abs(1.0 - state.attributes.get("probability")), 7) == 0
assert state.state == 'on'
assert state.state == "on"
def test_multiple_observations(self):
"""Test sensor with multiple observations of same entity."""
config = {
'binary_sensor': {
'name':
'Test_Binary',
'platform':
'bayesian',
'observations': [{
'platform': 'state',
'entity_id': 'sensor.test_monitored',
'to_state': 'blue',
'prob_given_true': 0.8,
'prob_given_false': 0.4
}, {
'platform': 'state',
'entity_id': 'sensor.test_monitored',
'to_state': 'red',
'prob_given_true': 0.2,
'prob_given_false': 0.4
}],
'prior':
0.2,
'probability_threshold':
0.32,
"binary_sensor": {
"name": "Test_Binary",
"platform": "bayesian",
"observations": [
{
"platform": "state",
"entity_id": "sensor.test_monitored",
"to_state": "blue",
"prob_given_true": 0.8,
"prob_given_false": 0.4,
},
{
"platform": "state",
"entity_id": "sensor.test_monitored",
"to_state": "red",
"prob_given_true": 0.2,
"prob_given_false": 0.4,
},
],
"prior": 0.2,
"probability_threshold": 0.32,
}
}
assert setup_component(self.hass, 'binary_sensor', config)
assert setup_component(self.hass, "binary_sensor", config)
self.hass.states.set('sensor.test_monitored', 'off')
self.hass.states.set("sensor.test_monitored", "off")
state = self.hass.states.get('binary_sensor.test_binary')
state = self.hass.states.get("binary_sensor.test_binary")
assert [] == state.attributes.get('observations')
assert 0.2 == state.attributes.get('probability')
assert [] == state.attributes.get("observations")
assert 0.2 == state.attributes.get("probability")
assert state.state == 'off'
assert state.state == "off"
self.hass.states.set('sensor.test_monitored', 'blue')
self.hass.states.set("sensor.test_monitored", "blue")
self.hass.block_till_done()
self.hass.states.set('sensor.test_monitored', 'off')
self.hass.states.set("sensor.test_monitored", "off")
self.hass.block_till_done()
self.hass.states.set('sensor.test_monitored', 'blue')
self.hass.states.set("sensor.test_monitored", "blue")
self.hass.block_till_done()
state = self.hass.states.get('binary_sensor.test_binary')
assert [{
'prob_true': 0.8,
'prob_false': 0.4
}] == state.attributes.get('observations')
assert round(abs(0.33-state.attributes.get('probability')), 7) == 0
state = self.hass.states.get("binary_sensor.test_binary")
assert [{"prob_true": 0.8, "prob_false": 0.4}] == state.attributes.get(
"observations"
)
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', 'blue')
self.hass.states.set("sensor.test_monitored", "blue")
self.hass.block_till_done()
self.hass.states.set('sensor.test_monitored', 'red')
self.hass.states.set("sensor.test_monitored", "red")
self.hass.block_till_done()
state = self.hass.states.get('binary_sensor.test_binary')
assert round(abs(0.11-state.attributes.get('probability')), 7) == 0
state = self.hass.states.get("binary_sensor.test_binary")
assert round(abs(0.11 - state.attributes.get("probability")), 7) == 0
assert state.state == 'off'
assert state.state == "off"
def test_probability_updates(self):
"""Test probability update function."""
@ -259,7 +249,7 @@ class TestBayesianBinarySensor(unittest.TestCase):
for pt, pf in zip(prob_true, prob_false):
prior = bayesian.update_probability(prior, pt, pf)
assert round(abs(0.720000-prior), 7) == 0
assert round(abs(0.720000 - prior), 7) == 0
prob_true = [0.8, 0.3, 0.9]
prob_false = [0.6, 0.4, 0.2]
@ -268,4 +258,4 @@ class TestBayesianBinarySensor(unittest.TestCase):
for pt, pf in zip(prob_true, prob_false):
prior = bayesian.update_probability(prior, pt, pf)
assert round(abs(0.9130434782608695-prior), 7) == 0
assert round(abs(0.9130434782608695 - prior), 7) == 0