feat: folder scan — depth-limited natural-sorted image listing
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,39 @@
|
||||
# tests/test_scan.py
|
||||
from gates import scan
|
||||
|
||||
def _touch(p, data=b"x"):
|
||||
p.parent.mkdir(parents=True, exist_ok=True)
|
||||
p.write_bytes(data)
|
||||
|
||||
def test_natural_sort_orders_numerically():
|
||||
items = ["img10.png", "img2.png", "img1.png"]
|
||||
assert sorted(items, key=scan.natural_key) == ["img1.png", "img2.png", "img10.png"]
|
||||
|
||||
def test_list_images_top_level_only_default(tmp_path):
|
||||
_touch(tmp_path / "a.png"); _touch(tmp_path / "b.jpg"); _touch(tmp_path / "note.txt")
|
||||
_touch(tmp_path / "sub" / "c.png")
|
||||
got = [p.split("/")[-1] for p in scan.list_images(str(tmp_path))]
|
||||
assert got == ["a.png", "b.jpg"] # depth 0: no sub/, no .txt
|
||||
|
||||
def test_list_images_depth_one(tmp_path):
|
||||
_touch(tmp_path / "a.png")
|
||||
_touch(tmp_path / "sub" / "c.png")
|
||||
_touch(tmp_path / "sub" / "deep" / "d.png")
|
||||
got = [p.split("/")[-1] for p in scan.list_images(str(tmp_path), depth=1)]
|
||||
assert got == ["a.png", "c.png"] # depth 1: include sub/, not sub/deep/
|
||||
|
||||
def test_list_images_unlimited_depth(tmp_path):
|
||||
_touch(tmp_path / "a.png"); _touch(tmp_path / "sub" / "deep" / "d.png")
|
||||
got = scan.list_images(str(tmp_path), depth=-1)
|
||||
assert len(got) == 2
|
||||
|
||||
def test_list_images_natural_sort_by_relpath(tmp_path):
|
||||
for n in ["img1.png", "img2.png", "img10.png"]:
|
||||
_touch(tmp_path / n)
|
||||
got = [p.split("/")[-1] for p in scan.list_images(str(tmp_path))]
|
||||
assert got == ["img1.png", "img2.png", "img10.png"]
|
||||
|
||||
def test_list_images_bad_path_raises(tmp_path):
|
||||
import pytest
|
||||
with pytest.raises(NotADirectoryError):
|
||||
scan.list_images(str(tmp_path / "nope"))
|
||||
Reference in New Issue
Block a user