|
5 | 5 | import pytest |
6 | 6 |
|
7 | 7 | from roborock import HomeDataProduct, RoborockCategory |
8 | | -from roborock.data.b01_q10.b01_q10_code_mappings import B01_Q10_DP |
| 8 | +from roborock.data.b01_q10.b01_q10_code_mappings import B01_Q10_DP, YXCleanType |
9 | 9 |
|
10 | 10 |
|
11 | 11 | def test_from_code() -> None: |
@@ -52,6 +52,22 @@ def test_invalid_from_value() -> None: |
52 | 52 | B01_Q10_DP.from_value("invalid_value") |
53 | 53 |
|
54 | 54 |
|
| 55 | +@pytest.mark.parametrize( |
| 56 | + ("raw_value", "expected"), |
| 57 | + [ |
| 58 | + ("bothwork", YXCleanType.VAC_AND_MOP), |
| 59 | + ("onlysweep", YXCleanType.VACUUM), |
| 60 | + ("onlymop", YXCleanType.MOP), |
| 61 | + ("BothWork", YXCleanType.VAC_AND_MOP), |
| 62 | + ("ONLYSWEEP", YXCleanType.VACUUM), |
| 63 | + ("OnlyMop", YXCleanType.MOP), |
| 64 | + ], |
| 65 | +) |
| 66 | +def test_yxcleantype_from_value_legacy_aliases(raw_value: str, expected: YXCleanType) -> None: |
| 67 | + """Ensure legacy clean type strings resolve to canonical enum members.""" |
| 68 | + assert YXCleanType.from_value(raw_value) is expected |
| 69 | + |
| 70 | + |
55 | 71 | @pytest.mark.parametrize( |
56 | 72 | "input, expected", |
57 | 73 | [ |
|
0 commit comments