Pin dev deps to latest, expand typecheck tests, simplify tox to testing-only

- Update dev dep versions: ty>=0.0.20, ruff>=0.15.4, mypy>=1.19.1,
  tox>=4.47.1, tox-uv>=1.33.1, ipykernel>=7.2.0
- Restructure tox.ini: remove lint/typecheck envs (redundant with
  pre-commit + CI), focus on runtime testing only
- Cross beartype compat with torch/jax backends (bt{020,021,022}×{torch,jax})
- Expand typecheck tests from 3 to 5 sample files covering all public API
  edge cases: imports, decorator signatures, Like types, make_array_type,
  and pyright-only annotations
- Update CI beartype-compat matrix to include backend combinations
- Fix missing B import in test_numpy.py (needed for ~B with PEP 563)

Author: acecchini

.github/workflows/ci.yml

@@ -39,7 +39,7 @@ jobs:
     steps:
       - uses: actions/checkout@v6
       - uses: astral-sh/setup-uv@v7
-      - run: uv run tox -e typecheck-compat
+      - run: uv run pytest tests/test_typecheck.py -v
 
   test:
     runs-on: ubuntu-latest
@@ -52,16 +52,19 @@ jobs:
     steps:
       - uses: actions/checkout@v6
       - uses: astral-sh/setup-uv@v7
-      - run: uv run tox -e "${{ matrix.python }}${{ matrix.suffix }}"
+      - run: uv run tox -e "py${{ matrix.python }}${{ matrix.suffix }}"
+        env:
+          TOX_PYTHON: python${{ matrix.python }}
 
   beartype-compat:
     runs-on: ubuntu-latest
     strategy:
       fail-fast: false
       matrix:
         version: ["bt020", "bt021", "bt022"]
-    name: beartype (${{ matrix.version }})
+        suffix: ["", "-torch", "-jax"]
+    name: beartype (${{ matrix.version }}${{ matrix.suffix }})
     steps:
       - uses: actions/checkout@v6
       - uses: astral-sh/setup-uv@v7
-      - run: uv run tox -e ${{ matrix.version }}
+      - run: uv run tox -e "${{ matrix.version }}${{ matrix.suffix }}"

pyproject.toml

@@ -8,19 +8,22 @@ requires-python = ">=3.12"
 dependencies = ["numpy>=2.2.0", "beartype>=0.22.9"]
 
 [dependency-groups]
+# CPU-only variants for dev/CI testing. Shapix is device-agnostic — it checks
+# .shape and .dtype only, so GPU/TPU builds (torch+cuda, jax[cuda12], etc.)
+# work without any extra configuration. Users install their own backend.
 torch = ["torch>=2.6.0"]
 jax = ["jax[cpu]>=0.5.0"]
 dev = [
-  "ruff>=0.14.11",
+  "ruff>=0.15.4",
   "pyright>=1.1.408",
-  "mypy>=1.15",
-  "ty>=0.0.1a7",
+  "mypy>=1.19.1",
+  "ty>=0.0.20",
   "pytest>=9.0.2",
   "pre-commit>=4.5.1",
-  "tox>=4.19",
-  "tox-uv>=1",
+  "tox>=4.47.1",
+  "tox-uv>=1.33.1",
   "codespell>=2.4.1",
-  "ipykernel>=7.1.0",
+  "ipykernel>=7.2.0",
   "matplotlib>=3.10.8",
 ]
 

tests/test_numpy.py

@@ -773,15 +773,15 @@ def f(x: F32[3, N]) -> F32[3, N]:
       return x
 
     with pytest.raises(BeartypeCallHintParamViolation):
-      f(np.ones((4, 5), dtype=np.int32))
+      f(np.ones((4, 5), dtype=np.int32))  # type: ignore
 
   def test_both_wrong(self) -> None:
     @beartype
     def f(x: F32[N]) -> F32[N]:
       return x
 
     with pytest.raises(BeartypeCallHintParamViolation):
-      f(np.ones(3, dtype=np.int64))
+      f(np.ones(3, dtype=np.int64))  # type: ignore
 
 
 # =====================================================================
@@ -824,7 +824,7 @@ def test_f32like_accepted(self, value: object) -> None:
     def f(x: F32Like) -> float:
       return float(np.asarray(x).sum())
 
-    f(value)
+    f(value)  # type: ignore
 
 
 class TestArrayLikeRejection:

tests/test_typecheck.py

@@ -3,6 +3,10 @@
 Runs pyright, mypy, and ty on sample files under ``tests/typing/`` to verify
 that shapix's public API type-checks cleanly across multiple checkers.
 
+Test matrix:
+- **Common files** (all checkers): imports, decorator, like types, make_array_type
+- **Pyright-only files**: full F32[N, C] annotations (TYPE_CHECKING stubs)
+
 These tests require ``pyright``, ``mypy``, and ``ty`` to be installed.
 """
 
@@ -16,6 +20,22 @@
 
 TYPING_DIR = Path(__file__).parent / "typing"
 
+# ---------------------------------------------------------------------------
+# File categories
+# ---------------------------------------------------------------------------
+
+# Files that should pass cleanly on ALL type checkers
+COMMON_FILES = [
+  "check_imports.py",
+  "check_decorator.py",
+  "check_like_types.py",
+  "check_make_array_type.py",
+]
+
+# Files that only pyright supports (F32[N, C] subscript annotations)
+PYRIGHT_ONLY_FILES = ["check_annotations.py"]
+
+
 # ---------------------------------------------------------------------------
 # Helpers
 # ---------------------------------------------------------------------------
@@ -24,7 +44,7 @@
 def _run(
   cmd: list[str], *, cwd: Path | None = None
 ) -> subprocess.CompletedProcess[str]:
-  return subprocess.run(cmd, capture_output=True, text=True, timeout=60, cwd=cwd)
+  return subprocess.run(cmd, capture_output=True, text=True, timeout=120, cwd=cwd)
 
 
 def _skip_if_missing(tool: str) -> None:
@@ -33,50 +53,48 @@ def _skip_if_missing(tool: str) -> None:
 
 
 # ---------------------------------------------------------------------------
-# Tests — common (all checkers should pass)
+# Pyright — common + pyright-only files
 # ---------------------------------------------------------------------------
 
-COMMON_FILES = ["check_imports.py", "check_decorator.py"]
-
 
 @pytest.mark.typecheck
-@pytest.mark.parametrize("filename", COMMON_FILES)
 class TestPyright:
+  @pytest.mark.parametrize("filename", COMMON_FILES + PYRIGHT_ONLY_FILES)
   def test_pyright(self, filename: str) -> None:
     _skip_if_missing("pyright")
     result = _run(["pyright", str(TYPING_DIR / filename)])
-    assert result.returncode == 0, f"pyright failed:\n{result.stdout}\n{result.stderr}"
+    assert result.returncode == 0, (
+      f"pyright failed on {filename}:\n{result.stdout}\n{result.stderr}"
+    )
+
+
+# ---------------------------------------------------------------------------
+# Mypy — common files only
+# ---------------------------------------------------------------------------
 
 
 @pytest.mark.typecheck
-@pytest.mark.parametrize("filename", COMMON_FILES)
 class TestMypy:
+  @pytest.mark.parametrize("filename", COMMON_FILES)
   def test_mypy(self, filename: str) -> None:
     _skip_if_missing("mypy")
     result = _run(["mypy", str(TYPING_DIR / filename), "--ignore-missing-imports"])
-    assert result.returncode == 0, f"mypy failed:\n{result.stdout}\n{result.stderr}"
-
-
-@pytest.mark.typecheck
-@pytest.mark.parametrize("filename", COMMON_FILES)
-class TestTy:
-  def test_ty(self, filename: str) -> None:
-    _skip_if_missing("ty")
-    result = _run(["ty", "check", str(TYPING_DIR / filename)])
-    assert result.returncode == 0, f"ty failed:\n{result.stdout}\n{result.stderr}"
+    assert result.returncode == 0, (
+      f"mypy failed on {filename}:\n{result.stdout}\n{result.stderr}"
+    )
 
 
 # ---------------------------------------------------------------------------
-# Tests — pyright-only (full F32[N, C] annotation pattern)
+# ty — common files only
 # ---------------------------------------------------------------------------
 
-PYRIGHT_FILES = ["check_annotations.py"]
-
 
 @pytest.mark.typecheck
-@pytest.mark.parametrize("filename", PYRIGHT_FILES)
-class TestPyrightAnnotations:
-  def test_pyright_annotations(self, filename: str) -> None:
-    _skip_if_missing("pyright")
-    result = _run(["pyright", str(TYPING_DIR / filename)])
-    assert result.returncode == 0, f"pyright failed:\n{result.stdout}\n{result.stderr}"
+class TestTy:
+  @pytest.mark.parametrize("filename", COMMON_FILES)
+  def test_ty(self, filename: str) -> None:
+    _skip_if_missing("ty")
+    result = _run(["ty", "check", str(TYPING_DIR / filename)])
+    assert result.returncode == 0, (
+      f"ty failed on {filename}:\n{result.stdout}\n{result.stderr}"
+    )