Fix tests

rytilahti · Jan 7, 2019 · f90d333 · f90d333
1 parent 199d1dd
commit f90d333
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/miio/tests/test_airhumidifier.py b/miio/tests/test_airhumidifier.py
@@ -147,6 +147,16 @@ def led_brightness():
         self.device.set_led_brightness(LedBrightness.Off)
         assert led_brightness() == LedBrightness.Off
 
+    def test_set_led(self):
+        def led_brightness():
+            return self.device.status().led_brightness
+
+        self.device.set_led(True)
+        assert led_brightness() == LedBrightness.Bright
+
+        self.device.set_led(False)
+        assert led_brightness() == LedBrightness.Off
+
     def test_set_buzzer(self):
         def buzzer():
             return self.device.status().buzzer
@@ -343,6 +353,16 @@ def led_brightness():
         self.device.set_led_brightness(LedBrightness.Off)
         assert led_brightness() == LedBrightness.Off
 
+    def test_set_led(self):
+        def led_brightness():
+            return self.device.status().led_brightness
+
+        self.device.set_led(True)
+        assert led_brightness() == LedBrightness.Bright
+
+        self.device.set_led(False)
+        assert led_brightness() == LedBrightness.Off
+
     def test_set_buzzer(self):
         def buzzer():
             return self.device.status().buzzer