17 lines
609 B
Python
17 lines
609 B
Python
|
"""Test the old numpy pickler, compatibility version."""
|
||
|
|
|
||
|
|
# numpy_pickle is not a drop-in replacement of pickle, as it takes
|
||
|
|
# filenames instead of open files as arguments.
|
||
|
|
from joblib import numpy_pickle_compat
|
||
|
|
|
||
|
|
|
||
|
|
def test_z_file(tmpdir):
|
||
|
|
# Test saving and loading data with Zfiles.
|
||
|
|
filename = tmpdir.join("test.pkl").strpath
|
||
|
|
data = numpy_pickle_compat.asbytes("Foo, \n Bar, baz, \n\nfoobar")
|
||
|
|
with open(filename, "wb") as f:
|
||
|
|
numpy_pickle_compat.write_zfile(f, data)
|
||
|
|
with open(filename, "rb") as f:
|
||
|
|
data_read = numpy_pickle_compat.read_zfile(f)
|
||
|
|
assert data == data_read
|