def test_mask(self):
data_x, data_y = np.meshgrid(np.arange(-50, 50, dtype=np.float32),
np.arange(-256, 0, dtype=np.float32))
data = np.stack((data_y, data_x), -1)
mask = np.zeros_like(data[:, :, 0], dtype=np.bool)
mask[45:50, 36:41] = True
s_mask = Signal2D(mask)
s = BeamShift(data)
s.change_dtype("float32")
s_orig = s.deepcopy()
s.data[45:50, 36:41, 0] = 100000
s.data[45:50, 36:41, 1] = -100000
s.make_linear_plane(mask=s_mask)
np.testing.assert_almost_equal(s.data, s_orig, decimal=7)
def test_simple(self):
data_x, data_y = np.meshgrid(np.arange(-50, 50, dtype=np.float32),
np.arange(-256, 0, dtype=np.float32))
data = np.stack((data_y, data_x), -1)
s = BeamShift(data)
s.change_dtype("float32")
s_orig = s.deepcopy()
s.make_linear_plane()
np.testing.assert_almost_equal(s.data, s_orig, decimal=7)
def test_to_dpcsignal1d(self):
probe_x = 200
s = BeamShift(np.zeros((probe_x, 2)))
s_dpc = s.to_dpcsignal()
assert s_dpc.axes_manager.shape == (2, probe_x)