✨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
@@ -12,7 +15,7 @@ class HasArrayNamespace(Protocol[NamespaceT_co]):
 
     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
+    array operations. This `Protocol` should not be used at runtime for type
     checking or as a base class.
 
     Example:
@@ -52,3 +55,26 @@ def __array_namespace__(
 
         """
         ...
+
+
+class Array(
+    HasArrayNamespace[NamespaceT_co],
+    # -------------------------
+    Protocol[NamespaceT_co],
+):
+    """Array API specification for array object attributes and methods.
+
+    The type is: ``Array[+NamespaceT = ModuleType] = Array[NamespaceT]`` where:
+
+    - `NamespaceT` is the type of the array namespace. It defaults to
+      `ModuleType`, which is the most common form of array namespace (e.g.,
+      `numpy`, `cupy`, etc.). However, it can be any type, e.g. a
+      `types.SimpleNamespace`, to allow for wrapper libraries to
+      semi-dynamically define their own array namespaces based on the wrapped
+      array type.
+
+    This type is intended for use in static typing to ensure that an object has
+    the attributes and methods defined in the array API specification. It should
+    not be used at runtime for type checking or as a base class.
+
+    """

tests/integration/test_numpy1p0.pyi

@@ -27,3 +27,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_numpy2p0.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