Skip to content

Commit f22a5b1

Browse files
committed
Fix tests
1 parent 4f7b192 commit f22a5b1

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

tests/python/bindings_test_getdata.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,4 +46,4 @@ def testGettingItems(self):
4646

4747
# After adding them, all labels should be retrievable
4848
returned_items = p.get_items(labels)
49-
self.assertSequenceEqual(data.tolist(), returned_items)
49+
self.assertTrue((data == returned_items).all())

tests/python/bindings_test_replace.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -94,10 +94,10 @@ def testRandomSelf(self):
9494
remaining_data = comb_data[remaining_labels_list]
9595

9696
returned_items = hnsw_index.get_items(remaining_labels_list)
97-
self.assertSequenceEqual(remaining_data.tolist(), returned_items)
97+
self.assertTrue((remaining_data == returned_items).all())
9898

9999
returned_items = hnsw_index.get_items(labels3_tr)
100-
self.assertSequenceEqual(data3_tr.tolist(), returned_items)
100+
self.assertTrue((data3_tr == returned_items).all())
101101

102102
# Check index serialization
103103
# Delete batch 3

0 commit comments

Comments
 (0)