Proto benchmark (WIP)

Author: sfmig

tests/test_unit/test_io/test_load_bboxes.py

@@ -744,3 +744,44 @@ def test_position_numpy_array_from_via_tracks_file(via_file_path):
         bboxes_arrays["position_array"],  # frames, xy, individuals
         np.stack(list_derived_centroids, axis=-1),
     )
+
+
+@pytest.mark.benchmark
+@pytest.mark.parametrize(
+    "via_file_path",
+    [
+        pytest.DATA_PATHS.get("VIA_multiple-crabs_5-frames_labels.csv"),
+        pytest.DATA_PATHS.get("VIA_single-crab_MOCA-crab-1.csv"),
+    ],
+)
+# @pytest.mark.parametrize("use_frame_numbers_from_file", [True, False])
+# small difference
+def test_benchmark_from_via_tracks_file(
+    via_file_path,
+    use_frame_numbers_from_file,
+    benchmark,
+):
+    """Benchmark the loading of a VIA tracks .csv file."""
+    kwargs = {
+        "file_path": via_file_path,
+        "use_frame_numbers_from_file": use_frame_numbers_from_file,
+    }
+    benchmark(load_bboxes.from_via_tracks_file, **kwargs)
+
+
+@pytest.mark.benchmark
+@pytest.mark.parametrize(
+    "via_file_path",
+    [
+        pytest.DATA_PATHS.get("VIA_multiple-crabs_5-frames_labels.csv"),
+        # multiple crabs present in all 5 frames
+        pytest.DATA_PATHS.get("VIA_single-crab_MOCA-crab-1.csv"),
+        # single crab present in 35 non-consecutive frames
+    ],
+)
+def test_benchmark_df_from_via_tracks_file(
+    via_file_path,
+    benchmark,
+):
+    """Benchmark the `_df_from_via_tracks_file` function."""
+    benchmark(load_bboxes._df_from_via_tracks_file, via_file_path)