def test_numpy_deser_from_npy():
array = np.ones((2, 3))
stream = io.BytesIO()
np.save(stream, array)
stream.seek(0)
result = numpy_deserializer(stream)
assert np.array_equal(array, result)
def test_numpy_deser_from_npy_object_array():
array = np.array(["one", "two"])
stream = io.BytesIO()
np.save(stream, array)
stream.seek(0)
result = numpy_deserializer(stream)
assert np.array_equal(array, result)
def test_numpy_deser_from_npy_object_array():
array = np.array(['one', 'two'])
stream = io.BytesIO()
np.save(stream, array)
stream.seek(0)
result = numpy_deserializer(stream)
assert np.array_equal(array, result)
def test_numpy_deser_from_npy():
array = np.ones((2, 3))
stream = io.BytesIO()
np.save(stream, array)
stream.seek(0)
result = numpy_deserializer(stream)
assert np.array_equal(array, result)
def test_numpy_deser_from_json_ragged():
arr = numpy_deserializer(io.BytesIO(b"[[1,2,3],\n[4,5,6,7]]"),
"application/json")
assert np.array_equal(arr, np.array([[1, 2, 3], [4, 5, 6, 7]]))
def test_numpy_deser_from_csv_ragged():
with pytest.raises(ValueError) as error:
numpy_deserializer(io.BytesIO(b"1,2,3\n4,5,6,7"), "text/csv")
assert "errors were detected" in str(error)
def test_numpy_deser_from_csv():
arr = numpy_deserializer(io.BytesIO(b"1,2,3\n4,5,6"), "text/csv")
assert np.array_equal(arr, np.array([[1, 2, 3], [4, 5, 6]]))
def test_numpy_deser_from_json():
arr = numpy_deserializer(io.BytesIO(b'[[1,2,3],\n[4,5,6]]'), 'application/json')
assert np.array_equal(arr, np.array([[1, 2, 3], [4, 5, 6]]))
def test_numpy_deser_from_json_ragged():
arr = numpy_deserializer(io.BytesIO(b'[[1,2,3],\n[4,5,6,7]]'), 'application/json')
assert np.array_equal(arr, np.array([[1, 2, 3], [4, 5, 6, 7]]))
def test_numpy_deser_from_csv_ragged():
with pytest.raises(ValueError) as error:
numpy_deserializer(io.BytesIO(b'1,2,3\n4,5,6,7'), 'text/csv')
assert "errors were detected" in str(error)
def test_numpy_deser_from_csv():
arr = numpy_deserializer(io.BytesIO(b'1,2,3\n4,5,6'), 'text/csv')
assert np.array_equal(arr, np.array([[1, 2, 3], [4, 5, 6]]))
