We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f22a5b1 commit db19931Copy full SHA for db19931
tests/python/bindings_test_getdata.py
@@ -45,5 +45,11 @@ def testGettingItems(self):
45
self.assertRaises(ValueError, lambda: p.get_items(labels[0]))
46
47
# After adding them, all labels should be retrievable
48
- returned_items = p.get_items(labels)
49
- self.assertTrue((data == returned_items).all())
+ returned_items_np = p.get_items(labels)
+ self.assertTrue((data == returned_items_np).all())
50
+
51
+ # check returned type of get_items
52
+ self.assertTrue(isinstance(returned_items_np, np.ndarray))
53
+ returned_items_list = p.get_items(labels, return_type="list")
54
+ self.assertTrue(isinstance(returned_items_list, list))
55
+ self.assertTrue(isinstance(returned_items_list[0], list))
0 commit comments