Add type check for return results of get_items

nmslib · Aug 13, 2023 · db19931 · db19931
1 parent f22a5b1
commit db19931
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/tests/python/bindings_test_getdata.py b/tests/python/bindings_test_getdata.py
@@ -45,5 +45,11 @@ def testGettingItems(self):
         self.assertRaises(ValueError, lambda: p.get_items(labels[0]))
 
         # After adding them, all labels should be retrievable
-        returned_items = p.get_items(labels)
-        self.assertTrue((data == returned_items).all())
+        returned_items_np = p.get_items(labels)
+        self.assertTrue((data == returned_items_np).all())
+
+        # check returned type of get_items
+        self.assertTrue(isinstance(returned_items_np, np.ndarray))
+        returned_items_list = p.get_items(labels, return_type="list")
+        self.assertTrue(isinstance(returned_items_list, list))
+        self.assertTrue(isinstance(returned_items_list[0], list))