OSSForks/faceswap
1
0
Fork 0
mirror of https://github.com/zebrajr/faceswap.git synced 2025-12-06 00:20:09 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

bugfix - utils unit test

Browse Source
This commit is contained in:
torzdf 2023-02-20 12:12:51 +00:00
parent f2c1086f94
commit 637dc3b55c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/lib/utils_test.py
View File

@ -139,12 +139,12 @@ def test_get_image_paths(tmp_path: str) -> None:
# Test getting any image paths from a folder with images and random files
exists = [os.path.join(test_folder, img)
for img in os.listdir(test_folder) if os.path.splitext(img)[-1] != ".txt"]
assert get_image_paths(test_folder) == exists
assert sorted(get_image_paths(test_folder)) == sorted(exists)
# Test getting image paths from a folder with images with a specific extension
exists = [os.path.join(test_folder, img)
for img in os.listdir(test_folder) if os.path.splitext(img)[-1] == ".png"]
assert get_image_paths(test_folder, extension=".png") == exists
assert sorted(get_image_paths(test_folder, extension=".png")) == sorted(exists)
_PARAMS = [("/path/to/file.txt", ["/", "path", "to", "file.txt"]), # Absolute