mirrors/facefusion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Minor updates for testing

Browse Source
This commit is contained in:
henryruhs 2024-12-22 00:35:39 +01:00
parent 487795fec4
commit bbc485c229
2 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/test_download.py
View File

@ -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'

10
tests/test_execution.py
View File

@ -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',