✨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
@@ -52,3 +55,10 @@ def __array_namespace__(
 
         """
         ...
+
+
+class Array(
+    HasArrayNamespace[NamespaceT_co],
+    Protocol[NamespaceT_co],
+):
+    """Array API specification for array object attributes and methods."""

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