Type and payload example of the time_state property updated (#293)

rytilahti · Apr 1, 2018 · 888a17c · 888a17c
1 parent dc43ed0
commit 888a17c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/miio/airqualitymonitor.py b/miio/airqualitymonitor.py
@@ -9,7 +9,7 @@
 class AirQualityMonitorStatus:
     """Container of air quality monitor status."""
     def __init__(self, data):
-        # {'power': 'on', 'aqi': 34, 'battery': 100, 'usb_state': 'off'}
+        # {'power': 'on', 'aqi': 34, 'battery': 100, 'usb_state': 'off', 'time_state': 'on'}
         self.data = data
 
     @property
@@ -38,9 +38,9 @@ def battery(self) -> int:
         return self.data["battery"]
 
     @property
-    def time_state(self) -> str:
+    def time_state(self) -> bool:
         """Current time state."""
-        return self.data["time_state"]
+        return self.data["time_state"] == "on"
 
     def __repr__(self) -> str:
         s = "<AirQualityMonitorStatus power=%s, " \

diff --git a/miio/tests/test_airqualitymonitor.py b/miio/tests/test_airqualitymonitor.py
@@ -14,7 +14,7 @@ def __init__(self, *args, **kwargs):
             'aqi': 34,
             'battery': 100,
             'usb_state': 'off',
-            'time_state': 'format unknown'
+            'time_state': 'on'
         }
         self.return_values = {
             'get_prop': self._get_state,
@@ -59,5 +59,5 @@ def test_status(self):
         assert self.is_on() is True
         assert self.state().aqi == self.device.start_state["aqi"]
         assert self.state().battery == self.device.start_state["battery"]
-        assert self.state().usb_power == (self.device.start_state["usb_state"] == 'on')
-        assert self.state().time_state == self.device.start_state["time_state"]
+        assert self.state().usb_power is (self.device.start_state["usb_state"] == 'on')
+        assert self.state().time_state is (self.device.start_state["time_state"] == 'on')