diff --git a/tests/tests_bec_ipython_client/test_lamni_tomo_alignment_scan.py b/tests/tests_bec_ipython_client/test_lamni_tomo_alignment_scan.py index 258dc920..4b6fa550 100644 --- a/tests/tests_bec_ipython_client/test_lamni_tomo_alignment_scan.py +++ b/tests/tests_bec_ipython_client/test_lamni_tomo_alignment_scan.py @@ -97,7 +97,7 @@ def test_tomo_alignment_scan_runs_12_projections_across_360(monkeypatch): lamni = make_lamni(monkeypatch, xray_eye_fit=[[0.0, 0.0, 0.0], [0.0, 0.0, 0.0]]) calls = [] - def _fake_projection(angle): + def _fake_projection(angle, _internal=False): calls.append(angle) builtins.__dict__["bec"].queue.next_scan_number += 1 @@ -115,7 +115,7 @@ def test_tomo_alignment_scan_runs_12_projections_across_360(monkeypatch): def test_tomo_alignment_scan_rotates_back_to_zero(monkeypatch): lamni = make_lamni(monkeypatch, xray_eye_fit=[[0.0, 0.0, 0.0], [0.0, 0.0, 0.0]]) - lamni.tomo_scan_projection = lambda angle: None + lamni.tomo_scan_projection = lambda angle, _internal=False: None umv_calls = [] monkeypatch.setattr(lamni_module, "umv", lambda *a: umv_calls.append(a), raising=False) diff --git a/tests/tests_bec_ipython_client/test_lamni_tomo_queue.py b/tests/tests_bec_ipython_client/test_lamni_tomo_queue.py index 6ceca9ec..a35f12e3 100644 --- a/tests/tests_bec_ipython_client/test_lamni_tomo_queue.py +++ b/tests/tests_bec_ipython_client/test_lamni_tomo_queue.py @@ -208,7 +208,7 @@ def test_at_each_angle_default_path_unchanged_without_hook(): behaviour lamni had before this port: tomo_scan_projection + tomo_reconstruct.""" lamni = make_lamni() calls = [] - lamni.tomo_scan_projection = lambda angle: calls.append(("projection", angle)) + lamni.tomo_scan_projection = lambda angle, _internal=False: calls.append(("projection", angle)) lamni.tomo_reconstruct = lambda: calls.append(("reconstruct",)) lamni._at_each_angle(30.0)