fix setup_top_images_extractor_env

BKDDFS · May 28, 2024 · 7bbc0ff · 7bbc0ff
1 parent d25b458
commit 7bbc0ff
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/tests/common.py b/tests/common.py
@@ -20,12 +20,12 @@ def top_images_dir(files_dir):
     return files_dir / "top_images"
 
 
-@pytest.fixture(scope="function")
+@pytest.fixture
 def setup_top_images_extractor_env(files_dir, top_images_dir) -> tuple[Path, Path]:
     assert files_dir.is_dir()
 
-    # found_files = [file for file in files_dir.iterdir() if file.suffix == ".jpg"]
-    # assert len(found_files) > 0, "No JPG files found in test directory"
+    found_files = [file for file in files_dir.iterdir() if file.suffix == ".jpg"]
+    assert len(found_files) > 0, "No JPG files found in test directory"
 
     if top_images_dir.is_dir():
         shutil.rmtree(top_images_dir)
@@ -35,7 +35,7 @@ def setup_top_images_extractor_env(files_dir, top_images_dir) -> tuple[Path, Pat
     return files_dir, top_images_dir
 
 
-@pytest.fixture(scope="function")
+@pytest.fixture
 def setup_best_frames_extractor_env(files_dir, best_frames_dir) -> tuple[Path, Path, Path]:
     video_filename = "test_video.mp4"
     expected_video_path = files_dir / f"frames_extracted_{video_filename}"

diff --git a/tests/test_files/best_frames/.gitkeep b/tests/test_files/best_frames/.gitkeep
diff --git a/tests/test_files/test_video.mp4 b/tests/test_files/test_video.mp4
diff --git a/tests/test_files/top_images/.gitkeep b/tests/test_files/top_images/.gitkeep