File size: 5,454 Bytes
1a07caf |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 |
import os.path
import pytest
from facefusion.download import conditional_download
from facefusion.filesystem import create_directory, filter_audio_paths, filter_image_paths, get_file_extension, get_file_format, get_file_size, has_audio, has_image, has_video, in_directory, is_audio, is_directory, is_file, is_image, is_video, remove_directory, resolve_file_paths, same_file_extension
from .helper import get_test_example_file, get_test_examples_directory, get_test_outputs_directory
@pytest.fixture(scope = 'module', autouse = True)
def before_all() -> None:
conditional_download(get_test_examples_directory(),
[
'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/source.jpg',
'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/source.mp3',
'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/target-240p.mp4'
])
def test_get_file_size() -> None:
assert get_file_size(get_test_example_file('source.jpg')) == 549458
assert get_file_size('invalid') == 0
def test_get_file_extension() -> None:
assert get_file_extension('source.jpg') == '.jpg'
assert get_file_extension('source.mp3') == '.mp3'
assert get_file_extension('invalid') is None
def test_get_file_format() -> None:
assert get_file_format('source.jpg') == 'jpeg'
assert get_file_format('source.jpeg') == 'jpeg'
assert get_file_format('source.mp3') == 'mp3'
assert get_file_format('invalid') is None
def test_same_file_extension() -> None:
assert same_file_extension('source.jpg', 'source.jpg') is True
assert same_file_extension('source.jpg', 'source.mp3') is False
assert same_file_extension('invalid', 'invalid') is False
def test_is_file() -> None:
assert is_file(get_test_example_file('source.jpg')) is True
assert is_file(get_test_examples_directory()) is False
assert is_file('invalid') is False
def test_is_audio() -> None:
assert is_audio(get_test_example_file('source.mp3')) is True
assert is_audio(get_test_example_file('source.jpg')) is False
assert is_audio('invalid') is False
def test_has_audio() -> None:
assert has_audio([ get_test_example_file('source.mp3') ]) is True
assert has_audio([ get_test_example_file('source.mp3'), get_test_example_file('source.jpg') ]) is True
assert has_audio([ get_test_example_file('source.jpg'), get_test_example_file('source.jpg') ]) is False
assert has_audio([ 'invalid' ]) is False
def test_is_image() -> None:
assert is_image(get_test_example_file('source.jpg')) is True
assert is_image(get_test_example_file('target-240p.mp4')) is False
assert is_image('invalid') is False
def test_has_image() -> None:
assert has_image([ get_test_example_file('source.jpg') ]) is True
assert has_image([ get_test_example_file('source.jpg'), get_test_example_file('source.mp3') ]) is True
assert has_image([ get_test_example_file('source.mp3'), get_test_example_file('source.mp3') ]) is False
assert has_image([ 'invalid' ]) is False
def test_is_video() -> None:
assert is_video(get_test_example_file('target-240p.mp4')) is True
assert is_video(get_test_example_file('source.jpg')) is False
assert is_video('invalid') is False
def test_has_video() -> None:
assert has_video([ get_test_example_file('target-240p.mp4') ]) is True
assert has_video([ get_test_example_file('target-240p.mp4'), get_test_example_file('source.mp3') ]) is True
assert has_video([ get_test_example_file('source.mp3'), get_test_example_file('source.mp3') ]) is False
assert has_video([ 'invalid' ]) is False
def test_filter_audio_paths() -> None:
assert filter_audio_paths([ get_test_example_file('source.jpg'), get_test_example_file('source.mp3') ]) == [ get_test_example_file('source.mp3') ]
assert filter_audio_paths([ get_test_example_file('source.jpg'), get_test_example_file('source.jpg') ]) == []
assert filter_audio_paths([ 'invalid' ]) == []
def test_filter_image_paths() -> None:
assert filter_image_paths([ get_test_example_file('source.jpg'), get_test_example_file('source.mp3') ]) == [ get_test_example_file('source.jpg') ]
assert filter_image_paths([ get_test_example_file('source.mp3'), get_test_example_file('source.mp3') ]) == []
assert filter_audio_paths([ 'invalid' ]) == []
def test_resolve_file_paths() -> None:
file_paths = resolve_file_paths(get_test_examples_directory())
for file_path in file_paths:
assert file_path == get_test_example_file(file_path)
assert resolve_file_paths('invalid') == []
def test_create_directory() -> None:
create_directory_path = os.path.join(get_test_outputs_directory(), 'create_directory')
assert create_directory(create_directory_path) is True
assert create_directory(get_test_example_file('source.jpg')) is False
def test_remove_directory() -> None:
remove_directory_path = os.path.join(get_test_outputs_directory(), 'remove_directory')
create_directory(remove_directory_path)
assert remove_directory(remove_directory_path) is True
assert remove_directory(get_test_example_file('source.jpg')) is False
assert remove_directory('invalid') is False
def test_is_directory() -> None:
assert is_directory(get_test_examples_directory()) is True
assert is_directory(get_test_example_file('source.jpg')) is False
assert is_directory('invalid') is False
def test_in_directory() -> None:
assert in_directory(get_test_example_file('source.jpg')) is True
assert in_directory('source.jpg') is False
assert in_directory('invalid') is False
|