def test_save_reload(self):
self.data.calculate_strain(self.q0)
data2 = copy.deepcopy(self.data)
data2.d1 += 1.00001
merged = ordered_merge([self.data, data2], [0, 1], 0.1)
merged.plot_slice("shear strain", phi=np.pi / 3)
plt.close()
merged.save_to_hdf5(fpath="pyxe/data/energy_test_pyxe.h5", overwrite=True)
merged_reload = PeakAnalysis(fpath="pyxe/data/energy_test_pyxe.h5")
merged_reload_b = DataViz(fpath="pyxe/data/energy_test_pyxe.h5")
merged_reload.plot_slice("shear strain", phi=np.pi / 3)
plt.close()
merged_reload_b.plot_slice("shear strain", phi=np.pi / 3)
plt.close()
assert np.array_equal(merged.peaks, merged_reload.peaks)
def test_save_reload(self):
self.data.calculate_strain(self.q0)
data2 = copy.deepcopy(self.data)
data2.d1 += 1.00001
merged = ordered_merge([self.data, data2], [0, 1], 0.1)
merged.plot_slice('shear strain', phi=np.pi / 3)
plt.close()
merged.save_to_hdf5(fpath='pyxe/data/energy_test_pyxe.h5',
overwrite=True)
merged_reload = PeakAnalysis(fpath='pyxe/data/energy_test_pyxe.h5')
merged_reload_b = DataViz(fpath='pyxe/data/energy_test_pyxe.h5')
merged_reload.plot_slice('shear strain', phi=np.pi / 3)
plt.close()
merged_reload_b.plot_slice('shear strain', phi=np.pi / 3)
plt.close()
assert np.array_equal(merged.peaks, merged_reload.peaks)
def test_no_analysis_save_load():
data, q0 = integration()
data.save_to_hdf5(fpath='pyxe/data/energy_test_pyxe.h5', overwrite=True)
PeakAnalysis(fpath='pyxe/data/energy_test_pyxe.h5')
merged = ordered_merge([self.data, data2], [0, 1], 0.1)
merged.plot_slice('shear strain', phi=np.pi / 3)
az_lst = [
'fwhm', 'fwhm error', 'peaks', 'peaks error', 'strain',
'strain error', 'stress'
]
phi_lst = ['strain', 'shear strain', 'stress', 'shear stress']
merged.save_to_txt('pyxe/data/test.csv', az_lst, az_idx=2)
merged.save_to_txt('pyxe/data/test.csv', phi_lst, phi=-np.pi / 3)
if __name__ == '__main__':
data, q0 = integration()
data.peak_fit(3.1, 1.)
q0.peak_fit(3.1, 1.)
data.calculate_strain(q0)
data.material_parameters(200 * 10**9, 0.3, G=None)
data.plot_intensity()
data.plot_strain_fit()
shift = 1.00001
data2 = copy.deepcopy(data)
data2.d1 += shift
merged = ordered_merge([data, data2], [0, 1], 0)
merged.plot_slice('shear strain', phi=0)
#plt.show()
merged.save_to_hdf5(overwrite=True)
merged_reload = PeakAnalysis('_pyxe.h5')
#print(merged_reload.analysis_state.decode())
merged_reload.plot_slice('shear strain', phi=0)
plt.show()
self.data.material_parameters(200 * 10 ** 9, 0.3)
data2 = copy.deepcopy(self.data)
data2.d1 += 1.00001
merged = ordered_merge([self.data, data2], [0, 1], 0.1)
merged.plot_slice("shear strain", phi=np.pi / 3)
az_lst = ["fwhm", "fwhm error", "peaks", "peaks error", "strain", "strain error", "stress"]
phi_lst = ["strain", "shear strain", "stress", "shear stress"]
merged.save_to_txt("pyxe/data/test.csv", az_lst, az_idx=2)
merged.save_to_txt("pyxe/data/test.csv", phi_lst, phi=-np.pi / 3)
if __name__ == "__main__":
data, q0 = integration()
data.peak_fit(3.1, 1.0)
q0.peak_fit(3.1, 1.0)
data.calculate_strain(q0)
data.material_parameters(200 * 10 ** 9, 0.3, G=None)
data.plot_intensity()
data.plot_strain_fit()
shift = 1.00001
data2 = copy.deepcopy(data)
data2.d1 += shift
merged = ordered_merge([data, data2], [0, 1], 0)
merged.plot_slice("shear strain", phi=0)
# plt.show()
merged.save_to_hdf5(overwrite=True)
merged_reload = PeakAnalysis("_pyxe.h5")
# print(merged_reload.analysis_state.decode())
merged_reload.plot_slice("shear strain", phi=0)
plt.show()
