Function test_capture_sequence_correctness

test/test_capture_sequence.py:37–51

()

def test_capture_sequence_correctness():
    output_dir_full = get_test_dir("scan")

    for view_index in range(9):

        filepath = Path(output_dir_full, f"led_map_2d_{view_index:04}.csv")

        leds = load_detections(filepath, view_index)

        if view_index in [0, 4, 8]:
            assert (  # If it's a straight on view, there should be 9 points
                len(leds) == 9
            )
        else:
            assert len(leds) == 15  # If it's a diagonal-ish view, then we see 15 points