feat/test: Update FPS widget to match netCDF file attributes and add corresponding tests

Author: Sparshr04

movement/napari/loader_widgets.py

@@ -185,6 +185,9 @@ def _on_load_clicked(self):
         if not success:
             return  # Stop execution if formatting/validation failed
 
+        # Update self.fps in case the file loading changed the spinbox value
+        self.fps = self.fps_spinbox.value()
+
         logger.info("Converted dataset to a napari Tracks array.")
         logger.debug(f"Tracks array shape: {self.data.shape}")
         if self.data_bboxes is not None:
@@ -217,6 +220,8 @@ def _format_data_for_layers(self) -> bool:
             ds = self._load_netcdf_file()
             if ds is None:
                 return False
+            if "fps" in ds.attrs:
+                self.fps_spinbox.setValue(ds.attrs["fps"])
 
         # Convert to napari arrays
         self.data, self.data_bboxes, self.properties = ds_to_napari_layers(ds)

tests/test_unit/test_napari_plugin/test_data_loader_widget.py

@@ -256,55 +256,63 @@ def test_on_load_clicked_without_file_path(make_napari_viewer_proxy, capsys):
 
 
 @pytest.mark.parametrize(
-    "filename, source_software, tracks_array_shape, is_bbox",
+    "filename, source_software, tracks_array_shape, is_bbox, expected_fps",
     [
         (
             "VIA_single-crab_MOCA-crab-1.csv",
             "VIA-tracks",
             (35, 4),
             True,
+            60.0,
         ),  # single individual, no keypoints (bboxes)
         (
             "VIA_multiple-crabs_5-frames_labels.csv",
             "VIA-tracks",
             (430, 4),
             True,
+            60.0,
         ),  # multiple individuals, no keypoints (bboxes)
         (
             "SLEAP_single-mouse_EPM.predictions.slp",
             "SLEAP",
             (110910, 4),
             False,
+            60.0,
         ),  # single individual, multiple keypoints
         (
             "DLC_single-wasp.predictions.h5",
             "DeepLabCut",
             (2170, 4),
             False,
+            60.0,
         ),  # single individual, multiple keypoints
         (
             "DLC_two-mice.predictions.csv",
             "DeepLabCut",
             (1439976, 4),
             False,
+            60.0,
         ),  # two individuals, multiple keypoints
         (
             "SLEAP_three-mice_Aeon_mixed-labels.analysis.h5",
             "SLEAP",
             (1803, 4),
             False,
+            60.0,
         ),  # three individuals, one keypoint
         (
             "MOVE_two-mice_octagon.analysis.nc",
             "movement (netCDF)",
             (126000, 4),
             False,
+            50.0,
         ),
         (
             "MOVE_single-crab_MOCA-crab-1_linear-interp.nc",
             "movement (netCDF)",
             (168, 4),
             True,
+            24.0,
         ),
     ],
 )
@@ -313,6 +321,7 @@ def test_on_load_clicked_with_valid_file_path(
     source_software,
     tracks_array_shape,
     is_bbox,
+    expected_fps,
     make_napari_viewer_proxy,
     caplog,
 ):
@@ -342,7 +351,7 @@ def test_on_load_clicked_with_valid_file_path(
     data_loader_widget._on_load_clicked()
 
     # Check the class attributes from the input data
-    assert data_loader_widget.fps == 60
+    assert data_loader_widget.fps == expected_fps
     assert data_loader_widget.source_software == source_software
     assert Path(data_loader_widget.file_path) == file_path
     assert data_loader_widget.file_name == file_path.name