From 79743014153d074887229dfd691e8ffab6402b2a Mon Sep 17 00:00:00 2001 From: henryruhs Date: Tue, 4 Aug 2026 09:27:37 +0200 Subject: [PATCH] fix test --- tests/test_inference_manager.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/tests/test_inference_manager.py b/tests/test_inference_manager.py index f1e30c12..dd69d669 100644 --- a/tests/test_inference_manager.py +++ b/tests/test_inference_manager.py @@ -59,9 +59,15 @@ def test_resolve_static_inference_providers(override_module : SimpleNamespace, a resolve_static_inference_providers.cache_clear() with patch('facefusion.inference_manager.importlib', Mock(import_module = Mock(return_value = override_module))): - assert resolve_static_inference_providers('override_module', 0) == [ ('CoreMLExecutionProvider', { 'ModelFormat': 'MLProgram' }) ] + inference_providers = resolve_static_inference_providers('override_module', 0) + + assert inference_providers == [ ('CoreMLExecutionProvider', { 'ModelFormat': 'MLProgram' }) ] with patch('facefusion.inference_manager.importlib', Mock(import_module = Mock(return_value = adjust_module))): - assert resolve_static_inference_providers('adjust_module', 0) == [ ('CoreMLExecutionProvider', { 'SpecializationStrategy': 'FastPrediction', 'ModelCacheDirectory': resolve_cache_path(), 'ModelFormat': 'MLProgram' }) ] + inference_providers = resolve_static_inference_providers('adjust_module', 0) - assert resolve_static_inference_providers('test', 0) == [ ('CoreMLExecutionProvider', { 'SpecializationStrategy': 'FastPrediction', 'ModelCacheDirectory': resolve_cache_path() }) ] + assert inference_providers == [ ('CoreMLExecutionProvider', { 'SpecializationStrategy': 'FastPrediction', 'ModelCacheDirectory': resolve_cache_path(), 'ModelFormat': 'MLProgram' }) ] + + inference_providers = resolve_static_inference_providers('test', 0) + + assert inference_providers == [ ('CoreMLExecutionProvider', { 'SpecializationStrategy': 'FastPrediction', 'ModelCacheDirectory': resolve_cache_path() }) ]