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_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)