diff --git a/django/curator/tests/test_spam.py b/django/curator/tests/test_spam.py index 952864c57..d722aefde 100644 --- 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):