tests: rework downloading the wheelhouse for isolation tests

.github/workflows/ci.yml

@@ -187,7 +187,7 @@ jobs:
 
       - name: Install min requirements
         run: |
-          uv pip install -e. --group=test --resolution=lowest-direct --system
+          uv pip install -e. --group=test-pybind11 --resolution=lowest-direct --system
 
       - name: Setup CMake 3.15
         uses: jwlawson/actions-setup-cmake@v2.0
@@ -223,7 +223,7 @@ jobs:
         run: python3.13t -m venv /venv
 
       - name: Install deps
-        run: /venv/bin/pip install -e . --group=test ninja
+        run: /venv/bin/pip install -e . --group=test-pybind11 ninja
 
       - name: Test package
         run: /venv/bin/pytest
@@ -246,7 +246,7 @@ jobs:
             cmake ninja git make gcc-g++ python39 python39-devel python39-pip
 
       - name: Install
-        run: python3.9 -m pip install . --group=test
+        run: python3.9 -m pip install . --group=test-pybind11
 
       - name: Test package
         run:
@@ -312,7 +312,7 @@ jobs:
           persist-credentials: false
 
       - name: Install
-        run: python -m pip install . --group=test
+        run: python -m pip install . --group=test-pybind11
 
       - name: Test package
         run: >-

pyproject.toml

@@ -1,5 +1,5 @@
 [build-system]
-requires = ["hatchling", "hatch-vcs"]
+requires = ["hatchling >=1.24", "hatch-vcs >=0.4"]
 build-backend = "hatchling.build"
 
 [project]
@@ -77,6 +77,9 @@ hatch.scikit-build = "scikit_build_core.hatch.hooks"
 test = [
     "build >=0.8",
     "cattrs >=22.2.0",
+    "filelock >=3.8.0",
+    "hatchling >=1.24",
+    "hatch-vcs >=0.4",
     "pip>=23; python_version<'3.13'",
     "pip>=24.1; python_version>='3.13'",
     "pytest >=7.2",
@@ -89,10 +92,6 @@ test = [
     "virtualenv >=20.20",
     "wheel >=0.40",
 ]
-test-hatchling = [
-    { include-group = "test" },
-    "hatchling >=1.24.0",
-]
 test-meta = [
     { include-group = "test" },
     "hatch-fancy-pypi-readme>=22.3",
@@ -120,7 +119,6 @@ cov = [
 ]
 dev = [
     { include-group = "cov" },
-    { include-group = "test-hatchling" },
     { include-group = "test-meta" },
     { include-group = "test-numpy" },
     { include-group = "test-pybind11" },

tests/conftest.py

@@ -14,6 +14,7 @@
 from typing import Any, Literal, overload
 
 import virtualenv as _virtualenv
+from filelock import FileLock
 
 if sys.version_info < (3, 11):
     import tomli as tomllib
@@ -26,6 +27,7 @@
     from typing import TypeGuard
 
 
+import download_wheels
 import pytest
 from packaging.requirements import Requirement
 from packaging.version import Version
@@ -37,53 +39,32 @@
 
 
 @pytest.fixture(scope="session")
-def pep518_wheelhouse(tmp_path_factory: pytest.TempPathFactory) -> Path:
-    wheelhouse = tmp_path_factory.mktemp("wheelhouse")
-
-    subprocess.run(
-        [
-            sys.executable,
-            "-m",
-            "pip",
-            "wheel",
-            "--wheel-dir",
-            str(wheelhouse),
-            f"{BASE}",
-        ],
-        check=True,
-    )
-    packages = [
-        "build",
-        "cython",
-        "hatchling",
-        "pip",
-        "setuptools",
-        "virtualenv",
-        "wheel",
-    ]
+def pep518_wheelhouse(pytestconfig: pytest.Config) -> Path:
+    wheelhouse = pytestconfig.cache.mkdir("wheelhouse")
+
+    main_lock = FileLock(wheelhouse / "main.lock")
+    with main_lock:
+        subprocess.run(
+            [
+                sys.executable,
+                "-m",
+                "pip",
+                "wheel",
+                "--wheel-dir",
+                str(wheelhouse),
+                "--no-build-isolation",
+                f"{BASE}",
+            ],
+            check=True,
+            capture_output=True,
+            text=True,
+        )
+
+    wheels_lock = FileLock(wheelhouse / "wheels.lock")
+    with wheels_lock:
+        if not all(list(wheelhouse.glob(f"{p}*.whl")) for p in download_wheels.WHEELS):
+            download_wheels.prepare(wheelhouse)
 
-    if importlib.util.find_spec("cmake") is not None:
-        packages.append("cmake")
-
-    if importlib.util.find_spec("ninja") is not None:
-        packages.append("ninja")
-
-    if importlib.util.find_spec("pybind11") is not None:
-        packages.append("pybind11")
-
-    subprocess.run(
-        [
-            sys.executable,
-            "-m",
-            "pip",
-            "download",
-            "-q",
-            "-d",
-            str(wheelhouse),
-            *packages,
-        ],
-        check=True,
-    )
     return wheelhouse
 
 

tests/utils/download_wheels.py

@@ -0,0 +1,56 @@
+# /// script
+# dependencies = ["pip"]
+# ///
+
+"""
+Download wheels into the pytest cache. Must be run from the pytest directory
+(project root, usually). If run in an environment, requires pip. Only downloads
+pybind11, ninja, or cmake if those are in the environment already.
+"""
+
+import importlib.util
+import subprocess
+import sys
+from pathlib import Path
+
+WHEELS = [
+    "build",
+    "cython",
+    "hatchling",
+    "pip",
+    "setuptools",
+    "virtualenv",
+    "wheel",
+]
+
+if importlib.util.find_spec("cmake") is not None:
+    WHEELS.append("cmake")
+
+if importlib.util.find_spec("ninja") is not None:
+    WHEELS.append("ninja")
+
+if importlib.util.find_spec("pybind11") is not None:
+    WHEELS.append("pybind11")
+
+
+def prepare(wheelhouse: Path) -> None:
+    subprocess.run(
+        [
+            sys.executable,
+            "-m",
+            "pip",
+            "download",
+            "-q",
+            "-d",
+            str(wheelhouse),
+            *WHEELS,
+        ],
+        check=True,
+    )
+    print(f"Downloaded wheels to {wheelhouse}")
+
+
+if __name__ == "__main__":
+    wheelhouse = Path(".pytest_cache/d/wheelhouse")
+    wheelhouse.mkdir(parents=True, exist_ok=True)
+    prepare(wheelhouse)