?HAMA

Edit File Size: 2.07 KB

//opt/cloudlinux/venv/lib/python3.11/site-packages/numpy/testing/overrides.py

Text Base64

"""Tools for testing implementations of __array_function__ and ufunc overrides

"""

from numpy.core.overrides import ARRAY_FUNCTIONS as _array_functions
from numpy import ufunc as _ufunc
import numpy.core.umath as _umath

def get_overridable_numpy_ufuncs():
    """List all numpy ufuncs overridable via `__array_ufunc__`

Parameters
    ----------
    None

Returns
    -------
    set
        A set containing all overridable ufuncs in the public numpy API.
    """
    ufuncs = {obj for obj in _umath.__dict__.values()
              if isinstance(obj, _ufunc)}
    return ufuncs

def allows_array_ufunc_override(func):
    """Determine 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.

"""
    return isinstance(func, np.ufunc)

def get_overridable_numpy_array_functions():
    """List 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__`.

"""
    # 'import numpy' doesn't import recfunctions, so make sure it's imported
    # so ufuncs defined there show up in the ufunc listing
    from numpy.lib import recfunctions
    return _array_functions.copy()

def allows_array_function_override(func):
    """Determine 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.
    """
    return func in _array_functions

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Name	Size	Perms	Modified	Actions
tests DIR	-	drwxr-xr-x	2026-02-06 08:01:08	Edit Download recursive recursive
_private DIR	-	drwxr-xr-x	2026-02-06 08:01:08	Edit Download recursive recursive
__pycache__ DIR	-	drwxr-xr-x	2026-02-06 08:01:08	Edit Download recursive recursive
overrides.py	2.07 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
print_coercion_tables.py	6.04 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
setup.py	709 B	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
__init__.py	595 B	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
__init__.pyi	1.63 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive

Directory Contents