File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -213,20 +213,20 @@ def test_filter_by_ids_reset_ids():
213213 instance = ArrayWithIds (** ARRAY_WITH_IDS_ARRAYS_OF_FLOAT_VALUES_CONFIG_NON_CONSECUTIVE )
214214 # Filter to keep only first and third elements (ids 2 and 6)
215215 instance .filter_by_ids ([2 , 6 ], reset_ids = True )
216-
216+
217217 # Values should be filtered correctly
218218 assert instance .values == [
219219 ARRAY_WITH_IDS_ARRAYS_OF_FLOAT_VALUES_CONFIG_NON_CONSECUTIVE ["values" ][0 ],
220220 ARRAY_WITH_IDS_ARRAYS_OF_FLOAT_VALUES_CONFIG_NON_CONSECUTIVE ["values" ][2 ],
221221 ]
222222 # IDs should be reset to consecutive integers starting from 0
223223 assert instance .ids == [0 , 1 ]
224-
224+
225225 # Test with invert=True
226226 instance = ArrayWithIds (** ARRAY_WITH_IDS_ARRAYS_OF_FLOAT_VALUES_CONFIG_NON_CONSECUTIVE )
227227 # Filter to exclude first element (id 2), keep second and third (ids 4, 6)
228228 instance .filter_by_ids ([2 ], invert = True , reset_ids = True )
229-
229+
230230 assert instance .values == [
231231 ARRAY_WITH_IDS_ARRAYS_OF_FLOAT_VALUES_CONFIG_NON_CONSECUTIVE ["values" ][1 ],
232232 ARRAY_WITH_IDS_ARRAYS_OF_FLOAT_VALUES_CONFIG_NON_CONSECUTIVE ["values" ][2 ],
You can’t perform that action at this time.
0 commit comments