18 lines
624 B
Python
18 lines
624 B
Python
"""Test the old numpy pickler, compatibility version."""
|
|
|
|
import random
|
|
|
|
# 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
|