Skip to content

Commit db19931

Browse files
committed
Add type check for return results of get_items
1 parent f22a5b1 commit db19931

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

tests/python/bindings_test_getdata.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,5 +45,11 @@ def testGettingItems(self):
4545
self.assertRaises(ValueError, lambda: p.get_items(labels[0]))
4646

4747
# After adding them, all labels should be retrievable
48-
returned_items = p.get_items(labels)
49-
self.assertTrue((data == returned_items).all())
48+
returned_items_np = p.get_items(labels)
49+
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

Comments
 (0)