File tree 2 files changed +3
-3
lines changed 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -46,4 +46,4 @@ def testGettingItems(self):
46
46
47
47
# After adding them, all labels should be retrievable
48
48
returned_items = p .get_items (labels )
49
- self .assertSequenceEqual ( data . tolist (), returned_items )
49
+ self .assertTrue (( data == returned_items ). all () )
Original file line number Diff line number Diff line change @@ -94,10 +94,10 @@ def testRandomSelf(self):
94
94
remaining_data = comb_data [remaining_labels_list ]
95
95
96
96
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 () )
98
98
99
99
returned_items = hnsw_index .get_items (labels3_tr )
100
- self .assertSequenceEqual ( data3_tr . tolist (), returned_items )
100
+ self .assertTrue (( data3_tr == returned_items ). all () )
101
101
102
102
# Check index serialization
103
103
# Delete batch 3
You can’t perform that action at this time.
0 commit comments