def test_extension(tmpdir):
file = tmpdir / 'file'
pb.save(mock_data, file)
assert mock_data == pb.load(tmpdir / 'file.pbz')
file = tmpdir / 'file.ext'
pb.save(mock_data, file)
assert mock_data == pb.load(tmpdir / 'file.ext')
def test_pickle_round_trip(lattice, tmpdir):
file_name = str(tmpdir.join('file.npz'))
pb.save(lattice, file_name)
from_file = pb.load(file_name)
assert lattice.sub_name_map == from_file.sub_name_map
assert lattice.hop_name_map == from_file.hop_name_map
assert pytest.fuzzy_equal(lattice, from_file)
def test_pickle_round_trip(lattice, tmpdir):
file_name = str(tmpdir.join('file.npz'))
pb.save(lattice, file_name)
from_file = pb.load(file_name)
assert lattice.sub_name_map == from_file.sub_name_map
assert lattice.hop_name_map == from_file.hop_name_map
assert pytest.fuzzy_equal(lattice, from_file)
def get_expected(result, group=''):
file = path_from_fixture(request, prefix='baseline_data', ext='.pbz',
override_group=group)
if not request.config.getoption("--savebaseline") and file.exists():
return pb.load(file)
elif not request.config.getoption("--readonly"):
if not file.parent.exists():
file.parent.mkdir(parents=True)
pb.save(result, file)
return result
else:
raise RuntimeError("Missing baseline data: {}".format(file))
def test_path_type(tmpdir):
# str
file = str(tmpdir / 'file')
assert round_trip(mock_data, file)
# py.path.local object
file = tmpdir / 'file'
assert round_trip(mock_data, file)
# pathlib object
file = pathlib.Path(str(tmpdir)) / 'file'
assert round_trip(mock_data, file)
# file object
file = str(tmpdir / 'file')
with open(file, 'wb') as f:
pb.save(mock_data, f)
with open(file, 'rb') as f:
assert mock_data == pb.load(f)
def round_trip(obj, file):
pb.save(obj, file)
return pb.load(file) == obj
def test_pickle_round_trip(solver, tmpdir):
file_name = str(tmpdir.join('file.npz'))
pb.save(solver, file_name)
from_file = pb.load(file_name)
assert pytest.fuzzy_equal(solver, from_file)
def test_pickle_round_trip(model, tmpdir):
file_name = str(tmpdir.join('file.npz'))
pb.save(model.system, file_name)
from_file = pb.load(file_name)
assert pytest.fuzzy_equal(model.system, from_file)
def test_pickle_round_trip(model, tmpdir):
file_name = str(tmpdir.join('file.npz'))
pb.save(model.system, file_name)
from_file = pb.load(file_name)
assert pytest.fuzzy_equal(model.system, from_file)