diff --git a/tests/test_unit/test_load_poses.py b/tests/test_unit/test_load_poses.py index 7256644a..edc67e64 100644 --- a/tests/test_unit/test_load_poses.py +++ b/tests/test_unit/test_load_poses.py @@ -301,14 +301,13 @@ def test_from_numpy_valid( ) self.assert_dataset(ds, expected_source_software=source_software) - def test_from_multi_view(self): """Test that the from_file() function delegates to the correct loader function according to the source_software. """ view_names = ["view_0", "view_1"] file_path_dict = { - view: POSE_DATA_PATHS.get("DLC_single-wasp.predictions.h5") + view: DATA_PATHS.get("DLC_single-wasp.predictions.h5") for view in view_names } @@ -318,4 +317,4 @@ def test_from_multi_view(self): assert isinstance(multi_view_ds, xr.Dataset) assert "view" in multi_view_ds.dims - assert multi_view_ds.view.values.tolist() == view_names \ No newline at end of file + assert multi_view_ds.view.values.tolist() == view_names