From 68da1cf2660b108dfca97cabbc59d698e57607eb Mon Sep 17 00:00:00 2001 From: henryruhs Date: Fri, 23 May 2025 09:17:51 +0200 Subject: [PATCH] More edge case testing --- tests/test_ffmpeg.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/tests/test_ffmpeg.py b/tests/test_ffmpeg.py index 5d055bb..a1e5968 100644 --- a/tests/test_ffmpeg.py +++ b/tests/test_ffmpeg.py @@ -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: