fix: replace assertListEqual with assertTrue for sets in the spam det…

…ection testcases
comses · Nov 28, 2023 · ad9a8ac · ad9a8ac
1 parent b99c8b0
commit ad9a8ac
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/django/curator/tests/test_spam.py b/django/curator/tests/test_spam.py
@@ -129,8 +129,8 @@ def test_get_untrained_df__dataset_loaded(self):
         self.assertTrue(
             set(self.processor.column_names).issubset(set(df.columns.unique()))
         )
-        self.assertListEqual(user_ids, list(df["user_id"].values))
-
+        # self.assertListEqual(user_ids, list(df["user_id"].values))
+        self.assertTrue(set(user_ids) == set(list(df["user_id"].values)))
         self.delete_labels(user_ids)
 
     def test_get_untrained_df__labels_updated(self):
@@ -147,8 +147,8 @@ def test_get_untrained_df__labels_updated(self):
         self.assertTrue(
             set(self.processor.column_names).issubset(set(df.columns.unique()))
         )
-        self.assertListEqual(existing_users, list(df["user_id"].values))
-
+        # self.assertListEqual(existing_users, list(df["user_id"].values))
+        self.assertTrue(set(existing_users) == set(list(df["user_id"].values)))
         self.delete_labels(existing_users)
 
     def test_get_untrained_df__no_labels_updated(self):