Fix tests

nmslib · Aug 12, 2023 · f22a5b1 · f22a5b1
1 parent 4f7b192
commit f22a5b1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/python/bindings_test_getdata.py b/tests/python/bindings_test_getdata.py
@@ -46,4 +46,4 @@ def testGettingItems(self):
 
         # After adding them, all labels should be retrievable
         returned_items = p.get_items(labels)
-        self.assertSequenceEqual(data.tolist(), returned_items)
+        self.assertTrue((data == returned_items).all())
diff --git a/tests/python/bindings_test_replace.py b/tests/python/bindings_test_replace.py
@@ -94,10 +94,10 @@ def testRandomSelf(self):
         remaining_data = comb_data[remaining_labels_list]
 
         returned_items = hnsw_index.get_items(remaining_labels_list)
-        self.assertSequenceEqual(remaining_data.tolist(), returned_items)
+        self.assertTrue((remaining_data == returned_items).all())
 
         returned_items = hnsw_index.get_items(labels3_tr)
-        self.assertSequenceEqual(data3_tr.tolist(), returned_items)
+        self.assertTrue((data3_tr == returned_items).all())
 
         # Check index serialization
         # Delete batch 3