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:37:41 +02:00
parent 4021ddf3b7
commit 441305afe5
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/test_ffmpeg.py
View File

@@ -33,10 +33,10 @@ def before_all() -> None:
state_manager.init_item('temp_path', tempfile.gettempdir()) state_manager.init_item('temp_path', tempfile.gettempdir())
state_manager.init_item('temp_frame_format', 'png') state_manager.init_item('temp_frame_format', 'png')
state_manager.init_item('output_audio_encoder', 'aac') state_manager.init_item('output_audio_encoder', 'aac')
state_manager.init_item('output_audio_quality', 80) state_manager.init_item('output_audio_quality', 100)
state_manager.init_item('output_audio_volume', 100) state_manager.init_item('output_audio_volume', 100)
state_manager.init_item('output_video_encoder', 'libx264') state_manager.init_item('output_video_encoder', 'libx264')
state_manager.init_item('output_video_quality', 80) state_manager.init_item('output_video_quality', 100)
state_manager.init_item('output_video_preset', 'ultrafast') state_manager.init_item('output_video_preset', 'ultrafast')
@@ -45,7 +45,6 @@ def before_each() -> None:
prepare_test_output_directory() prepare_test_output_directory()
@pytest.mark.stub
def get_available_encoder_set() -> EncoderSet: def get_available_encoder_set() -> EncoderSet:
if os.getenv('CI'): if os.getenv('CI'):
return\ return\
@@ -146,9 +145,11 @@ def test_restore_audio() -> None:
output_audio_encoders = get_available_encoder_set().get('audio') output_audio_encoders = get_available_encoder_set().get('audio')
for target_path in target_paths: for target_path in target_paths:
create_temp_directory(target_path)
for output_audio_encoder in output_audio_encoders: for output_audio_encoder in output_audio_encoders:
state_manager.init_item('output_audio_encoder', output_audio_encoder) state_manager.init_item('output_audio_encoder', output_audio_encoder)
create_temp_directory(target_path)
copy_file(target_path, get_temp_file_path(target_path)) copy_file(target_path, get_temp_file_path(target_path))
assert restore_audio(target_path, output_path, 0, 270) is True assert restore_audio(target_path, output_path, 0, 270) is True
@@ -173,9 +174,10 @@ def test_replace_audio() -> None:
output_audio_encoders = get_available_encoder_set().get('audio') output_audio_encoders = get_available_encoder_set().get('audio')
for target_path in target_paths: for target_path in target_paths:
create_temp_directory(target_path)
for output_audio_encoder in output_audio_encoders: for output_audio_encoder in output_audio_encoders:
state_manager.init_item('output_audio_encoder', output_audio_encoder) state_manager.init_item('output_audio_encoder', output_audio_encoder)
create_temp_directory(target_path)
copy_file(target_path, get_temp_file_path(target_path)) copy_file(target_path, get_temp_file_path(target_path))
assert replace_audio(target_path, get_test_example_file('source.mp3'), output_path) is True assert replace_audio(target_path, get_test_example_file('source.mp3'), output_path) is True