瀏覽代碼

tests: update for removal of deprecated timer number entities

Jason Rumney 1 月之前
父節點
當前提交
9c46baf4f3

+ 0 - 7
tests/devices/test_beok_tr9b_thermostat.py

@@ -87,12 +87,6 @@ class TestBeokTR9BThermostat(
                     "device_class": BinarySensorDeviceClass.PROBLEM,
                     "testdata": (1, 0),
                 },
-                {
-                    "dps": VALVE_DPS,
-                    "name": "binary_sensor_valve",
-                    "device_class": BinarySensorDeviceClass.OPENING,
-                    "testdata": ("open", "close"),
-                },
             ],
         )
         self.setUpMultiNumber(
@@ -122,7 +116,6 @@ class TestBeokTR9BThermostat(
         self.mark_secondary(
             [
                 "binary_sensor_problem",
-                "binary_sensor_valve",
                 "lock_child_lock",
                 "number_minimum_temperature",
                 "number_maximum_temperature",

+ 3 - 25
tests/devices/test_digoo_dgsp202.py

@@ -1,6 +1,5 @@
 """Tests for Digoo DSSP202 dual switch with timers and energy monitoring"""
 
-from homeassistant.components.number import NumberDeviceClass
 from homeassistant.components.sensor import SensorDeviceClass
 from homeassistant.components.switch import SwitchDeviceClass
 from homeassistant.const import (
@@ -11,7 +10,6 @@ from homeassistant.const import (
 )
 
 from ..const import DIGOO_DGSP202_SOCKET_PAYLOAD
-from ..mixins.number import MultiNumberTests
 from ..mixins.sensor import MultiSensorTests
 from ..mixins.switch import MultiSwitchTests
 from .base_device_tests import TuyaDeviceTestCase
@@ -26,7 +24,9 @@ VOLTAGE_DPS = "20"
 
 
 class TestDigooDGSP202Switch(
-    MultiNumberTests, MultiSensorTests, MultiSwitchTests, TuyaDeviceTestCase
+    MultiSensorTests,
+    MultiSwitchTests,
+    TuyaDeviceTestCase,
 ):
     __test__ = True
 
@@ -48,26 +48,6 @@ class TestDigooDGSP202Switch(
                 },
             ]
         )
-        self.setUpMultiNumber(
-            [
-                {
-                    "dps": TIMER1_DPS,
-                    "name": "number_timer_1",
-                    "max": 1440,
-                    "scale": 60,
-                    "device_class": NumberDeviceClass.DURATION,
-                    "unit": UnitOfTime.MINUTES,
-                },
-                {
-                    "dps": TIMER2_DPS,
-                    "name": "number_timer_2",
-                    "max": 1440,
-                    "scale": 60,
-                    "device_class": NumberDeviceClass.DURATION,
-                    "unit": UnitOfTime.MINUTES,
-                },
-            ]
-        )
         self.setUpMultiSensors(
             [
                 {
@@ -97,8 +77,6 @@ class TestDigooDGSP202Switch(
         )
         self.mark_secondary(
             [
-                "number_timer_1",
-                "number_timer_2",
                 "sensor_voltage",
                 "sensor_current",
                 "sensor_power",

+ 0 - 12
tests/devices/test_goldair_gpph_heater.py

@@ -1,6 +1,5 @@
 from homeassistant.components.binary_sensor import BinarySensorDeviceClass
 from homeassistant.components.climate.const import ClimateEntityFeature, HVACMode
-from homeassistant.components.number import NumberDeviceClass
 from homeassistant.components.sensor import SensorDeviceClass
 from homeassistant.const import (
     PERCENTAGE,
@@ -15,7 +14,6 @@ from ..mixins.binary_sensor import BasicBinarySensorTests
 from ..mixins.climate import TargetTemperatureTests
 from ..mixins.light import BasicLightTests
 from ..mixins.lock import BasicLockTests
-from ..mixins.number import BasicNumberTests
 from ..mixins.sensor import BasicSensorTests
 from .base_device_tests import TuyaDeviceTestCase
 
@@ -37,7 +35,6 @@ class TestGoldairHeater(
     BasicBinarySensorTests,
     BasicLightTests,
     BasicLockTests,
-    BasicNumberTests,
     BasicSensorTests,
     TargetTemperatureTests,
     TuyaDeviceTestCase,
@@ -55,14 +52,6 @@ class TestGoldairHeater(
         )
         self.setUpBasicLight(LIGHT_DPS, self.entities.get("light_display"))
         self.setUpBasicLock(LOCK_DPS, self.entities.get("lock_child_lock"))
-        self.setUpBasicNumber(
-            TIMER_DPS,
-            self.entities.get("number_timer"),
-            max=1440,
-            step=60,
-            device_class=NumberDeviceClass.DURATION,
-            unit=UnitOfTime.MINUTES,
-        )
         self.setUpBasicSensor(
             POWERLEVEL_DPS,
             self.entities.get("sensor_power_level"),
@@ -81,7 +70,6 @@ class TestGoldairHeater(
                 "binary_sensor_problem",
                 "light_display",
                 "lock_child_lock",
-                "number_timer",
                 "sensor_power_level",
                 "time_timer",
             ]

+ 0 - 23
tests/devices/test_grid_connect_double_power_point.py

@@ -1,6 +1,5 @@
 """Tests for the switch entity."""
 
-from homeassistant.components.number import NumberDeviceClass
 from homeassistant.components.sensor import SensorDeviceClass
 from homeassistant.components.switch import SwitchDeviceClass
 from homeassistant.const import (
@@ -13,7 +12,6 @@ from homeassistant.const import (
 
 from ..const import GRIDCONNECT_2SOCKET_PAYLOAD
 from ..mixins.lock import BasicLockTests
-from ..mixins.number import MultiNumberTests
 from ..mixins.select import BasicSelectTests
 from ..mixins.sensor import MultiSensorTests
 from ..mixins.switch import MultiSwitchTests
@@ -40,7 +38,6 @@ MASTER_DPS = "101"
 class TestGridConnectDoubleSwitch(
     BasicLockTests,
     BasicSelectTests,
-    MultiNumberTests,
     MultiSensorTests,
     MultiSwitchTests,
     TuyaDeviceTestCase,
@@ -117,29 +114,9 @@ class TestGridConnectDoubleSwitch(
                 },
             ]
         )
-        self.setUpMultiNumber(
-            [
-                {
-                    "name": "number_timer_1",
-                    "dps": COUNTDOWN1_DPS,
-                    "max": 86400,
-                    "device_class": NumberDeviceClass.DURATION,
-                    "unit": UnitOfTime.SECONDS,
-                },
-                {
-                    "name": "number_timer_2",
-                    "dps": COUNTDOWN2_DPS,
-                    "max": 86400,
-                    "device_class": NumberDeviceClass.DURATION,
-                    "unit": UnitOfTime.SECONDS,
-                },
-            ]
-        )
         self.mark_secondary(
             [
                 "lock_child_lock",
-                "number_timer_1",
-                "number_timer_2",
                 "select_initial_state",
                 "switch_master",
                 "sensor_energy",

+ 0 - 52
tests/devices/test_logicom_powerstrip.py

@@ -1,11 +1,9 @@
 """Tests for the Logicom Strippy 4-way+USB powerstrip."""
 
-from homeassistant.components.number import NumberDeviceClass
 from homeassistant.components.switch import SwitchDeviceClass
 from homeassistant.const import UnitOfTime
 
 from ..const import LOGICOM_STRIPPY_PAYLOAD
-from ..mixins.number import MultiNumberTests
 from ..mixins.switch import MultiSwitchTests
 from .base_device_tests import TuyaDeviceTestCase
 
@@ -22,7 +20,6 @@ TIMERUSB_DPS = "13"
 
 
 class TestLogicomPowerstrip(
-    MultiNumberTests,
     MultiSwitchTests,
     TuyaDeviceTestCase,
 ):
@@ -59,57 +56,8 @@ class TestLogicomPowerstrip(
                 },
             ]
         )
-        self.setUpMultiNumber(
-            [
-                {
-                    "dps": TIMER1_DPS,
-                    "name": "number_timer_1",
-                    "max": 1440,
-                    "scale": 60,
-                    "device_class": NumberDeviceClass.DURATION,
-                    "unit": UnitOfTime.MINUTES,
-                },
-                {
-                    "dps": TIMER2_DPS,
-                    "name": "number_timer_2",
-                    "max": 1440,
-                    "scale": 60,
-                    "device_class": NumberDeviceClass.DURATION,
-                    "unit": UnitOfTime.MINUTES,
-                },
-                {
-                    "dps": TIMER3_DPS,
-                    "name": "number_timer_3",
-                    "max": 1440,
-                    "scale": 60,
-                    "device_class": NumberDeviceClass.DURATION,
-                    "unit": UnitOfTime.MINUTES,
-                },
-                {
-                    "dps": TIMER4_DPS,
-                    "name": "number_timer_4",
-                    "max": 1440,
-                    "scale": 60,
-                    "device_class": NumberDeviceClass.DURATION,
-                    "unit": UnitOfTime.MINUTES,
-                },
-                {
-                    "dps": TIMERUSB_DPS,
-                    "name": "number_usb_timer",
-                    "max": 1440,
-                    "scale": 60,
-                    "device_class": NumberDeviceClass.DURATION,
-                    "unit": UnitOfTime.MINUTES,
-                },
-            ]
-        )
         self.mark_secondary(
             [
-                "number_timer_1",
-                "number_timer_2",
-                "number_timer_3",
-                "number_timer_4",
-                "number_usb_timer",
                 "time_timer_1",
                 "time_timer_2",
                 "time_timer_3",

+ 0 - 12
tests/devices/test_moes_rgb_socket.py

@@ -5,7 +5,6 @@ from homeassistant.components.light import (
     ColorMode,
     LightEntityFeature,
 )
-from homeassistant.components.number import NumberDeviceClass
 from homeassistant.components.sensor import SensorDeviceClass
 from homeassistant.components.switch import SwitchDeviceClass
 from homeassistant.const import (
@@ -17,7 +16,6 @@ from homeassistant.const import (
 
 from ..const import MOES_RGB_SOCKET_PAYLOAD
 from ..helpers import assert_device_properties_set
-from ..mixins.number import BasicNumberTests
 from ..mixins.sensor import MultiSensorTests
 from ..mixins.switch import BasicSwitchTests
 from .base_device_tests import TuyaDeviceTestCase
@@ -40,7 +38,6 @@ VOLTAGE_DPS = "106"
 
 
 class TestMoesRGBSocket(
-    BasicNumberTests,
     MultiSensorTests,
     BasicSwitchTests,
     TuyaDeviceTestCase,
@@ -58,14 +55,6 @@ class TestMoesRGBSocket(
             power_dps=POWER_DPS,
             power_scale=10,
         )
-        self.setUpBasicNumber(
-            TIMER_DPS,
-            self.entities.get("number_timer"),
-            max=1440.0,
-            unit=UnitOfTime.MINUTES,
-            device_class=NumberDeviceClass.DURATION,
-            scale=60,
-        )
         self.setUpMultiSensors(
             [
                 {
@@ -95,7 +84,6 @@ class TestMoesRGBSocket(
         )
         self.mark_secondary(
             [
-                "number_timer",
                 "sensor_current",
                 "sensor_power",
                 "sensor_voltage",

+ 4 - 11
tests/devices/test_nedis_htpl20f_heater.py

@@ -5,14 +5,12 @@ from homeassistant.components.climate.const import (
     ClimateEntityFeature,
     HVACMode,
 )
-from homeassistant.components.number import NumberDeviceClass
 from homeassistant.const import UnitOfTemperature, UnitOfTime
 
 from ..const import NEDIS_HTPL20F_PAYLOAD
 from ..helpers import assert_device_properties_set
 from ..mixins.climate import TargetTemperatureTests
 from ..mixins.lock import BasicLockTests
-from ..mixins.number import BasicNumberTests
 from .base_device_tests import TuyaDeviceTestCase
 
 HVACMODE_DPS = "1"
@@ -26,7 +24,9 @@ UNKNOWN101_DPS = "101"
 
 
 class TestNedisHtpl20fHeater(
-    BasicLockTests, BasicNumberTests, TargetTemperatureTests, TuyaDeviceTestCase
+    BasicLockTests,
+    TargetTemperatureTests,
+    TuyaDeviceTestCase,
 ):
     __test__ = True
 
@@ -43,14 +43,7 @@ class TestNedisHtpl20fHeater(
             LOCK_DPS,
             self.entities.get("lock_child_lock"),
         )
-        self.setUpBasicNumber(
-            TIMER_DPS,
-            self.entities.get("number_timer"),
-            max=1440,
-            device_class=NumberDeviceClass.DURATION,
-            unit=UnitOfTime.MINUTES,
-        )
-        self.mark_secondary(["lock_child_lock", "number_timer", "time_timer"])
+        self.mark_secondary(["lock_child_lock", "time_timer"])
 
     def test_supported_features(self):
         self.assertEqual(

+ 0 - 16
tests/devices/test_qoto_03_sprinkler.py

@@ -1,14 +1,12 @@
 """Tests for the Quto 03 Sprinkler."""
 
 from homeassistant.components.binary_sensor import BinarySensorDeviceClass
-from homeassistant.components.number import NumberDeviceClass
 from homeassistant.components.valve import ValveDeviceClass, ValveEntityFeature
 from homeassistant.const import UnitOfTime
 
 from ..const import QOTO_SPRINKLER_PAYLOAD
 from ..helpers import assert_device_properties_set
 from ..mixins.binary_sensor import BasicBinarySensorTests
-from ..mixins.number import MultiNumberTests
 from ..mixins.sensor import MultiSensorTests
 from .base_device_tests import TuyaDeviceTestCase
 
@@ -21,7 +19,6 @@ ERROR_DPS = "108"
 
 class TestQotoSprinkler(
     BasicBinarySensorTests,
-    MultiNumberTests,
     MultiSensorTests,
     TuyaDeviceTestCase,
 ):
@@ -36,17 +33,6 @@ class TestQotoSprinkler(
             device_class=BinarySensorDeviceClass.PROBLEM,
             testdata=(1, 0),
         )
-        self.setUpMultiNumber(
-            [
-                {
-                    "name": "number_timer",
-                    "dps": TIMER_DPS,
-                    "max": 86399,
-                    "device_class": NumberDeviceClass.DURATION,
-                    "unit": UnitOfTime.SECONDS,
-                },
-            ]
-        )
         self.setUpMultiSensors(
             [
                 {
@@ -64,9 +50,7 @@ class TestQotoSprinkler(
         )
         self.mark_secondary(
             [
-                "number",
                 "binary_sensor_problem",
-                "number_timer",
                 "sensor_open",
                 "sensor_time_remaining",
                 "time_timer",

+ 1 - 12
tests/devices/test_rgbcw_lightbulb.py

@@ -3,12 +3,10 @@ from homeassistant.components.light import (
     ColorMode,
     LightEntityFeature,
 )
-from homeassistant.components.number import NumberDeviceClass
 from homeassistant.const import UnitOfTime
 
 from ..const import RGBCW_LIGHTBULB_PAYLOAD
 from ..helpers import assert_device_properties_set
-from ..mixins.number import BasicNumberTests
 from ..mixins.text import TEXT_PATTERN_HEX, BasicTextTests
 from .base_device_tests import TuyaDeviceTestCase
 
@@ -21,21 +19,13 @@ SCENE_DPS = "25"
 TIMER_DPS = "26"
 
 
-class TestRGBCWLightbulb(BasicNumberTests, BasicTextTests, TuyaDeviceTestCase):
+class TestRGBCWLightbulb(BasicTextTests, TuyaDeviceTestCase):
     __test__ = True
 
     def setUp(self):
         self.setUpForConfig("rgbcw_lightbulb.yaml", RGBCW_LIGHTBULB_PAYLOAD)
         self.subject = self.entities.get("light")
 
-        self.setUpBasicNumber(
-            TIMER_DPS,
-            self.entities.get("number_timer"),
-            max=1440.0,
-            unit=UnitOfTime.MINUTES,
-            device_class=NumberDeviceClass.DURATION,
-            scale=60,
-        )
         self.setUpBasicText(
             SCENE_DPS,
             self.entities.get("text_scene"),
@@ -43,7 +33,6 @@ class TestRGBCWLightbulb(BasicNumberTests, BasicTextTests, TuyaDeviceTestCase):
         )
         self.mark_secondary(
             [
-                "number_timer",
                 "select_scene",
                 "text_scene",
                 "time_timer",

+ 2 - 12
tests/devices/test_simple_switch_with_timer.py

@@ -1,10 +1,8 @@
 """Tests for a simple switch with timer"""
 
-from homeassistant.components.number import NumberDeviceClass
 from homeassistant.const import UnitOfTime
 
 from ..const import TIMED_SOCKET_PAYLOAD
-from ..mixins.number import BasicNumberTests
 from ..mixins.switch import SwitchableTests
 from .base_device_tests import TuyaDeviceTestCase
 
@@ -12,22 +10,14 @@ SWITCH_DPS = "1"
 TIMER_DPS = "11"
 
 
-class TestTimedSwitch(BasicNumberTests, SwitchableTests, TuyaDeviceTestCase):
+class TestTimedSwitch(SwitchableTests, TuyaDeviceTestCase):
     __test__ = True
 
     def setUp(self):
         self.setUpForConfig("simple_switch_timer.yaml", TIMED_SOCKET_PAYLOAD)
         self.subject = self.entities.get("switch")
         self.setUpSwitchable(SWITCH_DPS, self.subject)
-        self.setUpBasicNumber(
-            TIMER_DPS,
-            self.entities.get("number_timer"),
-            max=1440,
-            scale=60,
-            unit=UnitOfTime.MINUTES,
-            device_class=NumberDeviceClass.DURATION,
-        )
-        self.mark_secondary(["number_timer", "time_timer"])
+        self.mark_secondary(["time_timer"])
 
     def test_extra_state_attributes_set(self):
         self.assertDictEqual(

+ 1 - 12
tests/devices/test_simple_switch_with_timerv2.py

@@ -1,10 +1,8 @@
 """Tests for a simple switch with timer"""
 
-from homeassistant.components.number import NumberDeviceClass
 from homeassistant.const import UnitOfTime
 
 from ..const import TIMED_SOCKETV2_PAYLOAD
-from ..mixins.number import BasicNumberTests
 from ..mixins.switch import SwitchableTests
 from .base_device_tests import TuyaDeviceTestCase
 
@@ -13,24 +11,15 @@ TIMER_DPS = "9"
 INITIAL_STATE_DPS = "38"
 
 
-class TestTimedSwitch(BasicNumberTests, SwitchableTests, TuyaDeviceTestCase):
+class TestTimedSwitch(SwitchableTests, TuyaDeviceTestCase):
     __test__ = True
 
     def setUp(self):
         self.setUpForConfig("simple_switch_timerv2.yaml", TIMED_SOCKETV2_PAYLOAD)
         self.subject = self.entities.get("switch")
         self.setUpSwitchable(SWITCH_DPS, self.subject)
-        self.setUpBasicNumber(
-            TIMER_DPS,
-            self.entities.get("number_timer"),
-            max=1440,
-            scale=60,
-            device_class=NumberDeviceClass.DURATION,
-            unit=UnitOfTime.MINUTES,
-        )
         self.mark_secondary(
             [
-                "number_timer",
                 "select_initial_state",
                 "time_timer",
             ]

+ 2 - 12
tests/devices/test_smartplug_encoded.py

@@ -1,11 +1,9 @@
 """Tests for the switch entity."""
 
-from homeassistant.components.number import NumberDeviceClass
 from homeassistant.components.switch import SwitchDeviceClass
 from homeassistant.const import UnitOfTime
 
 from ..const import SMARTPLUG_ENCODED_PAYLOAD
-from ..mixins.number import BasicNumberTests
 from ..mixins.switch import SwitchableTests
 from .base_device_tests import TuyaDeviceTestCase
 
@@ -16,22 +14,14 @@ CIRCULATE_DPS = "102"
 SCHEDULE_DPS = "103"
 
 
-class TestSwitchEncoded(BasicNumberTests, SwitchableTests, TuyaDeviceTestCase):
+class TestSwitchEncoded(SwitchableTests, TuyaDeviceTestCase):
     __test__ = True
 
     def setUp(self):
         self.setUpForConfig("smartplug_encoded.yaml", SMARTPLUG_ENCODED_PAYLOAD)
         self.subject = self.entities.get("switch_outlet")
         self.setUpSwitchable(SWITCH_DPS, self.subject)
-        self.setUpBasicNumber(
-            TIMER_DPS,
-            self.entities.get("number_timer"),
-            max=1440.0,
-            unit=UnitOfTime.MINUTES,
-            device_class=NumberDeviceClass.DURATION,
-            scale=60,
-        )
-        self.mark_secondary(["number_timer", "time_timer"])
+        self.mark_secondary(["time_timer"])
 
     def test_device_class_is_outlet(self):
         self.assertEqual(self.subject.device_class, SwitchDeviceClass.OUTLET)

+ 3 - 12
tests/devices/test_smartplugv2.py

@@ -1,6 +1,5 @@
 """Tests for the switch entity."""
 
-from homeassistant.components.number import NumberDeviceClass
 from homeassistant.components.sensor import SensorDeviceClass
 from homeassistant.components.switch import SwitchDeviceClass
 from homeassistant.const import (
@@ -11,7 +10,6 @@ from homeassistant.const import (
 )
 
 from ..const import KOGAN_SOCKET_PAYLOAD2
-from ..mixins.number import BasicNumberTests
 from ..mixins.sensor import MultiSensorTests
 from ..mixins.switch import SwitchableTests
 from .base_device_tests import TuyaDeviceTestCase
@@ -24,7 +22,9 @@ VOLTAGE_DPS = "20"
 
 
 class TestSwitchV2(
-    BasicNumberTests, MultiSensorTests, SwitchableTests, TuyaDeviceTestCase
+    MultiSensorTests,
+    SwitchableTests,
+    TuyaDeviceTestCase,
 ):
     __test__ = True
 
@@ -32,14 +32,6 @@ class TestSwitchV2(
         self.setUpForConfig("smartplugv2.yaml", KOGAN_SOCKET_PAYLOAD2)
         self.subject = self.entities.get("switch_outlet")
         self.setUpSwitchable(SWITCH_DPS, self.subject)
-        self.setUpBasicNumber(
-            TIMER_DPS,
-            self.entities.get("number_timer"),
-            max=1440.0,
-            unit=UnitOfTime.MINUTES,
-            device_class=NumberDeviceClass.DURATION,
-            scale=60,
-        )
         self.setUpMultiSensors(
             [
                 {
@@ -70,7 +62,6 @@ class TestSwitchV2(
         self.mark_secondary(
             [
                 "binary_sensor_problem",
-                "number_timer",
                 "sensor_current",
                 "sensor_power",
                 "sensor_voltage",

+ 0 - 12
tests/devices/test_smartplugv2_energy.py

@@ -1,7 +1,6 @@
 """Tests for the switch entity."""
 
 from homeassistant.components.binary_sensor import BinarySensorDeviceClass
-from homeassistant.components.number import NumberDeviceClass
 from homeassistant.components.sensor import SensorDeviceClass
 from homeassistant.components.switch import SwitchDeviceClass
 from homeassistant.const import (
@@ -15,7 +14,6 @@ from homeassistant.const import (
 from ..const import SMARTSWITCH_ENERGY_PAYLOAD
 from ..helpers import assert_device_properties_set
 from ..mixins.binary_sensor import BasicBinarySensorTests
-from ..mixins.number import BasicNumberTests
 from ..mixins.select import BasicSelectTests
 from ..mixins.sensor import MultiSensorTests
 from ..mixins.switch import MultiSwitchTests
@@ -44,7 +42,6 @@ ALT_OVERCHARGE_DPS = "51"
 
 class TestSwitchV2Energy(
     BasicBinarySensorTests,
-    BasicNumberTests,
     BasicSelectTests,
     MultiSensorTests,
     MultiSwitchTests,
@@ -73,14 +70,6 @@ class TestSwitchV2Energy(
             device_class=BinarySensorDeviceClass.PROBLEM,
             testdata=(1, 0),
         )
-        self.setUpBasicNumber(
-            TIMER_DPS,
-            self.entities.get("number_timer"),
-            max=1440.0,
-            unit=UnitOfTime.MINUTES,
-            device_class=NumberDeviceClass.DURATION,
-            scale=60,
-        )
         self.setUpBasicSelect(
             INITIAL_DPS,
             self.entities.get("select_initial_state"),
@@ -127,7 +116,6 @@ class TestSwitchV2Energy(
             [
                 "binary_sensor_problem",
                 "lock_child_lock",
-                "number_timer",
                 "select_initial_state",
                 "select_light_mode",
                 "sensor_current",

+ 2 - 12
tests/devices/test_tmwf02_fan.py

@@ -1,10 +1,8 @@
 from homeassistant.components.fan import FanEntityFeature
-from homeassistant.components.number import NumberDeviceClass
 from homeassistant.const import UnitOfTime
 
 from ..const import TMWF02_FAN_PAYLOAD
 from ..helpers import assert_device_properties_set
-from ..mixins.number import BasicNumberTests
 from ..mixins.switch import SwitchableTests
 from .base_device_tests import TuyaDeviceTestCase
 
@@ -14,22 +12,14 @@ LEVEL_DPS = "3"
 SPEED_DPS = "4"
 
 
-class TestTMWF02Fan(BasicNumberTests, SwitchableTests, TuyaDeviceTestCase):
+class TestTMWF02Fan(SwitchableTests, TuyaDeviceTestCase):
     __test__ = True
 
     def setUp(self):
         self.setUpForConfig("tmwf02_fan.yaml", TMWF02_FAN_PAYLOAD)
         self.subject = self.entities["fan"]
         self.setUpSwitchable(SWITCH_DPS, self.subject)
-        self.setUpBasicNumber(
-            TIMER_DPS,
-            self.entities.get("number_timer"),
-            max=1440,
-            scale=60,
-            device_class=NumberDeviceClass.DURATION,
-            unit=UnitOfTime.MINUTES,
-        )
-        self.mark_secondary(["number_timer", "time_timer"])
+        self.mark_secondary(["time_timer"])
 
     def test_supported_features(self):
         self.assertEqual(

+ 2 - 12
tests/devices/test_treatlife_ds02f.py

@@ -1,10 +1,8 @@
 from homeassistant.components.fan import FanEntityFeature
-from homeassistant.components.number import NumberDeviceClass
 from homeassistant.const import UnitOfTime
 
 from ..const import TREATLIFE_DS02F_PAYLOAD
 from ..helpers import assert_device_properties_set
-from ..mixins.number import BasicNumberTests
 from ..mixins.switch import SwitchableTests
 from .base_device_tests import TuyaDeviceTestCase
 
@@ -13,22 +11,14 @@ TIMER_DPS = "2"
 SPEED_DPS = "3"
 
 
-class TestTreatlifeFan(SwitchableTests, BasicNumberTests, TuyaDeviceTestCase):
+class TestTreatlifeFan(SwitchableTests, TuyaDeviceTestCase):
     __test__ = True
 
     def setUp(self):
         self.setUpForConfig("treatlife_ds02_fan.yaml", TREATLIFE_DS02F_PAYLOAD)
         self.subject = self.entities["fan"]
         self.setUpSwitchable(SWITCH_DPS, self.subject)
-        self.setUpBasicNumber(
-            TIMER_DPS,
-            self.entities.get("number_timer"),
-            max=1440.0,
-            scale=60,
-            device_class=NumberDeviceClass.DURATION,
-            unit=UnitOfTime.MINUTES,
-        )
-        self.mark_secondary(["number_timer", "time_timer"])
+        self.mark_secondary(["time_timer"])
 
     def test_supported_features(self):
         self.assertEqual(

+ 0 - 43
tests/devices/test_woox_r4028_powerstrip.py

@@ -1,11 +1,9 @@
 """Tests for the Woox R4028 powerstrip."""
 
-from homeassistant.components.number import NumberDeviceClass
 from homeassistant.components.switch import SwitchDeviceClass
 from homeassistant.const import UnitOfTime
 
 from ..const import WOOX_R4028_SOCKET_PAYLOAD
-from ..mixins.number import MultiNumberTests
 from ..mixins.switch import MultiSwitchTests
 from .base_device_tests import TuyaDeviceTestCase
 
@@ -20,7 +18,6 @@ TIMERUSB_DPS = "105"
 
 
 class TestWooxR4028Powerstrip(
-    MultiNumberTests,
     MultiSwitchTests,
     TuyaDeviceTestCase,
 ):
@@ -52,48 +49,8 @@ class TestWooxR4028Powerstrip(
                 },
             ]
         )
-        self.setUpMultiNumber(
-            [
-                {
-                    "dps": TIMER1_DPS,
-                    "name": "number_timer_1",
-                    "max": 1440,
-                    "scale": 60,
-                    "unit": UnitOfTime.MINUTES,
-                    "device_class": NumberDeviceClass.DURATION,
-                },
-                {
-                    "dps": TIMER2_DPS,
-                    "name": "number_timer_2",
-                    "max": 1440,
-                    "scale": 60,
-                    "unit": UnitOfTime.MINUTES,
-                    "device_class": NumberDeviceClass.DURATION,
-                },
-                {
-                    "dps": TIMER3_DPS,
-                    "name": "number_timer_3",
-                    "max": 1440,
-                    "scale": 60,
-                    "unit": UnitOfTime.MINUTES,
-                    "device_class": NumberDeviceClass.DURATION,
-                },
-                {
-                    "dps": TIMERUSB_DPS,
-                    "name": "number_usb_timer",
-                    "max": 1440,
-                    "scale": 60,
-                    "unit": UnitOfTime.MINUTES,
-                    "device_class": NumberDeviceClass.DURATION,
-                },
-            ]
-        )
         self.mark_secondary(
             [
-                "number_timer_1",
-                "number_timer_2",
-                "number_timer_3",
-                "number_usb_timer",
                 "time_timer_1",
                 "time_timer_2",
                 "time_timer_3",

+ 1 - 1
tests/test_number.py

@@ -20,7 +20,7 @@ async def test_init_entry(hass):
     entry = MockConfigEntry(
         domain=DOMAIN,
         data={
-            CONF_TYPE: "goldair_gpph_heater",
+            CONF_TYPE: "anko_fan",
             CONF_DEVICE_ID: "dummy",
             CONF_PROTOCOL_VERSION: "auto",
         },