Просмотр исходного кода

Fix CI tests

- test_moebot, test_lawn_mower: reformat to satisy black.
- test_device_config: add expectations for lawn_mower entities.
Jason Rumney 2 лет назад
Родитель
Сommit
4e2916567c
3 измененных файлов с 15 добавлено и 41 удалено
  1. 10 40
      tests/devices/test_moebot.py
  2. 1 0
      tests/test_device_config.py
  3. 4 1
      tests/test_lawn_mower.py

+ 10 - 40
tests/devices/test_moebot.py

@@ -81,55 +81,25 @@ class TestMoebot(TuyaDeviceTestCase):
 
     def test_lawnmower_activity(self):
         self.dps[STATUS_DP] = "ERROR"
-        self.assertEqual(
-            self.mower.activity,
-            LawnMowerActivity.ERROR
-        )
+        self.assertEqual(self.mower.activity, LawnMowerActivity.ERROR)
         self.dps[STATUS_DP] = "EMERGENCY"
-        self.assertEqual(
-            self.mower.activity,
-            LawnMowerActivity.ERROR
-        )
+        self.assertEqual(self.mower.activity, LawnMowerActivity.ERROR)
         self.dps[STATUS_DP] = "PAUSED"
-        self.assertEqual(
-            self.mower.activity,
-            LawnMowerActivity.PAUSED
-        )
+        self.assertEqual(self.mower.activity, LawnMowerActivity.PAUSED)
         self.dps[STATUS_DP] = "PARK"
-        self.assertEqual(
-            self.mower.activity,
-            LawnMowerActivity.PAUSED
-        )
+        self.assertEqual(self.mower.activity, LawnMowerActivity.PAUSED)
         self.dps[STATUS_DP] = "MOWING"
-        self.assertEqual(
-            self.mower.activity,
-            LawnMowerActivity.MOWING
-        )
+        self.assertEqual(self.mower.activity, LawnMowerActivity.MOWING)
         self.dps[STATUS_DP] = "FIXED_MOWING"
-        self.assertEqual(
-            self.mower.activity,
-            LawnMowerActivity.MOWING
-        )
+        self.assertEqual(self.mower.activity, LawnMowerActivity.MOWING)
         self.dps[STATUS_DP] = "STANDBY"
-        self.assertEqual(
-            self.mower.activity,
-            LawnMowerActivity.DOCKED
-        )
+        self.assertEqual(self.mower.activity, LawnMowerActivity.DOCKED)
         self.dps[STATUS_DP] = "CHARGING"
-        self.assertEqual(
-            self.mower.activity,
-            LawnMowerActivity.DOCKED
-        )
+        self.assertEqual(self.mower.activity, LawnMowerActivity.DOCKED)
         self.dps[STATUS_DP] = "LOCKED"
-        self.assertEqual(
-            self.mower.activity,
-            LawnMowerActivity.DOCKED
-        )
+        self.assertEqual(self.mower.activity, LawnMowerActivity.DOCKED)
         self.dps[STATUS_DP] = "CHARGING_WITH_TASK_SUSPEND"
-        self.assertEqual(
-            self.mower.activity,
-            LawnMowerActivity.DOCKED
-        )
+        self.assertEqual(self.mower.activity, LawnMowerActivity.DOCKED)
 
     async def test_async_start_mowing(self):
         async with assert_device_properties_set(

+ 1 - 0
tests/test_device_config.py

@@ -67,6 +67,7 @@ KNOWN_DPS = {
         "optional": ["switch", "oscillate", "direction"],
     },
     "humidifier": {"required": ["switch", "humidity"], "optional": ["mode"]},
+    "lawn_mower": {"required": ["activity", "command"], "optional": []},
     "light": {
         "required": [{"or": ["switch", "brightness", "effect"]}],
         "optional": ["color_mode", "color_temp", "rgbhsv"],

+ 4 - 1
tests/test_lawn_mower.py

@@ -10,7 +10,10 @@ from custom_components.tuya_local.const import (
     CONF_TYPE,
     DOMAIN,
 )
-from custom_components.tuya_local.lawn_mower import TuyaLocalLawnMower, async_setup_entry
+from custom_components.tuya_local.lawn_mower import (
+    TuyaLocalLawnMower,
+    async_setup_entry,
+)
 
 
 @pytest.mark.asyncio