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

part 1

Browse Source
This commit is contained in:
harisreedhar
2025-11-25 16:46:56 +05:30
committed by henryruhs
parent 549de72bef
commit 35c2022700
9 changed files with 48 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_ffmpeg.py
+11 -11
View File
@@ -82,12 +82,12 @@ def test_extract_frames() -> None:
]
for target_path, output_path, trim_frame_start, trim_frame_end, frame_total in test_set:
create_temp_directory(output_path)
create_temp_directory(output_path, state_manager.get_item('temp_path'))
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
assert len(resolve_temp_frame_paths(output_path, state_manager.get_item('temp_path'))) == frame_total
clear_temp_directory(output_path)
clear_temp_directory(output_path, state_manager.get_item('temp_path'))
def test_merge_video() -> None:
@@ -108,12 +108,12 @@ def test_merge_video() -> None:
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(output_path)
create_temp_directory(output_path, state_manager.get_item('temp_path'))
extract_frames(target_path, output_path, (452, 240), 25.0, 0, 1)
assert merge_video(target_path, output_path, 25.0, (452, 240), 0, 1) is True
clear_temp_directory(output_path)
clear_temp_directory(output_path, state_manager.get_item('temp_path'))
state_manager.init_item('output_video_encoder', 'libx264')
@@ -150,15 +150,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(output_path)
create_temp_directory(output_path, state_manager.get_item('temp_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(output_path))
copy_file(target_path, get_temp_file_path(output_path, state_manager.get_item('temp_path')))
assert restore_audio(target_path, output_path, 0, 270) is True
clear_temp_directory(output_path)
clear_temp_directory(output_path, state_manager.get_item('temp_path'))
state_manager.init_item('output_audio_encoder', 'aac')
@@ -177,15 +177,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(output_path)
create_temp_directory(output_path, state_manager.get_item('temp_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(output_path))
copy_file(target_path, get_temp_file_path(output_path, state_manager.get_item('temp_path')))
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(output_path)
clear_temp_directory(output_path, state_manager.get_item('temp_path'))
state_manager.init_item('output_audio_encoder', 'aac')
tests/test_temp_helper.py
+3 -6
View File
@@ -21,15 +21,12 @@ def before_all() -> None:
def test_get_temp_file_path() -> None:
state_manager.init_item('output_path', 'temp.mp4')
temp_directory = tempfile.gettempdir()
assert get_temp_file_path(get_test_example_file('target-240p.mp4')) == os.path.join(temp_directory, 'facefusion', 'target-240p', 'temp.mp4')
assert get_temp_file_path(get_test_example_file('target-240p.mp4'), state_manager.get_item('temp_path')) == os.path.join(temp_directory, 'facefusion', 'target-240p', 'temp.mp4')
def test_get_temp_directory_path() -> None:
temp_directory = tempfile.gettempdir()
assert get_temp_directory_path(get_test_example_file('target-240p.mp4')) == os.path.join(temp_directory, 'facefusion', 'target-240p')
assert get_temp_directory_path(get_test_example_file('target-240p.mp4'), state_manager.get_item('temp_path')) == os.path.join(temp_directory, 'facefusion', 'target-240p')
def test_get_temp_frames_pattern() -> None:
temp_directory = tempfile.gettempdir()
assert get_temp_frames_pattern(get_test_example_file('target-240p.mp4'), '%04d') == os.path.join(temp_directory, 'facefusion', 'target-240p', '%04d.png')
assert get_temp_frames_pattern(get_test_example_file('target-240p.mp4'), '%04d', state_manager.get_item('temp_path')) == os.path.join(temp_directory, 'facefusion', 'target-240p', '%04d.png')