github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-05 19:15:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Test image_utils.yield_images returns the same files as image_utils.list_images

Browse Source
This commit is contained in:
Samuel J. Woodward 2025-01-06 10:11:10 -05:00
parent 2aa8ebfec8
commit 6a7505269c
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tests/test_find.py
View File

@ -95,12 +95,23 @@ def test_filetype_for_find():
def test_filetype_for_find_bulk_embeddings(): def test_filetype_for_find_bulk_embeddings():
imgs = image_utils.list_images("dataset") # List
list_imgs = image_utils.list_images("dataset")
assert len(imgs) > 0 assert len(list_imgs) > 0
# img47 is webp even though its extension is jpg # img47 is webp even though its extension is jpg
assert "dataset/img47.jpg" not in imgs assert "dataset/img47.jpg" not in list_imgs
# Generator
gen_imgs = list(image_utils.yield_images("dataset"))
assert len(gen_imgs) > 0
# img47 is webp even though its extension is jpg
assert "dataset/img47.jpg" not in gen_imgs
assert gen_imgs == list_imgs
def test_find_without_refresh_database(): def test_find_without_refresh_database():