Switch to PyTest WIP

Author: JerilynFranz

pyproject.toml

@@ -51,26 +51,29 @@ classifiers = [
 requires-python = ">=3.10"
 dependencies = []
 
-optional-dependencies = {
-    dev = [
-        "flake8",
-        "black",
-        "pylint",
-        "pytest",
-        "pytest-cov",
-        "pytest-mock",
-        "pytest-xdist",
-        "testplan",
-        "coverage",
-        "mypy",
-    ],
-    docs = [
-        "sphinx",
-        "sphinx-rtd-theme",
-        "sphinx-autodoc-typehints",
-    ],
-}
-
+[project.optional-dependencies]
+tests = [
+    "pytest",
+    "testplan",
+    "coverage"
+]
+dev = [
+    "flake8",
+    "black",
+    "pylint",
+    "pytest",
+    "pytest-cov",
+    "pytest-mock",
+    "pytest-xdist",
+    "testplan",
+    "coverage",
+    "mypy"
+]
+docs = [
+    "sphinx",
+    "sphinx-rtd-theme",
+    "sphinx-autodoc-typehints"
+]
 
 [project.urls]
 Homepage = "https://github.com/JerilynFranz/python-gen-tries"

test_plan.py

@@ -1,32 +1,47 @@
 #!/usr/bin/env python3
-"""Testplan for testing the gentri module."""
+"""Pytest launcher for gentrie tests.
 
-# pylint: disable=import-error
-import sys
-
-from testplan import test_plan  # type: ignore
-from testplan.testing import pyunit  # type: ignore
+Usage:
+  ./test_plan.py                # run all tests under tests/gentrie
+  ./test_plan.py KeyToken       # pattern filter (class or test name substring / -k expression)
 
-from tests.gentrie import test_gentri
+Environment (optional):
+  GENTRIE_FAIL_FAST=1           # enable fail-fast (-x)
+"""
 
-# pylint: disable=missing-function-docstring
+from __future__ import annotations
 
-
-@test_plan(name='PyUnitGentri', description='PyUnit gentri tests')
-def main(plan):  # type: ignore
-    plan.add(  # type: ignore
-        pyunit.PyUnit(
-            name="gen-trie tests",
-            description="PyUnit testcases for the gentri module",
-            testcases=[test_gentri.TestTrieKeyToken,
-                       test_gentri.TestTrieId,
-                       test_gentri.TestGeneralizedKey,
-                       test_gentri.TestGeneralizedTrie],
-        )
-    )
+import os
+import sys
+import pytest
+
+
+def build_pytest_args(pattern: str | None) -> list[str]:
+    args: list[str] = []
+    # Fail fast
+    if os.environ.get("GENTRIE_FAIL_FAST") == "1":
+        args.append("-x")
+    # Quiet by default; adjust as needed
+    args.extend(["-q", "--disable-warnings"])
+    # Add test path
+    args.append("tests/gentrie")
+    # Pattern handling: allow either simple substring or full -k expression
+    if pattern:
+        # If user already passed pytest expression operators, trust it
+        if any(op in pattern for op in (" and ", " or ", " not ", "(", ")")):
+            args.extend(["-k", pattern])
+        else:
+            # Simple substring: wrap to match anywhere in node id
+            expr = pattern
+            args.extend(["-k", expr])
+    return args
+
+
+def main() -> int:
+    pattern = sys.argv[1] if len(sys.argv) > 1 else None
+    args = build_pytest_args(pattern)
+    return pytest.main(args)
 
 
 if __name__ == "__main__":
-    # pylint: disable=invalid-name, no-value-for-parameter
-    res = main()  # type: ignore
-    sys.exit(res.exit_code)  # type: ignore
+    sys.exit(main())