def test_lazy_input_error(self):
s = LazyBeamShift(da.zeros((50, 40, 2)))
with pytest.raises(ValueError):
s.make_linear_plane()
s.compute()
s.make_linear_plane()
def test_to_dpcsignal1d_lazy(self):
probe_x = 200
s = LazyBeamShift(da.zeros((probe_x, 2)))
s_dpc = s.to_dpcsignal()
assert s_dpc._lazy
assert s_dpc.axes_manager.shape == (2, probe_x)