1je d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlZ d dl
mZmZm
Z
mZ d dlmZmZmZ d dlmZ d dlZd Z ed d Z ed
d Z G d d
Z G d d Z G d d Z G d d Zd Z G d d Z G d d Z G d d Z G d d Z!d Z"d Z#dS ) NStringIO)mock)assert_assert_equal
assert_raisesassert_raises_regex)_get_implementing_argsarray_function_dispatchverify_matching_signatures)picklec t S NNotImplementedselfargskwargss s/builddir/build/BUILD/cloudlinux-venv-1.0.11/venv/lib64/python3.11/site-packages/numpy/core/tests/test_overrides.py_return_not_implementedr s c | fS r arrays r r s x r c dS
Docstring.originalr r s r dispatched_one_argr" :r c
| |fS r r array1array2s r r r s 0@ r c dS r r r% s r dispatched_two_argr) r# r c 2 e Zd Zd Zd Zd Zd Zd Zd ZdS )TestGetImplementingArgsc t j d }t |g }t t | |g t ||g }t t | |g t |dg }t t | |g t d|g }t t | |g d S )N npr r
r list)r r r s r test_ndarrayz$TestGetImplementingArgs.test_ndarray% s %ug..T$ZZ%)))%uen55T$ZZ%)))%uaj11T$ZZ%)))%q%j11T$ZZ%)))))r c G d dt j } G d dt j }t j d t j }t j d | }t j d | }t ||g }t t
| ||g t ||g }t t
| ||g t ||g }t t
| ||g d S )Nc e Zd ZeZdS )DTestGetImplementingArgs.test_ndarray_subclasses..OverrideSubN__name__
__module____qualname__r __array_function__r r r OverrideSubr4 6 !8r r: c e Zd ZdS )FTestGetImplementingArgs.test_ndarray_subclasses..NoOverrideSubNr6 r7 r8 r r r
NoOverrideSubr= 9 Dr r? r- )r/ ndarrayr viewr
r r0 )r r: r? r override_subno_override_subr s r test_ndarray_subclassesz/TestGetImplementingArgs.test_ndarray_subclasses4 sA 9 9 9 9 9"* 9 9 9 BJ ,,x{{''44(1++**=99%ul&;<<T$ZZ,!6777%uo&>??T$ZZ/5!9:::%
?+- -T$ZZ,!@AAAAAr c G d d }t j d } | }t ||g }t t | ||g t ||g }t t | ||g d S )Nc e Zd ZeZdS )BTestGetImplementingArgs.test_ndarray_and_duck_array..OtherNr5 r r r OtherrH L r; r rI r- r. )r rI r otherr s r test_ndarray_and_duck_arrayz3TestGetImplementingArgs.test_ndarray_and_duck_arrayJ s 9 9 9 9 9 9 9 9 %uen55T$ZZ%000%uen55T$ZZ%00000r c f G d dt j } G d d }t j d }t j d | } | }t t |||g |||g t t |||g |||g d S )Nc e Zd ZeZdS )QTestGetImplementingArgs.test_ndarray_subclass_and_duck_array..OverrideSubNr5 r r r r: rN Z r; r r: c e Zd ZeZdS )KTestGetImplementingArgs.test_ndarray_subclass_and_duck_array..OtherNr5 r r r rI rP ] r; r rI r- )r/ rA r rB r r
)r r: rI r subarrayrJ s r $test_ndarray_subclass_and_duck_arrayz