🚚 move HasArrayNamespace

Signed-off-by: nstarman <nstarman@users.noreply.github.com>

Author: nstarman

.github/workflows/ci.yml

@@ -88,11 +88,15 @@ jobs:
           python-version: "3.11"
           activate-environment: true
 
-      - name: get major numpy version
-        id: numpy-major
+      - name: get major.minor numpy version
+        id: numpy-version
         run: |
-          version=$(echo ${{ matrix.numpy-version }} | cut -c 1)
-          echo "::set-output name=version::$version"
+          version="${{ matrix.numpy-version }}"
+          major=$(echo "$version" | cut -d. -f1)
+          minor=$(echo "$version" | cut -d. -f2)
+
+          echo "major=$major" >> $GITHUB_OUTPUT
+          echo "minor=$minor" >> $GITHUB_OUTPUT
 
       - name: install deps
         run: |
@@ -101,10 +105,29 @@ jobs:
 
       # NOTE: `uv run --with=...` will be ignored by mypy (and `--isolated` does not help)
       - name: mypy
-        run: >
-          uv run --no-sync --active
-          mypy --tb --no-incremental --cache-dir=/dev/null
-          tests/integration/test_numpy${{ steps.numpy-major.outputs.version }}.pyi
+        run: |
+          major="${{ steps.numpy-version.outputs.major }}"
+          minor="${{ steps.numpy-version.outputs.minor }}"
+
+          # Directory containing versioned test files
+          prefix="tests/integration"
+          files=""
+
+          # Find all test files matching the current major version
+          for path in $(find "$prefix" -name "test_numpy${major}p*.pyi"); do
+            # Extract file name
+            fname=$(basename "$path")
+            # Parse the minor version from the filename
+            fminor=$(echo "$fname" | sed -E "s/test_numpy${major}p([0-9]+)\.pyi/\1/")
+            # Include files where minor version ≤ NumPy's minor
+            if [ "$fminor" -le "$minor" ]; then
+              files="$files $path"
+            fi
+          done
+
+          uv run --no-sync --active \
+            mypy --tb --no-incremental --cache-dir=/dev/null \
+            $files
 
       # TODO: (based)pyright
 

pyproject.toml

@@ -124,6 +124,7 @@ ignore = [
   "FBT", # flake8-boolean-trap
   "FIX", # flake8-fixme
   "ISC001", # Conflicts with formatter
+  "PYI041", # Use `float` instead of `int | float`
 ]
 
 [tool.ruff.lint.pylint]

src/array_api_typing/__init__.py

@@ -6,5 +6,5 @@
     "__version_tuple__",
 )
 
-from ._namespace import HasArrayNamespace
+from ._array import HasArrayNamespace
 from ._version import version as __version__, version_tuple as __version_tuple__

src/array_api_typing/_namespace.py src/array_api_typing/_array.pysrc/array_api_typing/_namespace.py renamed to src/array_api_typing/_array.py

@@ -4,12 +4,17 @@
 from typing import Literal, Protocol
 from typing_extensions import TypeVar
 
-T_co = TypeVar("T_co", covariant=True, default=ModuleType)
+NamespaceT_co = TypeVar("NamespaceT_co", covariant=True, default=ModuleType)
 
 
-class HasArrayNamespace(Protocol[T_co]):
+class HasArrayNamespace(Protocol[NamespaceT_co]):
     """Protocol for classes that have an `__array_namespace__` method.
 
+    This `Protocol` is intended for use in static typing to ensure that an
+    object has an `__array_namespace__` method that returns a namespace for
+    array operations. This `Protocol` should not be used at runtime, for type
+    checking or as a base class.
+
     Example:
     >>> import array_api_typing as xpt
     >>>
@@ -27,4 +32,4 @@ class HasArrayNamespace(Protocol[T_co]):
 
     def __array_namespace__(
         self, /, *, api_version: Literal["2021.12"] | None = None
-    ) -> T_co: ...
+    ) -> NamespaceT_co: ...

tests/integration/test_numpy1.pyi

@@ -0,0 +1,29 @@
+# mypy: disable-error-code="no-redef"
+
+from types import ModuleType
+from typing import TypeAlias
+
+import numpy.array_api as np  # type: ignore[import-not-found, unused-ignore]
+
+import array_api_typing as xpt
+
+# Define NDArrays against which we can test the protocols
+# Note that `np.array_api` doesn't support boolean arrays.
+nparr = np.eye(2)
+nparr_i32 = np.asarray([1], dtype=np.int32)
+nparr_f32 = np.asarray([1.0], dtype=np.float32)
+
+# =========================================================
+# `xpt.HasArrayNamespace`
+
+_: xpt.HasArrayNamespace[ModuleType] = nparr
+_: xpt.HasArrayNamespace[ModuleType] = nparr_i32
+_: xpt.HasArrayNamespace[ModuleType] = nparr_f32
+
+# Check `__array_namespace__` method
+a_ns: xpt.HasArrayNamespace[ModuleType] = nparr
+ns: ModuleType = a_ns.__array_namespace__()
+
+# Incorrect values are caught when using `__array_namespace__` and
+# backpropagated to the type of `a_ns`
+_: xpt.HasArrayNamespace[dict[str, int]] = nparr  # not caught

tests/integration/test_numpy1p0.pyi

@@ -0,0 +1,36 @@
+# mypy: disable-error-code="no-redef"
+
+from types import ModuleType
+from typing import Any, TypeAlias
+
+import numpy as np
+import numpy.typing as npt
+
+import array_api_typing as xpt
+
+# DType aliases
+F32: TypeAlias = np.float32
+I32: TypeAlias = np.int32
+
+# Define NDArrays against which we can test the protocols
+nparr: npt.NDArray[Any]
+nparr_i32: npt.NDArray[I32]
+nparr_f32: npt.NDArray[F32]
+nparr_b: npt.NDArray[np.bool_]
+
+# =========================================================
+# `xpt.HasArrayNamespace`
+
+# Check assignment
+_: xpt.HasArrayNamespace[ModuleType] = nparr
+_: xpt.HasArrayNamespace[ModuleType] = nparr_i32
+_: xpt.HasArrayNamespace[ModuleType] = nparr_f32
+_: xpt.HasArrayNamespace[ModuleType] = nparr_b
+
+# Check `__array_namespace__` method
+a_ns: xpt.HasArrayNamespace[ModuleType] = nparr
+ns: ModuleType = a_ns.__array_namespace__()
+
+# Incorrect values are caught when using `__array_namespace__` and
+# backpropagated to the type of `a_ns`
+_: xpt.HasArrayNamespace[dict[str, int]] = nparr  # not caught

tests/integration/test_numpy2.pyi

@@ -0,0 +1,5 @@
+from test_numpy2p0 import nparr
+
+import array_api_typing as xpt
+
+_: xpt.HasArrayNamespace[dict[str, int]] = nparr  # type: ignore[assignment]