CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-06-08 05:33:53 +02:00
Code Issues Packages Projects Releases Wiki Activity

refactor temp handling from target-path to output-path

Browse Source
This commit is contained in:
harisreedhar
2025-11-23 08:14:51 +05:30
committed by henryruhs
parent d548353cf6
commit 5ca7f3d0dc
6 changed files with 48 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_ffmpeg.py
+18 -16
View File
@@ -82,12 +82,13 @@ def test_extract_frames() -> None:
]
for target_path, trim_frame_start, trim_frame_end, frame_total in test_set:
create_temp_directory(target_path)
output_path = get_test_output_file(f'test-extract-frames-{trim_frame_start}-{trim_frame_end}.mp4')
create_temp_directory(output_path)
assert extract_frames(target_path, (452, 240), 30.0, trim_frame_start, trim_frame_end) is True
assert len(resolve_temp_frame_paths(target_path)) == frame_total
assert extract_frames(target_path, output_path, (452, 240), 30.0, trim_frame_start, trim_frame_end) is True
assert len(resolve_temp_frame_paths(output_path)) == frame_total
clear_temp_directory(target_path)
clear_temp_directory(output_path)
def test_merge_video() -> None:
@@ -105,15 +106,16 @@ def test_merge_video() -> None:
for target_path in target_paths:
for output_video_encoder in output_video_encoders:
output_path = get_test_output_file(f'test-merge-video-{output_video_encoder}.mp4')
state_manager.init_item('output_path', target_path)
state_manager.init_item('output_video_fps', 25.0)
state_manager.init_item('output_video_encoder', output_video_encoder)
create_temp_directory(target_path)
extract_frames(target_path, (452, 240), 25.0, 0, 1)
create_temp_directory(output_path)
extract_frames(target_path, output_path, (452, 240), 25.0, 0, 1)
assert merge_video(target_path, 25.0, (452, 240), 0, 1) is True
assert merge_video(target_path, output_path, 25.0, (452, 240), 0, 1) is True
clear_temp_directory(target_path)
clear_temp_directory(output_path)
state_manager.init_item('output_video_encoder', 'libx264')
@@ -150,15 +152,15 @@ def test_restore_audio() -> None:
output_audio_encoders = get_available_encoder_set().get('audio')
for target_path, output_path in test_set:
create_temp_directory(target_path)
create_temp_directory(output_path)
for output_audio_encoder in output_audio_encoders:
state_manager.init_item('output_audio_encoder', output_audio_encoder)
copy_file(target_path, get_temp_file_path(target_path))
copy_file(target_path, get_temp_file_path(output_path))
assert restore_audio(target_path, output_path, 0, 270) is True
clear_temp_directory(target_path)
clear_temp_directory(output_path)
state_manager.init_item('output_audio_encoder', 'aac')
@@ -177,15 +179,15 @@ def test_replace_audio() -> None:
output_audio_encoders = get_available_encoder_set().get('audio')
for target_path, output_path in test_set:
create_temp_directory(target_path)
create_temp_directory(output_path)
for output_audio_encoder in output_audio_encoders:
state_manager.init_item('output_audio_encoder', output_audio_encoder)
copy_file(target_path, get_temp_file_path(target_path))
copy_file(target_path, get_temp_file_path(output_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.wav'), output_path) is True
assert replace_audio(get_test_example_file('source.mp3'), output_path) is True
assert replace_audio(get_test_example_file('source.wav'), output_path) is True
clear_temp_directory(target_path)
clear_temp_directory(output_path)
state_manager.init_item('output_audio_encoder', 'aac')