vnyash
24 lines
1.1 KiB
Python
24 lines
1.1 KiB
Python
import pytest
|
|
|
|
from deepfuze.download import conditional_download, get_download_size, is_download_done
|
|
|
|
|
|
@pytest.fixture(scope = 'module', autouse = True)
|
|
def before_all() -> None:
|
|
conditional_download('../../models/facefusion/examples',
|
|
[
|
|
'https://github.com/facefusion/facefusion-assets/releases/download/examples/target-240p.mp4'
|
|
])
|
|
|
|
|
|
def test_get_download_size() -> None:
|
|
assert get_download_size('https://github.com/facefusion/facefusion-assets/releases/download/examples/target-240p.mp4') == 191675
|
|
assert get_download_size('https://github.com/facefusion/facefusion-assets/releases/download/examples/target-360p.mp4') == 370732
|
|
assert get_download_size('invalid') == 0
|
|
|
|
|
|
def test_is_download_done() -> None:
|
|
assert is_download_done('https://github.com/facefusion/facefusion-assets/releases/download/examples/target-240p.mp4', '../../models/facefusion/examples/target-240p.mp4') is True
|
|
assert is_download_done('https://github.com/facefusion/facefusion-assets/releases/download/examples/target-240p.mp4', 'invalid') is False
|
|
assert is_download_done('invalid', 'invalid') is False
|