🧙 split numpy 1 and 2 integration tests with black voodoo magic

Author: jorenham

.github/workflows/ci.yml

@@ -70,8 +70,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        # numpy<2 does not support the array api
-        numpy-version: ["2.0.0", "2.1.3", "2.2.6", "2.3.1"]
+        numpy-version: ["1.25.0", "1.26.4", "2.0.2", "2.1.3", "2.2.6", "2.3.1"]
 
     steps:
       - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683  # v4.2.2
@@ -81,6 +80,12 @@ jobs:
           python-version: "3.11"
           activate-environment: true
 
+      - name: get major numpy version
+        id: numpy-major
+        run: |
+          version=$(echo ${{ matrix.numpy-version }} | cut -c 1-7)
+          echo "::set-output name=version::$version"
+
       - name: install deps
         run: |
           uv sync --no-editable --group=mypy
@@ -90,10 +95,10 @@ jobs:
       - name: mypy
         run: >
           uv run --no-sync --active
-          mypy --tb --no-incremental --cache-dir=/dev/null tests/integration/test_numpy.pyi
+          mypy --tb --no-incremental --cache-dir=/dev/null
+          tests/integration/test_numpy${{ steps.numpy-major.outputs.version }}.pyi
 
       # TODO: (based)pyright
 
-  # TODO: integration tests for array-api-compat + numpy<2
   # TODO: integration tests for array-api-strict
   # TODO: integration tests for 3rd party libs such as cupy, pytorch, tensorflow, dask, etc.

tests/integration/test_numpy1.pyi

@@ -0,0 +1,11 @@
+from typing import Any
+
+import numpy.array_api as np
+
+import array_api_typing as xpt
+
+###
+# Ensure that `np.ndarray` instances are assignable to `xpt.HasArrayNamespace`.
+
+arr: np.Array
+arr_namespace: xpt.HasArrayNamespace[Any] = arr

tests/integration/test_numpy.pyi renamed to tests/integration/test_numpy2.pyi

@@ -1,13 +1,9 @@
-from typing import Any, TypeVar
+from typing import Any
 
 import numpy.typing as npt
 
 import array_api_typing as xpt
 
-_T = TypeVar("_T", bound=npt.NDArray[Any])
-
-def get_namespace(obj: xpt.HasArrayNamespace[_T], /) -> _T: ...
-
 ###
 # Ensure that `np.ndarray` instances are assignable to `xpt.HasArrayNamespace`.