refactor(images): add directory check

yzx9 · Apr 5, 2024 · a8007ac · a8007ac
1 parent c6b042f
commit a8007ac
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/swcgeom/images/folder.py b/swcgeom/images/folder.py
@@ -62,6 +62,9 @@ def _read(self, fname: str) -> npt.NDArray[ScalarType]:
 
     @staticmethod
     def scan(root: str, *, pattern: Optional[str] = None) -> List[str]:
+        if not os.path.isdir(root):
+            raise NotADirectoryError(f"not a directory: {root}")
+
         is_valid = re.compile(pattern).match if pattern is not None else truthly
 
         fs = []