Add video search to "robustify" (excessively) fragile test (#1262)

Add video search to fix fragile test
talmolab · Apr 6, 2023 · 4cd8c0d · 4cd8c0d
1 parent 11a71ff
commit 4cd8c0d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/tests/fixtures/datasets.py b/tests/fixtures/datasets.py
@@ -60,7 +60,7 @@ def min_labels_robot():
 @pytest.fixture
 def siv_robot():
     """Created before grayscale attribute was added to SingleImageVideo backend."""
-    return Labels.load_file(TEST_SLP_SIV_ROBOT)
+    return Labels.load_file(TEST_SLP_SIV_ROBOT, video_search="tests/data/videos/")
 
 
 @pytest.fixture

diff --git a/tests/io/test_video.py b/tests/io/test_video.py
@@ -495,7 +495,7 @@ def test_reset_video_mp4(small_robot_mp4_vid: Video):
     assert_video_params(video=video, filename=filename, bgr=True, reset=True)
 
 
-def test_reset_video_siv(small_robot_single_image_vid: Video, siv_robot: Labels):
+def test_reset_video_siv(small_robot_single_image_vid: Video):
     video = small_robot_single_image_vid
     filename = video.backend.filename
 
@@ -548,7 +548,9 @@ def test_reset_video_siv(small_robot_single_image_vid: Video, siv_robot: Labels)
     assert_video_params(video=video, filenames=filenames, reset=True)
 
     # Test reset does not break deserialization of older slp
-    labels: Labels = Labels.load_file(TEST_SLP_SIV_ROBOT)
+    labels: Labels = Labels.load_file(
+        TEST_SLP_SIV_ROBOT, video_search="tests/data/videos/"
+    )
     video: Video = labels.video
     filename = labels.video.backend.filename
     labels.video.backend.reset(filename=filename, grayscale=True)