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 2

Browse Source
This commit is contained in:
harisreedhar
2025-11-25 17:46:47 +05:30
committed by henryruhs
parent 320131772b
commit 6cbcd327bf
9 changed files with 47 additions and 47 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, state_manager.get_item('temp_path'))
create_temp_directory(state_manager.get_item('temp_path'), output_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, state_manager.get_item('temp_path'), state_manager.get_item('temp_frame_format'))) == frame_total
assert len(resolve_temp_frame_paths(state_manager.get_item('temp_path'), output_path, state_manager.get_item('temp_frame_format'))) == frame_total
clear_temp_directory(output_path, state_manager.get_item('temp_path'))
clear_temp_directory(state_manager.get_item('temp_path'), output_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, state_manager.get_item('temp_path'))
create_temp_directory(state_manager.get_item('temp_path'), output_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, state_manager.get_item('temp_path'))
clear_temp_directory(state_manager.get_item('temp_path'), output_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, state_manager.get_item('temp_path'))
create_temp_directory(state_manager.get_item('temp_path'), 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(output_path, state_manager.get_item('temp_path')))
copy_file(target_path, get_temp_file_path(state_manager.get_item('temp_path'), output_path))
assert restore_audio(target_path, output_path, 0, 270) is True
clear_temp_directory(output_path, state_manager.get_item('temp_path'))
clear_temp_directory(state_manager.get_item('temp_path'), output_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, state_manager.get_item('temp_path'))
create_temp_directory(state_manager.get_item('temp_path'), 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(output_path, state_manager.get_item('temp_path')))
copy_file(target_path, get_temp_file_path(state_manager.get_item('temp_path'), output_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, state_manager.get_item('temp_path'))
clear_temp_directory(state_manager.get_item('temp_path'), output_path)
state_manager.init_item('output_audio_encoder', 'aac')
tests/test_temp_helper.py
+3 -3
View File
@@ -21,12 +21,12 @@ def before_all() -> None:
def test_get_temp_file_path() -> None:
state_manager.init_item('output_path', 'temp.mp4')
assert get_temp_file_path(get_test_example_file('target-240p.mp4'), state_manager.get_item('temp_path')) == os.path.join(state_manager.get_item('temp_path'), 'facefusion', 'target-240p', 'temp.mp4')
assert get_temp_file_path(state_manager.get_item('temp_path'), get_test_example_file('target-240p.mp4')) == os.path.join(state_manager.get_item('temp_path'), 'facefusion', 'target-240p', 'temp.mp4')
def test_get_temp_directory_path() -> None:
assert get_temp_directory_path(get_test_example_file('target-240p.mp4'), state_manager.get_item('temp_path')) == os.path.join(state_manager.get_item('temp_path'), 'facefusion', 'target-240p')
assert get_temp_directory_path(state_manager.get_item('temp_path'), get_test_example_file('target-240p.mp4')) == os.path.join(state_manager.get_item('temp_path'), 'facefusion', 'target-240p')
def test_get_temp_frames_pattern() -> None:
assert get_temp_frames_pattern(get_test_example_file('target-240p.mp4'), '%04d', state_manager.get_item('temp_path'), state_manager.get_item('temp_frame_format')) == os.path.join(state_manager.get_item('temp_path'), 'facefusion', 'target-240p', '%04d.png')
assert get_temp_frames_pattern(state_manager.get_item('temp_path'), get_test_example_file('target-240p.mp4'), '%04d', state_manager.get_item('temp_frame_format')) == os.path.join(state_manager.get_item('temp_path'), 'facefusion', 'target-240p', '%04d.png')