Skip to content

Commit 393c4e5

Browse files
committed
fix(q10): refactor test for canonical status names using a dictionary
1 parent 7fddf43 commit 393c4e5

File tree

1 file changed

+24
-17
lines changed

1 file changed

+24
-17
lines changed

tests/data/b01_q10/test_b01_q10_containers.py

Lines changed: 24 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -5,23 +5,30 @@
55

66
def test_q10_status_values_are_canonical() -> None:
77
"""Q10 status enum values should expose canonical names."""
8-
assert YXDeviceState.SLEEP_STATE.value == "sleeping"
9-
assert YXDeviceState.STANDBY_STATE.value == "standby"
10-
assert YXDeviceState.CLEANING_STATE.value == "cleaning"
11-
assert YXDeviceState.TO_CHARGE_STATE.value == "going_to_charge"
12-
assert YXDeviceState.REMOTEING_STATE.value == "remote_control"
13-
assert YXDeviceState.CHARGING_STATE.value == "charging"
14-
assert YXDeviceState.PAUSE_STATE.value == "paused"
15-
assert YXDeviceState.FAULT_STATE.value == "fault"
16-
assert YXDeviceState.UPGRADE_STATE.value == "updating"
17-
assert YXDeviceState.CREATING_MAP_STATE.value == "creating_map"
18-
assert YXDeviceState.MAP_SAVE_STATE.value == "saving_map"
19-
assert YXDeviceState.RE_LOCATION_STATE.value == "relocating"
20-
assert YXDeviceState.ROBOT_SWEEPING.value == "sweeping"
21-
assert YXDeviceState.ROBOT_MOPING.value == "mopping"
22-
assert YXDeviceState.ROBOT_SWEEP_AND_MOPING.value == "sweep_and_mop"
23-
assert YXDeviceState.ROBOT_TRANSITIONING.value == "transitioning"
24-
assert YXDeviceState.ROBOT_WAIT_CHARGE.value == "waiting_to_charge"
8+
expected_values = {
9+
YXDeviceState.UNKNOWN: "unknown",
10+
YXDeviceState.SLEEP_STATE: "sleeping",
11+
YXDeviceState.STANDBY_STATE: "standby",
12+
YXDeviceState.CLEANING_STATE: "cleaning",
13+
YXDeviceState.TO_CHARGE_STATE: "going_to_charge",
14+
YXDeviceState.REMOTEING_STATE: "remote_control",
15+
YXDeviceState.CHARGING_STATE: "charging",
16+
YXDeviceState.PAUSE_STATE: "paused",
17+
YXDeviceState.FAULT_STATE: "fault",
18+
YXDeviceState.UPGRADE_STATE: "updating",
19+
YXDeviceState.DUSTING: "dusting",
20+
YXDeviceState.CREATING_MAP_STATE: "creating_map",
21+
YXDeviceState.MAP_SAVE_STATE: "saving_map",
22+
YXDeviceState.RE_LOCATION_STATE: "relocating",
23+
YXDeviceState.ROBOT_SWEEPING: "sweeping",
24+
YXDeviceState.ROBOT_MOPING: "mopping",
25+
YXDeviceState.ROBOT_SWEEP_AND_MOPING: "sweep_and_mop",
26+
YXDeviceState.ROBOT_TRANSITIONING: "transitioning",
27+
YXDeviceState.ROBOT_WAIT_CHARGE: "waiting_to_charge",
28+
}
29+
30+
assert {state: state.value for state in expected_values} == expected_values
31+
assert all(not value.endswith("state") for value in expected_values.values())
2532

2633

2734
def test_q10_status_codes_map_to_canonical_values() -> None:

0 commit comments

Comments
 (0)