yd Z?ej0 9 e d ej0 1 d d ej< d z e;dej< z g d Z@d! ZAej0 9 e!d ej0 1 d"ejB jC ejB jD ej d# d$% ej d& d'% g d( ZEd) ZFej0 1 d*ejG d+ d, ZHej0 1 d*ejG d+ ej0 1 d-d.d/g d0 ZI G d1 d2 ZJdS )3 N)param) assert_ assert_equal
assert_raisesassert_array_equalassert_almost_equalassert_array_almost_equalassert_no_warningsassert_allclose HAS_REFCOUNTsuppress_warningsIS_WASMIS_PYPY)requires_memory)picklec F g | ] }t |t j | S )
isinstancenpufunc).0objs o/builddir/build/BUILD/cloudlinux-venv-1.0.12/venv/lib64/python3.11/site-packages/numpy/core/tests/test_ufunc.py
r s8 2 2 2!#rx002 2 2 2 c $ g | ]
}d |j v |S )zO->O)types)r ufs r r r s$ GGGbFbh4F4Fr4F4F4Fr c & e Zd Zd Zd Zd Zd ZdS )TestUfuncKwargsc D t t t j ddd t t t j ddt t t t j dddg t t t j ddd t t t j ddd
t t t j ddd t t t j ddd
t t t j dddg d S )N safe)castingx)dtypex )extobjx)outxii->i)sigx)
signaturexF)subokxT)wherexr TypeErrorr addintselfs r test_kwarg_exactz TestUfuncKwargs.test_kwarg_exact s iA????iAc::::iAv>>>>iAD9999iAG<<<<iA'BBBBiAe<<<<iAtf======r c N t t t j dddd d S )Nr" r# r* )sig signaturer r0 r r1 r3 s r test_sig_signaturez"TestUfuncKwargs.test_sig_signature' s2 iA7 ' ) ) ) ) ) )r c t t t j dddt t t t j dddt d S )Nr" r# r* )r7 dtype)r8 r< r/ r3 s r test_sig_dtypez TestUfuncKwargs.test_sig_dtype+ sX iA7 ! ! ! !iA ! ! ! ! ! !r c P t t t j dddgd d S )Nr" r# r' T)extobjparrotr9 r3 s r test_extobj_refcountz$TestUfuncKwargs.test_extobj_refcount1 s' iAtfTJJJJJJr N)__name__
__module__ __qualname__r5 r: r= rA r r r r r sU > > >) ) )! ! ! K K K K Kr r c ( e Zd ZdZej ej fej ej fej ej fej ej fej ej fej ej fej ej fej
ej
fgZ e j
de ej ddfd Zd Ze j
de eddfd Z G d d Zd
Zefd Zd Zefd
ZefdZd Ze j
de d ZdS )TestUfuncGenericLoopsab Test generic loops.
The loops to be tested are:
PyUFunc_ff_f_As_dd_d
PyUFunc_ff_f
PyUFunc_dd_d
PyUFunc_gg_g
PyUFunc_FF_F_As_DD_D
PyUFunc_DD_D
PyUFunc_FF_F
PyUFunc_GG_G
PyUFunc_OO_O
PyUFunc_OO_O_method
PyUFunc_f_f_As_d_d
PyUFunc_d_d
PyUFunc_f_f
PyUFunc_g_g
PyUFunc_F_F_As_D_D
PyUFunc_F_F
PyUFunc_D_D
PyUFunc_G_G
PyUFunc_O_O
PyUFunc_O_O_method
PyUFunc_On_Om
Where:
f -- float
d -- double
g -- long double
F -- complex float
D -- complex double
G -- complex long double
O -- python object
It is difficult to assure that each of these loops is entered from the
Python level as the special cased loops are a moving target and the
corresponding types are architecture dependent. We probably need to
define C level testing ufuncs to get at them. For the time being, I've
just looked at the signatures registered in the build directory to find
relevant functions.
zinput_dtype,output_dtyper r" c t j d || | } || d d d }t || t |j | d S N
r< r# r fullr r r< r4 input_dtype output_dtypefxyxsyss r test_unary_PyUFuncz(TestUfuncGenericLoops.test_unary_PyUFunci s_
WRQ |
<
<
<
QrUU33Q3ZA RX|,,,,,r c | |z S Nr )rQ rR s r f2zTestUfuncGenericLoops.f2p s !t r c t j d || | } ||| d d d }t || t |j | d S rH rK rM s r test_binary_PyUFuncz)TestUfuncGenericLoops.test_binary_PyUFuncs sa
WRQ |
<
<
<
Qr2YYsss^A RX|,,,,,r c e Zd Zd Zd ZdS )TestUfuncGenericLoops.fooc * t j d S Nr" r bool_r3 s r conjugatez#TestUfuncGenericLoops.foo.conjugate| 8A;; r c * t j d S r^ r_ r4 r s r logical_xorz%TestUfuncGenericLoops.foo.logical_xor rb r N)rB rC rD ra re r r r foor\ { s2 r rf c t j dt }t t j t j | dk d S )NrI rJ r" )r onesobjectr allabsr4 rQ s r test_unary_PyUFunc_O_Oz,TestUfuncGenericLoops.test_unary_PyUFunc_O_O s@ GBf % % %rvayyA~&&'''''r c t j d | t }t t j t j | dk d S )NrI rJ T)r rL ri r rj ra r4 rf rQ s r $test_unary_PyUFunc_O_O_method_simplez:TestUfuncGenericLoops.test_unary_PyUFunc_O_O_method_simple sI GBV , , ,r|A$.//00000r c t j dt }t t j t j || dk d S rH )r rh ri r rj r1 rl s r test_binary_PyUFunc_OO_Oz.TestUfuncGenericLoops.test_binary_PyUFunc_OO_O sC GBf % % %rva||q())*****r c t j d | t }t t j t j || d S )NrI rJ r rL ri r rj re ro s r test_binary_PyUFunc_OO_O_methodz5TestUfuncGenericLoops.test_binary_PyUFunc_OO_O_method sH GBV , , ,r~a++,,-----r c t j d | t }t t j t j || d S )N)rI r# rJ rt ro s r test_binary_PyUFunc_On_Om_methodz6TestUfuncGenericLoops.test_binary_PyUFunc_On_Om_method sH GJV 4 4 4r~a++,,-----r c " t j ddgd }t |d t sJ t j | }|j t j d k sJ t
|t j ddgd d S )Ny ? @y @ OrJ r y ? y @ @)r arrayr complexra r< r r4 arrress r test_python_complex_conjugatez3TestUfuncGenericLoops.test_python_complex_conjugate s hd|3///#a&'*****l3 yBHSMM))))3$ S A A ABBBBBr r c z t j t j dz G fddt j }t j t j }t j | d }t j d 5 || } || }t | d | nV# t $ rI}t t | 5 || ddd n # 1 swxY w Y Y d}~nd}~ww xY wddd dS # 1 swxY w Y dS ) z9Compare the result of the object loop with non-object one c " e Zd Z fdZ xZS )ITestUfuncGenericLoops.test_unary_PyUFunc_O_O_method_full..MyFloatc v t S # t $ r fdcY S w xY w)Nc T t t j j S rW )getattrr coreumath)attrvals r zgTestUfuncGenericLoops.test_unary_PyUFunc_O_O_method_full..MyFloat.__getattr__.. s #?727=$#?#?#D#D r )super __getattr__AttributeError)r4 r __class__r s `r r zUTestUfuncGenericLoops.test_unary_PyUFunc_O_O_method_full..MyFloat.__getattr__ sY E 77..t444% E E EDDDDDDDDEs % 88)rB rC rD r
__classcell__)r r s @r MyFloatr sH
E
E
E
E
E
E
E
E
E
Er r rJ rz raiserj N)
r float64pir{ errstater astype Exceptionr type) r4 r r num_arrobj_arrres_numres_objexcr s @r "test_unary_PyUFunc_O_O_method_fullz8TestUfuncGenericLoops.test_unary_PyUFunc_O_O_method_full s jq!! E E E E E E Ebj E E E (3bj111(773< RZ 2>2>"B DI [ 7CC>@fQ - - - DC- [ 7CC?AQ! - - - DC- ( ( ( 8; 1 1 1 1+ + + 36 . . . . 47 . . . .C C C [ W&9::H H ;:H H Hr rF c d S rW r r r r _pickleable_module_globalr s Dr c F e Zd Zd Zd Zej ed d Z d Z
d Z dZ d Z
d
Zd Zd Zd
Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Z d Z d Z d Z d Z ej ! d d d g d Z"ej ! d e#d! d"# e#d$ d%# g d& Z"ej ! d'g d( d) Z$d* Z%d+ Z&d, Z'd- Z(ej ! d.e)j* e)j+ g d/ Z,d0 Z-d1 Z.ej e/d2 d3 Z0d4 Z1d5 Z2d6 Z3d7 Z4d8 Z5d9 Z6ej ! d:d;gdg e)j7 d< fd=dg e)j7 d= f e)j8 d> e)j7 d? fgg d@A dB Z9dC Z:dD Z;dE ZdH Z?dI Z@dJ ZAdK ZBdL ZCdM ZDdN ZEdO ZFdP ZGdQ ZHdR ZIdS ZJdT ZKdU ZLdV ZMdW ZNdX ZOdY ZPdZ ZQd[ ZRd\ ZSd] ZTd^ ZUd_ ZVd` ZWda ZXdb ZY eZdc dd Z[de Z\df Z]dg Z^dh Z_di Z`dj Zadk Zbdl Zcej ! dmdn ej ! do e)jd g dp e)jd dqgdrgdqgg e)jd g dsg dtg dpg f du Zeej ! dvdw e)jd g dx fd=g dyfdzf ej ! d{e)jf d|f d} Zgd~ Zhd Zid Zjd Zkd Zld Zmd Znd Zod Zpd Zqej ! d e)jr des e)jr detju f d Zve)j* e)jw e)jx e)jy e)jz e)j{ e)j| e)j} gZ~ej ! de)j d e)j d z ej ! d.e~ d Zej ! de)j d ej ! d.e)j* e)jw e)jx g d Zd Zd Zd Zd Zej ! d e)j8 d= e)j8 d< e)j d dg d Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zej ! d.e)j e)j g ej ! d%ddefeddfdedfg d Zej ! d.e)j e)j e)j g ej ! d%edefedefdeefg d Zej ! d.e)j e)j e)j g d Zej ! d.e)j e)j e)j g d Zej ! d.e)j e)j e)j g d Zd Zd Zd Zd Zej ! ddqdrg d Zej ! dg d ej ! ddqdrg ej ! de)j* j e)j{ j g d Zd Zd Zd Zd Zej ! d e)j d e)j d g d Zej ! d e)j d e)j d g d Zej ! d e)j d e)j d g d ZdS ) TestUfuncc v t dt j dz D ]}t t j t j t j | t j u t j t j t j | }t |t j u d S )Nr# r" )protocol)
ranger HIGHEST_PROTOCOLr loadsdumpsr sin_rational_teststest_add)r4 protor s r test_picklezTestUfunc.test_pickle s 1f 5 9:: 5 5E FL bf7<"> "> "> ? ?BD&I
J
J
J
,v|O,D5: < < < = =C C?33 4 4 4 4 5 5r c d d}t t j | t j u d S )NsM cnumpy.core
_ufunc_reconstruct
p0
(S'numpy.core.umath'
p1
S'cos'
p2
tp3
Rp4
.)r r r r cos)r4 astrings r test_pickle_withstringz TestUfunc.test_pickle_withstring s/ H W%%/00000r z 'is' check does not work on PyPyreasonc t j t _ t j t j t j }|t j u sJ d S rW )umt _pickleable_module_global_ufuncr r r r r rd s r test_pickle_name_is_qualnamez&TestUfunc.test_pickle_name_is_qualname sG +.*M!'l6<(A(GHHIIc9999999r c d}t j | }t j t t t j |dz
t j |dz
dz }t
t j || d d d g d d S )N r# )r" rw ) r aranger{ listzipravelr r1 reduceat)r4 LrQ idxs r test_reduceat_shifting_sumz$TestUfunc.test_reduceat_shifting_sum s
IaLLhtC !a% 0 0")AE2B2BQ2F G GHHIIOOQQ26??1c2233Q37 FFFFFr c dS )aL Try to check presence and results of all ufuncs.
The list of ufuncs comes from generate_umath.py and is as follows:
===== ==== ============= =============== ========================
done args function types notes
===== ==== ============= =============== ========================
n 1 conjugate nums + O
n 1 absolute nums + O complex -> real
n 1 negative nums + O
n 1 sign nums + O -> int
n 1 invert bool + ints + O flts raise an error
n 1 degrees real + M cmplx raise an error
n 1 radians real + M cmplx raise an error
n 1 arccos flts + M
n 1 arccosh flts + M
n 1 arcsin flts + M
n 1 arcsinh flts + M
n 1 arctan flts + M
n 1 arctanh flts + M
n 1 cos flts + M
n 1 sin flts + M
n 1 tan flts + M
n 1 cosh flts + M
n 1 sinh flts + M
n 1 tanh flts + M
n 1 exp flts + M
n 1 expm1 flts + M
n 1 log flts + M
n 1 log10 flts + M
n 1 log1p flts + M
n 1 sqrt flts + M real x < 0 raises error
n 1 ceil real + M
n 1 trunc real + M
n 1 floor real + M
n 1 fabs real + M
n 1 rint flts + M
n 1 isnan flts -> bool
n 1 isinf flts -> bool
n 1 isfinite flts -> bool
n 1 signbit real -> bool
n 1 modf real -> (frac, int)
n 1 logical_not bool + nums + M -> bool
n 2 left_shift ints + O flts raise an error
n 2 right_shift ints + O flts raise an error
n 2 add bool + nums + O boolean + is ||
n 2 subtract bool + nums + O boolean - is ^
n 2 multiply bool + nums + O boolean * is &
n 2 divide nums + O
n 2 floor_divide nums + O
n 2 true_divide nums + O bBhH -> f, iIlLqQ -> d
n 2 fmod nums + M
n 2 power nums + O
n 2 greater bool + nums + O -> bool
n 2 greater_equal bool + nums + O -> bool
n 2 less bool + nums + O -> bool
n 2 less_equal bool + nums + O -> bool
n 2 equal bool + nums + O -> bool
n 2 not_equal bool + nums + O -> bool
n 2 logical_and bool + nums + M -> bool
n 2 logical_or bool + nums + M -> bool
n 2 logical_xor bool + nums + M -> bool
n 2 maximum bool + nums + O
n 2 minimum bool + nums + O
n 2 bitwise_and bool + ints + O flts raise an error
n 2 bitwise_or bool + ints + O flts raise an error
n 2 bitwise_xor bool + ints + O flts raise an error
n 2 arctan2 real + M
n 2 remainder ints + real + O
n 2 hypot real + M
===== ==== ============= =============== ========================
Types other than those listed will be accepted, but they are cast to
the smallest compatible type for which the function is defined. The
casting rules are:
bool -> int8 -> float32
ints -> double
Nr r3 s r test_all_ufunczTestUfunc.test_all_ufunc s
b
r r# r c t j ddd \ }}}}}t |d t |d t |d t || j f t |d d S )Nr# r" (i),(i)->())r" r" r r r r test_signaturer
size_inferredr4 enablednum_dimsixsflagssizess r test_signature0zTestUfunc.test_signature04 s /2/A
q-0! 0!,3uWa X{+++S&!!!UT / 1222UE"""""r c t j ddd \ }}}}}t |d t |d t |d t |d t |d d S )Nr# r" z (),()->()r r r r r r r r r s r test_signature1zTestUfunc.test_signature1> s| /2/A
q+0 0 ,3uWa X{+++S" UB UB r c t j ddd \ }}}}}t |d t |d t |d t || j fdz t |d d S )Nr# r" z(i1,i2),(J_1)->(_kAB)r# r" r" )r r" r# rw r r r r r r r s r test_signature2zTestUfunc.test_signature2H s /2/A
q)0+ 0+,3uWa Xy)))S,'''UT / 1! 3444U ,-----r c t j ddd \ }}}}}t |d t |d t |d t || j fdz t |d d S )Nr# r" z (i1, i12), (J_1)->(i12, i2)r# r" r# )r r" r# r" rw r r r r s r test_signature3zTestUfunc.test_signature3R s /2/A
q103 03,3uWa Xy)))S/***UT / 1! 3444U ,-----r c t j ddd \ }}}}}t |d t |d t |d t || j fdz t |d d S )Nr# r" z(n,k),(k,m)->(n,m)r# r# r# r r" r" r# r r# rw r r r r r s r test_signature4zTestUfunc.test_signature4[ s /2/A
q&0( 0(,3uWa Xy)))S,---UT / 1! 3444UL)))))r c " t j ddd \ }}}}}t |d t |d t |d t || j | j z | j | j | j z f t |d d S )Nr# r" z(n?,k),(k,m?)->(n?,m?)r r r )r r r r
can_ignorer s r test_signature5zTestUfunc.test_signature5e s /2/A
q*0, 0,,3uWa Xy)))S,---UT /$/ A ! / ! /$/ A C D D D UL)))))r c t j ddd \ }}}}}t |d t |d t |d t |d t |d d S )Nr" z(3)->()r" r r rw r r s r test_signature6zTestUfunc.test_signature6q s| /2/A
q)0 0 ,3uWa Xv&&&S$ UD!!!UD!!!!!r c t j ddd \ }}}}}t |d t |d t |d t |d| j df t |d d S )Nrw r" z(3),(03,3),(n)->(9)r" r# r" r" r r r r" r# r rw r r r s r test_signature7zTestUfunc.test_signature7z s /2/A
q'0) 0),3uWa X|,,,S/***UQ 2A 6777UJ'''''r c t j ddd \ }}}}}t |d t |d t |d t || j | j df t |d d S )Nrw r" z(3?),(3?,3?),(n)->(9)r r r r r r r r r r s r test_signature8zTestUfunc.test_signature8 s /2/A
q)0+ 0+,3uWa X|,,,S/***UT_d.@! DEEEUJ'''''r c t j ddd \ }}}}}t |d t |d t |d t |d t |d d S )Nr" z
( 3) -> ( )r r r r r s r test_signature9zTestUfunc.test_signature9 s| /2/A
q/0# 0#,3uWa Xv&&&S$ UD!!!UD!!!!!r c t j ddd \ }}}}}t |d t |d t |d t || j | j df t |d d S )Nrw r" z ( 3? ) , (3? , 3?) ,(n )-> ( 9)r r r r r r s r test_signature10zTestUfunc.test_signature10 s /2/A
q406 06,3uWa X|,,,S/***UT_d.@! DEEEUJ'''''r c t t 5 t j ddd d d d d S # 1 swxY w Y d S )Nr# r" z
((i)),(i)->()r
ValueErrorr r r3 s r (test_signature_failure_extra_parenthesisz2TestUfunc.test_signature_failure_extra_parenthesis s
:
&
& 6 6 q!_ 5 5 5 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6
9= =c t t 5 t j ddd d d d d S # 1 swxY w Y d S )Nr# r" z (i),)i(->()r
r3 s r .test_signature_failure_mismatching_parenthesisz8TestUfunc.test_signature_failure_mismatching_parenthesis
:
&
& 4 4 q!] 3 3 3 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4r
c t t 5 t j ddd d d d d S # 1 swxY w Y d S )Nr# r" z(i),->()r
r3 s r 2test_signature_failure_signature_missing_input_argz > >?Hyy J J J
HT & & &
q#1 Q
&&&!!!
\,
-
- = = F1cs8 < < < < = = = = = = = = = = = = = = =Q """!!!
q#78<<<<Q """!!!
\,
-
- A A F1cwAx @ @ @ @ A A A A A A A A A A A A A A AQ """!!!
q#-1hGGGGQ """""sH )B
BB)DDD GG
G
5II "I c t j dgdgd }t j dgdg }t || t j dgdgd }t j dgdg }t || t j t 5 t j ddd d d d d S # 1 swxY w Y d S )Nrw r NNNr7 rW r8 )r r1 r maximumr raisesr0 )r4 res1res2s r test_signature_all_Nonez!TestUfunc.test_signature_all_None s vqcA3$6777vqcA3 4&&&z1#s(:;;;z1#s##4&&&
]9
%
% , , F1a7 + + + + , , , , , , , , , , , , , , , , , ,s B??CCc t t j t j }t j dd||d f d S )Nrw r r- )r r r< r r1 )r4 float_dtypes r test_signature_dtype_typez#TestUfunc.test_signature_dtype_type sA 28BJ//00
q! [$ ?@@@@@@r get_kwargc , t t S NrJ dictrQ dts r r zTestUfunc. s t!}}} r c 2 t t d d f S Nr- r9 r; s r r zTestUfunc. s tq$o666 r c H t j d }t j t j | }||usJ t j di || j |k sJ t j dd }t j ||fi |d j dk sJ d S )Nint64r" r# sm8m8[s]r" r# )r r< r r r r1 timedelta)r4 r6 r@ int64_2tds r &test_signature_dtype_instances_allowedz0TestUfunc.test_signature_dtype_instances_allowed s !!,v|E 2 233G####v11ii 0 01175@@@@
\!S
!
!vb"00 $006'AAAAAAr c " t | S r8 r: rQ s r r zTestUfunc. s DqMMM r r< idc ( t | d d f S r> rJ rK s r r zTestUfunc. s DAtT? ; ; ; r r8 c
F d}t j t | 5 t j di |t j d d d d n # 1 swxY w Y t j t | 5 t j di |t j d d d d n # 1 swxY w Y t j t | 5 t j di |d d d d d S # 1 swxY w Y d S )Nz/The `dtype` and `signature` arguments to ufuncsmatchrw r r@ zm8[ns])rw r )r r/ r0 r r1 r< newbyteorder)r4 r6 msgs r rH z0TestUfunc.test_signature_dtype_instances_allowed s @
]9C
0
0
0 H H F G G99RXg%6%6%C%C%E%EFF G G G H H H H H H H H H H H H H H H
]9C
0
0
0 : : F 9 999RXh%7%788 9 9 9 : : : : : : : : : : : : : : :
]9C
0
0
0 0 0 F / /99X.. / / / 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0s5 ?A))A-0A--CC C .DD Dr )r same_kindr$ c t j t j d t j d d }|j dk sJ t j t j d t j d d }|j dk sJ t j t 5 t j dt j d d d d d n # 1 swxY w Y t j t 5 t j dt j d d d d d d S # 1 swxY w Y d S )N ?r# drJ )NNrW r- rw ) r ldexpfloat32int_r< r r/ r0 uint64)r4 r r s r test_partial_signature_mismatchz)TestUfunc.test_partial_signature_mismatch s hrz"~~rwqzz===yC hrz"~~rwqzz=NOOOyC ]9
%
% 2 2 HR1 S 1 1 1 1 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2
]9
%
% D D HR1 1B C C C C D D D D D D D D D D D D D D D D D Ds$ **C C$'C$*D;;D?D?c * t j t 5 t j t j d t j d d d d d n # 1 swxY w Y t j t j d t j d t j t 5 t j t j d t j d d d d d d S # 1 swxY w Y d S )Nr" r# )erW Nr, )r r/ r0 r r1 float16r[ r r3 s r *test_partial_signature_mismatch_with_cachez4TestUfunc.test_partial_signature_mismatch_with_cache st
]9
%
% F F F2:a==")A,,4D E E E E F F F F F F F F F F F F F F F rz!}}bjmm,,,
]9
%
% F F F2:a==")A,,4D E E E E F F F F F F F F F F F F F F F F F Fs# #s ii-izii>i)Nr s ->i)r r/ r0 r r1 r UnicodeDecodeErrorr3 s r test_signature_errorsz TestUfunc.test_signature_errors$ s
]9MO O O ) ) F1a4 ( ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ]:
&
& , , F1a7 + + + + , , , , , , , , , , , , , , , ]:
&
& , , F1a7 + + + + , , , , , , , , , , , , , , , ]:
&
& + + F1a6 * * * * + + + + + + + + + + + + + + + ]:
&
& 1 1 F1a< 0 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ]-
.
. 3 3 F1a> 2 2 2 2 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3sk A AA$BB B ,CCC4DD D <E E$'E$F))F-0F-c . t j g dd }t j ||d t j |d t j dd }t j ||d t j ||d d S )N)z
2010-01-02z
1999-03-14z1833-03z>M8[D]rJ MrI rC m)r r{ r. reducer r1 )r4 r' r~ s r test_forced_dtype_timesz!TestUfunc.test_forced_dtype_times9 s
H<<.call_ufuncI s! 53&59,777 7r )rV @ @rJ nor Nequivrb r$ ) r r{ rY r r< rR r r r/ r0 r )r4 r rs r~ arr_bsexpectedr s ` r test_cast_safetyzTestUfunc.test_cast_safetyD s
8 8 8 8 8 h|||2:666CI 2 2 4 455 :c??jd+++8S)))
]9
%
% - - Jvt , , , , - - - - - - - - - - - - - - - j1118S))) ]9
%
% B B JvRZ A A A A B B B B B B B B B B B B B B B jrz6BBB :cjj 4 4558S)))))s$ B00B47B41DD Dc
t j d }t j d }t j d }dD ]}t j | }| | }| | }t j || g|| g D ]#\ }} |dv r t
| t
| z }n t | t | z }t j || }
t t j
|
j d } t |
|| |dv r t |
j j d k n"t |
j j |j k dD ]9} t j | }
t t t j || |
:d D ]} t j | }
|dv r$t t t j || |
?t | t | z }t t j
|
j d } t# 5 }| t& t j t j
|
j s(t t j
|
j d }nd }d d d n # 1 swxY w Y t j d
d
5 t j || |
}
d d d n # 1 swxY w Y t j |
s| dk r{t |
|| | t |
j j |
j k dD ]T} t j | }
t
| t
| z }t t j
|
j d } t# 5 }| t& t j t j
|
j s(t t j
|
j d }nd }d d d n # 1 swxY w Y t j || |
}
t j |
s t |
|| | t |
j j |
j k V%t j dt j
}t j || }
t |
dk t |
j j d k t j | | }
t |
dk t |
j j d k d S )NrI r bhilqBHILQefdgFDGFDGgV瞯<)rtol
bhilqBHILQr rJ efdggwMh ignore)invalidoverr^ )r atolr rV )r r{ r< r itertoolsproductr| float true_dividemaxfinfo
resolutionr r namer r0 r
filter UserWarningisnantinyr isfiniterh r` )r4 r' r( tgttcr< aabbrQ rR r r tcoutdtoutsupr s r test_true_dividezTestUfunc.test_true_dividec s HRLL HRLLhsmm% G : G :B" B" B" B ")2s)b2#Y?? A
: A
:1 ;;!!**WQZZ/CC ((588+CnQ**28C==3U;; St4444 %%CINi 78888CINbg 5777 * P PE HUOOE!)R^QOOOOO # > >E HUOOEU{{%i AUSSSSS #AhhuQxx / "28E??#=u E E . 0 0 .C JJ{ 3 3 3#%8BHUOO,@#A#A .'*28E??+?'H'H'- . . . . . . . . . . . . . . . [ I I I D D"$.AU"C"C"CCD D D D D D D D D D D D D D D !{3 / / %ESLL $'St$GGGG %* <==== " : :E HUOOE!!**WQZZ/C rx 95AAD*,, *
;/// !x(< = = *#&rx';V#D#DDD#)D * * * * * * * * * * * * * * * .AU;;;C ;s++ ! #C4dCCCCCINej 89999 :eA
:H
GBbh ' ' 'nQ""s
)+,,,naR##s
)+,,,,,s8 A0K K
K
$L L
L
A0Q Q
Q
c J t j dt j }t |dz |j dz z
dd t j dt j }t |dz |j dz z
dd d S )N rJ $@r r
)r rh rY r sumsizer r4 r' s r test_sum_stabilityz TestUfunc.test_sum_stability s GCrz * * *QWMMOOafsl :AqAAA GCrz * * *QWMMOOafsl :ArBBBBBr fp errors don't work in wasmc & t t j t j t j t j fD ]}dD ]:}t
j d 5 }t
j dt |||dz z dz }t j
| }t t | d|z t j
d|dz | }t t j | | t t | d|z t t j |d d d | t t | d |z d d d n # 1 swxY w Y > >VUVWWW!\\ # # # # #r c t t j ddgddggddg d t t j ddgddggdd ddg
d d g d S ) NrV rt ru @TFwherer @)r r r "@)r r r r3 s r test_sum_wherezTestUfunc.test_sum_where s RVb"XBx 0u
FFFKKKRVb"XBx 0q"#'-1 1 135r( < < < < ()z
(2,)->(1,)z(4, 3)->(4, 1))idsc R t j t d 5 t j || d d d n # 1 swxY w Y t j t d 5 t j t j d || d d d d S # 1 swxY w Y d S )Nznon-broadcastablerP r! r )r r/ r r positiver1 rh r s r test_out_broadcast_errorsz#TestUfunc.test_out_broadcast_errorsN s ]:-@
A
A
A & & K % % % % & & & & & & & & & & & & & & & ]:-@
A
A
A . . F272;; - - - - . . . . . . . . . . . . . . . . . .s! ?AA%*B B #B c d}t j dd d }t t j || t j ||z d | d }t j d d }|d
z }t t j || t j ||z d | d S ) Nz type castr shortrJ r r r r ztype cast on one argumentr )r r r r r r r r )r4 rS r' r( s r test_type_castzTestUfunc.test_type_cast\ s Iaw ' ' ' / / 7 73;q!,,bfQqSr.B.B.B#& ( ( ( () IaLL ( (
G!#+a"3"3RVAaCb5I5I5I*- / / / / / /r c d}t j dd d }t t j || t j ||z d | d }t j dd
d }t t j || t j ||z d | t j d d }t j d d
}t ||z j t j d t ||z j t j d t ||z j t j d t ||z j t j d t t j | j t j d t t j | j t j d t t j
| j t j d t t j
| j t j d d S )Nz
big endianr z>i4rJ r r r r z
little endianzf8zQ'("d;;;;3;q!,,bfQqSr.B.B.BCPPPP IbMM ! !!Q * *
C
C'
QwZT22223;q!,,bfQqSr.B.B.BCPPPPPPr c d}t j d d }t j d d dz }t j dd }t j ||| t
|t j ||z d
| d
|d d < t j |||
t
|t j ||z d
| d}t j dd }t j ||| t
|t j ||z d
| d
|d d < t j |||
t
|t j ||z d
| d}t j dd }t j |||d t
|d t j ||z d
| d
|d d < t j |||d
t
|d t j ||z d
| d S )Nzoutput argumentr r# rw r# r r r" r r2 rJ r r r r! z output argument with type castint16z(output argument with incontiguous layoutr# rw r ).r )r r r r% r r r r )r4 rS r' r( r s r test_output_argumentz TestUfunc.test_output_argument s6 IbMM ! !) , , IaLL + +a / HV5 ) ) ) Aq! 1bfQqSr 2 2 2C@@@@!!! Aqa 1bfQqSr 2 2 2C@@@@. HV7 + + + Aq! 1bfQqSr 2 2 2C@@@@!!! Aqa 1bfQqSr 2 2 2C@@@@8 HYg . . . Aq!F)$$$1V9bfQqSr&:&:&:CHHHH!!! Aqai((((1V9bfQqSr&:&:&:CHHHHHHr c t j }t j d d }t j dd d } ||| }t |||z d |||g d }t |||z d |||g d }t |||z d |||g d
}t |||z d |||ddg }t |||z d |||t j d t j dt j g }t |||z d |||d d g }t |||z d |||d d
g }t ||
dd
d |z d t t |||d t t |||dg t t |||ddg t t |||ddg t t |||d dg t t j |||g d t t j |||g d t t j |||g d t t |||d dg t j }t j d d }t j d d dz } ||| }t |t j || |||g d }t |t j || |||g d }t |t j || |||g d }t |t j |
dd d
|
d ddd
t j d } ||||g d }t% ||u t |t j || t j d } ||||g d }t% ||u t |t j ||
dd dd
t t |||d t t |||d t t |||d t t |||g t t |||d!g t t |||g d" t t |||d#dgd#dgd#dgg t t |||d!d!d#dgg t t |||g d$ t t j |||g d t t j |||g d% t t |||g d& t j d' }t t |||d t t ||||d d d f ( t t ||d |d dg t t ||||d d dg t t t j d)d)d g t t |||d dgd*+ d S ),N ;@rw rw rw r 3@rw r" rw r r r r )axes)r r r )r r r rJ r r# r" )r r )r r r )r r r r r r r r# r# r# r" )r
r
r
)rD r r ))r r rD r
rw )r# r# rw r" )r" r )r" r# r# rw )r
r
)rw r r r
r+ r )r
r
N)r
r
r )r
r
)r r r r! rV T)r r@ )r r r r r r r int8r{ int32 transposer r0 r AxisErrormatrix_multiplymatmulr r r% r1 )r4 r r' r( r mmrW zs r test_axes_argumentz TestUfunc.test_axes_argument s + IcNN " "9 - - Ic3 ' ' 2 2 GAqMM1q1ukk"oo... GAq 1 1 1 2 2 21q1ukk"oo... GAq||| , , ,1q1ukk"oo... GAq / / /1q1ukk"oo... GAqBx ( ( (1q1ukk"oo... GAq RXb-I-I-I J K K K1q1ukk"oo... GAq1v & & &1q1ukk!nn--- GAq1v & & &1q{{1a 3 3a 7 < > >Q1bi1oo 7 71a C CDDD iQ2222iQ2222iQ0NOOOOj"a4444j"a(<<<<iQ0B0B0BCCCCi!Qr2hR2r(%C E E E Ei!Qh2r(%C E E E EiQ0J0J0JKKKKblB1<<<@@@@blB13K3K3KLLLLj"a1O1O1OPPPP HV j"a1...j"a!!!Q$8888j"adAQF;;;;j"a!Aq6BBBBiRqc::::iQAtDDDDDDr c t j }t j d d }t j dd d } ||| }t |||z d |||d }t |||z d t j | } |||d| }t ||u t || |||d }t |||z d t j d
d }t j d d
d }t j dd d }t t j |||d t j ||z |z d t t j
|d t j
|d t t j
|d t j
|d t j | }t j
||d }t ||u t |t j
|d t j
||d }t ||u t |t j
|d t t |||d d d g t t |||d g t j }t t |||d t j d|j }t t" t j
||d t t t j ddd d S )Nr r r r r r r )r r" r r r rI r r} r" r r" )r r r" r# rw rJ rV )r r r r r r r
zeros_liker innerwtcumsum
empty_liker r0 r r r< r r1 ) r4 r r' r( r r" rW r r s r test_axis_argumentz TestUfunc.test_axis_argument sc + IcNN " "9 - - Ic3 ' ' 2 2 GAqMM1q1ukk"oo... GAqr " " "1q1ukk"oo...mA GAqrs + + +S1a GAqq ! ! !1q1ukk!nn--- IaLL ( ( Ib" % %f - - Ib" % %f - -3;q!QQ777 6!a%!)!444 6 6 63:aa000")AA2F2F2FGGG3:ab11129QR3H3H3HIIImA Jqc * * *q1bi 2 2 2333 Jqc * * *q1bi 3 3 3444 i!QQaVDDDDi!QaS9999
iQ2222hy000j#*aSqAAAAiRa888888r c t j }t j d d }t j dd d } ||| }t |||z d |||d }t |||z d |||d }t |||z dd t j | } |||d |
}t ||u t || |||dd }t |||z dd |||dd }t |||z dd |||d d }t |||z d d |||d d }t |||z d d |||g d
d }t |||z d |||g dd }t |||z dd |||d d gd }t |||z d |||g dd }t |||z d d |||d dgd }t || ddd |z d |||d dgd }t || ddd |z dd |||g dd }t || ddd |z dd |||g dd }t ||| dd d z d d |||g dd }t || dd d | d dd z dd t j
d t j d d d t j t j f z }t j
| }t j
|d }t || t j
|d }t ||d d t j t j f t j
d t j d d d t j t j f z }t j | \ }} t j |d \ }
} t |
| t | | t j |d \ }
} t |
|d d t j t j f t | | d d t j t j f t j d d }t j dd d }t j d d d } t t j ||| d t j ||z | z dd t t j ||| d d t j ||z | z d d t! t" |||d t j }
t! t" |
||d t! t" |
||d t! t" t j d d d d S ) Nr r r r r r FkeepdimsT)r# r" r r# r r
)r r# )r r r r r# r" )r r# r# )r r# r )r r# r" rw r r r rI r r} r truerV )r r r r r r r r r r eyenewaxisumldetslogdetr r r0 r r1 )r4 r r' r( r r" rW rz
expected_s
expected_lcsclr r s r test_keepdims_argumentz TestUfunc.test_keepdims_argument> s + IcNN " "9 - - Ic3 ' ' 2 2 GAqMM1q1ukk"oo... GAq5 ) ) )1q1ukk"oo... GAq4 ( ( (1q1ukk"tk < <===mA GAq4S 1 1 1S1a GAqrE 2 2 21q1ukk"uk = =>>> GAqrD 1 1 11q1ukk"tk < <=== GAqq5 1 1 11q1ukk!ek < <=== GAqq4 0 0 01q1ukk!dk ; ;<<< GAq 1 1 1E B B B1q1ukk"oo... GAq 4 4 4t D D D1q1ukk"tk < <=== GAq1v 6 6 61q1ukk!nn--- GAqyyy4 8 8 81q1ukk!dk ; ;<<< GAq1v 6 6 61q{{1a 3 3a 7 < >> 6!a%!)"tDDD F F F3;q!QQFFF 6!a%!)!dCCC E E E i!Q@@@@
iQD9999iQE::::iR%@@@@@@r c t j d d }t j dd d }t j dd d }t t j ||| t j ||z |z d t j d d
d }t j d d
d }t j dd d }t t j ||| t j ||z |z d d S )Nr r rI r r} r r r d | r , iD )r r r r r r r r4 r' r( r s r test_innerwtzTestUfunc.test_innerwt s( IaLL ( ( Ib" % %f - - Ib" % %f - -3;q!Q//!AB1G1G1GHHH Ic3 ' ' 2 2 Ic3 ' ' 2 2 Ic3 ' ' 2 23;q!Q//!AB1G1G1GHHHHHr c t j g d }t j g d }t j g d }t t j ||| t j ||z |z d dS )z/Test generalized ufunc with zero-sized operandsr rJ r r N)r r{ r r r r r6 s r test_innerwt_emptyz TestUfunc.test_innerwt_empty sy HRt $ $ $ HRt $ $ $ HRt $ $ $3;q!Q//!AB1G1G1GHHHHHr c t j d }t t j || t j d t j d }t j |d || }t
||u t |t j t j d |d |d f t t t j t j d t j d t t t j |t j
d t t t j |t j
d t j d t t t j |t j
d t j d d
S ) z Test with fixed-sized signature.rw rw rw r r# r" r r ru )rw r N) r r& r r cross1dr% r vstackr r r )r4 r' r" results r test_cross1dzTestUfunc.test_cross1d s< F1II3;q!,,bhv.>.>???hv QqT1c**#
629bhqkk1Q4!A$-G#H#HIIIj#+rvayy"&))DDDj#+q")B--@@@j#+q")B--&AQAQRRRj#+q")B--!MMMMMr c T t j d d }t j d }|d d t j f }t j d d dz }t j || }t j || }t || t t t j || t j || }t ||
t j || }t j || }t || t j d }|t j d d f }t j d d dz } t j || }
t j || } t | |
t t t j || t j || } t | |
t j | | }
t j | | } t | |
|j }| d
d }t j || }
t j || }t ||
t t t j || t j || }t ||
t j | | }t j | | }t || t j
| }t j | || }t || d
|d d < t j | || }t || t j
|
}t t t j ||| t j ||| }t ||
d S )Nr r r# r r r" rw r )r r# r" r" rw r r r! r )r r r r' r r r r r r squeezer swapaxesr )r4 mat
single_veccol_vec
col_vec_array
mm_col_vecmatmul_col_vec
matmul_colrow_vec
row_vec_array
mm_row_vecmatmul_row_vec
matmul_rowmm_row_col_vecmatmul_row_col_vecmatmul_row_colmm_row_col_arraymatmul_row_col_arrayr" s r test_can_ignore_signaturez#TestUfunc.test_can_ignore_signature s imm##I..Yq\\
QQQ
]+ ! ,,\::Q>
(g66
C11>:666 j#"5sJGGGZZ00
:z'9'9';';<<< (m<<
C 77>:666Yq\\
RZ]+ "
--o>>B
(#66
GS11>:666j#"5z3GGGZ
C00
:z'9'9';';<<<( <<
M377>:666 )%..r266
,Wg>> Z 9 9-~>>>j#"5z:NNN J
;;>>+A+A+C+CDDD .}mLL "z- G G /1ABBB m ,--!- CHHH3 0111AAAj
3???3 0111mN++j#"5z: jZ553 6 6 8 899999r c | t j | t j d S rW ) compare_matrix_multiply_resultsr r@ r r3 s r test_matrix_multiplyz TestUfunc.test_matrix_multiply s4 ,,RX666 ,,RY77777r c r t j t j d t j d }t |t j d t j t j d t j d }t |t j d d S )N)r rI )rI r r )rI rI )r r r rh r r% re s r test_matrix_multiply_umath_emptyz*TestUfunc.test_matrix_multiply_umath_empty s !"''"2"2BGG4D4DEE3 0 0111!"''"2"2BGG4D4DEE3 2 233333r c , t j t j ddd | }t j t j ddd | }d|j j z }fdfdd } d } d t
d d d fdz fz }d
}|D ]F} |D ]?}
|D ]8} |D ]1} | | | }
| |
| }|o|
j d u}|o|j d u}|
j d |j d k r ||
j d |j d rt t j |
| t j
|d
t j f dd |
d
t j d d f z d |dt! |
j dt! |j z 3:AHt# |d
d d S )Nr# rw r rJ zmatrix multiply on type %sc | dk rdgfS d} | dz
}|D ]9}t | D ]'}|| dz
gz }|| || dz
< | dz
||< ||fz
}(:|S )Nr" r r )r )nretbasepermr new permute_ns r ra z.permute_n s Avvv
C 9QqS>>D
"
" q " "A !A#,C "1vC!H qSCFC6MCC "
Jr c | dk rdS d} | dz
}|D ]1}||t d fz fz
}||t dd fz fz
}2|S )Nr )r r r" )slice)r\ r] r^ slslice_ns r re z:TestUfunc.compare_matrix_multiply_results..slice_n ss Avv uC71Q3<)++E!QKK>)++Jr c & | |k p | dk p|dk S r^ r )s1s2s r
broadcastablez@TestUfunc.compare_matrix_multiply_results..broadcastable s 81rQw1"' 1r r Tr r .r r r zreference check)r r{ randomrandr< r rc r r^ shaper r r r r' rB strr )r4 tpd1d2rS ri permute_3slice_3refp1p2rg rh a1a2ra re s @@r rV z)TestUfunc.compare_matrix_multiply_results s~
XbinnQ1--R
8
8
8
XbinnQ1--R
8
8
8*RX]: 2 2 2 IaLL '!**tT2!6!6 8 : << J JB
J
J!
J
JB % J J \\" - -b 1 \\" - -b 1 ! 9bgT&9 ! 9bgT&9 HRLBHRL 8 8 -
bhqk28A; G G 9 5 # 3B ; ; "r#rz/':'C'CB'K'K')#rz!!!*;'<(=CE!G !G !G(+#bh----:=bh---/I )I J J J J J
J
J S$(9::::::r c . t j dt dd }t j |j d |j d dz
z dz f|j }t j || t j t j
|d d d f |z
dz d }|t j |j d t }t || t t t j | d S )
Nr rJ r rw r r" r# r r )r r r r r rm r< r euclidean_pdistsqrtr triboolr r r )r4 r' r" r( s r test_euclidean_pdistz TestUfunc.test_euclidean_pdist4 s Ib & & & . .q! 4 4h
agaj1n5: s r test_cumsumzTestUfunc.test_cumsum> s: IbMMA 6188::.....r c t j g dt }t t j |d t j d |D t t t j |d t j d |D t t t j |d t j d |D t t t j |d t j d |D t t t j |d t j d
|D t t t j |d t j d |D t t t j |d t j d |D t t t j |d t j d
|D t t t j | t j d |D t t t j | d t t j | d d S )N)rw NTFtest rJ c g | ]}|pd S rW r r rQ s r r z1TestUfunc.test_object_logical..F !7!7!7!)t!7!7!7r Tc g | ]}|pd S Tr r s r r z1TestUfunc.test_object_logical..H r r r c g | ]}|pd S r r r s r r z1TestUfunc.test_object_logical..J s !5!5!5a!'r!5!5!5r blahc g | ]}|pd S r r r s r r z1TestUfunc.test_object_logical..L s !9!9!9!!+v!9!9!9r c g | ]}|od S rW r r s r r z1TestUfunc.test_object_logical..O !8!8!8!*!8!8!8r c g | ]}|od S r r r s r r z1TestUfunc.test_object_logical..Q r r c g | ]}|od S r r r s r r z1TestUfunc.test_object_logical..S s !6!6!6q!(!6!6!6r c g | ]}|od S r r r s r r z1TestUfunc.test_object_logical..U s !:!:!:1!,!:!:!:r c g | ]}| S r r r s r r z1TestUfunc.test_object_logical..X s !3!3!3Aa%!3!3!3r rw )r r{ ri r
logical_or logical_and logical_notrm r s r test_object_logicalz TestUfunc.test_object_logicalC s H777v F F FR]1d++!7!7Q!7!7!7vFFF H H HR]1d++!7!7Q!7!7!7vFFF H H HR]1b))!5!51!5!5!5VDDD F F FR]1f--!9!9q!9!9!9HHH J J J R^At,,!8!8a!8!8!8GGG I I IR^At,,!8!8a!8!8!8GGG I I IR^Ar**!6!6A!6!6!6fEEE G G GR^Av..!:!:!:!:!:&III K K K R^A&&!3!3!3!3!36BBB D D D R]))!,,a000R^**1--t44444r c 4 G d d }t j | }t ||k d t t j || d t j | g }t ||k t j dg t t j || t j dg t t j ||t t j dg d S )Nc e Zd Zd ZdS )8TestUfunc.test_object_comparison..HasComparisonsc dS )N==r )r4 others r __eq__z?TestUfunc.test_object_comparison..HasComparisons.__eq___ s tr N)rB rC rD r r r r HasComparisonsr ^ s#
r r TrJ r )r r{ r equalri )r4 r arr0darr1ds r test_object_comparisonz TestUfunc.test_object_comparison] s ))**Ue^T***RXeU++T222..**+,,Ue^RXtf%5%5666RXeU++RXtf-=-=>>>RXeU&99928TF;K;KLLLLLr c t j g dt }t t j | d t t j | d t t j | d t j g dt }t t j | d t t j | d t t j | d t t j | d
t t j | d t t j | d
t t j d ggt d t t j d dgggt d d dg t t j d gt d
d t t j d gddggt dgd
d g g d d S )N)r' r( r rJ abcr r' TFTr# r TFr" r r" r rw )r r )r r# rw )
r r{ ri r r r minprodanyrj r s r test_object_array_reductionz%TestUfunc.test_object_array_reductionk s H___F 3 3 3RVAYY&&&RVAYY$$$RVAYY$$$ H((( 7 7 7RVAYY"""RWQZZ###RVAYY%%%RVAYY&&&RVAYY%%%RVAYY&&&RXse62226688!<<<RXAxj777;;FCCaVLLLRXqc00044Q4??CCCRXsQFm6:::c1#eT]c;;YYY H H H H Hr c t j dt }d t d D |d d < t j || t j || t
|t j d dD t t j dt }d t d D |d d d f< t j ||d
t j ||d
t
|d d d f t j d dD t d S )
Nr rJ c g | ]}d gS r r r r s r r zBTestUfunc.test_object_array_accumulate_inplace.. (((!1#(((r r! c g | ]}d g|z S r r r s r r zBTestUfunc.test_object_array_accumulate_inplace.. $B$B$BqaSU$B$B$Br )r" rw r rI r# r c g | ]}d gS r# r r s r r zBTestUfunc.test_object_array_accumulate_inplace.. +++QaS+++r r r r c g | ]}d g|z S r r r s r r zBTestUfunc.test_object_array_accumulate_inplace.. r r )r rh ri r r1
accumulater r{ r4 r~ s r $test_object_array_accumulate_inplacez.TestUfunc.test_object_array_accumulate_inplace~ sb gav&&&((uQxx(((AAA
#3'''
#3'''3 8$B$BM$B$B$B&QQQ
gfF+++++%((+++AqqqD
#3R000
#3R0003q!!!t9 8$B$BM$B$B$B&QQQ r c v t j t j g dt }t |t j g dt t
j t 5 t j g d d d d d S # 1 swxY w Y d S )N)r" r r# rJ )r" r" rw r" Nr# ) r r1 r r{ ri r r r/ r0 re s r $test_object_array_accumulate_failurez.TestUfunc.test_object_array_accumulate_failure s f & A A ABB3& A A ABBB
]9
%
% , , F lll + + + , , , , , , , , , , , , , , , , , ,s ?"B..B25B2c t j dt }d t d D |d d < t j dt }d t d D |d d < t j |t j d | t j |t j d | t || t j dt }d t d D |dd d f< t j dt }d t d D |dd d f< t j |t j d |d
t j |t j d |d
t || d S ) Nr rJ c g | ]}d gS r r r s r r z@TestUfunc.test_object_array_reduceat_inplace.. r r c g | ]}d gS r r r s r r z@TestUfunc.test_object_array_reduceat_inplace.. r r r! r c g | ]}d gS r r r s r r z@TestUfunc.test_object_array_reduceat_inplace.. r r r c g | ]}d gS r r r s r r z@TestUfunc.test_object_array_reduceat_inplace.. r r r r ) r r ri r r1 r r r rh r s r "test_object_array_reduceat_inplacez,TestUfunc.test_object_array_reduceat_inplace s hq'''((uQxx(((AAAhq'''((uQxx(((AAA
RYq\\s333
RYq\\s3333$$$ gfF+++++%((+++AqqqD gfF+++++%((+++AqqqD
RYq\\s<<<
RYq\\s<<<3$$$$$r c t j t j g dt ddg }t |t j d dgt t
j t 5 t j g dddg d d d d S # 1 swxY w Y d S )Nr rJ r" r# r ) r r1 r r{ ri r r r/ r0 re s r "test_object_array_reduceat_failurez,TestUfunc.test_object_array_reduceat_failure s foobh|||6 B B BQFKK3$& A A ABBB
]9
%
% 2 2 FOOLLL1a& 1 1 1 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2s %B44B8;B8c g t j g t fD ]}t t j | d t t j | d t t j | d t t j | d t t t j
| t t t j | d S )NrJ r r" FT) r r{ ri r r r r rj r r r r r s r test_zerosize_reductionz!TestUfunc.test_zerosize_reduction s bhr0001 1 1A A & & & Q ' ' ' E * * * D ) ) ) *bfa 0 0 0 *bfa 0 0 0 0
1 1r c t j ddg }t t j |j d t j ddg }t t j |j d t j ddg }t t j |j d t j ddg }t t j |j d d S )NFr" r r )r r{ r r rj r r s r test_axis_out_of_boundsz!TestUfunc.test_axis_out_of_bounds s HeU^ $ $blAE2222 HeU^ $ $blAE3333 HeU^ $ $blAE2222 HeU^ $ $blAE333333r c t t j dd d t t j dd d t t j dd d t t j dd d t t j dd d t t j dd d t t j d d t t t j t j d d t j u t t t j t j d d t j u t t t j t j d d t j u t t t j t j d d t j u t t t j dd t j
u G d d t j }t j d
| }t t t j | |u d S ) Nrw r r @TFg @c e Zd ZdS )0TestUfunc.test_scalar_reduction..MyArrayNrB rC rD r r r MyArrayr Dr r r" )r r r r r rj r r rc rm r r rY r` ndarrayr{ view)r4 r r' s r test_scalar_reductionz TestUfunc.test_scalar_reduction sL RVAA&&&***RWSq)))3///RVDq)))4000RVE***E222RVAA&&&***RVCa(((#... RX__Q''+++ RWRZ__155566"*DEEERVBJsOO!44455CDDDRVBJsOO!44455CDDDRVBJsOO!44455CDDD RVAA&&&''283444 bj HQKK W % %RVAYY7*+++++r c t j dt j }t j dt j }t j dt j }t j ||| t |d t j d }t j dt j }t j |d| t |d d S )N)r3 r1 r! r# i rJ r ) r rh r@ r r1 r r% rY subtractr4 r' r( r s r test_casting_out_paramz TestUfunc.test_casting_out_param s GJ ) ) GJ ) ) GJ
+ +
q!Q HUOO HU"* - - -
Aqa Qr c t j d }t j d }t j d }t j ||||dz dk t |g d t j d dd dz }t j |ddg|ddgddgg t |dd gd
d gg t j |d|d d
g t |dd gdd gg d S )Nr r# r" r" r )r r# r r r r r r rw r r r TF ) r r rh r% r1 r r rc r r s r test_where_paramzTestUfunc.test_where_param s
IaLL GAJJ HQKK
q!1q5A:////Q---... IaLL A & & *
QF1a&1a&)9::::Q!R2q'*+++
Aqae}5555Q!R2q'*+++++r c t j dt j }t j dt j }dt j dt j z }t j |||g d t |g d d S )NrI r )
r" r r r" r r r" r" r" r r )
r# r r r# r r r# r# r# r )r rh r@ r r1 r r s r test_where_param_buffer_outputz(TestUfunc.test_where_param_buffer_output s
GB ! ! GB ! ! "'"bj)) )
q!"@"@"@AAAAQ@@@AAAAAr c t j dgt j }t j dgt }t t j || dg t j dgt j }t j dgt }t t j || dg d S )Nr" rJ Tr )r r{ r@ r} r r{ r r4 r' rl s r test_where_param_allocz TestUfunc.test_where_param_alloc s HaS ) ) ) HdV4 ( ( (RWQa(((1#...
HaS
+ + + HdV4 ( ( (RWQa(((1#.....r c t j d }t j d }|dk }t j |d }t j |||| t j ||j | }t
|| |k || t || rJ d S )N) r )r r" r r )r r" )
r rk full_likeless broadcast_torm r r r} r )r4 r' r( r r" b_wheres r test_where_with_broadcastingz&TestUfunc.test_where_with_broadcasting s I Y ' ' I Y ' 'Cl1a
1Es++++/!QW--e4AeHw .U1B1B41H1HIIIv;??$$$$$$$r c d|d< d|d< t t j |d d t t j |d g d t t j |d g d t t j |d
ddg t t j |d g dg d g d g t t j |d g d g dg t t j |d g d
g d g t t j |d | d|d< d|d< t t j |d d t t j |d g d t t j |d g d t t j |d
ddg t t j |d g d g dg d g t t j |d g dg d g t t j |d g dg d
g t t j |d | d|d< d|d< t t j |d d t t j |d g d t t j |d g d t t j |d
ddg t t j |d g dg d g d g t t j |d g dg d g t t j |d g d g d
g t t j |d | d S )Nr" .r )r" r r r r )r r" r" r" r )r r" r" rD )r" r" r" r" r# )r" r" r" r )r r" r )r" r r" r )r" r r" r" )r r minimumrm r s r check_identityless_reductionz&TestUfunc.check_identityless_reduction% s #'
RZ&&qt&44a888RZ&&qv&66 EEERZ&&qv&66 BBBRZ&&qv&66A???RZ&&qq&11%1\\<<< $N P P PRZ&&qq&11%1\\<<<$@ B B BRZ&&qq&11%.YY $: < < <RZ&&qr&22A666#'
RZ&&qt&44a888RZ&&qv&66 EEERZ&&qv&66 BBBRZ&&qv&66A???RZ&&qq&11%1\\<<< $N P P PRZ&&qq&11%1\\<<<$@ B B BRZ&&qq&11%.YY $: < < <RZ&&qr&22A666#'
RZ&&qt&44a888RZ&&qv&66 EEERZ&&qv&66 BBBRZ&&qv&66A???RZ&&qq&11%1\\<<< $N P P PRZ&&qq&11%1\\<<<$@ B B BRZ&&qq&11%.YY $: < < <RZ&&qr&22A66666r l c t j dd }ddg|d d df< ddg|d d df< t j |d }~|d dk sJ |d dk sJ d S ) N)r# l uint8r" rw r r r r )r r% r. rm r} s r &test_identityless_reduction_huge_arrayz0TestUfunc.test_identityless_reduction_huge_arrayS s hz7++FAAAqD VAAArE
j ! ,, 1v{{{{2w!||||||r c \ t j dd }| | d S )Nr Corderr r r r s r "test_identityless_reduction_corderz,TestUfunc.test_identityless_reduction_corder^ 0 HYc * * * ))!,,,,,r c \ t j dd }| | d S )Nr Fr r r s r "test_identityless_reduction_forderz,TestUfunc.test_identityless_reduction_forderb r r c t j dd dd }| | d S )N)r# r rw r r r" r# r r rB r r s r &test_identityless_reduction_otherorderz0TestUfunc.test_identityless_reduction_otherorderf s@ HYc * * * 3 3Aq 9 9 ))!,,,,,r c t j dd dd }|dd dd dd f }| | d S )N)rw r r r r r" r# r r s r %test_identityless_reduction_noncontigz/TestUfunc.test_identityless_reduction_noncontigj s[ HYc * * * 3 3Aq 9 9
abb!""abbjM ))!,,,,,r c t j dd }|dd d }d|_ |dd dd dd f }| | d S )N)i i1rJ r" r rw r r )r r r rm r r s r /test_identityless_reduction_noncontig_unalignedz9TestUfunc.test_identityless_reduction_noncontig_unalignedo sm H^4 0 0 0
abbEJJTJ " "
abb!""abbjM ))!,,,,,r c H t t j g t t
u sJ t j dt }t j g |t j t |d t u sJ dS )z
The type of the result should always depend on the selected loop, not
necessarily the output (only relevant for object arrays).
rJ Nr" r< r ) r r r1 rm ri r2 r{ r r )r4 r" s r $test_reduce_identity_depends_on_loopz.TestUfunc.test_reduce_identity_depends_on_loopv s BFMM"FM3344;;;;ht6***
bc
444CG}}%%%%%%r c L t t j g d d t t j g t j t j t t j g t j t j t t j dgd d t t j dgd d t t j dgd d t t j dgd d t
t t j j g d t
t t j j g d t j
t 5 t j g d t d d d n # 1 swxY w Y t t j g t j d t j
dgt }t j |d }t |d d S ) Nr r r r )r r< rI rJ r )r r r. rm r infr r r1 r r/ ri _NoValuer{ r4 r' r s r test_initial_reductionz TestUfunc.test_initial_reduction sB
RZ&&r1&55q999 RZ&&r26&::BFCCCRZ&&rBF7&;;bfWEEE RZ&&sA&66:::RZ&&sA&66:::RZ&&sA&66:::RZ&&sA&66::: j"*"3RFFFFj"&-TBBBB
]:
&
& : : FMM"d&M 9 9 9 : : : : : : : : : : : : : : : RV]]2r{];;Q???
HbT ( ( (fmmAqm))S" s (HHHc ~ t j d }t j |d j dk sJ t j t 5 t j |d d d d n # 1 swxY w Y t j d }t j t 5 t j |d d d d n # 1 swxY w Y t j |dd }t |t j d d S )N)r r r" r r r )r r r r )
r r% r rm rm r r/ r r rh r} s r test_empty_reduction_and_idenityz*TestUfunc.test_empty_reduction_and_idenity s hv ~$$Sq$1174????
]:
&
& / / N ! !#A ! . . . / / / / / / / / / / / / / / / hy!!
]:
&
& / / N ! !#A ! . . . / / / / / / / / / / / / / / / n##Ca#;;300000s$ "BB B "C..C25C2r )r r" Nr )FTTTF)TFFFTFc t j d dd }| }t j | }t j ||| t j ||| }| | }t || t || d S )Nr rw r r r ) r r r copyr r r1 rm r r r )r4 r r r' a_copya_checkr checks r test_reduction_with_wherez#TestUfunc.test_reduction_with_where s
IbMM ! !!Q ' '-""
A7%0000fmmADm66 D!!S% 1f%%%%%r r
r r )TTF)NTr r c t j d dd }| }t j |j t j }t j ||| t j |||| }|
|| }t || d S )Nr rw r )r r r r ) r r r r rL rm r r r. rm r r ) r4 r r r r' r r r r s r %test_reduction_with_where_and_initialz/TestUfunc.test_reduction_with_where_and_initial s
IbMM ! !!Q ' ''!'BF7++
A7%0000j E7 KK D' 22S% r c t j d dd }g d}t t t j j || d S )Nr rw r r )r r r r r r. rm r s r #test_reduction_where_initial_neededz-TestUfunc.test_reduction_where_initial_needed sL IbMM ! !!Q ' ' j"*"3Qa@@@@@@r c t j g dg dg }t j |d }t |g d t j |d }t |ddg t j |d }t || t t t j j |d
d S ) N) @rt rt )rV r g ?r r )r r r r" rt r r r )r r{ dividerm r r r r s r *test_identityless_reduction_nonreorderablez4TestUfunc.test_identityless_reduction_nonreorderable s Hooo'7'7'78 9 9i qq ))S///***i qq ))S3*%%%i qr **S! j")"2AFCCCCCCr c d }d }d }t t j j d ||t j j dd ||t j j dd ||t j j dd ||t j j dd |t j j g t j t j d d ||t j j dd ||t j j dd ||t j j dd ||t j j dd t j g t j t j d d t j t j fD ]}| t j d d | t j d
d | t j d d | t j d d d S ) Nc | |i | d S rW r rP argsrr s r okz+TestUfunc.test_reduce_zero_axis..ok s
At v r c 2 t t | g|R i | d S rW )r r r s r errz,TestUfunc.test_reduce_zero_axis..err s( *a 9$ 9 9 9& 9 9 9 9 9r c | |t j ||f d | |t j ||f d | |t j |dz |dz |f d | |t j |dz ||dz f d | |t j ||dz |dz f d | |t j |dz ||dz f d | |t j |dz |dz |dz |dz |dz f d | |t j d ||f d
| |t j d ||f d | |t j |d |f d | |t j d ||f d | |t j d ||f d d S ) Nr" r r r# rD r rw r r" r# rI r )r r% )expectfuncr\ rl s r tz*TestUfunc.test_reduce_zero_axis..t s' F41a&)) 2 2 2 2 F41a&)) 2 2 2 2 F416161"566Q ? ? ? ? F4161a1f"566Q ? ? ? ? F41a1fa1f"566V D D D D F4161a1f"566V D D D D F4161616"#q&!q 3 3&/
1
1
1
1
F42q!*--F ; ; ; ; F42q!*--F ; ; ; ; F41b!*--A 6 6 6 6 F42q!*--A 6 6 6 6 F42q!*--A 6 6 6 6 6 6r r r r r )r' r )r r' )r' r' )r r r. identityrm r% r1 r )r4 r r! r& r s r test_reduce_zero_axisz TestUfunc.test_reduce_zero_axis s5 : : : 7 7 7& RZ($/// "bj R((( "bj B''' #rz "a((( #rz !Q''' BJ r"""
"(6**444
"bfmR$$$ "bfmQ### "bfmR### "bfmQ"""
b
bhv&&R
000
:rv& 4 4B MM"(7++!M 4 4 4 MM"(7++!M 4 4 4 MM"(8,,1M 5 5 5 MM"(6**M 3 3 3 3 4 4r c p t j g dt }t t t j |d g d t
t t j |d| d }t
t ||d t t j |d|d t |g d d S )
Nr rJ g?)g @g@gffffff@r! c | |z
} d S rW r )r' r( s r add_inplacez0TestUfunc.test_safe_casting..add_inplace4 s
FAAAr r r" r r )r r{ r2 r r
r1 r r0 )r4 r' r, s r test_safe_castingzTestUfunc.test_safe_casting) s
HYYYc * * *-bfa==*?? , , ,iCQ7777 ia555261cq(CCCC1iii(((((r c t j g dd }t j g dd }t j dt j }t j ||| }t j g dt j }t
|| t j || }t
|| t j || t j | }t
|| t t 5 t j |t j d d d d d S # 1 swxY w Y d S )Nr# i8rJ rw )r r# r r! r# ) r r{ r r rationalr r r uint16r r0 )r4 r' r( r r> targets r test_ufunc_custom_outz TestUfunc.test_ufunc_custom_out< sv
HYYYd + + + HYYYd + + + HQo 6 7 7 7 !)!Q22)))?+CDDDVV$$$ !)!Q//VV$$$ !)!QXXbi-@-@aHHHVV$$$ 9
%
% 6 6 $Q ! 5 5 5 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6s (E EEc
t j dd dd }t j dd dd }t j |d dd df | t |t j g dg d g d
g d gd t j d }t j |d t |d t j |ddg t |d
d S )Nr lrJ r r rw r )r r# r rw )r r r )r r r7 )r r r r r rI )r r r opflag_tests inplace_addr r{ r& s r test_operand_flagsz TestUfunc.test_operand_flagsX s Ib $ $ $ , ,Q 2 2 Ias # # # + +Aq 1 1 3B38a000Q,,,
. . ."07:< < < = = =
HQKK A&&&Q QF+++Qr c dd l mc m} t j dgd }t j dgd }| || }t
|t j dgd t t |j d S )Nr r zu8,u8,u8rJ )r# r r )
numpy.core._struct_ufunc_testsr _struct_ufunc_testsr r{ add_tripletr r RuntimeError
register_fail)r4 struct_ufuncr' r( r> s r test_struct_ufunczTestUfunc.test_struct_ufunce s ========= Hi[
3 3 3 Hi[
3 3 3 ))!Q//VRXyk D D DEEElL$>?????r c v t j t j dd t j dd t j dd gt j }t j t j dd t j dd t j dd gt j }t j || }t j t j d t j dd t j dd gt j }t || d S )Nr" r# rw r rJ )r r{ r r1 test_add_rationalsr )r4 r' r( r> rz s r test_custom_ufunczTestUfunc.test_custom_ufunco s' H
%a
+
+
%a
+
+
%a
+
+
- "*
,
,
,
H
%a
+
+
%a
+
+
%a
+
+
- "*
,
,
, !3Aq998
%a
(
(
%a
+
+
%a
+
+
- "* , , ,
VX&&&&&r c t t 5 t j t j d dt j t d f d d d d S # 1 swxY w Y d S )Nr" r- )r r0 r multiplyr r1 r2 r3 s r test_custom_ufunc_forced_sigz&TestUfunc.test_custom_ufunc_forced_sig s
9
%
% I I K033Q#2#;S$"G
I
I
I
I I I I I I I I I I I I I I I I I I Is .MyThing r c || _ d S rW rm )r4 rm s r __init__z:TestUfunc.test_custom_array_like..MyThing.__init__ s
"
r c | j d S Nr rO r3 s r __len__z9TestUfunc.test_custom_array_like..MyThing.__len__ s z!}$r c xj dz
c_ t |t s|f}t | | j k rt d | j t | d S )Nr" boo)
getitem_countr tupler ndim
IndexErrorrm )r4 r MyThings r __getitem__z=TestUfunc.test_custom_array_like..MyThing.__getitem__ sr %%*%%!!U++ Aq66DI%%$U+++ wtz#a&&'' 2333r c ( xj dz
c_ | S r^ )
rmul_count)r4 r rZ s r __rmul__z:TestUfunc.test_custom_array_like..MyThing.__rmul__ s ""a'"" r N)
rB rC rD __array_priority__r] rV rP rS r[ r^ )rZ s r rZ rL su !% J M
#
#
#
%
%
%
4
4
4
4
4
r rZ r r; r" r# )r r r r] rV )r4 rZ s @r test_custom_array_likez TestUfunc.test_custom_array_like s 2
1
ggfoo%%"a');<<<%*G,ABBBBBr r' rI rJ c | }t j |g dd t |g d t j t 5 t j |g d d d d n # 1 swxY w Y | }t j |g d t |g d | }t j g d }t j |g d| t |g d t j t 5 t j |g dg d d d d n # 1 swxY w Y t j t 5 t j |g ddd gdg d d d d S # 1 swxY w Y d S )
Nr# r r# r" )
r r" r rw r r r r r r )r# r rw )
r r" r r r r r r r r1 r1 r1 )
r r" rw r i r r r r r r# )
r r r1 atr r r/ r r r{ )r4 r' r r( s r test_ufunc_at_basicz TestUfunc.test_ufunc_at_basic su VVXX
"iii###R777888
]:
&
& % % FIIb))) $ $ $ % % % % % % % % % % % % % % % VVXX
r999%%%R:::;;;
VVXX H___ % %
"iii###R;;;<<<
]:
&
& 4 4 KNN1iii 3 3 3 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 ]:
&
& 1 1 FIIaaVQK 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1s6 ##BBB &FFF3(G((G,/G,typecode
AllIntegerFloatc
|t j u rn|t j d v rZt j d| }t j ddt j }t j dd| }nXt j d| }t j dd t j }t j d | }| }t j
d
5 }t j d | ||| d d d n # 1 swxY w Y t j
d
5 }t j d t || D ]%\ } }
||| |
|| | dz d
& d d d n # 1 swxY w Y t || t | dk r{t | dk sJ |d j |d j k sJ t# |d j d d t# |d j d d k sJ d S d S )Nrj r1 rJ r' )r r< r" rM i Tr r r r- r rI )r r typecodesrh rk randintrd r r r r r rg r r r categoryrn message) r4 ri r r' indxvalsatagw_atw_loopr r s r test_ufunc_at_inner_loopsz#TestUfunc.test_ufunc_at_inner_loops s BI (bl<.H"H"H8,,,A9$$Sr$AAD9Q(333DDH---A9$$TBG$DDD9T222Dvvxx
$D
1
1
1 $T !( + + + HHQd # # # $ $ $ $ $ $ $ $ $ $ $ $ $ $ $
$D
1
1
1 EV !( + + +D$
E
E1 d1gqd1QqS5k8DDDDD
E E E E E E E E E E E E E E E E T1 v;;??t99q====7#vay'99999tAw'',F1I4E0F0Fss0KKKKK ? LKs% >,D66D:=D:A F..F25F2Complexc D t j d| }t j t j dt j t j ddt j g }|j d }| ||| t j | }|t j u r dx|d< |d< nH|dxx d|t j
u r|n| z z
cc< |dxx d|t j
u r|n| z z
cc< t || d S ) NrI rJ r r8 r r r r" ) r rh concatenaterd rL r< r rg ones_likerH r1 r )r4 ri r r' rr valuerz s r !test_ufunc_at_inner_loops_complexz+TestUfunc.test_ufunc_at_inner_loops_complex s
GBh ' ' '~rwq 8 8 8 !wr1BG < < < > ? ? R
D% <?? BK (* *HQK(1++ QKKK1"& ufE EKKK QKKK2%26//vF FKKK1h'''''r c t j d }t j |t d t j d t
|t j d d S )Nr )r r% r1 rg rc rh r r s r test_ufunc_at_ellipsisz TestUfunc.test_ufunc_at_ellipsis sU hqkk
#uT{{BGAJJ///3
+++++r c t j dt j }t j d }t j || t j |g dk sJ d S )Nr rJ )r r r i8r )r rh r r r indexed_negativerg rj )r4 r~ rr s r test_ufunc_at_negativez TestUfunc.test_ufunc_at_negative sf garx(((y|| T***vc 3 3 334444444r c V t j dt j }t j dt }d|d< d|d< d|dd < t j dt }t j ||| |d | k sJ d S ) Ni rJ rI r r r" r1 i )r r% r r r1 rg r )r4 indicesr( r' s r test_ufunc_at_largez TestUfunc.test_ufunc_at_large
s (4rx000 HT ' ' '!!$%% HQe $ $ $
!Wa tquuwwr c ( t j d }t j d }t j t | t j }t j ||| |d t | k sJ d S )NrI i rJ r )r r% rh r r r1 rg )r4 r~ valuesindexs r test_cast_index_fastpathz"TestUfunc.test_cast_index_fastpath sq hrll VBH555
#uf%%%1vV$$$$$$r r| rV c
t j d }t j t j d d }t j ||| t
|t j |d|z d S )NrM r# )r r% repeatr r1 rg r r )r4 r| r~ r s r #test_ufunc_at_scalar_value_fastpathz-TestUfunc.test_ufunc_at_scalar_value_fastpath sh htnn ")D//1--
#ue$$$3 S!e) < <=====r c t j d dd }t j g dg dg dg }t j |t
d g df| t |g dg dg d g t j d
ddd }t j g d }t j |t
d t
d g df| t |g d g dg d
gg dg dg dgg dg dg dgg t j d dd }t j g dg dg dg }t j |g dt
d f| t |g dg dg dg t j d
ddd }t j g d }t j |t
d g dt
d f| t |g dg dg dgg dg dg dgg d g d g d gg t j d dd }t j g d }t j |d g df| t |g d g d g d!g t j d
ddd }t j g d }t j |g dd t
d f| t |g dg d g d!gg d"g d#g d$gg d%g d&g d'gg t j d
ddd }t j g d }t j |t
d t
d t
d f| t |g d(g d)g dgg d*g d+g dgg d%g d,g d gg d S )-Nr rw rd )r3 r3 r3 )r5 r5 r5 r" r# r" )r f )rw )r i_ 4 r )r1 r3 r5 )r i re )r i )r )r )r i )r8 i ) )r i r# )i r i ) r ) r i] )j r r )r rI r7 ) r if )s i= )r8 r r} ) r io )r2 iF r r )r r r ) r ic )r r r )r r )v i@ )r )r r )r1 r i. )g i1 )m i7 )p i: )y iC )r r r r{ r1 rg rc r r& s r test_ufunc_at_multiDz TestUfunc.test_ufunc_at_multiD& s IaLL A & & HoooH I I
!eDkk999-q111Q
}}}EFFF IbMM ! !!Q * * H___ % %
!eDkk5;; :A>>>Qmmmmmm mm nn nn nn nn nn
IaLL A & & HoooH I I
!iiit-q111QOOO___EFFF IbMM ! !!Q * * H___ % %
!eDkk999eDkk:A>>>Qkk oo oo ll oo oo ll oo oo
! ! !
IaLL A & & H___ % %
!a^Q'''Q 999=>>> IbMM ! !!Q * * H___ % %
!iiiE$KK0!444Qkkkkkk oommmm oommmm
IbMM ! !!Q * * H___ % %
!eDkk5;;d . r r r r r r )r r rc rg r r{ r s r test_ufunc_at_dtypesz TestUfunc.test_ufunc_at_dtypes~ sR IbMM
A|||S)))Q!C!C!CDDEEEEEr c > t j d }|dz dk }t j ||g d t |g d t j dd }t j |g d t |g d d S )
NrI r# r )r r# r r r )
r" r" r" rw r" r r" r r" r u4rJ rb )
r r" r# rw r l r r r r )r r r rg r invert)r4 r' r s r test_ufunc_at_booleanz TestUfunc.test_ufunc_at_boolean s IbMMA
Auooo...Q666777
Ib % % %
Q """QCCCDDDDDr c F t j d }|d d d f d d ddf }t j |ddgd t |t j d t j g dt j d }t j g dt j d }t j ||d t |g d t j d
dgt }t t t j j |ddgd t |t j d
dgt t t t j j t j d dg t j g d
}t j
|dgd t |t j g d
d S )Nr r r" rw r r )r" r# rw r rP )r" r r r r' rJ rI r )r r r1 rg r r{ r< rR ri r r0 r modfr. r )r4 origr' r r s r test_ufunc_at_advancedz TestUfunc.test_ufunc_at_advanced s y|| DM!!!QqS& !
!aVQ 41 ... BHSMM$>$>$@$@AA,,,
(B(B(D(DEE
&%###6<<<000 3(&111iFQFA>>>628S!HF#C#C#CDDD j"'*bimmaSAAA
HYYY
a!a Q++,,,,,r c > t j d }t j d }t t t j j |dg| t j ddgddggg }t t t j j j j |dg d S )Nr r r r" r# rw r )
r rh r r0 r rg r{ linalg
_umath_linalgr) r& s r test_at_not_none_signaturez$TestUfunc.test_at_not_none_signature s GI GI i q1#q999 H1v1v&' ( (i!8!A;c t j dg }t j |dgdg |d dk sJ d S )Nr r )r r{ r rg r s r test_at_output_castingz TestUfunc.test_at_output_casting sB htnn
C!qc"""1v{{{{{{r c t j d }t j t 5 t j |ddgg d d d d d S # 1 swxY w Y d S )Nr r r" r )r r r r/ r r1 rg r s r test_at_broadcast_failurez#TestUfunc.test_at_broadcast_failure s ill
]:
&
& . . FIIcAq6999 - - - . . . . . . . . . . . . . . . . . .s &A!!A%(A%c t j j }t j dt }t j d|j }|dz }t
|| | t
||d | t
||d | t
||dd | t
||dd | t
||dd | t
||dd | t
||dd d | t
||dd | | t
||dd | | t
||dd d | t
||dd d d
| t
||dd d d | d |j z t
||dd d d d | t
||dd d d d
| t
||dd d d dd | t
||dd d d dd | t
||dd d d | t
||dd d d | t
||dd d d | t
||dd d d dd | t t | t t ||dd d d dd d
t t ||d t t ||d t t ||dd d t t ||dd t t ||d t t ||dd t t ||dd d t t ||d t t ||dd t t ||dddd t t ||dd d t t ||d t t ||dd dd t t ||dd d d d t t ||dd d d d S ) N)r r# rJ r r r r r r! Fr" Tr r r )r r< r" r# )r< r" r# )r" r# )r" r# r r r" r )r r< r# )r< r" r )r# r< r" )r r< r r )r# r r" )r r< r# r" r )r r< r" r ) r r1 rm rh r2 r< r r rm r r0 )r4 rP rW ors r test_reduce_argumentsz TestUfunc.test_reduce_arguments s FM GE % % % GD ( ( (
EQQqTT1 QQqq\\\1%%%QQq!WWa QQq!4(((!,,,QQq!3'''+++QQq!S\\1%%%QQq!T]]A&&&QQq!Tt,,,a000QQq!Tq)))1---QQq!T1%%q)))QQq!T4((!,,,QQq!T4%888!<<<QQq!T4.. $.0I0IJJJQQq!T422A666QQq!T4:::A>>>QQq!T4488!<<<QQq!T4>>>BBBQQqq$GGGKKKQQq!4TEBBBAFFFQQq!Tte<<.MyAc B t || d |D i |S )Nc 3 T K | ]#}| t j V $d S rW )r r r )r inputs r zOTestUfunc.test_structured_equal..MyA.__array_ufunc__.. sL 0C 0C27 16
2:0F0F 0C 0C 0C 0C 0C 0Cr )r )r4 r methodinputsrr s r __array_ufunc__z.MyA.__array_ufunc__ sM -wuf-- 0C 0C;A0C 0C 0C NFLN N Nr N)rB rC rD r r r r MyAr s(
N
N
N
N
Nr r (@r rw zf8,f8,f8rJ )TFFFr T)
r r r r r rA r{ r} r rj )r4 r r' ramrar3 s r test_structured_equalz TestUfunc.test_structured_equal s N N N N N"* N N N
IcNN " "1Q ' '
VV:V
'
'
/
/
1
1ggcll666dCCCRVFsbe| 455t<<<<27 GRWbh bm J
BF J("'2= Hbl$
HSMM
Hb"X . .A )Q1 - - - )Q1 - - - - . .r Nc t j g dt } |||| }|j t k sJ d S )NTNFrJ r- )r r{ ri r< )r4 r r8 r' r s r %test_logical_ufuncs_object_signaturesz/TestUfunc.test_logical_ufuncs_object_signatures/ sN
H((( 7 7 7eAqI...yF""""""r c t j g d }t j t 5 |||| d d d d S # 1 swxY w Y d S )Nr r- )r r{ r r/ r0 )r4 r r8 r' s r +test_logical_ufuncs_mixed_object_signaturesz5TestUfunc.test_logical_ufuncs_mixed_object_signatures9 s
H((( ) )
]9
%
% - - E!Q) , , , , - - - - - - - - - - - - - - - - - -s A AAc t j dd }t j ddg }t ||| |ddgd | | dk sJ t j dt j } |ddgd |j }t |||| | t j d t j }| || dk sJ t j d
gd }t j d |j }| || d k sJ d S )
N 1V3rJ rV rt Tr# r! r rw r r" )r r{ r rm r% r r r< )r4 r r' r r" rz s r $test_logical_ufuncs_support_anythingz.TestUfunc.test_logical_ufuncs_support_anythingD sX
HT & & & Hb"X 55A;;tTlD(A(ABBB||A$&&&&hq)))5$ t,,33CI>>55A3///:::hr***||A3|''4////
HaS $ $ $hr)))||A3|''1,,,,,,r c t j t d 5 |dgdg ddd n # 1 swxY w Y t j t d 5 | g d ddd dS # 1 swxY w Y dS )a
Logical ufuncs are normally well defined by working with the boolean
equivalent, i.e. casting all inputs to bools should work.
However, casting strings to bools is *currently* weird, because it
actually uses `bool(int(str))`. Thus we explicitly reject strings.
This test should succeed (and can probably just be removed) as soon as
string to bool casts are well defined in NumPy.
z contain a loop with signaturerP r 3N)r 20)r r/ r0 rm )r4 r s r !test_logical_ufuncs_reject_stringz+TestUfunc.test_logical_ufuncs_reject_stringX s ]9,K
L
L
L E3%#
]9,K
L
L
L * * LL ) ) ) * * * * * * * * * * * * * * * * * *s 7;; BB Bc t j d }t j ddg }| }t j t
5 ||||d d d d d S # 1 swxY w Y d S )Nr rV rt rx r- )r r{ r r r/ r0 )r4 r r' r r" s r "test_logical_ufuncs_out_cast_checkz,TestUfunc.test_logical_ufuncs_out_cast_checki s
HSMM Hb"X ffhh
]9
%
% 2 2 E!QC 1 1 1 1 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2s A55A9<A9c t j dd }t j dd }t j | t j | k sJ t t j | t j | t t j |dg t j |dg d S )NrI z>i8rJ z(>(@(@AAA6$$S))
#3'''8S)))))r c t j dt j ddd }|dddd ddd
f dd d }t j d
t j dd }|d d d d d f }| }t j || d }t j ||d }t || t |dd d f |dd d f t |dd d f |dd d f d S )Ni rJ r r r r r r" r7 r r# r r r rw ) r r r r r r r1 rm r )r4 rQ y_baserR y_base_copyr0r1s r test_reduce_noncontig_outputz&TestUfunc.test_reduce_noncontig_output sJ
IfBH - - - 5 5aQ ? ?
ac!Bq&1n ' '1a 0 03bh///771== 33Q3qqq5M kkmm
V]]1!&&((]
3
3
V]]1!!]
,
, R VAaaaC[+ac"2333VAaaaC[+ac"233333r with_castc t j d }t j d }|r | t j }t j t d 5 t j |ddg| d d d n # 1 swxY w Y t j t d 5 t j || d d d d S # 1 swxY w Y d S )Nr rw z (shape|size)rP r r! )
r r r r r r/ r r1 r r )r4 r r~ r" s r /test_reduceat_and_accumulate_out_shape_mismatchz9TestUfunc.test_reduceat_and_accumulate_out_shape_mismatch sY illill ) **RZ((C
]:^
<
<
< 2 2 FOOC!QSO 1 1 1 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 ]:^
<
<
< , , F cs + + + , , , , , , , , , , , , , , , , , ,s$ %%BB B<"C++C/2C/ out_shape)r r r r" r" )r" rw r r# f_reducec t j d dd }t j ||j } ||d| }||j k r>t
t 5 ||d|| d d d d S # 1 swxY w Y d S ||d|| }t ||u t || d S )Nr r rw r r$ r r" r# )
r r r r r< rm r r r r )r4 r r# r r' r" correct_outr s r "test_reduce_wrong_dimension_outputz,TestUfunc.test_reduce_wrong_dimension_output s9
IcNN " "1a ( (hy!'** hqq8<<< ) ) ) z**
@
@h????
@
@
@
@
@
@
@
@
@
@
@
@
@
@
@
@
@
@ HQQC(CCCE ESL ! ! ! uk 2 2 2 2 2s 0B
BBc p t j d }t j d }t j d }t j |d|d t j |d|d d df d t t 5 t j |d|d d d d n # 1 swxY w Y t t 5 t j |d|d d df d d d d d S # 1 swxY w Y d S ) N)r" rI r )rw r" r Tr" r F)r rh r r1 rm r r )r4 r' out_correct
out_incorrects r +test_reduce_output_does_not_broadcast_inputz5TestUfunc.test_reduce_output_does_not_broadcast_input s
GG x'' ((
abkD
AAA
abk!!!Q$&7%
HHH
:
&
& H H FMM!"-$M G G G H H H H H H H H H H H H H H H
:
&
& O O FMM!"-1*=M N N N O O O O O O O O O O O O O O O O O Os$ !$CCC0.D++D/2D/c G d dt j }t j d }t j t j d | | | }t j t j d | |u sJ t t j | |u sJ d S )Nc e Zd ZdS )7TestUfunc.test_reduce_output_subclass_ok..MyArrNr r r r MyArrr+ r r r, r r r! )r r r r1 rm rh r r )r4 r, r" s r test_reduce_output_subclass_okz(TestUfunc.test_reduce_output_subclass_ok s BJ hrll
bgajjc
***hhuoov}}RWQZZS}11S8888BFMM#&&''5000000r c F t dt j j v d S )N
)r r inner1d_no_docr r3 s r test_no_doc_stringz TestUfunc.test_no_doc_string s! C . 6677777r c ~ t j t t j d }| d sJ d S )Nz loop of ufunc does not support)r r/ r0 r r{ rQ )r4 r s r test_invalid_argszTestUfunc.test_invalid_args s6 mIrw55yy9:::::::r natc V t j | rJ d S # t $ r Y d S w xY wrW )r r r0 r4 r4 s r test_nat_is_not_finitez TestUfunc.test_nat_is_not_finite
sE {3'' ' '' ' ' DD
((c V t j | sJ d S # t $ r Y d S w xY wrW )r r r0 r6 s r test_nat_is_nanzTestUfunc.test_nat_is_nan
sB 8C== = DD r8 c V t j | rJ d S # t $ r Y d S w xY wrW )r isinfr0 r6 s r test_nat_is_not_infz TestUfunc.test_nat_is_not_inf
sC x}} $ $$ $ $ DD r8 )rB rC rD r r r r skipifr r r r r r r r r r r r r r r r r r r r r r r) r2 r5 r rH r r\ r` rf ri rn r r1 r{ r{ r r r r r r r r r r r rh r r r r r r r r/ r7 r9 r? rT rW rY rV r~ r r r r r r r r r r r r r r r r r r r r r r r r r r r
r{ r r r r r r) r. r4 r; rC rF rI r` r r2 r r1 rh r rH r r. r r r indexed_ufuncsrn rw r} r r r r r r r r r r r r r r r r r r r r r ri r re r} r r r r r r r r r rm r$ r( r- r1 r3
datetime64 timedelta64r7 r: r= r r r r r s 5 5 51 1 1
[(JKK: : LK:G G G Q
Q
Q
h MJ# # # . . .. . .* * *
*
*
*" " "( ( (( ( (" " "( ( (6 6 64 4 41 1 14 4 4; ; ; # # #< , , , A A A [ [ $ $ 6 6+8 9 9
B
B9 9
B [ [ E))g 6 6 6 E;; L L L+N O O0 0O O0 [ Y(G(G(GHH D D IH D F F F . . .3 3 3* ( ( ( [ Wrvrw&788 * * 98 * >
. . > >
. / / /< < <.Q Q Q I I I6gE gE gER*9 *9 *9XMA MA MA^I I II I I
N
N
N E: E: E:N8 8 84 4 4 4; 4; 4;l: : :/ / /
5 5 54 M M M H H H& (, , ,% % %&2 2 21 1 1 4 4 4, , ,8
,
,
, B B B / / /
%
%
%,7 ,7 ,7\ _[!! "! - - -- - -- - -- - -
- - - & & & @1 1 1" [ V\22 [ Wxrx0C0C0C'D'D'/rx$%4&0I'J'J'/rx1E1E1E1E1E1E1D1D1D1F (G (G'H I I
&
& I I 32
& [ . ("(+>+>+>"?"? @ "5"5"5 6 * ,- - [ Y"&"
66! ! 76 - -
!A A A
D D D ?4 ?4 ?4B) ) )&6 6 68 @ @ @' ' '(I I I C C C> [ S &RYr 5 5 5 &RYr 1I J J J# 1 1 18 fbk2;j"*bgrw@N [ BL 6 g9N NP P [ Wn55L L 65P PL: [ Z i)@AA [ Wrvr{BK&HII
(
( JI BA
( , , ,5 5 5 % % % [ W
GBGBKKB '5 6 6> >6 6>N! N! N!`7 7 7F F F
E
E
E- - -4I I I+ + +
. . . >+ >+ >+@ = = = . . .& [ Wnbm
,. . [ [T6"VT4$8VT"$% %# #% %. .
#
[ W
^R]BN ;= = [ [v&t(<&() )- -) )= =
- [ W
^R]BN ;= =- -= =-$ [ W
^R]BN ;= =
*
*= =
* [ WnbmR^
<> >2 2> >2 H H H! ! !.* * *(4 4 4* [ [4-88
,
, 98
, [ [ E E EG G [ Z$ 77 [ Z"&-9J)KLL
3
3 ML 87G G
3 O O O1 1 18 8 8; ; ; [ U]R]5%9%9>2>%;P;P$QRR SR [ U]R]5%9%9>2>%;P;P$QRR SR [ U]R]5%9%9>2>%;P;P$QRR SR r r r c g | ]D}t t t | t j /t t | ES r r r r r r s r r r
O #I #I #Ia#-gb!nnbh#G#G#I72q>> #I #I #Ir c : | j D ]}d|v sd|v r | d \ }}d |D }t j d 5 t j d | | }ddd n # 1 swxY w Y t |t rgt
| }t | t | k sJ t || D ]$\ }}|j t j | k sJ %|j t j | k sJ dS ) z
Check all ufuncs that the correct type is returned. Avoid
object and boolean types since many operations are not defined for
for them.
Choose the shape so even dot and matmul will succeed
rz ?->c 8 g | ]}t j d | S )r; )r rh r r& s r r z$test_ufunc_types..'
s$ 000q""000r Tr r N) r splitr r filterwarningsr rW r r r< r ) r typinpr" r r outsr r& s r test_ufunc_typesrO
sx { . . #:: 99T??S00C000
$D
1
1
1 #H - - -%,C c5 ! ! .::Ds88s4yy((((C
.
.1w"(1++-----
. 9
----- . .s
A33A7 :A7 c g | ]D}t t t | t j /t t | ES r rC r s r r r 3
rD r c | j D ]v}t t d t | z r0| d \ }}d |D }d |D }|D ]}t dd |_ |D ]}t dd |_ t
j d 5 t
j d | | } | | }d
d
d
n # 1 swxY w Y t | dk r|f}|f}t || D ]m\ } }
| j } t j
| t j r2t j | j } d | z }
t# |||
|
]t% | |
nxd
S )
zs
Check that contiguous and non-contiguous calls to ufuncs
have the same results for values in range(9)
zO?mMrG c 8 g | ]}t j d | S )r r r rI s r r z,test_ufunc_noncontiguous..A
s" ...Q"(1a.....r c J g | ] }t j d | ddd !S )r8 Nrw rS rI s r r z,test_ufunc_noncontiguous..B
s- 4441"(2q//##A#&444r r" r Tr r Nr# )r r )r r setrJ r flatr r rK r r r< r
issubdtypefloatingr epsr r )r rL rM r" args_cargs_nr' res_cres_nc_arn_arr< res_epstols r test_ufunc_noncontiguousrb 3
s { ) ) s6{{SXX% & & 99T??S..#...44444 A1QZZAFF A1QZZAFF
$D
1
1
1 # # #H - - -E6NEE6NE # # # # # # # # # # # # # # # s88q==HEHE eU++
)
)JD$B}R--
) (2,,*i u3SAAAAA T4((((
)' ) )s 1 C C #C c $ t j dgd d }t j | sJ | j dk r
| | d S | j dk r | || d S t
d )Nl r0 r r" r# z ufunc with more than 2 inputs)r r{ r r rq r r )r r( s r test_ufunc_warn_with_nanrd [
s $%t,,11$77A
8A;;; yA~~
a a
a8999r zPython lacks refcountsr c " d} t j | gt t j dz z dgz | gt dt j z z z t }t j t
| t j }t j | }t j t 5 t j
|||d d d d n # 1 swxY w Y |t j | k sJ |d dk sJ t j t 5 t j
|||t j d d d d n # 1 swxY w Y |t j | k sJ |d dk sJ d S )
N{ r stringrJ r r- r r" )r" r< r )r r{ r2 BUFSIZEri rh r rd sys getrefcountr r/ r r1 )r| r~ r" counts r test_ufunc_out_casterrorsrl l
s
E
(E7Sc!1 2 22 : 7Srz!1 2 223:@ B B BC
'#c(("'
*
*
*C OE " "E z " " 4 4
sCS(33334 4 4 4 4 4 4 4 4 4 4 4 4 4 4 COE** * * * * r7a<<<< z " " C C
sCSBBBBC C C C C C C C C C C C C C C COE** * * * * r7a<<<<<dr- )r r/ r0 r r1 r{ r r r test_trivial_loop_invalid_castrv
s y,
.
.
. 9 9 rx5!!188889 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9s +AAAoffsetr# c d}t j |g| z dgz |gt dt j z z z t }t j dt j }t
j | }t j t d 5 t j |t j |d d d d n # 1 swxY w Y |t
j | k sJ |d || z k sJ d S )
Nrf rg r rJ r zinvalid literalrP )r< r" r r )
r r{ r2 rh ri rd ri rj r r/ r r1 rm )rw r| r~ r" rk s r test_reduce_casterrorsry
sW
E
(E7V# : 7Srz!1 2 223:@ B B BC
(2RW
%
%
%C OE " "E z): ; ; ; A A
cc4
@@@ A A A A A A A A A A A A A A A COE** * * * *
r7UV^ # # # # # #s .CCCc T t j t 5 t j g dd d d d n # 1 swxY w Y t j t 5 t j g d d d d d S # 1 swxY w Y d S )N)r" r# Nr r )r r/ r0 r r1 rm r r r %test_object_reduce_cleanup_on_failurer{
s y ! ! / /
lllA
.../ / / / / / / / / / / / / / /
y ! ! $ $
lll###$ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $s# $A
AA."B B!$B!r c D t j | dg S rR )r r1 r rK s r r r
s A3 7 7 r r rL c D t j | dg S )Nr# )r logrg rK s r r r
s !aS 1 1 r rg c ` t j t j dt j g }t j d 5 t j t d 5 | | d d d n # 1 swxY w Y d d d n # 1 swxY w Y t j t j dt j g }t j d 5 t j t 5 | | d d d n # 1 swxY w Y d d d d S # 1 swxY w Y d S )Nr warnr z
invalid valuerP r ) r r{ r r r r# r r/ rs )r r~ s r test_ufunc_methods_floaterrorsr
s
(BFAw'
(
(C
\.
@
@
@ F3KKK
(BFAw'
(
(C ! ! !
]-
.
. F3KKK sl B
A6* B
6A: :B
=A: >B
BBD#3 D ? D# D D#D D##D'*D'c | dk rdS t j | j sdS | j j dk rt j | j S dS )Nr Fr T)r signbitr r< kindr )r| s r _check_neg_zeror
sP ||u
:ej ! ! u {3 z%*%%% 4r r< AllFloatc Z t j | } | j dk r$| t dd }n| d }t j | }t j | }t
||z sJ ||z
}t
| sJ d S )Nr )r r< r r r| r{ r )r< neg_zeror~ arr2s r test_addition_negative_zeror
s HUOOE zS::gdD 1 122::d##
(8
C
8H D 3: & &&& &4KC 3 r use_initialTFc t j | } | j dk r$| t dd }n| d }i }|r||d< nt j d t dd D ]~}t j |g|z | }t j |fi |}|dk s|rt | sJ Ht j |j rJ t j |j
rJ d S )Nr r r z1-0. propagation in sum currently requires initialr rJ )r r< r r r| r xfailr r{ r r r r r )r< r r rr r r~ r s r "test_addition_reduce_negative_zeror
s%
HUOOE zS::gdD 1 122::d##
F J $y HIII 1c]] , ,hzA~U333fS##F## q55K5"3'' ' '' ' z#(++ + ++z#(++ + ++ +, ,r c
e Zd Zd Zd Zd Zej d e j
d e j
d fd e j
d e j
d f e j
d d e j
d fdg d Z d
Zd Z ej
eed d
d Zej dg d d Zd Zd ZdS )TestLowlevelAPIAccessc F t j d }t j d }t j d }t j ||d f }||||fk sJ t j ||d fd }||||fk sJ t j |t d f }||||fk sJ t j t 5 t j ||d fd d d d d S # 1 swxY w Y d S )Nr f4r )NNr r- rv rw )r r< r1 resolve_dtypesr2 r r/ r0 )r4 r r r r s r test_resolve_dtypes_basicz/TestLowlevelAPIAccess.test_resolve_dtypes_basic s|
Xd^^
Xd^^
Xd^^ F ! !2r4. 1 1RRL
F ! !B# );
"
=
=RRL
F ! !2sD/ 2 2RRL
]9
%
% @ @ F ! !2r4.$ ! ? ? ? @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @s $%DD Dc t j d }t j ||d f }|||t j t fk sJ t j d }t j t 5 t j ||d f d d d d S # 1 swxY w Y d S )NS010i)r r< r r r} r r/ r0 )r4 r r dtss r test_weird_dtypesz'TestLowlevelAPIAccess.test_weird_dtypes/ s
Xd^^ H # #RTN 3 3RRXd^^,,,,, huoo
]9
%
% 6 6 H # #S#t$4 5 5 5 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6s #B66B:=B:c t j d }t j t 5 t j |||fd d d d d S # 1 swxY w Y d S )Nr T reduction)r r< r r/ NotImplementedErrorr1 r )r4 r s r test_resolve_dtypes_reductionz3TestLowlevelAPIAccess.test_resolve_dtypes_reduction; s
Xd^^
].
/
/ @ @ F ! !2r2,$ ! ? ? ? @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @s %A A$'A$dtypesr NrP )r r Nc t j t 5 t j | d d d d S # 1 swxY w Y d S rW )r r/ r0 r r1 r )r4 r s r test_resolve_dtypes_errorsz0TestLowlevelAPIAccess.test_resolve_dtypes_errors@ s ]9
%
% * * F ! !& ) ) ) * * * * * * * * * * * * * * * * * *s AA A c t j d }t j d }t j d |d fd }||||fk sJ d S )Ni2longTr )r r< r1 r )r4 r long_r s r r z3TestLowlevelAPIAccess.test_resolve_dtypes_reductionI s]
Xd^^ f##T2t$4#EEueU+++++++r c v t j d }t j t 5 t j d ||f d d d n # 1 swxY w Y t j t 5 t j d d d d d S # 1 swxY w Y d S )Nr )NNr )r r< r r/ r0 r1 r r8 )r4 r s r $test_resolve_dtypes_reduction_errorsz:TestLowlevelAPIAccess.test_resolve_dtypes_reduction_errorsP s
Xd^^
]9
%
% 2 2 F ! !4R. 1 1 1 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 ]9
%
% 1 1 F / 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1s# #A A!$A! B..B25B2 pythonapiz2`ctypes.pythonapi` required for capsule unpacking.r c
t j t j d }t j t j d }t j t j d }t j t j t j |||t j
G
fddt j }t j d }t j
||f \ }}|||fk sJ t j
| t j t j
j _ t j
t j | t j d }t j |t j | j }t j d| } | |j || j j | j j | j j || j j d | j j d |j t9 | t j d| d S )
Nr# r" c Z e Zd Zd fdej fdej fdej fdej fgZdS );TestLowlevelAPIAccess.test_loop_access..call_info_t strided_loopcontextauxdatarequires_pyapino_floatingpoint_errorsN)rB rC rD ctc_void_pc_byte_fields_)strided_loop_ts r call_info_tr c sF 0BK(BK(!29-*BI6 HHHr r r s numpy_1.24_ufunc_call_inforI rJ r ) r ARRAYc_char_p c_ssize_t CFUNCTYPEc_intr Structurer r< r _resolve_dtypes_and_context_get_strided_loopr PyCapsule_GetPointerrestype py_objectcastPOINTERcontentsr r r ctypesdatarm r r r ) r4 data_tdim_t strides_tr r r<
call_info_obj call_infor~ r s @r test_loop_accessz&TestLowlevelAPIAccess.test_loop_accessY s "+q)) q))HR\1-- "+vuiN N ", Xd^^ K C CRH M MMb"X~~~~ %%m44446K )1L55 ]++ 9::< < GIrz+'>'>??H i"%%% !sz
88
#*,Q/1CA1FGG! # # # 32R!8!8!8 899999r r )r" r )r" r c t j d }t j ||f \ }}t j t d 5 t j || d d d d S # 1 swxY w Y d S )Nr z fixed_strides.*tuple.*or NonerP )
fixed_strides)r r< r r r r/ r0 r )r4 r r r< r s r )test__get_strided_loop_errors_bad_stridesz?TestLowlevelAPIAccess.test__get_strided_loop_errors_bad_strides s
Xd^^ ??RII
I
]9,K
L
L
L L L K ) ))7 ) K K K L L L L L L L L L L L L L L L L L Ls "BB
Bc t j d }t j ||f \ }}t j t d 5 t j d d d d n # 1 swxY w Y t j t d 5 t j | d d d n # 1 swxY w Y t j | t j t 5 t j | d d d d S # 1 swxY w Y d S )Nr PyCapsulerP znot the capsule!z.*incompatible context)
r r< r r r r/ r r r0 r1 )r4 r r< r s r +test__get_strided_loop_errors_bad_call_infozATestLowlevelAPIAccess.test__get_strided_loop_errors_bad_call_info s
Xd^^ ??RII
I
]:[
9
9
9 > > K ) )*< = = = > > > > > > > > > > > > > > > ]9,D
E
E
E 0 0 F $ $Y / / / 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 %%i000
]9
%
% 5 5 K ) )) 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5s6 B BB& CCC EE Ec t j dt j }d|d d< d|d d< t j | }t |d d |d d d S )N)i i rJ r" r i )r r% r r r )r4 r& r s r test_long_arraysz&TestLowlevelAPIAccess.test_long_arrays se H[ 2 2 2!Q"c
VAYYRU1Xr"vc{+++++r )rB rC rD r r r r r r r r< r r r> hasattrr r r r r r r r r r s @ @ @,
6
6
6@ @ @
[ X
RXc]]HBHSMM *
828C==("(3-- 0
RXc]]D("(3-- 0 ( ! !
* * ! !
*, , ,1 1 1 [GGB 4 44G I I(: (:I I(:T [ Y(@(@(@AAL L BAL
5
5
5 , , , , ,r r )Kr r ri r r r r numpyr numpy.core._umath_testsr _umath_testsr numpy.linalg._umath_linalgr r r( numpy.core._operand_flag_tests_operand_flag_testsr9 numpy.core._rational_testsr
numpy.testingr r r r r r r
r r r
r r numpy.testing._private.utilsr numpy.compatr r __dict__r UNARY_UFUNCSr r rF r r r r dirrO _no_nep50_warningrb signr rd r> rl r2 rh rp rt rv ry r{ r1 r rm r r rn r r r r r r r sP
% % % % % % % % % ( ( ( ( ( ( ( ( ( 5 5 5 5 5 5 5 5 5 4 4 4 4 4 4 4 4 4
9 8 8 8 8 8 2 2rw} 5 < < > > 2 2 2 GGLGGG K K K K K K K K4@H @H @H @H @H @H @H @HF X% X% X% X% X% X% X% X%vJ #I #I33r77 #I #I #I J J. .J J.2 #I #I33r77 #I #I #I J J") ") J J")J 27BH"566
:
: 76
: $-EFF GF > 33rzC/?+@+@'ABB: : CB: G$BCC 33rzC/?+@+@'ABB: : CB DC:9 9 9 $-EFF
BJM33s2:~../1 1$ $1 1 GF$4$ $ $ G$BCC BFM 77J G G G 11d ; ; ; => >
> > DC
",z":;;
<;
",z":;; u
66, , 76 <;,2I, I, I, I, I, I, I, I, I, I,r