@@ -34,7 +34,7 @@ def wash_towel_mode_trait(
3434 ],
3535)
3636async def test_wash_towel_mode_available (
37- wash_towel_mode : WashTowelModeTrait | None ,
37+ wash_towel_mode : WashTowelModeTrait ,
3838 mock_rpc_channel : AsyncMock ,
3939 dock_type_code : RoborockDockTypeCode ,
4040) -> None :
@@ -82,7 +82,7 @@ async def test_unsupported_wash_towel_mode(
8282 ],
8383)
8484async def test_set_wash_towel_mode (
85- wash_towel_mode : WashTowelModeTrait | None ,
85+ wash_towel_mode : WashTowelModeTrait ,
8686 mock_rpc_channel : AsyncMock ,
8787 wash_mode : WashTowelModes ,
8888 dock_type_code : RoborockDockTypeCode ,
@@ -102,7 +102,7 @@ async def test_set_wash_towel_mode(
102102 [(RoborockDockTypeCode .s8_dock )],
103103)
104104async def test_start_wash (
105- wash_towel_mode : WashTowelModeTrait | None ,
105+ wash_towel_mode : WashTowelModeTrait ,
106106 mock_rpc_channel : AsyncMock ,
107107 dock_type_code : RoborockDockTypeCode ,
108108) -> None :
@@ -119,7 +119,7 @@ async def test_start_wash(
119119 [(RoborockDockTypeCode .s8_dock )],
120120)
121121async def test_stop_wash (
122- wash_towel_mode : WashTowelModeTrait | None ,
122+ wash_towel_mode : WashTowelModeTrait ,
123123 mock_rpc_channel : AsyncMock ,
124124 dock_type_code : RoborockDockTypeCode ,
125125) -> None :
@@ -180,17 +180,14 @@ async def test_stop_wash(
180180 ],
181181)
182182async def test_wash_towel_mode_options (
183- wash_towel_mode : WashTowelModeTrait | None ,
183+ wash_towel_mode : WashTowelModeTrait ,
184184 dock_type_code : RoborockDockTypeCode ,
185185 is_super_deep_wash_supported : bool ,
186186 is_dirty_replenish_clean_supported : bool ,
187187 expected_modes : list [WashTowelModes ],
188188) -> None :
189189 """Test what modes are available based on device features."""
190190 assert wash_towel_mode is not None
191- # We need to clear the cached property to ensure it re-reads the features
192- if "wash_towel_mode_options" in wash_towel_mode .__dict__ :
193- del wash_towel_mode .__dict__ ["wash_towel_mode_options" ]
194191
195192 # Mock the device features
196193 assert wash_towel_mode .device_feature_trait is not None
0 commit comments