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)