Add test for ffmpeg in base images

replicate · Jan 23, 2025 · 13b22a6 · 13b22a6
1 parent b7aa7c3
commit 13b22a6
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/test-integration/test_integration/fixtures/ffmpeg-package/cog.yaml b/test-integration/test_integration/fixtures/ffmpeg-package/cog.yaml
@@ -0,0 +1,9 @@
+build:
+  gpu: true
+  python_version: "3.9"
+  python_packages:
+    - "torch==2.5.1"
+  cuda: "12.4"
+  run:
+    - command: ffmpeg --help
+predict: "predict.py:Predictor"
diff --git a/test-integration/test_integration/test_build.py b/test-integration/test_integration/test_build.py
@@ -434,3 +434,15 @@ def test_pydantic2(docker_image):
     )
 
     assert build_process.returncode == 0
+
+
+def test_ffmpeg_base_image(docker_image):
+    project_dir = Path(__file__).parent / "fixtures/ffmpeg-package"
+
+    build_process = subprocess.run(
+        ["cog", "build", "-t", docker_image],
+        cwd=project_dir,
+        capture_output=True,
+    )
+
+    assert build_process.returncode == 0