CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-07-06 10:37:54 +02:00
Code Issues Packages Projects Releases Wiki Activity

asset validation and image encoder lookup (#1058)

Browse Source 
* asset validation and image encoder lookup

* asset validation and image encoder lookup

* asset validation and image encoder lookup
This commit is contained in:
Henry Ruhs
2026-03-11 10:52:01 +01:00
committed by henryruhs
parent 246d48e079
commit efab505adb
9 changed files with 140 additions and 100 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_asset_helper.py
+4 -4
View File
@@ -1,6 +1,6 @@
import pytest
from facefusion.apis.asset_helper import detect_media_type, extract_audio_metadata, extract_image_metadata, extract_video_metadata
from facefusion.apis.asset_helper import detect_media_type_by_path, extract_audio_metadata, extract_image_metadata, extract_video_metadata
from facefusion.download import conditional_download
from .assert_helper import get_test_example_file, get_test_examples_directory
@@ -16,9 +16,9 @@ def before_all() -> None:
def test_detect_media_type() -> None:
assert detect_media_type(get_test_example_file('source.jpg')) == 'image'
assert detect_media_type(get_test_example_file('target-240p.mp4')) == 'video'
assert detect_media_type(get_test_example_file('source.mp3')) == 'audio'
assert detect_media_type_by_path(get_test_example_file('source.jpg')) == 'image'
assert detect_media_type_by_path(get_test_example_file('target-240p.mp4')) == 'video'
assert detect_media_type_by_path(get_test_example_file('source.mp3')) == 'audio'
def test_extract_image_metadata() -> None:
tests/test_ffmpeg.py
+2
View File
@@ -52,6 +52,7 @@ def get_available_encoder_set() -> EncoderSet:
return\
{
'audio': [ 'aac' ],
'image': [ 'png' ],
'video': [ 'libx264' ]
}
return facefusion.ffmpeg.get_available_encoder_set()
@@ -61,6 +62,7 @@ def test_get_available_encoder_set() -> None:
available_encoder_set = get_available_encoder_set()
assert 'aac' in available_encoder_set.get('audio')
assert 'png' in available_encoder_set.get('image')
assert 'libx264' in available_encoder_set.get('video')