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

Testing for restore audio

Browse Source
This commit is contained in:
henryruhs 2024-10-12 22:51:25 +02:00
parent 56f7bbcf7f
commit 784cb6c330
1 changed files with 26 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
tests/test_ffmpeg.py
View File

@ -5,7 +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, read_audio_buffer, replace_audio
from facefusion.ffmpeg import concat_video, extract_frames, read_audio_buffer, restore_audio, replace_audio
from facefusion.filesystem import copy_file
from facefusion.temp_helper import clear_temp_directory, create_temp_directory, get_temp_directory_path, get_temp_file_path
from .helper import get_test_example_file, get_test_examples_directory, get_test_output_file, prepare_test_output_directory
@ -24,6 +24,8 @@ def before_all() -> None:
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') ])
subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vf', 'fps=60', get_test_example_file('target-240p-60fps.mp4') ])
subprocess.run([ 'ffmpeg', '-i', get_test_example_file('source.mp3'), '-i', get_test_example_file('target-240p.mp4'), '-ar', '16000', get_test_example_file('target-240p-16khz.mp4') ])
subprocess.run([ 'ffmpeg', '-i', get_test_example_file('source.mp3'), '-i', get_test_example_file('target-240p.mp4'), '-ar', '48000', get_test_example_file('target-240p-48khz.mp4') ])
state_manager.init_item('temp_frame_format', 'jpg')
state_manager.init_item('output_audio_encoder', 'aac')
@ -55,14 +57,14 @@ def test_extract_frames() -> None:
def test_extract_frames_with_trim_start() -> None:
state_manager.init_item('trim_frame_start', 224)
providers =\
target_paths =\
[
(get_test_example_file('target-240p-25fps.mp4'), 55),
(get_test_example_file('target-240p-30fps.mp4'), 100),
(get_test_example_file('target-240p-60fps.mp4'), 212)
]
for target_path, frame_total in providers:
for target_path, frame_total in target_paths:
temp_directory_path = get_temp_directory_path(target_path)
create_temp_directory(target_path)
@ -75,14 +77,14 @@ def test_extract_frames_with_trim_start() -> None:
def test_extract_frames_with_trim_start_and_trim_end() -> None:
state_manager.init_item('trim_frame_start', 124)
state_manager.init_item('trim_frame_end', 224)
providers =\
target_paths =\
[
(get_test_example_file('target-240p-25fps.mp4'), 120),
(get_test_example_file('target-240p-30fps.mp4'), 100),
(get_test_example_file('target-240p-60fps.mp4'), 50)
]
for target_path, frame_total in providers:
for target_path, frame_total in target_paths:
temp_directory_path = get_temp_directory_path(target_path)
create_temp_directory(target_path)
@ -94,14 +96,14 @@ def test_extract_frames_with_trim_start_and_trim_end() -> None:
def test_extract_frames_with_trim_end() -> None:
state_manager.init_item('trim_frame_end', 100)
providers =\
target_paths =\
[
(get_test_example_file('target-240p-25fps.mp4'), 120),
(get_test_example_file('target-240p-30fps.mp4'), 100),
(get_test_example_file('target-240p-60fps.mp4'), 50)
]
for target_path, frame_total in providers:
for target_path, frame_total in target_paths:
temp_directory_path = get_temp_directory_path(target_path)
create_temp_directory(target_path)
@ -128,6 +130,23 @@ def test_read_audio_buffer() -> None:
assert read_audio_buffer(get_test_example_file('invalid.mp3'), 1, 1) is None
def test_restore_audio() -> None:
target_paths =\
[
get_test_example_file('target-240p-16khz.mp4'),
get_test_example_file('target-240p-48khz.mp4')
]
output_path = get_test_output_file('test-restore-audio.mp4')
for target_path in target_paths:
create_temp_directory(target_path)
copy_file(target_path, get_temp_file_path(target_path))
assert restore_audio(target_path, output_path, 30) is True
clear_temp_directory(target_path)
def test_replace_audio() -> None:
target_path = get_test_example_file('target-240p.mp4')
output_path = get_test_output_file('test-replace-audio.mp4')