예제 #1
0
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)
예제 #2
0
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)
예제 #3
0
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)