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

Improve testing

Browse Source
This commit is contained in:
henryruhs
2025-05-25 15:45:38 +02:00
parent 441305afe5
commit 84e86ad91e
2 changed files with 22 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
facefusion/ffmpeg.py
View File

@@ -167,8 +167,8 @@ def restore_audio(target_path : str, output_path : str, trim_frame_start : int,
temp_video_format = get_file_format(temp_video_path)
temp_video_duration = detect_video_duration(temp_video_path)
if temp_video_format in [ 'avi', 'mov' ] and output_audio_encoder == 'flac':
output_audio_encoder = 'aac'
if temp_video_format == 'webm':
output_audio_encoder = 'libopus'
commands = ffmpeg_builder.chain(
ffmpeg_builder.set_input(temp_video_path),
@@ -194,8 +194,8 @@ def replace_audio(target_path : str, audio_path : str, output_path : str) -> boo
temp_video_format = get_file_format(temp_video_path)
temp_video_duration = detect_video_duration(temp_video_path)
if temp_video_format in [ 'avi', 'mov' ] and output_audio_encoder == 'flac':
output_audio_encoder = 'aac'
if temp_video_format == 'webm':
output_audio_encoder = 'libopus'
commands = ffmpeg_builder.chain(
ffmpeg_builder.set_input(temp_video_path),

38
tests/test_ffmpeg.py
View File

@@ -131,20 +131,19 @@ def test_read_audio_buffer() -> None:
def test_restore_audio() -> None:
target_paths =\
restore_set =\
[
get_test_example_file('target-240p-16khz.avi'),
get_test_example_file('target-240p-16khz.m4v'),
get_test_example_file('target-240p-16khz.mkv'),
get_test_example_file('target-240p-16khz.mov'),
get_test_example_file('target-240p-16khz.mp4'),
get_test_example_file('target-240p-48khz.mp4'),
get_test_example_file('target-240p-16khz.webm')
(get_test_example_file('target-240p-16khz.avi'), get_test_output_file('target-240p-16khz.avi')),
(get_test_example_file('target-240p-16khz.m4v'), get_test_output_file('target-240p-16khz.m4v')),
(get_test_example_file('target-240p-16khz.mkv'), get_test_output_file('target-240p-16khz.mkv')),
(get_test_example_file('target-240p-16khz.mov'), get_test_output_file('target-240p-16khz.mov')),
(get_test_example_file('target-240p-16khz.mp4'), get_test_output_file('target-240p-16khz.mp4')),
(get_test_example_file('target-240p-48khz.mp4'), get_test_output_file('target-240p-48khz.mp4')),
(get_test_example_file('target-240p-16khz.webm'), get_test_output_file('target-240p-16khz.webm'))
]
output_path = get_test_output_file('test-restore-audio.mp4')
output_audio_encoders = get_available_encoder_set().get('audio')
for target_path in target_paths:
for target_path, output_path in restore_set:
create_temp_directory(target_path)
for output_audio_encoder in output_audio_encoders:
@@ -160,20 +159,19 @@ def test_restore_audio() -> None:
def test_replace_audio() -> None:
target_paths =\
replace_set =\
[
get_test_example_file('target-240p-16khz.avi'),
get_test_example_file('target-240p-16khz.m4v'),
get_test_example_file('target-240p-16khz.mkv'),
get_test_example_file('target-240p-16khz.mov'),
get_test_example_file('target-240p-16khz.mp4'),
get_test_example_file('target-240p-48khz.mp4'),
get_test_example_file('target-240p-16khz.webm')
(get_test_example_file('target-240p-16khz.avi'), get_test_output_file('target-240p-16khz.avi')),
(get_test_example_file('target-240p-16khz.m4v'), get_test_output_file('target-240p-16khz.m4v')),
(get_test_example_file('target-240p-16khz.mkv'), get_test_output_file('target-240p-16khz.mkv')),
(get_test_example_file('target-240p-16khz.mov'), get_test_output_file('target-240p-16khz.mov')),
(get_test_example_file('target-240p-16khz.mp4'), get_test_output_file('target-240p-16khz.mp4')),
(get_test_example_file('target-240p-48khz.mp4'), get_test_output_file('target-240p-48khz.mp4')),
(get_test_example_file('target-240p-16khz.webm'), get_test_output_file('target-240p-16khz.webm'))
]
output_path = get_test_output_file('test-replace-audio.mp4')
output_audio_encoders = get_available_encoder_set().get('audio')
for target_path in target_paths:
for target_path, output_path in replace_set:
create_temp_directory(target_path)
for output_audio_encoder in output_audio_encoders: