TYP: add a common._typing.Capabilities typed dict type

Author: jorenham

array_api_compat/common/_typing.py

@@ -1,10 +1,11 @@
 from __future__ import annotations
 
 from types import ModuleType as Namespace
-from typing import Any, Protocol, TypeAlias, TypeVar
+from typing import Any, Protocol, TypeAlias, TypedDict, TypeVar
 
 _T_co = TypeVar("_T_co", covariant=True)
 
+
 class NestedSequence(Protocol[_T_co]):
     def __getitem__(self, key: int, /) -> _T_co | NestedSequence[_T_co]: ...
     def __len__(self, /) -> int: ...
@@ -19,6 +20,16 @@ class HasShape(Protocol[_T_co]):
     def shape(self, /) -> _T_co: ...
 
 
+Capabilities = TypedDict(
+    "Capabilities",
+    {
+        "boolean indexing": bool,
+        "data-dependent shapes": bool,
+        "max dimensions": int,
+    },
+)
+
+
 SupportsBufferProtocol: TypeAlias = Any
 Array: TypeAlias = Any
 Device: TypeAlias = Any
@@ -27,12 +38,13 @@ def shape(self, /) -> _T_co: ...
 
 __all__ = [
     "Array",
-    "SupportsArrayNamespace",
+    "Capabilities",
     "DType",
     "Device",
     "HasShape",
     "Namespace",
     "NestedSequence",
+    "SupportsArrayNamespace",
     "SupportsBufferProtocol",
 ]