From bbc485c22920fa19b39a72b183ff42f38e898490 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Sun, 22 Dec 2024 00:35:39 +0100 Subject: [PATCH] Minor updates for testing --- tests/test_download.py | 7 ++++++- tests/test_execution.py | 10 +++++----- 2 files changed, 11 insertions(+), 6 deletions(-) diff --git a/tests/test_download.py b/tests/test_download.py index 49865a6d..48698aad 100644 --- a/tests/test_download.py +++ b/tests/test_download.py @@ -1,4 +1,4 @@ -from facefusion.download import get_static_download_size, ping_static_url +from facefusion.download import get_static_download_size, ping_static_url, resolve_download_url_by_provider def test_get_static_download_size() -> None: @@ -11,3 +11,8 @@ def test_static_ping_url() -> None: assert ping_static_url('https://github.com') is True assert ping_static_url('https://huggingface.co') is True assert ping_static_url('invalid') is False + + +def test_resolve_download_url_by_provider() -> None: + assert resolve_download_url_by_provider('github', 'models-3.0.0', 'fairface.onnx') == 'https://github.com/facefusion/facefusion-assets/releases/download/models-3.0.0/fairface.onnx' + assert resolve_download_url_by_provider('huggingface', 'models-3.0.0', 'fairface.onnx') == 'https://huggingface.co/facefusion/models-3.0.0/resolve/main/fairface.onnx' diff --git a/tests/test_execution.py b/tests/test_execution.py index 1d1743c9..1ce83344 100644 --- a/tests/test_execution.py +++ b/tests/test_execution.py @@ -1,16 +1,16 @@ from facefusion.execution import create_inference_execution_providers, get_available_execution_providers, has_execution_provider -def test_get_execution_provider_set() -> None: - assert 'cpu' in get_available_execution_providers() - - def test_has_execution_provider() -> None: assert has_execution_provider('cpu') is True assert has_execution_provider('openvino') is False -def test_multiple_execution_providers() -> None: +def test_get_available_execution_providers() -> None: + assert 'cpu' in get_available_execution_providers() + + +def test_create_inference_execution_providers() -> None: execution_providers =\ [ ('CUDAExecutionProvider',