Skip to content

Commit e48caed

Browse files
author
Balaji Veeramani
committed
Update test_deserializers.py
1 parent eda4633 commit e48caed

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

tests/unit/sagemaker/test_deserializers.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -151,6 +151,18 @@ def test_numpy_deserializer_from_npy_object_array(numpy_deserializer):
151151
assert np.array_equal(array, result)
152152

153153

154+
def test_numpy_deserializer_from_npy_object_array_with_allow_pickle_false():
155+
numpy_deserializer = NumpyDeserializer(allow_pickle=False)
156+
157+
array = np.array([{"a": "", "b": ""}, {"c": "", "d": ""}])
158+
stream = io.BytesIO()
159+
np.save(stream, array)
160+
stream.seek(0)
161+
162+
with pytest.raises(ValueError):
163+
numpy_deserializer.deserialize(stream, "application/x-npy")
164+
165+
154166
@pytest.fixture
155167
def json_deserializer():
156168
return JSONDeserializer()

0 commit comments

Comments
 (0)