✨: make HasNamespace

Signed-off-by: Nathaniel Starkman <[email protected]>

Author: nstarman

.gitignore

@@ -22,3 +22,6 @@ venv/
 
 # Build docs
 /src/array_api_typing/_version.py
+
+# Mac files
+.DS_Store

src/array_api_typing/__init__.py

@@ -1,5 +1,6 @@
 """Static typing support for the array API standard."""
 
-__all__ = ()
+__all__ = ["HasNamespace"]
 
 from ._version import version as __version__ , version_tuple as __version_tuple__
+from ._namespace import HasNamespace

src/array_api_typing/_namespace.py

@@ -0,0 +1,27 @@
+"""Static typing support for the array API standard."""
+
+__all__ = ["HasNamespace"]
+
+from typing import Protocol, runtime_checkable
+
+
+@runtime_checkable
+class HasNamespace(Protocol):
+    """Protocol for classes that have an `__array_namespace__` method.
+
+    This is a runtime checkable protocol.
+
+    Example:
+    >>> import array_api_typing as xpt
+    >>>
+    >>> class MyArray:
+    ...     def __array_namespace__(self):
+    ...         return "my namespace"
+    >>>
+    >>> x = MyArray()
+    >>> print(isinstance(x, xpt.HasNamespace))
+    True
+
+    """
+
+    def __array_namespace__(self) -> object: ...  # noqa: PLW3201