mirrors/facefusion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Basic testing for detect_fps

Browse Source
This commit is contained in:
henryruhs 2023-08-21 16:37:10 +02:00
parent 8d1da5ad49
commit bead4a6c5f
1 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
tests/test_utilities.py Normal file
View File

@ -0,0 +1,21 @@
import subprocess
import pytest
from facefusion.utilities import conditional_download, detect_fps
@pytest.fixture(scope = 'module', autouse = True)
def setup() -> None:
conditional_download('.assets/examples',
[
'https://github.com/facefusion/facefusion-assets/releases/download/examples/target-1080p.mp4'
])
subprocess.run([ 'ffmpeg', '-i', '.assets/examples/target-1080p.mp4', '-vf', 'fps=25', '.assets/examples/target-1080p-25fps.mp4' ])
subprocess.run([ 'ffmpeg', '-i', '.assets/examples/target-1080p.mp4', '-vf', 'fps=30', '.assets/examples/target-1080p-30fps.mp4' ])
subprocess.run([ 'ffmpeg', '-i', '.assets/examples/target-1080p.mp4', '-vf', 'fps=60', '.assets/examples/target-1080p-60fps.mp4' ])
def test_detect_fps() -> None:
assert detect_fps('.assets/examples/target-1080p-25fps.mp4') == 25.0
assert detect_fps('.assets/examples/target-1080p-30fps.mp4') == 30.0
assert detect_fps('.assets/examples/target-1080p-60fps.mp4') == 60.0