✨ HasNDim

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

Author: nstarman

src/array_api_typing/_array.py

@@ -97,11 +97,26 @@ def mT(self) -> Self:  # noqa: N802
         ...
 
 
+class HasNDim(Protocol):
+    """Protocol for array classes that have a number of dimensions attribute."""
+
+    @property
+    def ndim(self) -> int:
+        """Number of array dimensions (axes).
+
+        Returns:
+            int: number of array dimensions (axes).
+
+        """
+        ...
+
+
 class Array(
     # ------ Attributes -------
     HasDType[DTypeT_co],
     HasDevice,
     HasMatrixTranspose,
+    HasNDim,
     # ------- Methods ---------
     HasArrayNamespace[NamespaceT_co],
     # -------------------------

tests/integration/test_numpy1p0.pyi

@@ -63,3 +63,7 @@ _: object = x_i32.device
 # Check Attribute `.mT`
 _: xpt.Array[dtype[Any]] = x_f32.mT
 _: xpt.Array[dtype[Any]] = x_i32.mT
+
+# Check Attribute `.ndim`
+_: int = x_f32.ndim
+_: int = x_i32.ndim

tests/integration/test_numpy2p0.pyi

@@ -71,3 +71,8 @@ _: object = x_b.device
 _: xpt.Array[np.dtype[F32]] = x_f32.mT
 _: xpt.Array[np.dtype[I32]] = x_i32.mT
 _: xpt.Array[np.dtype[B]] = x_b.mT
+
+# Check Attribute `.ndim`
+_: int = x_f32.ndim
+_: int = x_i32.ndim
+_: int = x_b.ndim