✨Array class

Signed-off-by: nstarman <[email protected]>

Author: nstarman

src/array_api_typing/__init__.py

@@ -1,10 +1,11 @@
 """Static typing support for the array API standard."""
 
 __all__ = (
+    "Array",
     "HasArrayNamespace",
     "__version__",
     "__version_tuple__",
 )
 
-from ._array import HasArrayNamespace
+from ._array import Array, HasArrayNamespace
 from ._version import version as __version__, version_tuple as __version_tuple__

src/array_api_typing/_array.py

@@ -1,4 +1,7 @@
-__all__ = ("HasArrayNamespace",)
+__all__ = (
+    "Array",
+    "HasArrayNamespace",
+)
 
 from types import ModuleType
 from typing import Literal, Protocol
@@ -33,3 +36,10 @@ class HasArrayNamespace(Protocol[NamespaceT_co]):
     def __array_namespace__(
         self, /, *, api_version: Literal["2021.12"] | None = None
     ) -> NamespaceT_co: ...
+
+
+class Array(
+    HasArrayNamespace[NamespaceT_co],
+    Protocol[NamespaceT_co],
+):
+    """Array API specification for array object attributes and methods."""

tests/integration/test_numpy1.pyi

@@ -32,3 +32,9 @@ 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
+
+# =========================================================
+# `xpt.Array`
+
+# Check NamespaceT_co assignment
+a_ns: xpt.Array[ModuleType] = nparr

tests/integration/test_numpy2.0.pyi

@@ -34,3 +34,9 @@ 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
+
+# =========================================================
+# `xpt.Array`
+
+# Check NamespaceT_co assignment
+a_ns: xpt.Array[ModuleType] = nparr