Skip to content

Commit e791aff

Browse files
committed
chore: address test comments
1 parent 38076f7 commit e791aff

File tree

1 file changed

+5
-8
lines changed

1 file changed

+5
-8
lines changed

tests/devices/traits/v1/test_wash_towel_mode.py

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def wash_towel_mode_trait(
3434
],
3535
)
3636
async 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
)
8484
async 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
)
104104
async 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
)
121121
async 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
)
182182
async 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

Comments
 (0)