testpick 0.2.0__py3-none-any.whl

testpick/reporter/terminal.py

@@ -0,0 +1,108 @@
+"""Terminal reporter that renders beautiful terminal output using rich."""
+
+from __future__ import annotations
+
+from pathlib import Path
+
+from rich.console import Console
+from rich.table import Table
+from rich.panel import Panel
+
+
+class TerminalReporter:
+    def __init__(self) -> None:
+        self.console = Console()
+
+    def render(
+        self,
+        changed_files: list[tuple[str, str]],
+        test_files: list[Path],
+        dependency_chains: dict[Path, list[Path]],
+        stats: dict,
+    ) -> None:
+        """Render the test selection results to the terminal.
+
+        Args:
+            changed_files: List of (status, filepath) tuples from git diff.
+            test_files: List of selected test filepaths.
+            dependency_chains: Mapping from test file to its dependency chain.
+            stats: Statistics about the analysis.
+        """
+        # Header
+        self.console.print()
+        self.console.print(
+            Panel(
+                "[bold]testpick - Test Selection Results[/bold]",
+                style="blue",
+            )
+        )
+        self.console.print()
+
+        # Changed files table
+        changed_table = Table(title="Changed Files", show_header=True)
+        changed_table.add_column("Status", style="bold", width=10)
+        changed_table.add_column("File", style="cyan")
+
+        status_colors = {
+            "A": "green",
+            "M": "yellow",
+            "D": "red",
+            "R": "blue",
+        }
+        status_names = {
+            "A": "Added",
+            "M": "Modified",
+            "D": "Deleted",
+            "R": "Renamed",
+        }
+
+        for status, filepath in changed_files:
+            color = status_colors.get(status, "white")
+            status_name = status_names.get(status, status)
+            changed_table.add_row(
+                f"[{color}]{status_name}[/{color}]",
+                filepath,
+            )
+
+        self.console.print(changed_table)
+        self.console.print()
+
+        # Test files table
+        if test_files:
+            test_table = Table(title="Selected Tests", show_header=True)
+            test_table.add_column("#", style="dim", width=5)
+            test_table.add_column("Test File", style="cyan")
+            test_table.add_column("Dependency Path", style="dim")
+
+            for i, test_file in enumerate(test_files, 1):
+                chain = dependency_chains.get(test_file, [])
+                chain_str = " -> ".join(str(p) for p in chain) if chain else "direct"
+                test_table.add_row(
+                    str(i),
+                    str(test_file),
+                    chain_str,
+                )
+
+            self.console.print(test_table)
+        else:
+            self.console.print(
+                Panel(
+                    "[yellow]No tests selected[/yellow]",
+                    style="yellow",
+                )
+            )
+
+        self.console.print()
+
+        # Stats
+        stats_table = Table(title="Statistics", show_header=False)
+        stats_table.add_column("Metric", style="bold")
+        stats_table.add_column("Value", style="green")
+
+        stats_table.add_row("Files Changed", str(stats.get("files_changed", 0)))
+        stats_table.add_row("Tests Selected", str(len(test_files)))
+        stats_table.add_row("Insertions", str(stats.get("insertions", 0)))
+        stats_table.add_row("Deletions", str(stats.get("deletions", 0)))
+
+        self.console.print(stats_table)
+        self.console.print()

testpick/selector/__init__.py

@@ -0,0 +1,6 @@
+"""Test selector modules for testpick."""
+
+from .matcher import TestMatcher
+from .walker import DependencyWalker
+
+__all__ = ["DependencyWalker", "TestMatcher"]

testpick/selector/matcher.py

@@ -0,0 +1,58 @@
+"""Test matcher that filters and sorts test files from affected files."""
+
+from __future__ import annotations
+
+from pathlib import Path
+
+from ..graph import DependencyGraph
+
+
+class TestMatcher:
+    def match(
+        self, affected_files: set[Path], graph: DependencyGraph
+    ) -> list[Path]:
+        """Filter affected files to only test files and sort by directness.
+
+        Args:
+            affected_files: Set of all affected filepaths.
+            graph: The dependency graph.
+
+        Returns:
+            List of test filepaths, sorted with direct imports first.
+        """
+        test_files: list[Path] = []
+        direct_tests: list[Path] = []
+        transitive_tests: list[Path] = []
+
+        for filepath in affected_files:
+            if graph.is_test_file(filepath):
+                # Check if this test directly imports from a changed file
+                if filepath in graph.nodes:
+                    node = graph.nodes[filepath]
+                    # Check if any of its imports are in the affected files
+                    is_direct = False
+                    for imp in node.imports:
+                        targets = graph.resolve_import(imp, filepath)
+                        for target in targets:
+                            if target in affected_files and not graph.is_test_file(target):
+                                is_direct = True
+                                break
+                        if is_direct:
+                            break
+
+                    if is_direct:
+                        direct_tests.append(filepath)
+                    else:
+                        transitive_tests.append(filepath)
+                else:
+                    # File not in graph, treat as transitive
+                    transitive_tests.append(filepath)
+
+        # Sort each group alphabetically for consistency
+        direct_tests.sort()
+        transitive_tests.sort()
+
+        # Direct imports first, then transitive
+        test_files = direct_tests + transitive_tests
+
+        return test_files

testpick/selector/walker.py

@@ -0,0 +1,42 @@
+"""Dependency walker that finds all files affected by changes."""
+
+from __future__ import annotations
+
+from pathlib import Path
+
+from ..graph import DependencyGraph
+
+
+class DependencyWalker:
+    def walk_from_files(
+        self, changed_files: list[tuple[str, str]], graph: DependencyGraph
+    ) -> tuple[set[Path], dict[Path, Path]]:
+        """Walk from changed files to every affected file, tracking how each was reached.
+
+        Returns the affected files and a parent map used to explain why a file
+        was selected. Deleted files are walked but never selected themselves.
+        """
+        affected: set[Path] = set()
+        parents: dict[Path, Path] = {}
+
+        for status, filepath in changed_files:
+            path = Path(filepath).resolve()
+            if not status.startswith("D"):
+                affected.add(path)
+
+            if path not in graph.nodes:
+                continue
+
+            queue = [path]
+            visited = {path}
+            while queue:
+                current = queue.pop(0)
+                for dependent in graph.get_dependents(current):
+                    if dependent in visited:
+                        continue
+                    visited.add(dependent)
+                    affected.add(dependent)
+                    parents.setdefault(dependent, current)
+                    queue.append(dependent)
+
+        return affected, parents

testpick-0.2.0.dist-info/METADATA

@@ -0,0 +1,211 @@
+Metadata-Version: 2.5
+Name: testpick
+Version: 0.2.0
+Summary: Pick the tests affected by your git diff before you run them
+Project-URL: Homepage, https://github.com/krauzX/testpick
+Project-URL: Repository, https://github.com/krauzX/testpick
+Project-URL: Issues, https://github.com/krauzX/testpick/issues
+License: MIT
+License-File: LICENSE
+Keywords: ci,git,pytest,static-analysis,testing
+Classifier: Development Status :: 4 - Beta
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Software Development :: Testing
+Requires-Python: >=3.10
+Requires-Dist: click>=8.0
+Requires-Dist: gitpython>=3.1
+Requires-Dist: readchar>=4.0
+Requires-Dist: rich>=13.0
+Provides-Extra: dev
+Requires-Dist: mypy>=1.10; extra == 'dev'
+Requires-Dist: pytest-cov>=5.0; extra == 'dev'
+Requires-Dist: pytest>=8.0; extra == 'dev'
+Requires-Dist: ruff>=0.5; extra == 'dev'
+Provides-Extra: tree-sitter
+Requires-Dist: tree-sitter-javascript>=0.23; extra == 'tree-sitter'
+Requires-Dist: tree-sitter-python>=0.23; extra == 'tree-sitter'
+Requires-Dist: tree-sitter-typescript>=0.23; extra == 'tree-sitter'
+Requires-Dist: tree-sitter>=0.25; extra == 'tree-sitter'
+Description-Content-Type: text/markdown
+
+# testpick
+
+[![CI](https://github.com/krauzX/testpick/actions/workflows/ci.yml/badge.svg)](https://github.com/krauzX/testpick/actions/workflows/ci.yml)
+
+A Python CLI that looks at your git diff and tells you which tests are going to break. Before you run them.
+
+## Why this exists
+
+You change a function in `utils.py`. Now what? You could run the full test suite and wait, or you could guess
+which tests matter and hope you're right. `pytest --lf` reruns failures only after they've already failed.
+Jest's `--changedSince` follows direct file changes.
+
+testpick parses your git diff, builds an import graph of your source files, and walks it backwards from what
+you changed to find every test that depends on it — transitively. Each selected test comes with the chain that
+explains why, so you can sanity-check the answer instead of trusting it.
+
+## Getting started
+
+```bash
+git clone https://github.com/krauzX/testpick.git
+cd testpick
+pip install -e .
+
+# optional but recommended: real AST parsing instead of the regex fallback
+pip install -e ".[tree-sitter]"
+
+# for development (tests, linter, type checker)
+pip install -e ".[dev]"
+```
+
+Python 3.10+, runs on Linux, macOS and Windows.
+
+## Usage
+
+```bash
+# what changed since your last commit, including uncommitted work?
+testpick
+
+# committed changes on your branch since it diverged from main
+testpick --base main
+
+# a single commit
+testpick --commit abc123
+
+# output for scripts and CI
+testpick --format json
+testpick --format github      # GitHub Actions annotations
+
+# only analyze Python and TypeScript files
+testpick --languages python,typescript
+```
+
+Everything resolves from the repository root, so it works from any directory inside the repo.
+
+### Example output
+
+```
+┌─────────────────────────────────────────────────────────────────────────────┐
+│ testpick - Test Selection Results                                           │
+└─────────────────────────────────────────────────────────────────────────────┘
+
+                     Changed Files
+┌────────────┬────────────────────────────────────────┐
+│ Status     │ File                                   │
+├────────────┼────────────────────────────────────────┤
+│ Modified   │ src/testpick/graph/builder.py          │
+│ Modified   │ src/testpick/reporter/github.py        │
+│ Modified   │ src/testpick/reporter/json_reporter.py │
+│ Added      │ tests/test_selection_pipeline.py       │
+└────────────┴────────────────────────────────────────┘
+
+                                Selected Tests
+┌───────┬──────────────────────────────────┬──────────────────────────────────┐
+│ #     │ Test File                        │ Dependency Path                  │
+├───────┼──────────────────────────────────┼──────────────────────────────────┤
+│ 1     │ tests/test_cli.py                │ tests/test_cli.py ->             │
+│       │                                  │ src/testpick/cli.py ->           │
+│       │                                  │ src/testpick/graph/__init__.py   │
+│       │                                  │ -> src/testpick/graph/builder.py │
+│ 2     │ tests/test_graph/test_builder.py │ tests/test_graph/test_builder.py │
+│       │                                  │ ->                               │
+│       │                                  │ src/testpick/graph/__init__.py   │
+│       │                                  │ -> src/testpick/graph/builder.py │
+│ 3     │ tests/test_selection_pipeline.py │ direct                           │
+│ 4     │ tests/test_selector/test_matche… │ tests/test_selector/test_matche… │
+│       │                                  │ ->                               │
+│       │                                  │ src/testpick/graph/__init__.py   │
+│       │                                  │ -> src/testpick/graph/builder.py │
+└───────┴──────────────────────────────────┴──────────────────────────────────┘
+
+       Statistics
+┌────────────────┬─────┐
+│ Files Changed  │ 4   │
+│ Tests Selected │ 4   │
+│ Insertions     │ 258 │
+│ Deletions      │ 25  │
+└────────────────┴─────┘
+```
+
+`direct` means the test file itself changed. The JSON format carries the same
+information, with chains as arrays:
+
+```json
+{
+  "changed_files": [
+    { "status": "M", "path": "src/utils.ts" }
+  ],
+  "test_files": [
+    {
+      "path": "tests/test_page.ts",
+      "dependency_chain": ["tests/test_page.ts", "src/page.ts", "src/utils.ts"]
+    }
+  ],
+  "stats": {
+    "files_changed": 1,
+    "insertions": 6,
+    "deletions": 2,
+    "tests_selected": 1
+  }
+}
+```
+
+## How it works
+
+1. Read the diff with `git diff --name-status` and resolve every path against the repository root.
+2. Parse each source file for imports. tree-sitter grammars are used when installed, otherwise a regex pass
+   handles Python, TypeScript and JavaScript.
+3. Resolve each import to a file: relative specifiers against the importing file's directory, absolute ones
+   against `src/`, the project root, then the file's own directory — preferring the importing file's language
+   when both a `.py` and a `.ts` file would match.
+4. Walk reverse edges from the changed files with a breadth-first search, recording the parent of every hit so
+   the chain can be explained afterwards.
+5. Report the files that are test files, direct changes first.
+
+## How it compares
+
+Other tools pick tests from a diff too; they differ in how far they can see:
+
+- **pytest-testmon** records which code each test actually executed, so it catches runtime coupling a static
+  graph misses — but it needs a baseline run and is pytest-only.
+- **Jest `--changedSince`** maps changed files to tests, but inside a Jest project and mostly direct
+  relationships.
+- **Nx and Turborepo affected tasks** pick packages to rebuild, which is coarser than picking individual tests.
+- **snob, pytest-impacted, pytest-git-selector** do static graph selection like this one, but for Python only.
+
+testpick's corner: one CLI covering Python and TypeScript/JavaScript, no baseline run required, and every
+selection arrives with the chain that produced it.
+
+## Limitations
+
+Static analysis only sees imports it can read:
+
+- Dynamic imports, dependency injection, plugin registries and other runtime wiring are invisible to it.
+- A file deleted from the working tree is gone from the parse tree, so testpick reports the deletion but
+  cannot trace what used to import it.
+- `--languages` scopes which files are analyzed for dependencies; changed files are always reported.
+
+Treat the output as a starting point. A full suite run still belongs in CI before a release.
+
+## Running tests
+
+```bash
+pytest tests/ -v
+pytest tests/ --cov=testpick
+```
+
+77 tests covering diff parsing, import resolution, graph building, the selection pipeline and all three
+reporters, with 82% branch coverage. CI runs ruff, mypy and the suite on Python 3.10-3.13 across Linux, macOS
+and Windows.
+
+## License
+
+MIT

testpick-0.2.0.dist-info/RECORD

@@ -0,0 +1,28 @@
+testpick/__init__.py,sha256=CDFgRH2D_Tuw3V1GXu-CZ_1hcPHn06MbMwks2nMKIg8,254
+testpick/__main__.py,sha256=FY7OCYd3IE4hQpDvI-p_lrfkDl7ZYAhLajJVF1YegPA,38
+testpick/cli.py,sha256=VcWsc9Cyq5xK6IkZj_S8aroOdthOuYRMGaluJI8EXlY,6831
+testpick/git/__init__.py,sha256=atIQtxlOih29JM_vnZcWTPXhC9VOs0Sraq8vNN_cn14,514
+testpick/git/commit.py,sha256=6Vphw1j0-7gaCV3V1V25fthM9vaHh0xVeZvZ6tPb-Jc,1045
+testpick/git/diff.py,sha256=WrA2XPRh3DxDKTjvG-BQNtDIhhG2mz89KBmS2tOnpUw,1702
+testpick/git/interactive.py,sha256=PmIaWxOJwNbbIET9Qv7lkedGCUKepfdIO_MRoTjlCR0,3711
+testpick/graph/__init__.py,sha256=xkV_gekG83ghOlPqSIaCfDbODDgenQQy7Vc0x5vntGU,199
+testpick/graph/builder.py,sha256=HQ10NcxbMRfRn_cu7cVe3jhJfNjsQvQBUkkW3t5Vxtk,7878
+testpick/graph/edge.py,sha256=UrEQEhRha25fAg52L4qsgTNixQ3IitAvj-Q8vfmcwss,301
+testpick/graph/node.py,sha256=geha69c4i0uHv_iZYN7JdXNZNZ1-aV0s9vApNYe7DRY,375
+testpick/parser/__init__.py,sha256=PdxY0L32mY_Rtp86Opbg82Mc5-fIcLZZOUiThVK67Nw,434
+testpick/parser/base.py,sha256=iN03ee9_n_eAxZm73N2aghi0NYpnOAxxTSMNhmHHm2c,1010
+testpick/parser/python.py,sha256=7xeIkcX9gkCRYOufVWmNT9N9FhV4U8MGcwcS6wRfpEA,8840
+testpick/parser/registry.py,sha256=y2vJUQ9znhuu6IdQdGWx3dVqUuVPJ83Eksb8OMa-RVg,621
+testpick/parser/typescript.py,sha256=qR5Tl80t-5LpP4QZSlj3_ZzyCe-tifW-phWrAgP-Cw0,8133
+testpick/reporter/__init__.py,sha256=Lew5Dg3pzFw8RfXRiwav28Vf3Qg1EoK3sKeIHxTQo3M,225
+testpick/reporter/github.py,sha256=KVMoyoZejfABOa_Wpw7oLspfjySnP6oDa9zCReAsk70,2052
+testpick/reporter/json_reporter.py,sha256=fj3OG6xtr6LZfUp2nIfo5JZenm4M1ThIKgax66ShxMg,1565
+testpick/reporter/terminal.py,sha256=9vZXY1yZzW0vRSyHFXmX88MHSByTFgDzthOyy_ZcEh0,3478
+testpick/selector/__init__.py,sha256=H-q1Htokm30NsgcdcBqUdP8bysy65_v4CqSmWK1k8cI,160
+testpick/selector/matcher.py,sha256=Kb923i9XcVZwUq4UymswRiCX5whB2HKLIs3d8Xwit6s,2038
+testpick/selector/walker.py,sha256=9rals56AcZEIBWOHNSeKAjrJZMqHsmU8F0XuvYtmNAs,1394
+testpick-0.2.0.dist-info/METADATA,sha256=E6I6Xq_l9FBdKKGSgw25J6MutIpsArcHW6NBUe5WeRY,9815
+testpick-0.2.0.dist-info/WHEEL,sha256=2iZrJkDIkeDh0k9SikKMqFuRP-ua6mb4Ir1AsER4BEw,87
+testpick-0.2.0.dist-info/entry_points.txt,sha256=2IbOPSnYjU9aUd5vlaZgvzeIUelZQcZ3TlgyFmWK3YU,47
+testpick-0.2.0.dist-info/licenses/LICENSE,sha256=bUxTf7MNRUy_sQGHcUt0CLpDV301088prTjhVffHC1U,1063
+testpick-0.2.0.dist-info/RECORD,,

testpick-0.2.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.33.0
+Root-Is-Purelib: true
+Tag: py3-none-any

testpick-0.2.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+testpick = testpick.cli:main

testpick-0.2.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 krauzX
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.