Add new logic

pre_commit_hooks/forbid_articles_in_test_filenames.py

@@ -0,0 +1,48 @@
+#!/usr/bin/env python3
+
+import subprocess
+import sys
+from pathlib import Path
+
+FORBIDDEN = {"a", "an", "the"}
+
+
+def git_ls_python_files():
+    result = subprocess.run(
+        ["git", "ls-files", "*.py"],
+        capture_output=True,
+        text=True,
+        check=True,
+    )
+    return [Path(p) for p in result.stdout.splitlines()]
+
+
+def is_test_file(path: Path) -> bool:
+    name = path.name
+    return (
+        name.startswith("test_")
+        or name.startswith("tests_")
+        or name.endswith("_test.py")
+    )
+
+
+def has_forbidden_article(path: Path) -> bool:
+    parts = path.stem.split("_")
+    return any(part in FORBIDDEN for part in parts)
+
+
+def main() -> int:
+    for path in git_ls_python_files():
+        if not is_test_file(path):
+            continue
+
+        if has_forbidden_article(path):
+            print("ERROR: Forbidden article in test filename:")
+            print(path)
+            return 1
+
+    return 0
+
+
+if __name__ == "__main__":
+    sys.exit(main())