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

hopefully fix test

Browse Source
This commit is contained in:
harisreedhar
2026-02-22 14:13:56 +05:30
committed by henryruhs
parent 887a6ac1c7
commit 1f0af5906d
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_api_process.py
+10 -4
View File
@@ -6,12 +6,12 @@ import numpy
import pytest
from starlette.testclient import TestClient
from facefusion import metadata, session_manager, state_manager
from facefusion import face_classifier, face_detector, face_landmarker, face_masker, face_recognizer, metadata, session_manager, state_manager
from facefusion.apis import asset_store
from facefusion.apis.core import create_api
from facefusion.args_helper import apply_args
from facefusion.core import common_pre_check, processors_pre_check
from facefusion.download import conditional_download
from facefusion.processors.core import get_processors_modules
from facefusion.program import collect_step_program
from .helper import get_test_example_file, get_test_examples_directory
@@ -35,8 +35,14 @@ def test_client() -> Iterator[TestClient]:
state_manager.init_item('temp_path', tempfile.gettempdir())
state_manager.init_item('download_providers', [ 'github', 'huggingface' ])
state_manager.init_item('face_selector_mode', 'many')
common_pre_check()
processors_pre_check()
face_classifier.pre_check()
face_detector.pre_check()
face_landmarker.pre_check()
face_masker.pre_check()
face_recognizer.pre_check()
for processor_module in get_processors_modules(state_manager.get_item('processors')):
processor_module.pre_check()
with TestClient(create_api()) as test_client:
yield test_client