コード例 #1
0
ファイル: test_emd.py プロジェクト: dzhou36/hyperspy
 def test_save_and_read(self):
     signal_ref = Signal(data_save)
     signal_ref.metadata.General.title = test_title
     signal_ref.axes_manager[0].name = 'x'
     signal_ref.axes_manager[1].name = 'y'
     signal_ref.axes_manager[2].name = 'z'
     signal_ref.axes_manager[0].scale = 2
     signal_ref.axes_manager[1].scale = 3
     signal_ref.axes_manager[2].scale = 4
     signal_ref.axes_manager[0].offset = 10
     signal_ref.axes_manager[1].offset = 20
     signal_ref.axes_manager[2].offset = 30
     signal_ref.axes_manager[0].units = 'nmx'
     signal_ref.axes_manager[1].units = 'nmy'
     signal_ref.axes_manager[2].units = 'nmz'
     signal_ref.save(os.path.join(my_path, 'emd_files', 'example_temp.emd'), overwrite=True,
                     signal_metadata=sig_metadata, user=user, microscope=microscope,
                     sample=sample, comments=comments)
     signal = load(os.path.join(my_path, 'emd_files', 'example_temp.emd'))
     np.testing.assert_equal(signal.data, signal_ref.data)
     np.testing.assert_equal(signal.axes_manager[0].name, 'x')
     np.testing.assert_equal(signal.axes_manager[1].name, 'y')
     np.testing.assert_equal(signal.axes_manager[2].name, 'z')
     np.testing.assert_equal(signal.axes_manager[0].scale, 2)
     np.testing.assert_equal(signal.axes_manager[1].scale, 3)
     np.testing.assert_equal(signal.axes_manager[2].scale, 4)
     np.testing.assert_equal(signal.axes_manager[0].offset, 10)
     np.testing.assert_equal(signal.axes_manager[1].offset, 20)
     np.testing.assert_equal(signal.axes_manager[2].offset, 30)
     np.testing.assert_equal(signal.axes_manager[0].units, 'nmx')
     np.testing.assert_equal(signal.axes_manager[1].units, 'nmy')
     np.testing.assert_equal(signal.axes_manager[2].units, 'nmz')
     np.testing.assert_equal(signal.metadata.General.title, test_title)
     np.testing.assert_equal(
         signal.metadata.General.user.as_dictionary(), user)
     np.testing.assert_equal(
         signal.metadata.General.microscope.as_dictionary(),
         microscope)
     np.testing.assert_equal(
         signal.metadata.General.sample.as_dictionary(), sample)
     np.testing.assert_equal(
         signal.metadata.General.comments.as_dictionary(), comments)
     for key, ref_value in sig_metadata.items():
         np.testing.assert_equal(
             signal.metadata.Signal.as_dictionary().get(key), ref_value)
     nt.assert_is_instance(signal, Signal)