def test_import_from_datadir(fn): print("import ", fn) filename = os.path.join(io_test.datadir, fn) with warnings.catch_warnings(): warnings.filterwarnings("ignore", category=FutureWarning) with open(filename, 'rb') as f: data = pickle.load(f) if 'version' in data: data_expected = io_test.gen_example_data(data['version']) else: data_expected = io_test.gen_example_data('0.4.0') io_test.assert_equal_data(data, data_expected)
def test_import_from_datadir(fn): print("import ", fn) filename = os.path.join(io_test.datadir, fn) with warnings.catch_warnings(): warnings.filterwarnings("ignore", category=FutureWarning) with h5py.File(filename, 'r') as f: data = hdf5_io.load_from_hdf5(f) if 'version' in data: data_expected = io_test.gen_example_data(data['version']) else: data_expected = io_test.gen_example_data('0.4.0') io_test.assert_equal_data(data, data_expected) io_test.assert_event_handler_example_works(data)
def test_pickle(): """Try subsequent export and import to pickle.""" data = io_test.gen_example_data() with tempfile.TemporaryDirectory() as tdir: filename = 'test.pkl' with open(os.path.join(tdir, filename), 'wb') as f: pickle.dump(data, f) with open(os.path.join(tdir, filename), 'rb') as f: data_imported = pickle.load(f) io_test.assert_equal_data(data_imported, data)
def test_hdf5_export_import(): """Try subsequent export and import to pickle.""" data = io_test.gen_example_data() with tempfile.TemporaryDirectory() as tdir: filename = 'test.hdf5' with h5py.File(os.path.join(tdir, filename), 'w') as f: hdf5_io.save_to_hdf5(f, data) with h5py.File(os.path.join(tdir, filename), 'r') as f: data_imported = hdf5_io.load_from_hdf5(f) io_test.assert_equal_data(data_imported, data)
def test_pickle(tmp_path): """Try subsequent export and import to pickle.""" data = io_test.gen_example_data() io_test.assert_event_handler_example_works( data) #if this fails, it's not import/export filename = tmp_path / 'test.pkl' with filename.open('wb') as f: pickle.dump(data, f) with filename.open('rb') as f: data_imported = pickle.load(f) io_test.assert_equal_data(data_imported, data) io_test.assert_event_handler_example_works(data_imported)
def test_hdf5_export_import(tmp_path): """Try subsequent export and import to pickle.""" data = io_test.gen_example_data() io_test.assert_event_handler_example_works( data) #if this fails, it's not import/export filename = tmp_path / 'test.hdf5' with h5py.File(str(filename), 'w') as f: hdf5_io.save_to_hdf5(f, data) with h5py.File(str(filename), 'r') as f: data_imported = hdf5_io.load_from_hdf5(f) io_test.assert_equal_data(data_imported, data) io_test.assert_event_handler_example_works(data_imported)
def export_to_datadir(): filename = io_test.get_datadir_filename("exported_from_tenpy_{0}.hdf5") data = io_test.gen_example_data() with warnings.catch_warnings(record=True) as caught: #warnings.filterwarnings("ignore", category=UserWarning) with h5py.File(filename, 'w') as f: hdf5_io.save_to_hdf5(f, data) for w in caught: msg = str(w.message) expected = "without explicit HDF5 format" in msg if expected: expected = any(t in msg for t in [ 'io_test.DummyClass', 'tenpy.tools.events.EventHandler', 'tenpy.tools.events.Listener', 'method' ]) if not expected: warnings.showwarning(w.message, w.category, w.filename, w.lineno, w.file, w.line)
def export_to_datadir(): filename = io_test.get_datadir_filename("exported_from_tenpy_{0}.pkl") data = io_test.gen_example_data() with open(filename, 'wb') as f: pickle.dump(data, f, protocol=pickle.HIGHEST_PROTOCOL)
def export_to_datadir(): filename = io_test.get_datadir_filename("exported_from_tenpy_{0}.hdf5") data = io_test.gen_example_data() with h5py.File(filename, 'w') as f: hdf5_io.save_to_hdf5(f, data)