diff --git a/tests/unit_tests/test_stitching.py b/tests/unit_tests/test_stitching.py index 017d9291..37ea7b1a 100644 --- a/tests/unit_tests/test_stitching.py +++ b/tests/unit_tests/test_stitching.py @@ -333,10 +333,19 @@ def test_stitch_all_scans_continues_after_error(mocker, caplog, smart_scan_thing mocker.Mock(name="scan1", dzi=None), mocker.Mock(name="scan2", dzi=None), mocker.Mock(name="scan3", dzi=None), + mocker.Mock(name="scan4", dzi=None), ] scans[0].name = "scan1" scans[1].name = "scan2" scans[2].name = "scan3" + scans[3].name = "scan4" + + # stitch_all skips any scans with less than 2 images, without logging + + scans[0].number_of_images = 10 + scans[1].number_of_images = 10 + scans[2].number_of_images = 1 + scans[3].number_of_images = 10 mocker.patch.object( smart_scan_thing, @@ -377,5 +386,5 @@ def test_stitch_all_scans_continues_after_error(mocker, caplog, smart_scan_thing "Stitching scan1", "Stitching scan2", "Stitching failed: test_message", - "Stitching scan3", + "Stitching scan4", ]