def test_save_load_cycle(): sig_reload = None signal = hs.load(file2) try: signal.save(save_path, overwrite=True) sig_reload = hs.load(save_path) np.testing.assert_equal(signal.data, sig_reload.data) assert_deep_almost_equal(signal.axes_manager.as_dictionary(), axes3, places=8) assert_deep_almost_equal(signal.original_metadata.as_dictionary(), sig_reload.original_metadata.as_dictionary(), places=8) nt.assert_is_instance(signal, hs.signals.Signal2D) finally: # Delete reference to close memmap file! del sig_reload gc.collect() remove_file(save_path)
def test_load2(): s = hs.load(file2) nt.assert_equal(s.data.shape, (2, 3, 5, 5)) assert_deep_almost_equal(s.axes_manager.as_dictionary(), axes2, places=8) np.testing.assert_allclose(s.data, ref_data2)
def test_load1(): s = hs.load(file1) nt.assert_equal(s.data.shape, (3, 2, 144, 144)) assert_deep_almost_equal(s.axes_manager.as_dictionary(), axes1, places=8)