Add testing for walk() function

tests/test_walk.py

@@ -0,0 +1,32 @@
+from pathlib import Path
+from random import choice
+from string import ascii_lowercase
+from typing import List, Callable
+
+from phototag.helpers import walk
+
+letter: Callable[[], str] = lambda: choice(ascii_lowercase)
+extension: Callable[[], str] = lambda: choice(['jpeg', 'jpg', 'png'])
+file: Callable[[], str] = lambda: f"{letter()}.{extension()}"
+
+
+def test_walk(tmp_path: Path):
+    files: List[Path] = [tmp_path / file(), tmp_path / file()]
+
+    current_dir: Path = tmp_path
+    sub_dir_counts: List[int] = [2, 4, 2]
+    for file_count in sub_dir_counts:
+        current_dir = current_dir / letter()
+        current_dir.mkdir()
+
+        for _ in range(file_count):
+            path = current_dir / file()
+            files.append(path)
+            path.touch()
+
+    assert len(list(walk(tmp_path))) == sum(sub_dir_counts), "All files"
+    assert len(list(walk(tmp_path, depth=1))) == sum(sub_dir_counts[:1]), "Depth 1"
+    assert len(list(walk(tmp_path, depth=2))) == sum(sub_dir_counts[:2]), "Depth 2"
+    assert len(list(walk(tmp_path, depth=3))) == sum(sub_dir_counts[:3]), "Depth 3"
+
+    return files