fKJdZddlmZddlmZddlmcm Z dZ dZ dZ dZy) zNTools for testing implementations of __array_function__ and ufunc overrides )ARRAY_FUNCTIONS)ufuncNctjjDchc]}t|tr|}}|Scc}w)zList all numpy ufuncs overridable via `__array_ufunc__` Parameters ---------- None Returns ------- set A set containing all overridable ufuncs in the public numpy API. )_umath__dict__values isinstance_ufunc)objufuncss 9/usr/lib/python3/dist-packages/numpy/testing/overrides.pyget_overridable_numpy_ufuncsr sA$__335*cC(*F* M*s>c6t|tjS)aDetermine if a function can be overridden via `__array_ufunc__` Parameters ---------- func : callable Function that may be overridable via `__array_ufunc__` Returns ------- bool `True` if `func` is overridable via `__array_ufunc__` and `False` otherwise. Notes ----- This function is equivalent to ``isinstance(func, np.ufunc)`` and will work correctly for ufuncs defined outside of Numpy. )r nprfuncs r allows_array_ufunc_overriders( dBHH %%c6ddlm}tjS)aList all numpy functions overridable via `__array_function__` Parameters ---------- None Returns ------- set A set containing all functions in the public numpy API that are overridable via `__array_function__`. r recfunctions) numpy.libr_array_functionscopyrs r %get_overridable_numpy_array_functionsr2s '  ""rc|tvS)ahDetermine if a Numpy function can be overridden via `__array_function__` Parameters ---------- func : callable Function that may be overridable via `__array_function__` Returns ------- bool `True` if `func` is a function in the Numpy API that is overridable via `__array_function__` and `False` otherwise. )rrs r allows_array_function_overriderEs # ##r)__doc__numpy.core.overridesrrnumpyrr numpy.core.umathcoreumathrrrrrrr r%s- E!!!"&.#&$r