初始化奇妙小屏幕控制器项目

2026-09-08 22:56:52 +08:00
commit 8d368de3b5
491 changed files with 67678 additions and 0 deletions
@@ -0,0 +1,82 @@
+import shutil
+import subprocess
+
+import pytest
+from PIL import Image
+
+from app.media.converter import analyze, converted_frames
+
+
+VIDEO_CASES = [
+    ("mp4", "mpeg4"),
+    ("mov", "mpeg4"),
+    ("mkv", "ffv1"),
+    ("avi", "mpeg4"),
+    ("webm", "libvpx-vp9"),
+]
+
+
+@pytest.mark.skipif(not shutil.which("ffmpeg"), reason="FFmpeg is not installed")
+@pytest.mark.parametrize(("extension", "codec"), VIDEO_CASES)
+def test_common_video_containers_probe_sample_and_merge_identical_frames(tmp_path, extension, codec):
+    source = tmp_path / ("constant." + extension)
+    result = subprocess.run([
+        "ffmpeg", "-nostdin", "-v", "error", "-f", "lavfi",
+        "-i", "color=c=red:s=96x54:r=30:d=0.3",
+        "-an", "-c:v", codec, "-y", str(source),
+    ], stdout=subprocess.DEVNULL, stderr=subprocess.PIPE, check=False, timeout=30)
+    if result.returncode:
+        pytest.skip(codec + " encoder unavailable")
+    metadata = analyze(source, tmp_path / ("previews-" + extension))
+    frames = list(converted_frames(source, metadata, {
+        "fit_mode": "contain", "center_x": 0.5, "center_y": 0.5, "zoom": 1,
+        "transparency_color": "#000000", "padding_color": "#010203",
+    }))
+    assert metadata["decoder"] == "ffmpeg"
+    assert metadata["dynamic"] is True
+    assert 250 <= metadata["duration_ms"] <= 350
+    assert len(frames) == 1
+    assert frames[0][1] >= 250
+    assert len(frames[0][0]) == 64 * 64 * 3
+
+
+@pytest.mark.skipif(not shutil.which("ffmpeg"), reason="FFmpeg is not installed")
+def test_ffmpeg_free_crop_keeps_the_complete_wide_frame_and_void_padding(tmp_path):
+    source = tmp_path / "wide.mkv"
+    result = subprocess.run([
+        "ffmpeg", "-nostdin", "-v", "error", "-f", "lavfi",
+        "-i", "color=c=red:s=96x48:r=20:d=0.1",
+        "-an", "-c:v", "ffv1", "-y", str(source),
+    ], stdout=subprocess.DEVNULL, stderr=subprocess.PIPE, check=False, timeout=30)
+    if result.returncode:
+        pytest.skip("ffv1 encoder unavailable")
+    metadata = analyze(source, tmp_path / "wide-previews")
+    frames = list(converted_frames(source, metadata, {
+        "fit_mode": "crop", "center_x": 0.5, "center_y": 0.5, "zoom": 1,
+        "transparency_color": "#00FF00", "padding_color": "#0000FF",
+    }))
+    output = Image.frombytes("RGB", (64, 64), frames[0][0])
+    assert output.getpixel((32, 0)) == (0, 0, 255)
+    assert output.getpixel((32, 32))[0] > 240
+    assert output.getpixel((32, 63)) == (0, 0, 255)
+
+
+@pytest.mark.skipif(not shutil.which("heif-enc"), reason="libheif examples are not installed")
+def test_heic_uses_local_libheif_decoder(tmp_path):
+    png = tmp_path / "source.png"
+    source = tmp_path / "source.heic"
+    Image.new("RGB", (32, 20), (4, 5, 6)).save(png)
+    result = subprocess.run(
+        ["heif-enc", "-q", "100", str(png), "-o", str(source)],
+        stdout=subprocess.DEVNULL, stderr=subprocess.PIPE, check=False, timeout=30,
+    )
+    if result.returncode:
+        pytest.skip("HEIC encoder unavailable")
+    metadata = analyze(source, tmp_path / "heic-previews")
+    frames = list(converted_frames(source, metadata, {
+        "fit_mode": "stretch", "center_x": 0.5, "center_y": 0.5, "zoom": 1,
+        "transparency_color": "#000000", "padding_color": "#000000",
+    }))
+    assert metadata["decoder"] in {"pillow", "heif"}
+    assert metadata["dynamic"] is False
+    assert len(frames) == 1