From 97076aa3fd9d72e526a0f651dd647e4a61eb551e Mon Sep 17 00:00:00 2001
From: Diogo Gomes <diogogomes@gmail.com>
Date: Fri, 18 May 2018 06:48:16 +0100
Subject: [PATCH] Fix probability_threshold in binary_sensor.bayesian (#14512)
 (Closes: #14362)

---
 .../components/binary_sensor/bayesian.py      |  2 +-
 .../components/binary_sensor/test_bayesian.py | 31 +++++++++++++++++++
 2 files changed, 32 insertions(+), 1 deletion(-)

diff --git a/homeassistant/components/binary_sensor/bayesian.py b/homeassistant/components/binary_sensor/bayesian.py
index f3dbc912ade..72110eb50c9 100644
--- a/homeassistant/components/binary_sensor/bayesian.py
+++ b/homeassistant/components/binary_sensor/bayesian.py
@@ -217,4 +217,4 @@ class BayesianBinarySensor(BinarySensorDevice):
     @asyncio.coroutine
     def async_update(self):
         """Get the latest data and update the states."""
-        self._deviation = bool(self.probability > self._probability_threshold)
+        self._deviation = bool(self.probability >= self._probability_threshold)
diff --git a/tests/components/binary_sensor/test_bayesian.py b/tests/components/binary_sensor/test_bayesian.py
index 3b403c3702f..c3242e09e78 100644
--- a/tests/components/binary_sensor/test_bayesian.py
+++ b/tests/components/binary_sensor/test_bayesian.py
@@ -154,6 +154,37 @@ class TestBayesianBinarySensor(unittest.TestCase):
 
         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,
+            }
+        }
+
+        assert setup_component(self.hass, 'binary_sensor', config)
+
+        self.hass.states.set('sensor.test_monitored', 'on')
+        self.hass.block_till_done()
+
+        state = self.hass.states.get('binary_sensor.test_binary')
+        self.assertAlmostEqual(1.0, state.attributes.get('probability'))
+
+        assert state.state == 'on'
+
     def test_multiple_observations(self):
         """Test sensor with multiple observations of same entity."""
         config = {