diff --git a/src/numpy-stubs/testing/overrides.pyi b/src/numpy-stubs/testing/overrides.pyi new file mode 100644 index 00000000..916154c1 --- /dev/null +++ b/src/numpy-stubs/testing/overrides.pyi @@ -0,0 +1,10 @@ +from collections.abc import Callable, Hashable +from typing import Any +from typing_extensions import TypeIs + +import numpy as np + +def get_overridable_numpy_ufuncs() -> set[np.ufunc]: ... +def get_overridable_numpy_array_functions() -> set[Callable[..., Any]]: ... +def allows_array_ufunc_override(func: object) -> TypeIs[np.ufunc]: ... +def allows_array_function_override(func: Hashable) -> bool: ...