Intelegentny_Pszczelarz/.venv/Lib/site-packages/joblib/test/test_numpy_pickle_compat.py
2023-06-19 00:49:18 +02:00

19 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