robotframework-testselection 0.2.0__tar.gz → 0.3.0__tar.gz

{robotframework_testselection-0.2.0 → robotframework_testselection-0.3.0}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: robotframework-testselection
-Version: 0.2.0
-Summary: Vector-based diverse test case selection for Robot Framework
+Version: 0.3.0
+Summary: Vector-based diverse test case selection for Robot Framework and pytest
 Project-URL: Homepage, https://github.com/manykarim/robotframework-testselection
 Project-URL: Documentation, https://github.com/manykarim/robotframework-testselection#readme
 Project-URL: Repository, https://github.com/manykarim/robotframework-testselection
@@ -10,8 +10,9 @@ Project-URL: Changelog, https://github.com/manykarim/robotframework-testselectio
 Author: Many Kasiriha
 License: Apache-2.0
 License-File: LICENSE
-Keywords: diversity,embeddings,machine-learning,nlp,robotframework,test-selection,testing
+Keywords: diversity,embeddings,machine-learning,nlp,pytest,robotframework,test-selection,testing
 Classifier: Development Status :: 3 - Alpha
+Classifier: Framework :: Pytest
 Classifier: Framework :: Robot Framework
 Classifier: Framework :: Robot Framework :: Library
 Classifier: Framework :: Robot Framework :: Tool
@@ -58,7 +59,7 @@ Description-Content-Type: text/markdown
 
 # robotframework-testselection
 
-Vector-based diverse test case selection for Robot Framework. Embeds test cases as semantic vectors and selects maximally diverse subsets to reduce test suite execution time while preserving coverage breadth.
+Vector-based diverse test case selection for Robot Framework and pytest. Embeds test cases as semantic vectors and selects maximally diverse subsets to reduce test suite execution time while preserving coverage breadth.
 
 ## How It Works
 
@@ -101,6 +102,8 @@ uv sync --extra vectorize
 uv sync --extra all
 ```
 
+The pytest plugin is included in the base install and activates when you pass `--diverse-k` to pytest.
+
 ### Dependency Groups
 
 | Group | Packages | Purpose |
@@ -182,6 +185,38 @@ testcase-select run --suite tests/ --k 50 \
 
 This works with both `run` and `execute` subcommands, including during graceful fallback (when selection fails and all tests are run).
 
+### pytest Support
+
+The diversity selection algorithm also works with pytest test suites, via a pytest plugin:
+
+```bash
+# Select 20 most diverse tests
+pytest --diverse-k=20 tests/
+
+# With custom strategy and seed
+pytest --diverse-k=30 --diverse-strategy=fps_multi --diverse-seed=123 tests/
+
+# Filter by markers before selection
+pytest --diverse-k=20 --diverse-include-markers slow integration tests/
+
+# Group parametrized tests (select at group level)
+pytest --diverse-k=20 --diverse-group-parametrize tests/
+```
+
+Or via the `testcase-select` CLI:
+
+```bash
+# Full pipeline with pytest
+testcase-select run --framework pytest --suite tests/ --k 20 --strategy fps
+
+# Stage-by-stage
+testcase-select vectorize --framework pytest --suite tests/ --output ./artifacts/
+testcase-select select --artifacts ./artifacts/ --k 20
+testcase-select execute --framework pytest --suite tests/ --selection selected_tests.json
+```
+
+The pytest plugin is installed automatically and activated by `--diverse-k`. It uses AST-based text representation with sentence embeddings for test diversity analysis.
+
 ### Direct Robot Framework Integration
 
 You can also use the components directly with `robot`:
@@ -347,6 +382,11 @@ src/TestSelection/
     prerun_modifier.py # DiversePreRunModifier (SuiteVisitor)
     listener.py        # DiverseDataDriverListener (Listener v3)
     runner.py          # ExecutionRunner
+  pytest/              # pytest integration
+    plugin.py          # pytest plugin (pytest11 entry point)
+    collector.py       # Programmatic test collection
+    text_builder.py    # AST-based text representation
+    runner.py          # pytest execution
   pipeline/            # Orchestration layer
     vectorize.py       # Stage 1 orchestrator
     select.py          # Stage 2 orchestrator
@@ -401,6 +441,68 @@ uv run pytest -m integration
 uv run pytest -m benchmark
 ```
 
+## Coverage Retention Analysis
+
+Measured on [robotframework-doctestlibrary](https://github.com/manykarim/robotframework-doctestlibrary) (5,045 statements in the DocTest package). All strategies use `all-MiniLM-L6-v2` embeddings (384-dim), seed=42.
+
+### pytest (594 tests, 70.1% full coverage)
+
+| Strategy | Selection | Tests | Coverage | Retention | Speedup |
+|---|---|---|---|---|---|
+| *(full suite)* | 100% | 594 | 70.1% | baseline | 1.0x |
+| **fps** | **50%** | **297** | **65.9%** | **93.9%** | **3.4x** |
+| dpp | 50% | 297 | 65.9% | 93.9% | 3.5x |
+| kmedoids | 50% | 297 | 65.8% | 93.8% | 3.1x |
+| fps_multi | 50% | 297 | 65.7% | 93.7% | 3.0x |
+| facility | 50% | 297 | 64.9% | 92.6% | 2.9x |
+| random | 50% | 297 | 64.2% | 91.5% | 2.3x |
+| **facility** | **20%** | **119** | **58.6%** | **83.6%** | **7.5x** |
+| kmedoids | 20% | 119 | 58.3% | 83.1% | 6.7x |
+| fps | 20% | 119 | 56.1% | 80.0% | 6.7x |
+| fps_multi | 20% | 119 | 53.8% | 76.8% | 8.0x |
+| random | 20% | 119 | 53.2% | 75.8% | 5.4x |
+| dpp | 20% | 119 | 47.6% | 67.8% | 6.0x |
+| **fps_multi** | **10%** | **60** | **47.7%** | **67.9%** | **18.5x** |
+| kmedoids | 10% | 60 | 46.9% | 66.9% | 14.9x |
+| facility | 10% | 60 | 46.8% | 66.8% | 11.5x |
+| random | 10% | 60 | 44.3% | 63.1% | 10.3x |
+| dpp | 10% | 60 | 42.9% | 61.1% | 13.7x |
+| fps | 10% | 60 | 40.6% | 57.9% | 21.4x |
+
+### Robot Framework (126 tests, 63.5% full coverage)
+
+| Strategy | Selection | Tests | Coverage | Retention | Speedup |
+|---|---|---|---|---|---|
+| *(full suite)* | 100% | 126 | 63.5% | baseline | 1.0x |
+| **fps_multi** | **50%** | **63** | **61.6%** | **97.0%** | **1.5x** |
+| facility | 50% | 63 | 61.6% | 96.9% | 1.5x |
+| fps | 50% | 63 | 61.2% | 96.4% | 1.6x |
+| kmedoids | 50% | 63 | 60.5% | 95.3% | 1.5x |
+| dpp | 50% | 63 | 58.5% | 92.0% | 1.7x |
+| random | 50% | 63 | 57.2% | 90.1% | 2.3x |
+| **facility** | **20%** | **26** | **51.8%** | **81.5%** | **4.4x** |
+| kmedoids | 20% | 26 | 47.5% | 74.8% | 5.5x |
+| random | 20% | 26 | 46.0% | 72.5% | 5.7x |
+| fps | 20% | 26 | 45.3% | 71.4% | 3.1x |
+| fps_multi | 20% | 26 | 45.3% | 71.4% | 2.9x |
+| dpp | 20% | 26 | 43.9% | 69.0% | 4.6x |
+| **kmedoids** | **10%** | **13** | **42.3%** | **66.6%** | **11.3x** |
+| dpp | 10% | 13 | 40.9% | 64.3% | 10.7x |
+| facility | 10% | 13 | 40.2% | 63.3% | 12.9x |
+| random | 10% | 13 | 37.6% | 59.2% | 11.3x |
+| fps_multi | 10% | 13 | 37.5% | 59.1% | 10.7x |
+| fps | 10% | 13 | 33.2% | 52.3% | 8.3x |
+
+### Key Findings
+
+- **50% selection retains 92-97% of coverage** across all strategies — the diversity algorithms select tests maximizing behavioral breadth, with minimal redundancy loss
+- **Facility location excels at 20% selection** — 83.6% retention on pytest, 81.5% on Robot Framework, making it the best choice for aggressive smoke-test suites
+- **kmedoids matches facility at 20%** on pytest (83.1% retention) and leads at 10% on Robot Framework (66.6%)
+- **fps_multi leads at 10% pytest** (67.9% retention) where initial-point sensitivity matters most
+- **DPP performs well at 50%** (93.9% retention, matching FPS) but drops at 20% due to FPS fallback for large k values
+- **All diversity strategies outperform random** at 50% by 2-7 percentage points of retention
+- At 50% robot selection, fps_multi achieves **97% retention** — meaning half the tests contribute almost no incremental coverage
+
 ## Performance
 
 Benchmarked on 384-dimensional normalized vectors:

{robotframework_testselection-0.2.0 → robotframework_testselection-0.3.0}/README.md

@@ -1,6 +1,6 @@
 # robotframework-testselection
 
-Vector-based diverse test case selection for Robot Framework. Embeds test cases as semantic vectors and selects maximally diverse subsets to reduce test suite execution time while preserving coverage breadth.
+Vector-based diverse test case selection for Robot Framework and pytest. Embeds test cases as semantic vectors and selects maximally diverse subsets to reduce test suite execution time while preserving coverage breadth.
 
 ## How It Works
 
@@ -43,6 +43,8 @@ uv sync --extra vectorize
 uv sync --extra all
 ```
 
+The pytest plugin is included in the base install and activates when you pass `--diverse-k` to pytest.
+
 ### Dependency Groups
 
 | Group | Packages | Purpose |
@@ -124,6 +126,38 @@ testcase-select run --suite tests/ --k 50 \
 
 This works with both `run` and `execute` subcommands, including during graceful fallback (when selection fails and all tests are run).
 
+### pytest Support
+
+The diversity selection algorithm also works with pytest test suites, via a pytest plugin:
+
+```bash
+# Select 20 most diverse tests
+pytest --diverse-k=20 tests/
+
+# With custom strategy and seed
+pytest --diverse-k=30 --diverse-strategy=fps_multi --diverse-seed=123 tests/
+
+# Filter by markers before selection
+pytest --diverse-k=20 --diverse-include-markers slow integration tests/
+
+# Group parametrized tests (select at group level)
+pytest --diverse-k=20 --diverse-group-parametrize tests/
+```
+
+Or via the `testcase-select` CLI:
+
+```bash
+# Full pipeline with pytest
+testcase-select run --framework pytest --suite tests/ --k 20 --strategy fps
+
+# Stage-by-stage
+testcase-select vectorize --framework pytest --suite tests/ --output ./artifacts/
+testcase-select select --artifacts ./artifacts/ --k 20
+testcase-select execute --framework pytest --suite tests/ --selection selected_tests.json
+```
+
+The pytest plugin is installed automatically and activated by `--diverse-k`. It uses AST-based text representation with sentence embeddings for test diversity analysis.
+
 ### Direct Robot Framework Integration
 
 You can also use the components directly with `robot`:
@@ -289,6 +323,11 @@ src/TestSelection/
     prerun_modifier.py # DiversePreRunModifier (SuiteVisitor)
     listener.py        # DiverseDataDriverListener (Listener v3)
     runner.py          # ExecutionRunner
+  pytest/              # pytest integration
+    plugin.py          # pytest plugin (pytest11 entry point)
+    collector.py       # Programmatic test collection
+    text_builder.py    # AST-based text representation
+    runner.py          # pytest execution
   pipeline/            # Orchestration layer
     vectorize.py       # Stage 1 orchestrator
     select.py          # Stage 2 orchestrator
@@ -343,6 +382,68 @@ uv run pytest -m integration
 uv run pytest -m benchmark
 ```
 
+## Coverage Retention Analysis
+
+Measured on [robotframework-doctestlibrary](https://github.com/manykarim/robotframework-doctestlibrary) (5,045 statements in the DocTest package). All strategies use `all-MiniLM-L6-v2` embeddings (384-dim), seed=42.
+
+### pytest (594 tests, 70.1% full coverage)
+
+| Strategy | Selection | Tests | Coverage | Retention | Speedup |
+|---|---|---|---|---|---|
+| *(full suite)* | 100% | 594 | 70.1% | baseline | 1.0x |
+| **fps** | **50%** | **297** | **65.9%** | **93.9%** | **3.4x** |
+| dpp | 50% | 297 | 65.9% | 93.9% | 3.5x |
+| kmedoids | 50% | 297 | 65.8% | 93.8% | 3.1x |
+| fps_multi | 50% | 297 | 65.7% | 93.7% | 3.0x |
+| facility | 50% | 297 | 64.9% | 92.6% | 2.9x |
+| random | 50% | 297 | 64.2% | 91.5% | 2.3x |
+| **facility** | **20%** | **119** | **58.6%** | **83.6%** | **7.5x** |
+| kmedoids | 20% | 119 | 58.3% | 83.1% | 6.7x |
+| fps | 20% | 119 | 56.1% | 80.0% | 6.7x |
+| fps_multi | 20% | 119 | 53.8% | 76.8% | 8.0x |
+| random | 20% | 119 | 53.2% | 75.8% | 5.4x |
+| dpp | 20% | 119 | 47.6% | 67.8% | 6.0x |
+| **fps_multi** | **10%** | **60** | **47.7%** | **67.9%** | **18.5x** |
+| kmedoids | 10% | 60 | 46.9% | 66.9% | 14.9x |
+| facility | 10% | 60 | 46.8% | 66.8% | 11.5x |
+| random | 10% | 60 | 44.3% | 63.1% | 10.3x |
+| dpp | 10% | 60 | 42.9% | 61.1% | 13.7x |
+| fps | 10% | 60 | 40.6% | 57.9% | 21.4x |
+
+### Robot Framework (126 tests, 63.5% full coverage)
+
+| Strategy | Selection | Tests | Coverage | Retention | Speedup |
+|---|---|---|---|---|---|
+| *(full suite)* | 100% | 126 | 63.5% | baseline | 1.0x |
+| **fps_multi** | **50%** | **63** | **61.6%** | **97.0%** | **1.5x** |
+| facility | 50% | 63 | 61.6% | 96.9% | 1.5x |
+| fps | 50% | 63 | 61.2% | 96.4% | 1.6x |
+| kmedoids | 50% | 63 | 60.5% | 95.3% | 1.5x |
+| dpp | 50% | 63 | 58.5% | 92.0% | 1.7x |
+| random | 50% | 63 | 57.2% | 90.1% | 2.3x |
+| **facility** | **20%** | **26** | **51.8%** | **81.5%** | **4.4x** |
+| kmedoids | 20% | 26 | 47.5% | 74.8% | 5.5x |
+| random | 20% | 26 | 46.0% | 72.5% | 5.7x |
+| fps | 20% | 26 | 45.3% | 71.4% | 3.1x |
+| fps_multi | 20% | 26 | 45.3% | 71.4% | 2.9x |
+| dpp | 20% | 26 | 43.9% | 69.0% | 4.6x |
+| **kmedoids** | **10%** | **13** | **42.3%** | **66.6%** | **11.3x** |
+| dpp | 10% | 13 | 40.9% | 64.3% | 10.7x |
+| facility | 10% | 13 | 40.2% | 63.3% | 12.9x |
+| random | 10% | 13 | 37.6% | 59.2% | 11.3x |
+| fps_multi | 10% | 13 | 37.5% | 59.1% | 10.7x |
+| fps | 10% | 13 | 33.2% | 52.3% | 8.3x |
+
+### Key Findings
+
+- **50% selection retains 92-97% of coverage** across all strategies — the diversity algorithms select tests maximizing behavioral breadth, with minimal redundancy loss
+- **Facility location excels at 20% selection** — 83.6% retention on pytest, 81.5% on Robot Framework, making it the best choice for aggressive smoke-test suites
+- **kmedoids matches facility at 20%** on pytest (83.1% retention) and leads at 10% on Robot Framework (66.6%)
+- **fps_multi leads at 10% pytest** (67.9% retention) where initial-point sensitivity matters most
+- **DPP performs well at 50%** (93.9% retention, matching FPS) but drops at 20% due to FPS fallback for large k values
+- **All diversity strategies outperform random** at 50% by 2-7 percentage points of retention
+- At 50% robot selection, fps_multi achieves **97% retention** — meaning half the tests contribute almost no incremental coverage
+
 ## Performance
 
 Benchmarked on 384-dimensional normalized vectors:

{robotframework_testselection-0.2.0 → robotframework_testselection-0.3.0}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "hatchling.build"
 [project]
 name = "robotframework-testselection"
 dynamic = ["version"]
-description = "Vector-based diverse test case selection for Robot Framework"
+description = "Vector-based diverse test case selection for Robot Framework and pytest"
 readme = "README.md"
 requires-python = ">=3.10"
 license = { text = "Apache-2.0" }
@@ -14,6 +14,7 @@ authors = [
 ]
 keywords = [
     "robotframework",
+    "pytest",
     "testing",
     "test-selection",
     "diversity",
@@ -26,6 +27,7 @@ classifiers = [
     "Framework :: Robot Framework",
     "Framework :: Robot Framework :: Library",
     "Framework :: Robot Framework :: Tool",
+    "Framework :: Pytest",
     "Intended Audience :: Developers",
     "License :: OSI Approved :: Apache Software License",
     "Operating System :: OS Independent",
@@ -76,6 +78,9 @@ all = [
 [project.scripts]
 testcase-select = "TestSelection.cli:main"
 
+[project.entry-points.pytest11]
+diverse-selection = "TestSelection.pytest.plugin"
+
 [tool.hatch.version]
 path = "src/TestSelection/__init__.py"
 
@@ -128,3 +133,9 @@ module = [
     "robot.*",
 ]
 ignore_missing_imports = true
+
+[dependency-groups]
+dev = [
+    "coverage>=7.13.4",
+    "pytest-cov>=7.0.0",
+]

{robotframework_testselection-0.2.0 → robotframework_testselection-0.3.0}/src/TestSelection/__init__.py

@@ -1,3 +1,3 @@
 """Vector-based diverse test case selection for Robot Framework."""
 
-__version__ = "0.2.0"
+__version__ = "0.3.0"

{robotframework_testselection-0.2.0 → robotframework_testselection-0.3.0}/src/TestSelection/cli.py

@@ -18,13 +18,24 @@ def _setup_logging(verbose: bool = False) -> None:
     )
 
 
+def _add_framework_arg(parser: argparse.ArgumentParser) -> None:
+    """Add --framework option to a parser."""
+    parser.add_argument(
+        "--framework",
+        choices=["robot", "pytest"],
+        default="robot",
+        help="Test framework: robot (default) or pytest.",
+    )
+
+
 def _add_vectorize_parser(subparsers: argparse._SubParsersAction) -> None:
     p = subparsers.add_parser(
         "vectorize", help="Stage 1: vectorize test suite",
     )
+    _add_framework_arg(p)
     p.add_argument(
         "--suite", required=True, type=Path,
-        help="Path to .robot suite",
+        help="Path to test suite directory",
     )
     p.add_argument(
         "--output", required=True, type=Path,
@@ -36,7 +47,7 @@ def _add_vectorize_parser(subparsers: argparse._SubParsersAction) -> None:
     )
     p.add_argument(
         "--resolve-depth", type=int, default=0,
-        help="Keyword resolve depth",
+        help="Keyword resolve depth (Robot Framework only)",
     )
     p.add_argument(
         "--force", action="store_true",
@@ -44,7 +55,7 @@ def _add_vectorize_parser(subparsers: argparse._SubParsersAction) -> None:
     )
     p.add_argument(
         "--datadriver-csv", nargs="*", type=Path,
-        help="DataDriver CSV files",
+        help="DataDriver CSV files (Robot Framework only)",
     )
     p.set_defaults(func=_cmd_vectorize)
 
@@ -99,15 +110,16 @@ def _add_execute_parser(subparsers: argparse._SubParsersAction) -> None:
         "execute",
         help="Stage 3: execute selected tests",
         epilog=(
-            "All arguments after -- are passed directly to robot. "
+            "All arguments after -- are passed directly to the test runner. "
             "Example: testcase-select execute --suite tests/ "
             "--selection sel.json "
             "-- --variable ENV:staging --loglevel DEBUG"
         ),
     )
+    _add_framework_arg(p)
     p.add_argument(
         "--suite", required=True, type=Path,
-        help="Path to .robot suite",
+        help="Path to test suite directory",
     )
     p.add_argument(
         "--selection", required=True, type=Path,
@@ -129,14 +141,15 @@ def _add_run_parser(subparsers: argparse._SubParsersAction) -> None:
         "run",
         help="Full pipeline: vectorize + select + execute",
         epilog=(
-            "All arguments after -- are passed directly to robot. "
+            "All arguments after -- are passed directly to the test runner. "
             "Example: testcase-select run --suite tests/ --k 20 "
             "-- --include smoke --variable ENV:staging"
         ),
     )
+    _add_framework_arg(p)
     p.add_argument(
         "--suite", required=True, type=Path,
-        help="Path to .robot suite",
+        help="Path to test suite directory",
     )
     p.add_argument(
         "--k", type=int, default=default_k,
@@ -156,7 +169,7 @@ def _add_run_parser(subparsers: argparse._SubParsersAction) -> None:
     )
     p.add_argument(
         "--resolve-depth", type=int, default=0,
-        help="Keyword resolve depth",
+        help="Keyword resolve depth (Robot Framework only)",
     )
     p.add_argument(
         "--force", action="store_true",
@@ -170,6 +183,11 @@ def _add_run_parser(subparsers: argparse._SubParsersAction) -> None:
 
 
 def _cmd_vectorize(args: argparse.Namespace) -> int:
+    framework = getattr(args, "framework", "robot")
+
+    if framework == "pytest":
+        return _cmd_vectorize_pytest(args)
+
     from TestSelection.pipeline.vectorize import run_vectorize
 
     try:
@@ -193,6 +211,29 @@ def _cmd_vectorize(args: argparse.Namespace) -> int:
         return 2
 
 
+def _cmd_vectorize_pytest(args: argparse.Namespace) -> int:
+    """Vectorize pytest tests."""
+    from TestSelection.pipeline.vectorize_pytest import run_vectorize_pytest
+
+    try:
+        indexed = run_vectorize_pytest(
+            suite_path=args.suite,
+            artifact_dir=args.output,
+            model_name=args.model,
+            force=args.force,
+        )
+        if indexed:
+            logger.info("[DIVERSE-SELECT] Vectorization complete (pytest)")
+        else:
+            logger.info(
+                "[DIVERSE-SELECT] Vectorization skipped (no changes)"
+            )
+        return 0
+    except Exception as exc:
+        logger.error("[DIVERSE-SELECT] Vectorization failed: %s", exc)
+        return 2
+
+
 def _cmd_select(args: argparse.Namespace) -> int:
     from TestSelection.pipeline.select import run_select
 
@@ -219,6 +260,11 @@ def _cmd_select(args: argparse.Namespace) -> int:
 
 
 def _cmd_execute(args: argparse.Namespace) -> int:
+    framework = getattr(args, "framework", "robot")
+
+    if framework == "pytest":
+        return _cmd_execute_pytest(args)
+
     from TestSelection.pipeline.execute import run_execute
 
     return run_execute(
@@ -229,7 +275,24 @@ def _cmd_execute(args: argparse.Namespace) -> int:
     )
 
 
+def _cmd_execute_pytest(args: argparse.Namespace) -> int:
+    """Execute pytest with selection."""
+    from TestSelection.pytest.runner import run_pytest_with_selection
+
+    return run_pytest_with_selection(
+        suite_path=args.suite,
+        selection_file=args.selection,
+        output_dir=str(args.output_dir),
+        extra_args=args.robot_passthrough,
+    )
+
+
 def _cmd_run(args: argparse.Namespace) -> int:
+    framework = getattr(args, "framework", "robot")
+
+    if framework == "pytest":
+        return _cmd_run_pytest(args)
+
     from TestSelection.pipeline.execute import run_execute
     from TestSelection.pipeline.select import run_select
     from TestSelection.pipeline.vectorize import run_vectorize
@@ -280,6 +343,29 @@ def _cmd_run(args: argparse.Namespace) -> int:
     )
 
 
+def _cmd_run_pytest(args: argparse.Namespace) -> int:
+    """Run full pipeline for pytest."""
+    from TestSelection.pytest.runner import run_pytest_pipeline
+
+    try:
+        return run_pytest_pipeline(
+            suite_path=args.suite,
+            k=args.k,
+            strategy=args.strategy,
+            seed=args.seed,
+            model_name=args.model,
+            output_dir=str(args.output_dir),
+            extra_args=args.robot_passthrough,
+        )
+    except Exception as exc:
+        logger.warning(
+            "[DIVERSE-SELECT] pytest pipeline failed, "
+            "falling back to all tests: %s",
+            exc,
+        )
+        return _fallback_execute_pytest(args)
+
+
 def _fallback_execute(args: argparse.Namespace) -> int:
     """Execute all tests without selection (graceful degradation)."""
     logger.info("[DIVERSE-SELECT] Running all tests (no selection)")
@@ -290,7 +376,6 @@ def _fallback_execute(args: argparse.Namespace) -> int:
             "--outputdir",
             str(args.output_dir),
         ]
-        # Pass through any robot options from after --
         if args.robot_passthrough:
             robot_args.extend(args.robot_passthrough)
         robot_args.append(str(args.suite))
@@ -302,13 +387,30 @@ def _fallback_execute(args: argparse.Namespace) -> int:
         return 2
 
 
+def _fallback_execute_pytest(args: argparse.Namespace) -> int:
+    """Execute all pytest tests without selection (graceful degradation)."""
+    logger.info("[DIVERSE-SELECT] Running all pytest tests (no selection)")
+    try:
+        import pytest
+
+        pytest_args = [str(args.suite)]
+        if args.robot_passthrough:
+            pytest_args.extend(args.robot_passthrough)
+        return pytest.main(pytest_args)
+    except Exception as exc:
+        logger.error(
+            "[DIVERSE-SELECT] Fallback execution failed: %s", exc,
+        )
+        return 2
+
+
 def build_parser() -> argparse.ArgumentParser:
     """Build the argument parser."""
     parser = argparse.ArgumentParser(
         prog="testcase-select",
         description=(
             "Vector-based diverse test case selection "
-            "for Robot Framework"
+            "for Robot Framework and pytest"
         ),
     )
     parser.add_argument(
@@ -328,10 +430,10 @@ def build_parser() -> argparse.ArgumentParser:
 def _split_robot_passthrough(
     argv: list[str],
 ) -> tuple[list[str], list[str]]:
-    """Split argv at -- into our args and robot passthrough args.
+    """Split argv at -- into our args and passthrough args.
 
-    Returns (our_args, robot_args). If no -- is present, robot_args
-    is empty.
+    Returns (our_args, passthrough_args). If no -- is present,
+    passthrough_args is empty.
     """
     try:
         sep = argv.index("--")
@@ -343,7 +445,7 @@ def _split_robot_passthrough(
 def main(argv: list[str] | None = None) -> int:
     """Main CLI entry point.
 
-    Arguments after -- are passed through to robot:
+    Arguments after -- are passed through to the test runner:
         testcase-select run --suite tests/ --k 20 \\
             -- --variable ENV:staging --loglevel DEBUG --include smoke
     """

{robotframework_testselection-0.2.0 → robotframework_testselection-0.3.0}/src/TestSelection/pipeline/cache.py

@@ -15,10 +15,16 @@ class CacheInvalidator:
     def __init__(self, hash_store_path: Path) -> None:
         self._hash_store_path = hash_store_path
 
-    def has_changes(self, suite_path: Path) -> bool:
+    def has_changes(
+        self,
+        suite_path: Path,
+        glob_patterns: tuple[str, ...] | None = None,
+    ) -> bool:
         """Compare current file hashes with stored hashes.
 
         Returns True if there are changes or no stored hashes exist.
+        If glob_patterns is provided, uses those instead of the default
+        .robot/.csv patterns.
         """
         if not self._hash_store_path.exists():
             logger.info(
@@ -28,7 +34,7 @@ class CacheInvalidator:
             return True
 
         stored = json.loads(self._hash_store_path.read_text())
-        current = self._compute_hashes(suite_path)
+        current = self._compute_hashes(suite_path, glob_patterns)
 
         if current != stored:
             changed = set(current.keys()) ^ set(stored.keys())
@@ -49,19 +55,31 @@ class CacheInvalidator:
         )
         return False
 
-    def save_hashes(self, suite_path: Path) -> None:
+    def save_hashes(
+        self,
+        suite_path: Path,
+        glob_patterns: tuple[str, ...] | None = None,
+    ) -> None:
         """Save current file hashes to the hash store."""
-        hashes = self._compute_hashes(suite_path)
+        hashes = self._compute_hashes(suite_path, glob_patterns)
         self._hash_store_path.parent.mkdir(parents=True, exist_ok=True)
         self._hash_store_path.write_text(json.dumps(hashes, indent=2))
 
-    def _compute_hashes(self, suite_path: Path) -> dict[str, str]:
-        """Compute MD5 hashes for all .robot and .csv files."""
+    def _compute_hashes(
+        self,
+        suite_path: Path,
+        glob_patterns: tuple[str, ...] | None = None,
+    ) -> dict[str, str]:
+        """Compute MD5 hashes for source files.
+
+        By default hashes .robot and .csv files. Pass glob_patterns
+        to override (e.g., ("**/*.py",) for pytest).
+        """
         hashes: dict[str, str] = {}
         target = suite_path if suite_path.is_dir() else suite_path.parent
 
         if suite_path.is_dir():
-            patterns = ["*.robot", "*.csv"]
+            patterns = list(glob_patterns) if glob_patterns else ["*.robot", "*.csv"]
             for pattern in patterns:
                 for p in sorted(target.rglob(pattern)):
                     md5 = hashlib.md5(p.read_bytes()).hexdigest()