def electron_diffraction1d():
"""A simple, multiuse diffraction profile, with dimensions:
ElectronDiffraction1D <2,2|12>
"""
data = np.array([
[[1.0, 0.25, 0.0, 0.0, 0.0], [1.0, 0.25, 0.0, 0.0, 0.0]],
[[1.0, 0.25, 0.0, 0.0, 0.16666667], [1.5, 0.5, 0.0, 0.0, 0.0]],
])
return ElectronDiffraction1D(data)
def test_2d_data_as_lazy(self):
data = np.random.random((100, 150))
s = ElectronDiffraction1D(data)
scale0, scale1, metadata_string = 0.5, 1.5, "test"
s.axes_manager[0].scale = scale0
s.axes_manager[1].scale = scale1
s.metadata.Test = metadata_string
s_lazy = s.as_lazy()
assert isinstance(s_lazy, LazyElectronDiffraction1D)
assert hasattr(s_lazy.data, "compute")
assert s_lazy.axes_manager[0].scale == scale0
assert s_lazy.axes_manager[1].scale == scale1
assert s_lazy.metadata.Test == metadata_string
assert data.shape == s_lazy.data.shape
def test_3d_data_as_lazy(self):
data = np.random.random((4, 10, 15))
s = ElectronDiffraction1D(data)
s_lazy = s.as_lazy()
assert isinstance(s_lazy, LazyElectronDiffraction1D)
assert data.shape == s_lazy.data.shape
def red_int_generator():
data = np.arange(10, 0, -1).reshape(1, 10) * np.arange(1, 5).reshape(4, 1)
data = data.reshape(2, 2, 10)
rp = ElectronDiffraction1D(data)
rigen = ReducedIntensityGenerator1D(rp)
return rigen