ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter5/video-edit/test_slowmo_null_factor.py

@@ -0,0 +1,49 @@
+"""slowmo factor:null must not TypeError (skip like non-positive)."""
+from unittest.mock import patch, MagicMock
+from blender_editor import _plan_fields, generate_bpy_script
+from video_editor import _apply_edit_ffmpeg
+
+
+def test_null_slowmo_factor_skipped_in_plan_fields():
+    start, end, subtitle, slowmo = _plan_fields(
+        {"start": 0.0, "end": 2.0, "effects": [{"type": "slowmo", "factor": None}]}
+    )
+    assert start == 0.0 and end == 2.0
+    assert subtitle is None
+    assert slowmo is None
+
+
+def test_missing_slowmo_factor_still_defaults():
+    _, _, _, slowmo = _plan_fields(
+        {"start": 0.0, "end": 2.0, "effects": [{"type": "slowmo"}]}
+    )
+    assert slowmo == 2.0
+
+
+def test_positive_slowmo_factor_kept():
+    _, _, _, slowmo = _plan_fields(
+        {"start": 0.0, "end": 2.0, "effects": [{"type": "slowmo", "factor": 3.0}]}
+    )
+    assert slowmo == 3.0
+
+
+def test_generate_bpy_script_with_null_factor():
+    script = generate_bpy_script(
+        "/tmp/in.mp4",
+        {"start": 0.0, "end": 1.0, "effects": [{"type": "slowmo", "factor": None}]},
+        "/tmp/out.mp4",
+    )
+    assert "SLOWMO = None" in script
+
+
+def test_apply_edit_ffmpeg_factor_speedup_and_bare_filename():
+    plan = {
+        "start": 0.0,
+        "end": 2.0,
+        "effects": [{"type": "slowmo", "factor": 0.5}]
+    }
+    with patch("video_editor.run") as mock_run:
+        _apply_edit_ffmpeg("input.mp4", plan, "output.mp4")
+        mock_run.assert_called_once()
+        cmd = mock_run.call_args[0][0]
+        assert "atempo=2.0" in cmd[cmd.index("-af") + 1]