OXIESEC PANEL

Name	Size	Modified	Perms
📁 ..	-	10/28/2024 05:59:26 AM	rwxr-xr-x
📄 __init__.py	0 bytes	10/28/2024 05:59:24 AM	rw-r--r--
📁 __pycache__	-	10/28/2024 05:59:26 AM	rwxr-xr-x
📄 _locales.py	2.14 KB	10/28/2024 05:59:24 AM	rw-r--r--
📁 data	-	10/28/2024 05:59:26 AM	rwxr-xr-x
📄 test__exceptions.py	1.96 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_abc.py	2.27 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_api.py	20.32 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_arrayprint.py	33.68 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_conversion_utils.py	4.86 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_cpu_features.py	6.63 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_datetime.py	105.66 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_defchararray.py	23.81 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_deprecations.py	23.62 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_dtype.py	48.14 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_einsum.py	44.14 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_errstate.py	2.02 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_extint128.py	5.51 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_function_base.py	12.84 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_getlimits.py	4.2 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_half.py	22.56 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_indexerrors.py	5.01 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_indexing.py	48.11 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_item_selection.py	3.5 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_longdouble.py	12.74 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_machar.py	1.04 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_mem_overlap.py	28.18 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_memmap.py	7.29 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_multiarray.py	312.3 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_nditer.py	109.67 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_numeric.py	121.21 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_numerictypes.py	20.36 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_overrides.py	14.09 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_print.py	6.58 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_protocols.py	1.14 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_records.py	19.23 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_regression.py	86.76 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_scalar_ctors.py	2.52 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_scalar_methods.py	3.97 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_scalarbuffer.py	3.97 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_scalarinherit.py	2.08 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_scalarmath.py	27.9 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_scalarprint.py	14.8 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_shape_base.py	24.1 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_ufunc.py	82.01 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_umath.py	118.6 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_umath_accuracy.py	2.97 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_umath_complex.py	22.6 KB	10/28/2024 05:59:24 AM	rw-r--r--
📄 test_unicode.py	12.26 KB	10/28/2024 05:59:24 AM	rw-r--r--

Editing: test_overrides.py

import inspect
import sys
from unittest import mock

import numpy as np
from numpy.testing import (
    assert_, assert_equal, assert_raises, assert_raises_regex)
from numpy.core.overrides import (
    _get_implementing_args, array_function_dispatch,
    verify_matching_signatures, ARRAY_FUNCTION_ENABLED)
from numpy.compat import pickle
import pytest

requires_array_function = pytest.mark.skipif(
    not ARRAY_FUNCTION_ENABLED,
    reason="__array_function__ dispatch not enabled.")

def _return_not_implemented(self, *args, **kwargs):
    return NotImplemented

# need to define this at the top level to test pickling
@array_function_dispatch(lambda array: (array,))
def dispatched_one_arg(array):
    """Docstring."""
    return 'original'

@array_function_dispatch(lambda array1, array2: (array1, array2))
def dispatched_two_arg(array1, array2):
    """Docstring."""
    return 'original'

class TestGetImplementingArgs:

def test_ndarray(self):
        array = np.array(1)

args = _get_implementing_args([array])
        assert_equal(list(args), [array])

args = _get_implementing_args([array, array])
        assert_equal(list(args), [array])

args = _get_implementing_args([array, 1])
        assert_equal(list(args), [array])

args = _get_implementing_args([1, array])
        assert_equal(list(args), [array])

def test_ndarray_subclasses(self):

class OverrideSub(np.ndarray):
            __array_function__ = _return_not_implemented

class NoOverrideSub(np.ndarray):
            pass

array = np.array(1).view(np.ndarray)
        override_sub = np.array(1).view(OverrideSub)
        no_override_sub = np.array(1).view(NoOverrideSub)

args = _get_implementing_args([array, override_sub])
        assert_equal(list(args), [override_sub, array])

args = _get_implementing_args([array, no_override_sub])
        assert_equal(list(args), [no_override_sub, array])

args = _get_implementing_args(
            [override_sub, no_override_sub])
        assert_equal(list(args), [override_sub, no_override_sub])

def test_ndarray_and_duck_array(self):

class Other:
            __array_function__ = _return_not_implemented

array = np.array(1)
        other = Other()

args = _get_implementing_args([other, array])
        assert_equal(list(args), [other, array])

args = _get_implementing_args([array, other])
        assert_equal(list(args), [array, other])

def test_ndarray_subclass_and_duck_array(self):

class OverrideSub(np.ndarray):
            __array_function__ = _return_not_implemented

class Other:
            __array_function__ = _return_not_implemented

array = np.array(1)
        subarray = np.array(1).view(OverrideSub)
        other = Other()

assert_equal(_get_implementing_args([array, subarray, other]),
                     [subarray, array, other])
        assert_equal(_get_implementing_args([array, other, subarray]),
                     [subarray, array, other])

def test_many_duck_arrays(self):

class A:
            __array_function__ = _return_not_implemented

class B(A):
            __array_function__ = _return_not_implemented

class C(A):
            __array_function__ = _return_not_implemented

class D:
            __array_function__ = _return_not_implemented

a = A()
        b = B()
        c = C()
        d = D()

assert_equal(_get_implementing_args([1]), [])
        assert_equal(_get_implementing_args([a]), [a])
        assert_equal(_get_implementing_args([a, 1]), [a])
        assert_equal(_get_implementing_args([a, a, a]), [a])
        assert_equal(_get_implementing_args([a, d, a]), [a, d])
        assert_equal(_get_implementing_args([a, b]), [b, a])
        assert_equal(_get_implementing_args([b, a]), [b, a])
        assert_equal(_get_implementing_args([a, b, c]), [b, c, a])
        assert_equal(_get_implementing_args([a, c, b]), [c, b, a])

def test_too_many_duck_arrays(self):
        namespace = dict(__array_function__=_return_not_implemented)
        types = [type('A' + str(i), (object,), namespace) for i in range(33)]
        relevant_args = [t() for t in types]

actual = _get_implementing_args(relevant_args[:32])
        assert_equal(actual, relevant_args[:32])

with assert_raises_regex(TypeError, 'distinct argument types'):
            _get_implementing_args(relevant_args)

class TestNDArrayArrayFunction:

@requires_array_function
    def test_method(self):

class Other:
            __array_function__ = _return_not_implemented

class NoOverrideSub(np.ndarray):
            pass

class OverrideSub(np.ndarray):
            __array_function__ = _return_not_implemented

array = np.array([1])
        other = Other()
        no_override_sub = array.view(NoOverrideSub)
        override_sub = array.view(OverrideSub)

result = array.__array_function__(func=dispatched_two_arg,
                                          types=(np.ndarray,),
                                          args=(array, 1.), kwargs={})
        assert_equal(result, 'original')

result = array.__array_function__(func=dispatched_two_arg,
                                          types=(np.ndarray, Other),
                                          args=(array, other), kwargs={})
        assert_(result is NotImplemented)

result = array.__array_function__(func=dispatched_two_arg,
                                          types=(np.ndarray, NoOverrideSub),
                                          args=(array, no_override_sub),
                                          kwargs={})
        assert_equal(result, 'original')

result = array.__array_function__(func=dispatched_two_arg,
                                          types=(np.ndarray, OverrideSub),
                                          args=(array, override_sub),
                                          kwargs={})
        assert_equal(result, 'original')

with assert_raises_regex(TypeError, 'no implementation found'):
            np.concatenate((array, other))

expected = np.concatenate((array, array))
        result = np.concatenate((array, no_override_sub))
        assert_equal(result, expected.view(NoOverrideSub))
        result = np.concatenate((array, override_sub))
        assert_equal(result, expected.view(OverrideSub))

def test_no_wrapper(self):
        # This shouldn't happen unless a user intentionally calls
        # __array_function__ with invalid arguments, but check that we raise
        # an appropriate error all the same.
        array = np.array(1)
        func = lambda x: x
        with assert_raises_regex(AttributeError, '_implementation'):
            array.__array_function__(func=func, types=(np.ndarray,),
                                     args=(array,), kwargs={})

@requires_array_function
class TestArrayFunctionDispatch:

def test_pickle(self):
        for proto in range(2, pickle.HIGHEST_PROTOCOL + 1):
            roundtripped = pickle.loads(
                    pickle.dumps(dispatched_one_arg, protocol=proto))
            assert_(roundtripped is dispatched_one_arg)

def test_name_and_docstring(self):
        assert_equal(dispatched_one_arg.__name__, 'dispatched_one_arg')
        if sys.flags.optimize < 2:
            assert_equal(dispatched_one_arg.__doc__, 'Docstring.')

def test_interface(self):

class MyArray:
            def __array_function__(self, func, types, args, kwargs):
                return (self, func, types, args, kwargs)

original = MyArray()
        (obj, func, types, args, kwargs) = dispatched_one_arg(original)
        assert_(obj is original)
        assert_(func is dispatched_one_arg)
        assert_equal(set(types), {MyArray})
        # assert_equal uses the overloaded np.iscomplexobj() internally
        assert_(args == (original,))
        assert_equal(kwargs, {})

def test_not_implemented(self):

class MyArray:
            def __array_function__(self, func, types, args, kwargs):
                return NotImplemented

array = MyArray()
        with assert_raises_regex(TypeError, 'no implementation found'):
            dispatched_one_arg(array)

@requires_array_function
class TestVerifyMatchingSignatures:

def test_verify_matching_signatures(self):

verify_matching_signatures(lambda x: 0, lambda x: 0)
        verify_matching_signatures(lambda x=None: 0, lambda x=None: 0)
        verify_matching_signatures(lambda x=1: 0, lambda x=None: 0)

with assert_raises(RuntimeError):
            verify_matching_signatures(lambda a: 0, lambda b: 0)
        with assert_raises(RuntimeError):
            verify_matching_signatures(lambda x: 0, lambda x=None: 0)
        with assert_raises(RuntimeError):
            verify_matching_signatures(lambda x=None: 0, lambda y=None: 0)
        with assert_raises(RuntimeError):
            verify_matching_signatures(lambda x=1: 0, lambda y=1: 0)

def test_array_function_dispatch(self):

with assert_raises(RuntimeError):
            @array_function_dispatch(lambda x: (x,))
            def f(y):
                pass

# should not raise
        @array_function_dispatch(lambda x: (x,), verify=False)
        def f(y):
            pass

def _new_duck_type_and_implements():
    """Create a duck array type and implements functions."""
    HANDLED_FUNCTIONS = {}

class MyArray:
        def __array_function__(self, func, types, args, kwargs):
            if func not in HANDLED_FUNCTIONS:
                return NotImplemented
            if not all(issubclass(t, MyArray) for t in types):
                return NotImplemented
            return HANDLED_FUNCTIONS[func](*args, **kwargs)

def implements(numpy_function):
        """Register an __array_function__ implementations."""
        def decorator(func):
            HANDLED_FUNCTIONS[numpy_function] = func
            return func
        return decorator

return (MyArray, implements)

@requires_array_function
class TestArrayFunctionImplementation:

def test_one_arg(self):
        MyArray, implements = _new_duck_type_and_implements()

@implements(dispatched_one_arg)
        def _(array):
            return 'myarray'

assert_equal(dispatched_one_arg(1), 'original')
        assert_equal(dispatched_one_arg(MyArray()), 'myarray')

def test_optional_args(self):
        MyArray, implements = _new_duck_type_and_implements()

@array_function_dispatch(lambda array, option=None: (array,))
        def func_with_option(array, option='default'):
            return option

@implements(func_with_option)
        def my_array_func_with_option(array, new_option='myarray'):
            return new_option

# we don't need to implement every option on __array_function__
        # implementations
        assert_equal(func_with_option(1), 'default')
        assert_equal(func_with_option(1, option='extra'), 'extra')
        assert_equal(func_with_option(MyArray()), 'myarray')
        with assert_raises(TypeError):
            func_with_option(MyArray(), option='extra')

# but new options on implementations can't be used
        result = my_array_func_with_option(MyArray(), new_option='yes')
        assert_equal(result, 'yes')
        with assert_raises(TypeError):
            func_with_option(MyArray(), new_option='no')

def test_not_implemented(self):
        MyArray, implements = _new_duck_type_and_implements()

@array_function_dispatch(lambda array: (array,), module='my')
        def func(array):
            return array

array = np.array(1)
        assert_(func(array) is array)
        assert_equal(func.__module__, 'my')

with assert_raises_regex(
                TypeError, "no implementation found for 'my.func'"):
            func(MyArray())

class TestNDArrayMethods:

def test_repr(self):
        # gh-12162: should still be defined even if __array_function__ doesn't
        # implement np.array_repr()

class MyArray(np.ndarray):
            def __array_function__(*args, **kwargs):
                return NotImplemented

array = np.array(1).view(MyArray)
        assert_equal(repr(array), 'MyArray(1)')
        assert_equal(str(array), '1')

class TestNumPyFunctions:

def test_set_module(self):
        assert_equal(np.sum.__module__, 'numpy')
        assert_equal(np.char.equal.__module__, 'numpy.char')
        assert_equal(np.fft.fft.__module__, 'numpy.fft')
        assert_equal(np.linalg.solve.__module__, 'numpy.linalg')

def test_inspect_sum(self):
        signature = inspect.signature(np.sum)
        assert_('axis' in signature.parameters)

@requires_array_function
    def test_override_sum(self):
        MyArray, implements = _new_duck_type_and_implements()

@implements(np.sum)
        def _(array):
            return 'yes'

assert_equal(np.sum(MyArray()), 'yes')

@requires_array_function
    def test_sum_on_mock_array(self):

# We need a proxy for mocks because __array_function__ is only looked
        # up in the class dict
        class ArrayProxy:
            def __init__(self, value):
                self.value = value
            def __array_function__(self, *args, **kwargs):
                return self.value.__array_function__(*args, **kwargs)
            def __array__(self, *args, **kwargs):
                return self.value.__array__(*args, **kwargs)

proxy = ArrayProxy(mock.Mock(spec=ArrayProxy))
        proxy.value.__array_function__.return_value = 1
        result = np.sum(proxy)
        assert_equal(result, 1)
        proxy.value.__array_function__.assert_called_once_with(
            np.sum, (ArrayProxy,), (proxy,), {})
        proxy.value.__array__.assert_not_called()

@requires_array_function
    def test_sum_forwarding_implementation(self):

class MyArray(np.ndarray):

def sum(self, axis, out):
                return 'summed'

def __array_function__(self, func, types, args, kwargs):
                return super().__array_function__(func, types, args, kwargs)

# note: the internal implementation of np.sum() calls the .sum() method
        array = np.array(1).view(MyArray)
        assert_equal(np.sum(array), 'summed')