Round off probability to 2 decimals. (#9365)

* Round off probablity to 2 decimals.

* Update tests

* remove debug print
This commit is contained in:
Alok Saboo 2017-09-12 00:44:51 +05:30 committed by Pascal Vizeli
parent cc1979691e
commit c7ecebfd07
2 changed files with 3 additions and 4 deletions

View file

@ -73,7 +73,7 @@ class TestBayesianBinarySensor(unittest.TestCase):
'prob_false': 0.1,
'prob_true': 0.9
}], state.attributes.get('observations'))
self.assertAlmostEqual(0.7714285714285715,
self.assertAlmostEqual(0.77,
state.attributes.get('probability'))
assert state.state == 'on'
@ -141,7 +141,7 @@ class TestBayesianBinarySensor(unittest.TestCase):
'prob_true': 0.8,
'prob_false': 0.4
}], state.attributes.get('observations'))
self.assertAlmostEqual(0.33333333, state.attributes.get('probability'))
self.assertAlmostEqual(0.33, state.attributes.get('probability'))
assert state.state == 'on'