🏷️ fix stubtest errors in lib._array_utils_impl

Author: jorenham

.mypyignore-todo

@@ -15,7 +15,6 @@ numpy._pyinstaller.hook-numpy
 numpy.ctypeslib._ctypeslib
 numpy.fft.helper
 
-numpy.lib.(_array_utils_impl|array_utils).normalize_axis_tuple
 numpy.lib._iotools.NameValidator.defaultdeletechars
 numpy.lib.format.open_memmap
 numpy.lib.format.read_array(_header_(1|2)_0)?

src/numpy-stubs/lib/_array_utils_impl.pyi

@@ -1,20 +1,30 @@
-from collections.abc import Iterable
-from typing import Any
-
-import numpy as np
-import numpy.typing as npt
+from collections.abc import Iterable, Mapping
+from typing import Any, Protocol, type_check_only
 
 __all__ = ["byte_bounds", "normalize_axis_index", "normalize_axis_tuple"]
 
+###
+
+@type_check_only
+class _HasSizeAndArrayInterface(Protocol):
+    @property
+    def size(self, /) -> int: ...
+    @property  # `TypedDict` cannot be used because it rejects `dict[str, Any]`
+    def __array_interface__(self, /) -> Mapping[str, Any]: ...
+
+###
+
 # NOTE: In practice `byte_bounds` can (potentially) take any object
 # implementing the `__array_interface__` protocol. The caveat is
 # that certain keys, marked as optional in the spec, must be present for
 #  `byte_bounds`. This concerns `"strides"` and `"data"`.
-def byte_bounds(a: np.generic | npt.NDArray[Any]) -> tuple[int, int]: ...
-def normalize_axis_index(axis: int = ..., ndim: int = ..., msg_prefix: str | None = ...) -> int: ...
+def byte_bounds(a: _HasSizeAndArrayInterface) -> tuple[int, int]: ...
+
+###
+def normalize_axis_index(axis: int, ndim: int, msg_prefix: str | None = None) -> int: ...
 def normalize_axis_tuple(
     axis: int | Iterable[int],
-    ndim: int = ...,
-    argname: str | None = ...,
-    allow_duplicate: bool | None = ...,
-) -> tuple[int, int]: ...
+    ndim: int,
+    argname: str | None = None,
+    allow_duplicate: bool = False,
+) -> tuple[int, ...]: ...