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

More edge case testing

Browse Source
This commit is contained in:
henryruhs
2025-05-23 09:17:51 +02:00
parent 52000114ba
commit 68da1cf266
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_ffmpeg.py
View File

@@ -5,8 +5,7 @@ import pytest
from facefusion import process_manager, state_manager
from facefusion.download import conditional_download
from facefusion.ffmpeg import concat_video, extract_frames, get_available_encoder_set, read_audio_buffer, replace_audio, \
restore_audio, merge_video
from facefusion.ffmpeg import concat_video, extract_frames, get_available_encoder_set, merge_video, read_audio_buffer, replace_audio, restore_audio
from facefusion.filesystem import copy_file
from facefusion.temp_helper import clear_temp_directory, create_temp_directory, get_temp_file_path, resolve_temp_frame_paths
from .helper import get_test_example_file, get_test_examples_directory, get_test_output_file, prepare_test_output_directory
@@ -22,6 +21,8 @@ def before_all() -> None:
'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/target-240p.mp4'
])
subprocess.run([ 'ffmpeg', '-i', get_test_example_file('source.mp3'), get_test_example_file('source.wav') ])
subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), get_test_example_file('target-240p.avi') ])
subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), get_test_example_file('target-240p.mov') ])
subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), get_test_example_file('target-240p.webm') ])
subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vf', 'fps=25', get_test_example_file('target-240p-25fps.mp4') ])
subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vf', 'fps=30', get_test_example_file('target-240p-30fps.mp4') ])
@@ -80,9 +81,12 @@ def test_extract_frames() -> None:
def test_merge_video() -> None:
merge_set =\
[
(get_test_example_file('target-240p.avi'), 'libx264'),
(get_test_example_file('target-240p.mov'), 'libx264'),
(get_test_example_file('target-240p.mp4'), 'libx264'),
(get_test_example_file('target-240p.mp4'), 'rawvideo'),
(get_test_example_file('target-240p.webm'), 'libx264')
(get_test_example_file('target-240p.webm'), 'libx264'),
(get_test_example_file('target-240p.webm'), 'rawvideo')
]
for target_path, output_video_encoder in merge_set: