def roundtrip_numpy_file_pointers(ndarray):
sio = StringIO()
sink = CompressedFPSink(sio)
pack_ndarray(ndarray, sink)
sio.seek(0)
source = CompressedFPSource(sio)
b = unpack_ndarray(source)
return npt.assert_array_equal, ndarray, b
def roundtrip_numpy_file_pointers(ndarray):
sio = StringIO()
sink = CompressedFPSink(sio)
pack_ndarray(ndarray, sink)
sio.seek(0)
source = CompressedFPSource(sio)
b = unpack_ndarray(source)
return npt.assert_array_equal, ndarray, b
def test_roundtrip_numpy_file_pointers(self):
sio = StringIO()
sink = CompressedFPSink(sio)
pack_ndarray(self.a, sink)
sio.seek(0)
source = CompressedFPSource(sio)
b = unpack_ndarray(source)
npt.assert_array_equal(self.a, b)
def roundtrip_numpy_memory(ndarray):
sink = CompressedMemorySink()
pack_ndarray(ndarray, sink)
source = CompressedMemorySource(sink)
b = unpack_ndarray(source)
return npt.assert_array_equal, ndarray, b
def roundtrip_numpy_memory(ndarray):
sink = CompressedMemorySink()
pack_ndarray(ndarray, sink)
source = CompressedMemorySource(sink)
b = unpack_ndarray(source)
return npt.assert_array_equal, ndarray, b
def roundtrip_ndarray(ndarray):
sink = CompressedMemorySink()
pack_ndarray(ndarray, sink)
source = CompressedMemorySource(sink)
result = unpack_ndarray(source)
npt.assert_array_equal(ndarray, result)
def test_roundtrip_numpy_memory(self):
sink = CompressedMemorySink()
pack_ndarray(self.a, sink)
source = CompressedMemorySource(sink)
b = unpack_ndarray(source)
npt.assert_array_equal(self.a, b)
