File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -162,6 +162,7 @@ class YXCleanType(RoborockModeEnum):
162162 VAC_AND_MOP = "vac_and_mop" , 1 # bothwork
163163 VACUUM = "vacuum" , 2 # onlysweep
164164 MOP = "mop" , 3 # onlymop
165+ CUSTOMIZED = "customized" , 4 # custom mode
165166
166167
167168class YXDeviceState (RoborockModeEnum ):
Original file line number Diff line number Diff line change @@ -97,9 +97,15 @@ def test_homedata_product_unknown_category():
9797 ("vac_and_mop" , YXCleanType .VAC_AND_MOP ),
9898 ("vacuum" , YXCleanType .VACUUM ),
9999 ("mop" , YXCleanType .MOP ),
100+ ("customized" , YXCleanType .CUSTOMIZED ),
100101 ],
101102)
102103def test_yx_clean_type_from_value_readable_values (readable_value : str , expected_clean_type : YXCleanType ) -> None :
103104 """Test YXCleanType accepts canonical readable values."""
104105 assert YXCleanType .from_value (readable_value ) is expected_clean_type
105106 assert expected_clean_type .value == readable_value
107+
108+
109+ def test_yx_clean_type_from_code_customized () -> None :
110+ """Test YXCleanType accepts custom mode code used by Q10 status updates."""
111+ assert YXCleanType .from_code (4 ) is YXCleanType .CUSTOMIZED
You can’t perform that action at this time.
0 commit comments