Skip to content

Commit cd053fc

Browse files
committed
Explicitly assert that data in BinaryVector.as_vector(return_numpy=True) isinstance of np.ndarray
1 parent b8d9719 commit cd053fc

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

test/test_bson.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -890,6 +890,7 @@ def test_vector_from_numpy(self):
890890
# as_numpy_vector
891891
vector_np = binary_vector_int8.as_vector(return_numpy=True)
892892
assert isinstance(vector_np, BinaryVector)
893+
assert isinstance(vector_np.data, np.ndarray)
893894
assert np.all(vector.data == arr)
894895
# PACKED_BIT
895896
binary_vector_uint8 = Binary.from_vector(arr, BinaryVectorDtype.PACKED_BIT)
@@ -900,6 +901,7 @@ def test_vector_from_numpy(self):
900901
# as_numpy_vector
901902
vector_np = binary_vector_uint8.as_vector(return_numpy=True)
902903
assert isinstance(vector_np, BinaryVector)
904+
assert isinstance(vector_np.data, np.ndarray)
903905
assert np.all(vector_np.data == arr)
904906
# FLOAT32
905907
binary_vector_float32 = Binary.from_vector(arr, BinaryVectorDtype.FLOAT32)
@@ -910,6 +912,7 @@ def test_vector_from_numpy(self):
910912
# as_numpy_vector
911913
vector_np = binary_vector_float32.as_vector(return_numpy=True)
912914
assert isinstance(vector_np, BinaryVector)
915+
assert isinstance(vector_np.data, np.ndarray)
913916
assert np.all(vector_np.data == arr)
914917

915918
# Invalid cases

0 commit comments

Comments
 (0)