feat(video_analysis): extract_keyframes.py with TDD (4 tests)

tests/test_video_analysis_extract_keyframes.py

@@ -0,0 +1,40 @@
+from __future__ import annotations
+
+from pathlib import Path
+from unittest.mock import MagicMock, patch
+
+from scripts.video_analysis.extract_keyframes import (
+ build_ffmpeg_scene_select_filter,
+ compute_phash,
+ dedupe_frames,
+ extract_keyframes,
+)
+
+
+def test_build_ffmpeg_scene_select_filter() -> None:
+ filter_str = build_ffmpeg_scene_select_filter(0.4)
+ assert "select=gt(scene\\,0.4)" in filter_str
+
+
+def test_compute_phash_returns_string() -> None:
+ from PIL import Image
+ img = Image.new("RGB", (100, 100), color="red")
+ h = compute_phash(img)
+ assert isinstance(h, str)
+ assert len(h) >= 8
+
+
+def test_dedupe_frames_keeps_unique() -> None:
+ hashes = ["aaaaaaaaaaaaaaaa", "bbbbbbbbbbbbbbbb", "aaaaaaaaaaaaaaaa", "cccccccccccccccc"]
+ result = dedupe_frames(hashes, hamming_threshold=5)
+ assert result == [True, True, False, True]
+
+
+def test_extract_keyframes_creates_output_dir(tmp_path: Path) -> None:
+ fake_video = tmp_path / "fake.mp4"
+ fake_video.write_bytes(b"fake")
+ with patch("subprocess.run") as mock_run:
+  mock_run.return_value = MagicMock(returncode=0, stdout="", stderr="")
+  result = extract_keyframes(fake_video, tmp_path / "frames", threshold=0.4)
+  assert result.is_ok()
+  assert (tmp_path / "frames").exists()