CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-08-12 20:10:21 +02:00
Code Issues Packages Projects Releases Wiki Activity

Remove upload queue & media chunk reader (#1076)

Browse Source 
* remove upload queue & media chunk reader

* remove macos large

---------

Co-authored-by: henryruhs <info@henryruhs.com>
This commit is contained in:
Harisreedhar
2026-04-08 15:52:04 +05:30
committed by GitHub
co-authored by henryruhs
parent a3ed37f50c
commit 5ce181b33d
5 changed files with 29 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/assert_helper.py
-7
View File
@@ -1,7 +1,5 @@
import io
import os
import tempfile
from functools import partial
from facefusion.filesystem import are_images, create_directory, is_directory, is_file, remove_directory, resolve_file_paths
from facefusion.types import JobStatus
@@ -48,8 +46,3 @@ def prepare_test_output_directory() -> bool:
remove_directory(test_outputs_directory)
create_directory(test_outputs_directory)
return is_directory(test_outputs_directory)
def create_media_reader(file_path : str) -> partial[bytes]: #todo: kill this
file_buffer = io.BytesIO(open(file_path, 'rb').read())
return partial(file_buffer.read, 1024)
tests/test_ffmpeg.py
+9 -6
View File
@@ -12,7 +12,7 @@ from facefusion.ffprobe import probe_entries
from facefusion.filesystem import copy_file, is_image
from facefusion.temp_helper import clear_temp_directory, create_temp_directory, get_temp_file_path, resolve_temp_frame_paths
from facefusion.types import EncoderSet
from .assert_helper import create_media_reader, get_test_example_file, get_test_examples_directory, get_test_output_path, prepare_test_output_directory
from .assert_helper import get_test_example_file, get_test_examples_directory, get_test_output_path, prepare_test_output_directory
@pytest.fixture(scope = 'module', autouse = True)
@@ -216,37 +216,40 @@ def test_replace_audio() -> None:
def test_sanitize_audio() -> None:
file_path = get_test_example_file('source.wav')
file_content = open(file_path, 'rb').read()
output_paths =\
[
get_test_output_path('test-sanitize-audio-strict.mp3'),
get_test_output_path('test-sanitize-audio-moderate.wav')
]
assert sanitize_audio(create_media_reader(file_path), output_paths[0], 'strict') is True
assert sanitize_audio(file_content, output_paths[0], 'strict') is True
assert probe_entries(output_paths[0], [ 'codec_name' ]).get('codec_name') == 'mp3'
assert sanitize_audio(create_media_reader(file_path), output_paths[1], 'moderate') is True
assert sanitize_audio(file_content, output_paths[1], 'moderate') is True
assert probe_entries(output_paths[1], [ 'codec_name' ]).get('codec_name') == 'pcm_s16le'
def test_sanitize_image() -> None:
file_path = get_test_example_file('source.jpg')
file_content = open(file_path, 'rb').read()
output_path = get_test_output_path('test-sanitize-image.jpg')
assert sanitize_image(create_media_reader(file_path), output_path) is True
assert sanitize_image(file_content, output_path) is True
assert is_image(output_path) is True
def test_sanitize_video() -> None:
file_path = get_test_example_file('target-240p-h265.mp4')
file_content = open(file_path, 'rb').read()
output_paths =\
[
get_test_output_path('test-sanitize-video-strict.mp4'),
get_test_output_path('test-sanitize-video-moderate.mp4')
]
assert sanitize_video(create_media_reader(file_path), output_paths[0], 'strict') is True
assert sanitize_video(file_content, output_paths[0], 'strict') is True
assert probe_entries(output_paths[0], [ 'codec_name' ]).get('codec_name') == 'h264'
assert sanitize_video(create_media_reader(file_path), output_paths[1], 'moderate') is True
assert sanitize_video(file_content, output_paths[1], 'moderate') is True
assert probe_entries(output_paths[1], [ 'codec_name' ]).get('codec_name') == 'hevc'