U
|
O ±dxÒ ã @ s d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlZd dl m
|
m Z d dl m mZ d dlm
|
mZ d dlm
|
mZ d dlmZmZmZmZmZm Z m Z m Z m Z m Z m!Z! d dl"m#Z# d dl$m%Z% dd ej
|
j&j' (¡ D Z)dd e)D Z*G d d
|
d
|
Z+G d d d Z,d d Z-G dd dZ.ej/ 0ddd e1eD ¡dd Z2ej/ 0ddd e1eD ¡e 3¡ dd Z4ej/ 0dej5ej6g¡dd Z7ej/j8e ddd d Z9ej/ 0d d e:ej;d g¡d d! Z<ej/j8e!d"dej/ 0d d e:ej;d g¡d#d$ Z=d%d& Z>ej/j8e ddej/ 0d'd ej;d( e:d ej; g¡d)d* Z?ej/j8e!d"dej/ 0d+ej@jAej@jBejd,d- d.d/ejd0d- d1d/g¡d2d3 ZCd4d5 ZDej/ 0d6ejEd7 ¡d8d9 ZFej/ 0d6ejEd7 ¡ej/ 0d:d;d<g¡d=d> ZGG d?d@ d@ZHdS )Aé N)Úparam) Úassert_Ú assert_equalÚ assert_raisesÚassert_array_equalÚassert_almost_equalÚassert_array_almost_equalÚassert_no_warningsÚassert_allcloseÚ HAS_REFCOUNTÚsuppress_warningsÚIS_WASM)Úrequires_memory)Úpicklec C s g | ]}t |tjr|qS © )Ú
|
isinstanceÚnpÚufunc)Ú.0Úobjr r úRd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/core/tests/test_ufunc.pyÚ
|
<listcomp> s ÿr c C s g | ]}d |j kr|qS )zO->O)Útypes)r Úufr r r r s
|
c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
ÚTestUfuncKwargsc C s¨ t ttjdddd t ttjddtd t ttjdddgd t ttjddd d t ttjddd d
|
t ttjddd d t ttjddd d t ttjdddgd d S )Né é Úsafe)Zcastingx)Zdtypexé )Zextobjx)Zoutxúii->i)Zsigx)Z
|
signaturexF)ZsubokxT)Zwherex©r Ú TypeErrorr ÚaddÚint©Úselfr r r Útest_kwarg_exact s z TestUfuncKwargs.test_kwarg_exactc C s t ttjddddd d S )Nr r r )ÚsigÚ signature©r r! r r" r$ r r r Útest_sig_signature' s ÿz"TestUfuncKwargs.test_sig_signaturec C s0 t ttjdddtd t ttjdddtd d S )Nr r r )r' Údtype)r( r+ r r$ r r r Útest_sig_dtype+ s ÿÿz TestUfuncKwargs.test_sig_dtypec C s t ttjdddgdd d S )Nr r r T)ZextobjÚparrotr) r$ r r r Útest_extobj_refcount1 s z$TestUfuncKwargs.test_extobj_refcountN)Ú__name__Ú
|
__module__Ú __qualname__r& r* r, r. r r r r r s
|
r c @ s
|
e Zd ZdZejejfejejfejejfejejfejejfej ej fejejfej
|
ej
|
fgZ e j de ¡ejddfddZdd Ze j de ¡eddfd d
|
ZG d d d Zd d ZefddZdd ZefddZefddZdd Ze j de¡dd Zd S ) Ú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 C s@ t jd|||d}||d d d
}t|| t|j| d S ©Né
|
©r+ r ©r Úfullr
|
r r+ ©r% Z input_dtypeZ output_dtypeÚfÚxÚyÚxsZysr r r Útest_unary_PyUFunci s
|
z(TestUfuncGenericLoops.test_unary_PyUFuncc C s | | S ©Nr )r: r; r r r Úf2p s zTestUfuncGenericLoops.f2c C sB t jd|||d}|||d d d
}t|| t|j| d S r3 r6 r8 r r r Útest_binary_PyUFuncs s
|
z)TestUfuncGenericLoops.test_binary_PyUFuncc @ s e Zd Zdd Zdd ZdS )zTestUfuncGenericLoops.fooc C s
|
t d¡S ©Nr ©r Úbool_r$ r r r Ú conjugate| s z#TestUfuncGenericLoops.foo.conjugatec C s
|
t d¡S rA rB ©r% r r r r Ú logical_xor s z%TestUfuncGenericLoops.foo.logical_xorN)r/ r0 r1 rD rF r r r r Úfoo{ s rG c C s* t jdtd}tt t |¡dk¡ d S )Nr4 r5 r )r ÚonesÚobjectr ÚallÚabs©r% r: r r r Útest_unary_PyUFunc_O_O s z,TestUfuncGenericLoops.test_unary_PyUFunc_O_Oc C s. t jd| td}tt t |¡dk¡ d S )Nr4 r5 T)r r7 rI r rJ rD ©r% rG r: r r r Ú$test_unary_PyUFunc_O_O_method_simple s z:TestUfuncGenericLoops.test_unary_PyUFunc_O_O_method_simplec C s, t jdtd}tt t ||¡dk¡ d S r3 )r rH rI r rJ r" rL r r r Útest_binary_PyUFunc_OO_O s z.TestUfuncGenericLoops.test_binary_PyUFunc_OO_Oc C s, t jd| td}tt t ||¡¡ d S )Nr4 r5 ©r r7 rI r rJ rF rN r r r Ú test_binary_PyUFunc_OO_O_method s z5TestUfuncGenericLoops.test_binary_PyUFunc_OO_O_methodc C s, t jd| td}tt t ||¡¡ d S )N)r4 r é r5 rQ rN r r r Ú test_binary_PyUFunc_On_Om_method s z6TestUfuncGenericLoops.test_binary_PyUFunc_On_Om_methodc C s^ t jddgdd}t|d ts$tt |¡}|jt d¡ksBtt|t jddgdd d S )Ny ð? @y @ ÀÚOr5 r y ð? Ày @ @)r Úarrayr ÚcomplexÚAssertionErrorrD r+ r ©r% ÚarrÚresr r r Ú test_python_complex_conjugate s
|
|
z3TestUfuncGenericLoops.test_python_complex_conjugater c sÌ t t jd ¡ G fdddt j}t j t jd}t j| dd}t jddn z ||}W nB tk
|
r¤ } z$tt| || W 5 Q R X W 5 d}~X Y nX ||}t| d¡| W 5 Q R X dS ) z9Compare the result of the object loop with non-object oneé c s e Zd Z fddZ ZS )zITestUfuncGenericLoops.test_unary_PyUFunc_O_O_method_full.<locals>.MyFloatc s6 zt ¡W S tk
|
r0 fdd Y S X d S )Nc s t tjj S r> )Úgetattrr ÚcoreÚumathr )ÚattrÚvalr r Ú<lambda>¨ ó zgTestUfuncGenericLoops.test_unary_PyUFunc_O_O_method_full.<locals>.MyFloat.__getattr__.<locals>.<lambda>)ÚsuperÚ __getattr__ÚAttributeError)r% ra )Ú __class__rb )ra r rf ¤ s zUTestUfuncGenericLoops.test_unary_PyUFunc_O_O_method_full.<locals>.MyFloat.__getattr__)r/ r0 r1 rf Ú __classcell__r ©rb )rh r ÚMyFloat£ s rk r5 rU Úraise©rJ N)
|
r Úfloat64ÚpirV ÚerrstateÚ Exceptionr Útyper Úastype)r% r rk Znum_arrZobj_arrZres_numÚexcZres_objr rj r Ú"test_unary_PyUFunc_O_O_method_full s $z8TestUfuncGenericLoops.test_unary_PyUFunc_O_O_method_fullN) r/ r0 r1 Ú__doc__r ÚsingleÚdoubleZcsingleZcdoubleÚ
|
longdoubleÚ clongdoubleZ np_dtypesÚpytestÚmarkÚ parametrizeÚexpr= r? r@ rG rM rO rP rR rT r\ ÚUNARY_OBJECT_UFUNCSru r r r r r2 6 s0 -
|
|
|
|
|
|
|
|
ü r2 c C s d S r> r r r r r Ú_pickleable_module_global¹ s r c @ sò e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd Zd Z d d Z
|
dd Z dd Z dd Z dd Zdd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Z e j d3d4d5 d6d5 g¡d7d8 Z e j d3e!d9d5 d:d;e!d<d5 d=d;g¡d>d8 Z e j d?d@dAdBg¡dCdD Z"dEdF Z#dGdH Z$dIdJ Z%dKdL Z&e j dMe'j(e'j)g¡dNdO Z*dPdQ Z+dRdS Z,e j j-e.dTdUdVdW Z/dXdY Z0dZd[ Z1d\d] Z2d^d_ Z3d`da Z4dbdc Z5e j j dddegd ge' 6df¡fdgd ge' 6dg¡fe' 7dh¡e' 6di¡fgdjdkdlgdmdndo Z8dpdq Z9drds Z:dtdu Z;dvdw Z<dxdy Z=dzd{ Z>d|d} Z?d~d Z@dd ZAdd ZBdd
ZCdd ZDdd ZEdd ZFdd ZGdd ZHdd ZIdd ZJdd ZKdd ZLdd ZMdd ZNdd ZOdd ZPd d¡ ZQd¢d£ ZRd¤d¥ ZSd¦d§ ZTd¨d© ZUdªd« ZVd¬d ZWd®d¯ ZXeYd°d±d² ZZd³d´ Z[dµd¶ Z\d·d¸ Z]d¹dº Z^d»d¼ Z_d½d¾ Z`e j d¿dÀ¡e j dÁe' adÂdÃdÃg¡e' adÃgdÂgdÃgg¡e' adÃdÂdÂgdÂdÃdÂgdÂdÃdÃgg¡f¡dÄdÅ Zbe j dÆdÇe' adÃdÂdÃg¡fdgdÃdÃdÂgfdÈf¡e j dÉe'jc dÊf¡dËdÌ ZddÍdÎ ZedÏdÐ ZfdÑdÒ ZgdÓdÔ ZhdÕdÖ Zid×dØ ZjdÙdÚ ZkdÛdÜ ZldÝdÞ Zmdßdà Zndádâ Zodãdä Zpdådæ Zqdçdè Zrdédê Zsdëdì Zte j dMe'jue'jvg¡e j d=dídíewfewdídífdíewdífg¡dîdï Zxe j dMe'jue'jve'jyg¡e j d=ezdíewfewdíezfdíewezfg¡dðdñ Z{e j dMe'jue'jve'jyg¡dòdó Z|e j dMe'jue'jve'jyg¡dôdõ Z}e j dMe'jue'jve'jyg¡död÷ Z~dødù Zdúdû Zdüdý Zdþdÿ Ze j d dÃdÂg¡dd Ze j ddfdddddhg¡e j ddÃdÂg¡e j d e'j(je'j
jg¡d
|
d Zd d Zdd Zdd Zdd Ze j de' d¡e' d¡g¡dd Ze j de' d¡e' d¡g¡dd Ze j de' d¡e' d¡g¡dd ZdíS ( Ú TestUfuncc C s^ t dtjd D ]H}tt tjtj|d¡tjk t tjtj |d¡}t|tj k qd S )Nr r )Úprotocol)
|
Úranger ÚHIGHEST_PROTOCOLr ÚloadsÚdumpsr ÚsinÚ_rational_testsÚtest_add)r% Úprotor[ r r r Ú test_pickle¾ s ÿÿ ÿzTestUfunc.test_picklec C s d}t t |¡tjk d S )NsM cnumpy.core
|
_ufunc_reconstruct
|
p0
|
(S'numpy.core.umath'
|
p1
|
S'cos'
|
p2
|
tp3
|
Rp4
|
.)r r r
r Úcos)r% Zastringr r r Útest_pickle_withstringÉ s z TestUfunc.test_pickle_withstringc C s, t jt_t t tj¡¡}|t jks(td S r> )ÚumtZ _pickleable_module_global_ufuncr r r r
r rX rE r r r Ú test_pickle_name_is_qualnameÎ s z&TestUfunc.test_pickle_name_is_qualnamec C sh d}t |¡}t ttt |d ¡t |d ¡d ¡ ¡ }tt j ||¡d d d
ddddg d S )Né r r rS é é ) r ÚarangerV ÚlistÚzipZravelr r" Úreduceat)r% ÚLr: Úidxr r r Útest_reduceat_shifting_sum× s
|
0z$TestUfunc.test_reduceat_shifting_sumc C s 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 r$ r r r Útest_all_ufuncÝ s QzTestUfunc.test_all_ufuncr r] c C sR t ddd¡\}}}}}t|d t|d t|d t|| jf t|d d S )Nr r ú (i),(i)->())r r r ©r r ©éÿÿÿÿ©r Útest_signaturer Ú size_inferred©r% ZenabledZnum_dimsZixsÚflagsÚsizesr r r Útest_signature03 s ÿ
|
|
|
zTestUfunc.test_signature0c C sN t 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¢ r r r Útest_signature1= s ÿ
|
|
|
|
zTestUfunc.test_signature1c C sV t ddd¡\}}}}}t|d t|d t|d t|| jfd t|d d S )Nr r z(i1,i2),(J_1)->(_kAB)©r r r )r r r rS r] ©r r r r r r¢ r r r Útest_signature2G s ÿ
|
|
|
zTestUfunc.test_signature2c C sV t ddd¡\}}}}}t|d t|d t|d t|| jfd t|d d S )Nr r z (i1, i12), (J_1)->(i12, i2)©r r r )r r r r rS r] rª r r¢ r r r Útest_signature3Q s ÿ
|
|
|
zTestUfunc.test_signature3c C sV t ddd¡\}}}}}t|d t|d t|d t|| jfd t|d d S )Nr r z(n,k),(k,m)->(n,m)©r r r ©r r r r r r rS ©r r r r r¢ r r r Útest_signature4Z s ÿ
|
|
|
zTestUfunc.test_signature4c C sf t ddd¡\}}}}}t|d t|d t|d t|| j| jB | j| j| jB f t|d d S )Nr r z(n?,k),(k,m?)->(n?,m?)r® r¯ r° )r r r r¡ Ú
|
can_ignorer¢ r r r Útest_signature5d s ÿ
|
|
|
|
þzTestUfunc.test_signature5c C sN t ddd¡\}}}}}t|d t|d t|d t|d t|d d S )Nr z(3)->()©r r ©r ©rS r§ r¢ r r r Útest_signature6p s ÿ
|
|
|
|
zTestUfunc.test_signature6c C sV t ddd¡\}}}}}t|d t|d t|d t|d| jdf t|d d S )NrS r z(3),(03,3),(n)->(9)©r r r r ©r r r r r r ©rS r é r r¢ r r r Útest_signature7y s ÿ
|
|
|
zTestUfunc.test_signature7c C sX t ddd¡\}}}}}t|d t|d t|d t|| j| jdf t|d d S )NrS r z(3?),(3?,3?),(n)->(9)r¸ r¹ r rº ©r r r r² r¡ r¢ r r r Útest_signature8 s ÿ
|
|
|
zTestUfunc.test_signature8c C sN t ddd¡\}}}}}t|d t|d t|d t|d t|d d S )Nr z ( 3) -> ( )r´ rµ r¶ r§ r¢ r r r Útest_signature9 s ÿ
|
|
|
|
zTestUfunc.test_signature9c C sX t ddd¡\}}}}}t|d t|d t|d t|| j| jdf t|d d S )NrS r z ( 3? ) , (3? , 3?) ,(n )-> ( 9)r¸ r¹ r rº r½ r¢ r r r Útest_signature10 s ÿ
|
|
|
zTestUfunc.test_signature10c C s& t t t ddd¡ W 5 Q R X d S )Nr r z ((i)),(i)->()©r Ú
|
ValueErrorr r r$ r r r Ú(test_signature_failure_extra_parenthesis s
|
z2TestUfunc.test_signature_failure_extra_parenthesisc C s& t t t ddd¡ W 5 Q R X d S )Nr r z (i),)i(->()rÁ r$ r r r Ú.test_signature_failure_mismatching_parenthesis¡ s
|
z8TestUfunc.test_signature_failure_mismatching_parenthesisc C s& t t t ddd¡ W 5 Q R X d S )Nr r z(i),->()rÁ r$ r r r Ú2test_signature_failure_signature_missing_input_arg¥ s
|
z<TestUfunc.test_signature_failure_signature_missing_input_argc C s& t t t ddd¡ W 5 Q R X d S )Nr r rÁ r$ r r r Ú3test_signature_failure_signature_missing_output_arg© s
|
z=TestUfunc.test_signature_failure_signature_missing_output_argc C s t tjjd d S )Nr )r r Úinner1dr( r$ r r r Útest_get_signature s z TestUfunc.test_get_signaturec C sì dt jddd }tt |d¡dddg t t¡$ tt j|dddd d
|
d
|
dg W 5 Q R X tt j|dd dd d
|
d
|
dg t t¡$ tt j|dd dd d
|
d
|
dg W 5 Q R X tt j|dd dd d
|
d
|
dg t jddd}t j|d|d t|dddg d
|
|d d
< t t¡ t j|dd|dd W 5 Q R X t|d
|
d
|
dg d
|
|d d
< t j|dd |dd t|d
|
d
|
dg d
|
|d d
< t t¡ t j|dd |dd W 5 Q R X t|d
|
d
|
dg d
|
|d d
< t j|dd |dd t|d
|
d
|
dg d S )Nç à?rS Úf8r5 r ç ø?ÚiÚunsafe)r' Úcastingr r )Úi4)rÏ rÏ rÏ r¶ ©Úout)r' rÑ rÎ )r r r r" r{ ÚwarnsÚDeprecationWarningÚzeros©r% ÚaÚbr r r Útest_forced_sig° s> ( ÿ ÿÿ zTestUfunc.test_forced_sigc C s t jdgdgdd}t dgdg¡}t|| t jdgdgdd}t dgdg¡}t|| t t¡ t jdddd W 5 Q R X d S )NrS r] )NNN©r' r> ©r( )r r" r Úmaximumr{ Úraisesr! )r% Zres1Zres2r r r Útest_signature_all_NoneÎ s
|
|
z!TestUfunc.test_signature_all_Nonec C s* t t tj¡}tjdd||d fd d S )NrS r] rÚ )rr r r+ rn r" )r% Z float_dtyper r r Útest_signature_dtype_typeÜ s z#TestUfunc.test_signature_dtype_typeÚ get_kwargc C s
|
t tdS ©Nr5 ©Údictr: ©Údtr r r rc ã rd zTestUfunc.<lambda>c C s t td d fdS ©NrÚ rá rã r r r rc ä rd c C sn t d¡}t t |¡¡}||k s&tt jd||j|ks@tt dd¡}t j||f|djdksjtd S )NÚint64r r ÚsZm8úm8[s])r r )r r+ r r
r rX r" Ú timedelta)r% rß ræ Zint64_2Útdr r r Ú&test_signature_dtype_instances_allowedâ s
|
z0TestUfunc.test_signature_dtype_instances_allowedc C s
|
t | dS rà ©râ ©r: r r r rc ò rd r+ ©Úidc C s t | d d fdS rå rì rí r r r rc ó rd r( c C s d}t jt|d tjd|t d¡ ¡ W 5 Q R X t jt|d tjd|t d¡ W 5 Q R X t jt|d tjd |d W 5 Q R X d S )
|
Nz/The `dtype` and `signature` arguments to ufuncs©ÚmatchrS r ræ zm8[ns])rS r )rS r )rS r )r{ rÜ r! r r" r+ Ú newbyteorder)r% rß Úmsgr r r rë ñ s $ rÎ rÍ Z same_kindr c C s° t jt d¡t d¡dd}|jdks*tt jt d¡t d¡dd}|jdksTtt t¡ t jdt d¡dd W 5 Q R X t t¡ t jdt d¡dd W 5 Q R X d S )Nç ð?r Údr5 )NNrõ rÚ rS )
|
r ÚldexpÚfloat32Úint_r+ rX r{ rÜ r! Úuint64)r% rÎ r[ r r r Ú test_partial_signature_mismatchþ s z)TestUfunc.test_partial_signature_mismatchc C s t t¡" tjt d¡t d¡dd W 5 Q R X t t d¡t d¡¡ t t¡" tjt d¡t d¡dd W 5 Q R X d S )Nr r )Úerõ NrÙ )r{ rÜ r! r r" Úfloat16rù rn r$ r r r Ú*test_partial_signature_mismatch_with_cache s
|
& z4TestUfunc.test_partial_signature_mismatch_with_cachec C sr t jddt jdd}|dks tt jddd d t jfdd}|dksFtt t¡ t jddt jd W 5 Q R X d S )NrË gffffff@rÍ ©r+ rÎ r )r( rÎ r5 )r ÚpowerÚintprX r{ rÜ r! ©r% r[ r r r Ú+test_use_output_signature_for_all_arguments s ÿ z5TestUfunc.test_use_output_signature_for_all_argumentsc C sì t jtdd tjdddd W 5 Q R X t t¡ tjdddd W 5 Q R X t t¡ tjdddd W 5 Q R X t t¡ tjddd d W 5 Q R X t t¡ tjddd
|
d W 5 Q R X t t¡ tjddd d W 5 Q R X d S ) Nz1the signature object to ufunc must be a string orrð rS r] g À^@rÚ z%^->#s ii-izii>i)NrÊ s ÿÿ->i)r{ rÜ r! r r" r ÚUnicodeDecodeErrorr$ r r r Útest_signature_errors# s ÿ z TestUfunc.test_signature_errorsc C sf t jdddgdd}t j||dd t jj|dd t jddd}t j||d d t j||d d d S )
|
Nz
|
2010-01-02z
|
1999-03-14z1833-03z>M8[D]r5 ÚMr4 rè Úm)r rV rÛ Úreducer r" )r% rÖ rZ r r r Útest_forced_dtype_times8 s z!TestUfunc.test_forced_dtype_timesr c sÜ fdd}t jdddgt jd}| |j ¡ ¡}||}||dd}t|| t t ¡ ||dd W 5 Q R X ||d d}t|| t t ¡ ||t j
|
d d
|
W 5 Q R X ||t j
|
d d
|
}|| t j
|
¡}t|| d S ) zBasic test for the safest casts, because ufuncs inner loops can
|
indicate a cast-safety as well (which is normally always "no").
|
c s | f j |S r> )Únin)rZ Úkwargs©r r r Ú
|
call_ufuncH s z.TestUfunc.test_cast_safety.<locals>.call_ufuncrô ç @ç @r5 Úno©rÎ Úequivrþ r N) r rV r÷ rs r+ rò r r{ rÜ r! rn )r% r r rZ Zarr_bsÚexpectedr[ r r r Útest_cast_safetyC s
|
|
zTestUfunc.test_cast_safetyc C sF t d¡}t d¡}t d¡}dD ]º}t |¡}| |¡}| |¡}t || g|| g¡D ]|\}} |dkrt|t| }nt|t| }t || ¡}
|
t t
|
|
|
¡j d} t |
|
|| d |dkrØt |
|
jjd k nt |
|
jj|jk dD ]"} t | ¡} ttt j|| | d
|
qîd D ] } t | ¡} |dkrFttt j|| | d
|
nÐt|t| }t t
|
| ¡j d} t <}| t¡ t t
|
| ¡j¡s¢t t
|
| ¡jd }nd }W 5 Q R X t jd d d t j|| | d
|
}
|
W 5 Q R X t |
|
¡sô| dkrôqt |
|
|| |d t |
|
jj| jk qdD ]¼} t | ¡} t|t| }t t
|
| ¡j d} t <}| t¡ t t
|
| ¡j¡st t
|
| ¡jd }nd }W 5 Q R X t j|| | d
|
}
|
t |
|
¡s¶q t |
|
|| |d t |
|
jj| jk q q^q"t jdt jd
|
}t ||¡}
|
t |
|
dk t |
|
jjd k t | |¡}
|
t |
|
dk t |
|
jjd k d S )Nr4 é rÉ ZbhilqBHILQefdgFDGZFDGgVç¯Ò<)ÚrtolZ
|
bhilqBHILQrn r5 ZefdggwºMh Úignore)ÚinvalidZoverrû )r Úatolr rô ç )r rV r+ rs Ú itertoolsÚproductrW ÚfloatÚ true_divideÚmaxÚfinfoÚ
|
resolutionr
|
r Únamer r! r ÚfilterÚ UserWarningÚisnanZtinyrp ÚisfiniterH rC )r% rÖ r× ÚtgtÚtcrä ZaaZbbr: r; r[ r ZtcoutZdtoutÚsupr r r r Útest_true_divideb st
|
|
|
|
|
|
|
"
|
|
|
|
|
|
|
zTestUfunc.test_true_dividec C s` t jdt jd}t|d ¡ |jd dd t jdt jd}t|d ¡ |jd dd d S )Néô r5 ç $@r r] é )r rH r÷ r ÚsumÚsizern ©r% rÖ r r r Útest_sum_stability¹ s z TestUfunc.test_sum_stabilityú fp errors don't work in wasm©Úreasonc C sâ t tjtjtjtjfD ]Ä}dD ]¼}tjdd¦}t dt ¡ |||d d }t
|
|¡ }t t |d| tj d|d |d}tt |¡| t t |d| tt |d d d
¡| t t |d | W 5 Q R X q tjd
|
|d}tt |d d d
¡d tt |dd d
¡d tt |d d d
¡d tt |dd d
¡d tt |d d d
¡d tt |dd d
¡d tt |d d d
¡d tt |dd d
¡d tjd|d}||7 }t|d qd S )N© r r r r é r» é é é é é é iÓ T©ÚrecordÚalwaysr r r5 r rS r* g @o@g àd@éþÿÿÿéýÿÿÿ©r r )r# r rü r÷ rn ry ÚwarningsÚcatch_warningsÚ simplefilterÚRuntimeWarningr% r Úlenr r r- rH )r% rä ÚvÚwr& Zoverflowrõ r r r Útest_sumÀ s0 zTestUfunc.test_sumc C s¶ t jt jt jfD ]}dD ]}|||d d |||d d d }t j||d}t d|d ¡|_t d|d ¡ |_tt |¡| tt |d d d
¡| qt j
|
d|dd }tt |d d d
¡d tt |dd d
¡d tt |d d d
¡d
|
tt |dd d
¡d
|
tt |d d d
¡d tt |dd d
¡d tt |d d d
¡d
|
tt |dd d
¡d
|
t j
|
d |dd }||7 }t|d qd S )Nr4 r r y ð?r5 r r* y @o@ @o@rS y àd@ àd@r? r@ rA y @ @) r Z complex64Z
|
complex128rz Úemptyr ÚrealÚimagr r- rH )r% rä rG r& rõ r r r Útest_sum_complexã s( , zTestUfunc.test_sum_complexc C sX t tjdgddd ttjdgddd t tjtjdtjd d
|
dd d d d g d S ) NrS r ©Úinitialr gÉ?ç¹?ç333333Ó?)r rS r r5 ©r r )ÚaxisrO é )r r r- r rH ræ r$ r r r Útest_sum_initialü s
|
ÿzTestUfunc.test_sum_initialc C sT t tjddgddggddgdd t tjddgddggdd ddgd
|
d d g d S ) Nrô r r ç @TF©Úwherer ç @)rS rO rX ç "@)r r r- r$ r r r Útest_sum_where s $ÿÿzTestUfunc.test_sum_wherec C sZ t d¡ d¡}tt ||¡t j|| dd t d¡}tt ||¡t || ¡ d S )Nr ©r rS r ©rS )r r Úreshaper r rÇ r- r/ r r r Ú test_inner1d s
|
zTestUfunc.test_inner1dc C sø d}t d¡ d¡}t d¡ d¡}tt ||¡t j|| dd|d d}t d¡ d ¡}tt ||¡t j|| dd|d t d
|
¡ d ¡}t d¡ d ¡}tttj|| t d
|
¡ d ¡}t d ¡}tttj|| t d¡ d¡}t d¡ d¡}tttj|| t d¡}t d¡ d ¡}t
|
||¡\}}t |j d d || }t jdd:}t d¡ ||7 }t t|d t|d |d k W 5 Q R X t d¡ dd¡}t d¡}t d¡}tttj||| t d¡} t ||| ¡}
|
t|
|
| k d S )NÚ broadcastr] r¬ ©r r r r r] ©Úerr_msgz"extend & broadcast loop dimensions©r r r5 )r] r ©r] r r r r© rS )rS r r r Tr< r> r r r r )r r r^ r r rÇ r- r r rV Zbroadcast_arraysr ÚstridesrB rC rD rF r rH rJ ) r% ró rÖ r× ÚurG r: rH rÑ Zout2Úcr r r Útest_broadcast sB $$
|
|
|
|
|
|
zTestUfunc.test_broadcastc C sj t d¡ dd¡}t d¡}t j|||d |t d¡d k ¡ sFttj|||d |dk ¡ sftd S )NrS r )r r] rS rÐ r r ) r r r^ rJ r" rJ rX r rÇ ©r% rZ rÑ r r r Útest_out_broadcasts> s
|
z TestUfunc.test_out_broadcastsrZ rÑ r r )r] rS re z(1,)->()z
|
(2,)->(1,)z(4, 3)->(4, 1))Úidsc C s\ t jtdd tj||d W 5 Q R X t jtdd tjt d¡||d W 5 Q R X d S )Nznon-broadcastablerð rÐ r )r{ rÜ r r Úpositiver" rH rj r r r Útest_out_broadcast_errorsM s z#TestUfunc.test_out_broadcast_errorsc C s d}t jddd d¡}tt ||¡t j|| dd|d d }t d¡ d¡}|d
|
}tt ||¡t j|| dd|d d S ) Nz type castr Úshortr5 r\ r r] rb ztype cast on one argumentrP )r r r^ r r rÇ r- r )r% ró rÖ r× r r r Útest_type_cast[ s ÿ ÿzTestUfunc.test_type_castc C sP d}t jddd d¡}tt ||¡t j|| dd|d d }t jdd
|
d d¡}tt ||¡t j|| dd|d t jd d d}t jd d d}t|| jt d¡ t|| jt d¡ t|| jt d¡ t|| jt d¡ tt |¡jt d¡ tt |¡jt d¡ tt
|
|¡jt d¡ tt
|
|¡jt d¡ d S )Nz
|
big endianr z>i4r5 r\ r r] rb z little endianz<i4r z>f8z<f8rÊ ) r r r^ r r rÇ r- r r+ ÚabsoluteÚnegative)r% ró rÖ ZBaZLar r r Ú test_endianf s( ÿ ÿzTestUfunc.test_endianc C sø d}t d¡ d¡}|d d
dd d
dd d
df }|d d
dd d
dd d
df }d|d< d}t|d d|d
|
tt ||¡t j|| dd |d
|
t d ¡ d dd¡}|j}|j}d|d< t|d d|d
|
tt ||¡t j|| dd |d
|
d S )Nz#incontiguous memory layout of arrayé@ )r r r r r r r r r r¦ z-make sure it references to the original array)r r r r r r rb r] é r rS r] ) r r r^ r r r rÇ r- ÚT)r% ró r: rÖ r× Zmsg2r r r Útest_incontiguous_array} s $z!TestUfunc.test_incontiguous_arrayc C s d}t d¡ d¡}t d¡ d¡d }t jddd }t |||¡ t|t j|| d
|
d |d d
|
|d d
< tj|||d t|t j|| d
|
d |d d}t jddd }t |||¡ t|t j|| d
|
d |d d
|
|d d
< tj|||d t|t j|| d
|
d |d d}t jddd }t |||d ¡ t|d t j|| d
|
d |d d
|
|d d
< tj|||d d t|d t j|| d
|
d |d d S )Nzoutput argumentrT ©r rS r r] r¬ r r\ r# r5 r r] rb rÐ z output argument with type castÚint16z(output argument with incontiguous layout©r rS r] ).r )r r r^ rÔ r rÇ r r- )r% ró rÖ r× rh r r r Útest_output_argument s. z TestUfunc.test_output_argumentc C s, t j}t d¡ d¡}t dd¡ d¡}|||}t||| d¡ |||dddgd }t||| d¡ |||dddgd }t||| d¡ |||dddgd }t||| d¡ |||ddgd }t||| d¡ |||t d¡tjdtj d
|
gd }t||| d¡ |||d d gd }t||| d ¡ |||d d gd }t||
|
d d d ¡| d¡ t t |||dd t t |||dgd t t |||ddgd t t |||ddgd t t |||d d gd t t |||dddgd t t |||dddgd t t |||dddgd t t |||d d gd t j}t d¡ d¡}t d¡ d¡d }|||}t|t ||¡ |||dddgd }t|t ||¡ |||dddgd }t|t ||¡ |||dddgd }t|t |
|
d d d ¡|
|
d dd d ¡¡ t d¡}||||dddgd}t||k t|t ||¡ t d¡}||||ddd gd}t||k t|t ||¡
|
dd d d ¡ t t |||d d t t |||d d t t |||d d t t |||g d t t |||dgd t t |||dddgd t t |||dddgd t t |||d dgd dgd dggd t t |||ddd dggd t t |||ddd gd t t |||ddd gd t d!¡}t t |||d t t ||||d d
d f d" t t ||d |d d gd t t ||||d d d gd t t tjd#d#d gd t t |||d d gd$d% d S )&Nç ;@©rS rS rS r+ ç 3@©rS r rS r r r )Úaxesr5 r r r ç ð¿)r ©r? r rT rx r5 ©r r r r ©r r r\ )r r rS )r r rS r )rÑ r )r r r rS )rS r r] )r r r r? )r? r? rd rÐ rô T)r r- )r rÇ r r r^ r r- Zint8rV Úint32Ú transposer r! r Úmatrix_multiplyÚmatmulrJ r rÔ r" )r% rÇ rÖ r× rh Úmmrõ Úzr r r Útest_axes_argument¨ s
|
$
|
ÿ
|
|
ÿ ÿ
|
z TestUfunc.test_axes_argumentc C s> t j}t d¡ d¡}t dd¡ d¡}|||}t||| d¡ |||dd}t||| d¡ t |¡}|||d|d}t||k t|| |||d d}t||| d ¡ t d
|
¡ d ¡}t d d ¡ d ¡}t dd¡ d ¡}tt j |||d dtj|| | d d tt j
|
|d dtj
|
|d d tt j
|
|ddtj
|
|dd t |¡}t j
|
||d d}t||k t|tj
|
|d d t j
|
||dd}t||k t|tj
|
|dd t t |||d d d gd t t |||d gd t j}t t |||dd tjd|jd}t tt j
|
||d d t t tjddd d d S )Nr| r} r+ r~ r r r] )rS rÑ r r r\ r4 r7 r é ©rÑ rS r )rS r ©r r rS r5 rô )r rÇ r r r^ r r- Ú
|
zeros_liker ÚinnerwtÚcumsumZ
|
empty_liker r! r rJ r+ r r" ) r% rÇ rÖ r× rh rÑ rõ rH r r r r Útest_axis_argument sF
|
|
|
ÿ
|
z TestUfunc.test_axis_argumentc C s t j}t d¡ d¡}t dd¡ d¡}|||}t||| d¡ |||dd}t||| d¡ |||d d}t||| jdd d t |¡}|||d |d
|
}t||k t|| |||ddd }t||| jddd |||dd d }t||| jdd d |||d dd }t||| jd dd |||d d d }t||| jd d d |||d d dgdd}t||| d¡ |||d d d gd d}t||| jdd d |||d d gdd}t||| d ¡ |||d d d gd d}t||| jd d d |||d dgdd}t|| ddd ¡| d¡ |||d dgd d}t|| ddd ¡| jdd d |||d ddgd d}t|| ddd ¡| jdd d |||d dd gd d}t||| dd d¡ jd d d |||d ddgd d}t|| dd d¡| d dd¡ jdd d t
|
d¡t d¡d d
tj tj f }t |¡}t j |dd}t|| t j |d d}t||d d
tj tj f t
|
d¡t d¡d d
tj tj f }t |¡\}} t j|dd\}
|
} t|
|
| t| | t j|d d\}
|
} t|
|
|d d
tj tj f t| | d d
tj tj f t d¡ d¡}t dd¡ d¡}t dd¡ d¡} tt j||| d dtj|| | dd d tt j||| d d d tj|| | d d d tt|||dd t j} tt| ||d d tt| ||dd tttjdddd d S ) Nr| r} r+ r~ r r F©ÚkeepdimsT)r rÑ ©rS r r r r )r r r r rS rV r r\ r4 r7 r r Útruerô )r rÇ r r r^ r r- r r r ÚeyeÚnewaxisÚumlÚdetZslogdetr r r! r r" )r% rÇ rÖ r× rh rÑ rõ r Z
|
expected_sZ
|
expected_lÚcsZclrH r r r r Útest_keepdims_argument; s
|
|
|
ÿÿ" ÿ&
|
|
&
|
|
ÿÿz TestUfunc.test_keepdims_argumentc C sº t d¡ d¡}t dd¡ d¡}t dd¡ d¡}tt |||¡t j|| | dd t d d
|
¡ d ¡}t d d ¡ d ¡}t dd¡ d ¡}tt |||¡t j|| | dd d S )Nr r\ r4 r7 r r r r] éd é| rz éÈ éà é, iD )r r r^ r r r r- ©r% rÖ r× rH r r r Ú test_innerwt s &zTestUfunc.test_innerwtc C sT t jg dd}t jg dd}t jg dd}tt |||¡t j|| | dd dS )z/Test generalized ufunc with zero-sized operandsrÊ r5 r r] N)r rV r r r r- r¢ r r r Útest_innerwt_empty s z TestUfunc.test_innerwt_emptyc C sâ t d¡}tt ||¡t d¡ t d¡}t |d ||¡}t||k t|t t d¡|d |d f¡ tt tjt d¡t d¡ tt tj|t
|
d¡ tt tj|t
|
d¡t d ¡ tt tj|t
|
d¡t d¡ d
|
S ) z Test with fixed-sized signature.rS ©rS rS r r r r] rV r )rS r] N) r r r r Zcross1drÔ r Zvstackr r r )r% rÖ rÑ Úresultr r r Ú test_cross1d s
|
|
& zTestUfunc.test_cross1dc C sR t d¡ d¡}t d¡}|d d
t jf }t d¡ d¡d }t ||¡}t ||¡}t|| tt tj|| t ||¡}t||
|
¡ t ||¡}t ||¡}t|| t d¡}|t jd d
f }t d¡ d ¡d } t ||¡}
|
t ||¡} t| |
|
tt tj|| t ||¡} t| |
|
|
¡ t | |¡}
|
t | |¡} t| |
|
|j }| d
|
d ¡}t ||¡} t ||¡}t|| tt tj|| t ||¡}t||
|
¡ t | |¡}t | |¡}t|| t |¡}tj| ||d }t|| d |d d
< tj| ||d }t|| t | ¡}tt tj||| t |||¡}t||
|
¡ d S )NrT rx r r5 r r rS ru )r] r r r rS r? r rÐ r )r r r^ r r r r r r r Úsqueezerv Úswapaxesr )r% ÚmatZ
|
single_vecZcol_vecZ col_vec_arrayZ
|
mm_col_vecZmatmul_col_vecZ
|
matmul_colZrow_vecZ row_vec_arrayZ
|
mm_row_vecZmatmul_row_vecZ
|
matmul_rowZmm_row_col_vecZmatmul_row_col_vecZmatmul_row_colZmm_row_col_arrayZmatmul_row_col_arrayrÑ r r r Útest_can_ignore_signatureª s`
|
|
|
|
|
|
|
|
|
|
|
|
ÿz#TestUfunc.test_can_ignore_signaturec C s | tj¡ | tj¡ d S r> )Ú compare_matrix_multiply_resultsr ræ rx r$ r r r Útest_matrix_multiplyñ s z TestUfunc.test_matrix_multiplyc C sT t t d¡t d¡¡}t|t d¡ t t d¡t d¡¡}t|t d¡ d S )N)r r4 )r4 r r )r4 r4 )r r r rH r rÔ r r r r Ú test_matrix_multiply_umath_emptyõ s z*TestUfunc.test_matrix_multiply_umath_emptyc s t jt j ddd¡|d}t jt j ddd¡|d}d|jj } fdd fdd d
|
d } d}dtd d d fd f }d }|D ]â} |D ]Ø}
|
|D ]Î} |D ]Ä} | | ¡| } | |
|
¡| }|oÐ| jd k }|oÞ|jd k }| j d |j d kr¤|| j d |j d r¤t
|
t | |¡t j |dt jf dd ¡| dt jd d
f d d|dt| j t|j f d q¤qqqt|d dd d S )Nr rS r] r5 zmatrix multiply on type %sc sp | dkrdgfS d} | d }|D ]F}t | D ]8}|| d g }|| || d < | d ||< ||f7 }q0q$|S )Nr r r )r )ÚnÚretÚbaseÚpermrÌ Únew)Ú permute_nr r r´ s z<TestUfunc.compare_matrix_multiply_results.<locals>.permute_nc sT | dkr dS d} | d }|D ].}||t d f f7 }||t ddf f7 }q |S )Nr )r r r )Úslice)r¯ r° r± Úsl)Úslice_nr r r· s z:TestUfunc.compare_matrix_multiply_results.<locals>.slice_nc S s | |kp| dkp|dkS rA r )Ús1Ús2r r r Ú broadcastable s z@TestUfunc.compare_matrix_multiply_results.<locals>.broadcastabler Tr? r .r@ r] z %s %srb zreference check)r rV ÚrandomZrandr+ r! rµ r r± Úshaper r r r- r r© Ústrr )r% ÚtpÚd1Zd2ró rº Z permute_3Zslice_3ÚrefÚp1Úp2r¸ r¹ Za1Za2r )r´ r· r r¬ û s@
|
ÿ
|
ÿÿ ÿüz)TestUfunc.compare_matrix_multiply_resultsc C sª t jdtd dd¡}t j|jd |jd d d f|jd}t ||¡ t t j
|
|d d
d f | d dd ¡}|t j |jd t d }t || tttj| d S )
|
NrT r5 r] rS r r r r r] )r r r r^ rJ r¼ r+ r Zeuclidean_pdistÚsqrtr- ZtriÚboolr r r )r% rÖ rÑ r× r r r Útest_euclidean_pdist1 s * (
|
z TestUfunc.test_euclidean_pdistc C s& t d¡}t |¡}t|| ¡ d S )Nr4 )r r r r r )r% rÖ r¦ r r r Ú test_cumsum; s
|
|
zTestUfunc.test_cumsumc C s t jdd ddddgtd}tt |d ¡t jdd |D td tt |d¡t jd d |D td tt |d
|
¡t jd d |D td tt |d ¡t jd d |D td tt |d ¡t jdd |D td tt |d¡t jdd |D td tt |d
|
¡t jdd |D td tt |d ¡t jdd |D td tt |¡t jdd |D td tt j |¡d tt j |¡d d S )NrS TFÚtestÚ r5 c S s g | ] }|pd qS r> r ©r r: r r r r C s z1TestUfunc.test_object_logical.<locals>.<listcomp>c S s g | ] }|pd qS ©Tr rÉ r r r r E s rT c S s g | ] }|pd qS ©rT r rÉ r r r r G s Úblahc S s g | ] }|pd qS ©rÌ r rÉ r r r r I s c S s g | ] }|od qS r> r rÉ r r r r L s c S s g | ] }|od qS rÊ r rÉ r r r r N s c S s g | ] }|od qS rË r rÉ r r r r P s c S s g | ] }|od qS rÍ r rÉ r r r r R s c S s g | ]
|
}| qS r r rÉ r r r r U s )r rV rI r Ú
|
logical_orÚ logical_andZ logical_notr r/ r r r Útest_object_logical@ s< ÿ ÿ ÿ ÿ ÿ ÿ ÿ ÿ
|
ÿz TestUfunc.test_object_logicalc C s G dd d}t | ¡}t||kd tt ||¡d t | g¡}t||kt dg¡ tt ||¡t dg¡ tt j||tdt dg¡ d S )Nc @ s e Zd Zdd ZdS )z8TestUfunc.test_object_comparison.<locals>.HasComparisonsc S s dS )Nú==r ©r% Úotherr r r Ú__eq__\ s z?TestUfunc.test_object_comparison.<locals>.HasComparisons.__eq__N)r/ r0 r1 rÔ r r r r ÚHasComparisons[ s rÕ Tr5 rÑ )r rV r ÚequalrI )r% rÕ Zarr0dZarr1dr r r Útest_object_comparisonZ s z TestUfunc.test_object_comparisonc C sP t jdddgtd}tt |¡d tt |¡d tt |¡d t jdddgtd}tt |¡d tt |¡d tt |¡d tt |¡d tt |¡d tt |¡d tt jd
|
ggtd ¡ d
|
tt jd
|
dgggtd d ¡d
|
dg tt jd
|
gtdjd
|
d d tt jd
|
gdd ggtdjd gddgdd dd g d S )NrÖ r× rh r5 ÚabcTFr r r ©r r rN rS )rO rX )
|
r rV rI r r- r ÚminÚprodÚanyrJ r/ r r r Útest_object_array_reductionh s( & ÿÿz%TestUfunc.test_object_array_reductionc C sä t jdtd}dd tdD |d d
< t jj||d t jj||d t|t jdd dD td t jdtd}d d tdD |d
|
d d
f< t jj||d d t jj||d d t|d
|
d d
f t jd d dD td d S )Nr] r5 c S s g | ]
|
}d gqS rA r ©r rÌ r r r r ~ s zBTestUfunc.test_object_array_accumulate_inplace.<locals>.<listcomp>rÐ c S s g | ]}d g| qS rA r rÞ r r r r s )r rS r r4 ©r r] c S s g | ]
|
}d gqS ©r r rÞ r r r r s r r r c S s g | ]}d g| qS rà r rÞ r r r r s )r rH rI r r" Ú
|
accumulater rV )r% rZ r r r Ú$test_object_array_accumulate_inplace{ s ÿ ÿz.TestUfunc.test_object_array_accumulate_inplacec C sb t j t jdddgtd¡}t|t jdddgtd t t¡ t j dd dg¡ W 5 Q R X d S )Nr r r r5 rS ) r r" rá rV rI r r{ rÜ r! r r r r Ú$test_object_array_accumulate_failure s z.TestUfunc.test_object_array_accumulate_failurec C s$ t jdtd}dd tdD |d d
< t jdtd}dd tdD |d d
< t jj|t d¡|d t jj|t d¡|d t|| t jdtd}dd tdD |d d d
f< t jdtd}d
|
d tdD |d d d
f< t jj|t d¡|d d t jj|t d¡|d d t|| d S ) Nr] r5 c S s g | ]
|
}d gqS rA r rÞ r r r r s z@TestUfunc.test_object_array_reduceat_inplace.<locals>.<listcomp>c S s g | ]
|
}d gqS rA r rÞ r r r r s rÐ rß c S s g | ]
|
}d gqS rà r rÞ r r r r £ s r c S s g | ]
|
}d gqS rà r rÞ r r r r ¥ s r r ) r rJ rI r r" r r r rH rj r r r Ú"test_object_array_reduceat_inplace s
|
z,TestUfunc.test_object_array_reduceat_inplacec C sl t j t jdd dgtdddg¡}t|t jd dgtd t t¡ t j dd dgddg¡ W 5 Q R X d S )Nr r r5 r ) r r" r rV rI r r{ rÜ r! r r r r Ú"test_object_array_reduceat_failureª s " z,TestUfunc.test_object_array_reduceat_failurec C sx g t jg tdfD ]`}tt |¡d tt |¡d tt |¡d tt |¡d tt t j
|
| tt t j | qd S )Nr5 r r FT) r rV rI r r- rÛ rÜ rJ r r r rÚ r/ r r r Útest_zerosize_reduction³ s z!TestUfunc.test_zerosize_reductionc C s t ddg¡}tt j|jdd t ddg¡}tt j|jdd t ddg¡}tt j|jdd t ddg¡}tt j|jdd d S )NFr r] r? )r rV r Z AxisErrorrJ rÜ r/ r r r Útest_axis_out_of_bounds½ s z!TestUfunc.test_axis_out_of_boundsc C sj t tjdddd t tjdddd t tjdddd t tjdddd t tjdddd t tjdddd t tj d¡d t
|
t tjt d¡ddtj k t
|
t tjt d¡ddtj k t
|
t tjt d¡ddtj k t
|
t tjt d¡ddtj k t
|
t tjdddtj k G dd d tj}t d
|
¡ |¡}t
|
t t |¡|k d S ) NrS r r] ç @TFg @c @ s e Zd ZdS )z0TestUfunc.test_scalar_reduction.<locals>.MyArrayN©r/ r0 r1 r r r r ÚMyArrayß s rê r )r r r- rÛ rÜ rJ r rÚ rÿ r r rr r÷ rC ÚndarrayrV Úview)r% rê rÖ r r r Útest_scalar_reductionÈ s """" z TestUfunc.test_scalar_reductionc C s| t dt j¡}t dt j¡}t dt j¡}t j|||d t|d t d¡}t jdt jd}t j|d|d t|d d S )N)r r rÐ r i r5 r ) r rH ræ rn r" r rÔ r÷ Úsubtract©r% rÖ r× rh r r r Útest_casting_out_paramä s
|
|
z TestUfunc.test_casting_out_paramc C sÎ t d¡}t d¡}t d¡}t j||||d dkd t|dddddddg t d¡ dd¡d }t j|ddg|ddgddggd t|dd gd
|
d gg t j|d|d d gd t|dd gdd gg d S )Nr r r ©rÑ rX r r] r rS é r7 r TFé ) r r rH rÔ r" r r^ rÿ rî rï r r r Útest_where_paramñ s
|
|
|
"zTestUfunc.test_where_paramc C sv t dt j¡}t dt j¡}dt dt j¡ }t j|||ddddddddddg
|
d t|ddddddddddg
|
d S )Nr4 rË r r rñ r )r rH ræ rn r" r rï r r r Ú test_where_param_buffer_output s
|
&z(TestUfunc.test_where_param_buffer_outputc C st t jdgt jd}t jdgtd}tt j||ddg t jdgt jd}t jdgtd}tt j||ddg d S )Nr r5 TrW )r rV ræ rÄ r rà rn ©r% rÖ r r r r Útest_where_param_alloc s z TestUfunc.test_where_param_allocc C s t j d¡}t j d¡}|dk}t |d¡}t j||||d t ||j¡| }t|| |k || t¡ || ¡ r~t
|
d S )N)é r] )rø r rQ r )rX rÑ ) r r» Z full_likeÚlessZ broadcast_tor¼ r rs rÄ rÜ rX )r% rÖ r× rX rÑ Zb_wherer r r Ú test_where_with_broadcasting s z&TestUfunc.test_where_with_broadcastingc C s@ d|d< d|d< t tjj|d dd t tjj|ddddddg t tjj|dddddg t tjj|ddddg t tjj|ddddddgddddgddddgg t tjj|ddddddgddddgg t tjj|d ddddgdddgg t tjj|d
|
d| d|d< d|d < t tjj|d dd t tjj|ddddddg t tjj|dddddg t tjj|ddddg t tjj|ddddddgddddgddddgg t tjj|ddddddgddddgg t tjj|d ddddgdddgg t tjj|d
|
d| d|d< d|d < t tjj|d dd t tjj|ddddddg t tjj|dddddg t tjj|ddddg t tjj|ddddddgddddgddddgg t tjj|ddddddgddddgg t tjj|d ddddgdddgg t tjj|d
|
d| d S ) Nr .r )r r r r] rÙ rR r r r )r r r )r r r )r r Úminimumr r/ r r r Ú check_identityless_reduction" s` ÿÿÿ ÿÿÿ ÿÿÿz&TestUfunc.check_identityless_reductionl c C sj t dd¡}ddg|d d
df< ddg|d d
df< t jj|dd}~|d dksVt|d dksftd S ) N)r l Úuint8r rS r r] r r] )r rÔ rÛ r rX rY r r r Ú&test_identityless_reduction_huge_arrayP s z0TestUfunc.test_identityless_reduction_huge_arrayc C s t jddd}| |¡ d S )Nrz ÚC©Úorder©r rJ rü r/ r r r Ú"test_identityless_reduction_corder[ s z,TestUfunc.test_identityless_reduction_corderc C s t jddd}| |¡ d S )Nrz ÚFr r r/ r r r Ú"test_identityless_reduction_forder_ s z,TestUfunc.test_identityless_reduction_forderc C s$ t jddd dd¡}| |¡ d S )N)r r] rS rÿ r r r ©r rJ r© rü r/ r r r Ú&test_identityless_reduction_otherorderc s z0TestUfunc.test_identityless_reduction_otherorderc C s> t jddd dd¡}|dd
dd
dd
f }| |¡ d S )N)rS r r] rÿ r r r r r/ r r r Ú%test_identityless_reduction_noncontigg s z/TestUfunc.test_identityless_reduction_noncontigc C sP t jddd}|dd
jdd}d|_|dd
dd
dd
f }| |¡ d S )N)iá Úi1r5 r rÊ )rS r] r )r rJ rì r¼ rü r/ r r r Ú/test_identityless_reduction_noncontig_unalignedl s
|
z9TestUfunc.test_identityless_reduction_noncontig_unalignedc C s t tjjg ddd t tjjg tjdtj t tjjg tj dtj t tjjdgddd t tjjdgddd t tjjdgddd t tjjdgddd tttjjg d d tttjjg d d t tjjg tj dd tj
|
dgt d}tjj|dd}t |d d S )Nr rN r r] r4 r5 r6 ) r r rÛ r rû Úinfr r r" Z_NoValuerV rI ©r% rÖ r[ r r r Útest_initial_reductions s z TestUfunc.test_initial_reductionrS )r r NrX FTc C sh t d¡ dd¡}| ¡ }t |¡}t j|||d t jj|||d}| |¡}t || t
|
|| d S )NrZ rS rñ ©rS rX ) r r r^ Úcopyr rm r" r r- r r )r% rS rX rÖ Úa_copyÚa_checkr[ Úcheckr r r Útest_reduction_with_where s
|
|
|
z#TestUfunc.test_reduction_with_wherer r )NTrO rY c C sl t d¡ dd¡}| ¡ }t |jt j ¡}t j|||d t jj ||||d}|j
|
||d}t || d S )NrZ rS rñ )rS rX rO rN ) r r r^ r r7 r¼ r rm rÛ r r r ) r% rS rX rO rÖ r r r[ r r r r Ú%test_reduction_with_where_and_initial¢ s z/TestUfunc.test_reduction_with_where_and_initialc C s4 t d¡ dd¡}dddg}ttt jj||d d S )NrZ rS FTrW )r r r^ r r rÛ r rö r r r Ú#test_reduction_where_initial_needed± s
|
z-TestUfunc.test_reduction_where_initial_neededc C s t dddgdddgg¡}t jj|dd}t|dddg t jj|d d}t|ddg t jj|d
|
d}t|| ttt jj|d d d S ) Ng @r rô rÉ g Ð?r r] rV r r rÙ )r rV Údivider r r r r r r r Ú*test_identityless_reduction_nonreorderable¶ s
|
z4TestUfunc.test_identityless_reduction_nonreorderablec C sd dd }dd }dd }t tjjd ||tjjdd ||tjjdd ||tjjdd ||tjjdd |tjjg tjjt d ¡d
|
d ||tjjdd ||tjjdd ||tjjdd ||tjjdd tj g ¡ tjjt d ¡d
|
d tjtjfD ]V}|jt d ¡dd |jt d ¡dd |jt d¡dd |jt d ¡dd qd S )Nc _ s | || d S r> r ©r9 Úargsr
|
r r r ÚokÒ s z+TestUfunc.test_reduce_zero_axis.<locals>.okc _ s t t| f|| d S r> )r r r r r r ÚerrÕ s z,TestUfunc.test_reduce_zero_axis.<locals>.errc
|
S sp | |t ||f¡dd | |t ||f¡dd | |t |d |d |f¡dd | |t |d ||d f¡dd | |t ||d |d f¡dd | |t |d ||d f¡dd | |t |d |d |d |d |d f¡dd | |t d ||f¡d
|
d | |t d ||f¡dd | |t |d |f¡dd | |t d ||f¡dd | |t d ||f¡dd d S ) Nr r] r r r rR rS )r r r r4 rÙ )r rÔ )ÚexpectÚfuncr¯ r r r r ÚtØ s" """" ÿþz*TestUfunc.test_reduce_zero_axis.<locals>.té r r r r] )r r )r r )r r )r r rÛ Úidentityr rÔ r" rá )r% r r r r r r r Útest_reduce_zero_axisÄ s* z TestUfunc.test_reduce_zero_axisc C s t jdddgtd}ttt j|ddddg ttt j|d|d d
|
d }tt||d tt j|d|d d t|dddg d S )Nr r rS r5 gñ?gÍÌÌÌÌÌ @gÍÌÌÌÌÌ@gffffff@rÐ c S s | |7 } d S r> r )rÖ r× r r r Ú add_inplace s z0TestUfunc.test_safe_casting.<locals>.add_inplacerÍ ©rÑ rÎ r] )r rV r# r r r" r r! )r% rÖ r" r r r Útest_safe_casting s ÿzTestUfunc.test_safe_castingc C sÈ t jdddgdd}t jdddgdd}t jdtjd}t |||¡}t jdddgtjd}t|| t ||¡}t|| tj|| t j¡|d}t|| t t
|
t |t d¡¡ W 5 Q R X d S ) Nr r r Úi8r5 rS r] rÐ ) r rV rJ r Úrationalr r rs Úuint16r r! )r% rÖ r× rh r¦ Útargetr r r Útest_ufunc_custom_out s
|
|
|
|
z TestUfunc.test_ufunc_custom_outc
|
C s t jddd dd¡}t jddd dd¡}t |d d
d d
f |¡ t|t jdd ddgd
|
dd d
|
gd dd d gddd dggdd t d¡}t |d¡ t|d t |ddg¡ t|d d S )Nr7 Úlr5 r] r» rS r r r r é ró é rT r, r6 r4 )r r r^ Ú opflag_testsZ inplace_addr rV rÕ r r r Útest_operand_flags4 s
|
|
ÿÿ
|
|
z TestUfunc.test_operand_flagsc C sd dd l m m} tjdgdd}tjdgdd}| ||¡}t|tjdgdd tt|j d S )Nr r zu8,u8,u8r5 )r r] r )
|
Z numpy.core._struct_ufunc_testsr_ Z_struct_ufunc_testsr rV Z add_tripletr r Ú RuntimeErrorZ register_fail)r% Z struct_ufuncrÖ r× r¦ r r r Útest_struct_ufuncA s zTestUfunc.test_struct_ufuncc C s¢ t jt dd¡t dd¡t dd¡gtjd}t jt dd¡t dd¡t dd¡gtjd}t ||¡}t jt d¡t dd¡t dd¡gtjd}t|| d S )Nr r rS r] r5 )r rV r r& Ztest_add_rationalsr )r% rÖ r× r¦ r r r r Útest_custom_ufuncK s.
|
|
|
þü
|
|
|
þü
|
|
þüzTestUfunc.test_custom_ufuncc C s6 t t$ tjt d¡dtjtd fd W 5 Q R X d S )Nr rÚ )r r! r Úmultiplyr r& r# r$ r r r Ú test_custom_ufunc_forced_sig_ s
|
|
ÿz&TestUfunc.test_custom_ufunc_forced_sigc sL G fddd t d¡ d t jdk j t jdk j d S )Nc s@ e Zd ZdZdZdZdd Zdd Z fddZ fd d
|
Z d S ) z1TestUfunc.test_custom_array_like.<locals>.MyThingéè r c S s
|
|| _ d S r> ©r¼ )r% r¼ r r r Ú__init__m s z:TestUfunc.test_custom_array_like.<locals>.MyThing.__init__c S s
|
| j d S ©Nr r5 r$ r r r Ú__len__p s z9TestUfunc.test_custom_array_like.<locals>.MyThing.__len__c sJ j d7 _ t|ts |f}t|| jkr4td | jt|d
S )Nr Zboo)Ú getitem_countr ÚtuplerF ÚndimÚ
|
IndexErrorr¼ )r% rÌ ©ÚMyThingr r Ú __getitem__s s
|
z=TestUfunc.test_custom_array_like.<locals>.MyThing.__getitem__c s j d7 _ | S rA )Ú
|
rmul_countrÒ r= r r Ú__rmul__| s z:TestUfunc.test_custom_array_like.<locals>.MyThing.__rmul__N)
|
r/ r0 r1 Z__array_priority__r@ r9 r6 r8 r? rA r r= r r r> g s r> r r¥ r r )r rn r r@ r9 r$ r r= r Útest_custom_array_likee s z TestUfunc.test_custom_array_likec C sê t d¡}t j |dddgd¡ t|dddddddd d
|
d g
|
t d¡}t d d d g¡}t j |dddg|¡ t|ddd ddddd d
|
d g
|
t d ¡ dd¡}t d d d gdddgdddgg¡}t j |td dddgf|¡ t|dddgdddgdddgg t d¡ ddd¡}t d ddg¡}t j |td td dddgf|¡ t|ddd gdddgdddggd dd gd d d gd d!d"ggd#d$d%gd&d'd(gd)d*d+ggg t d ¡ dd¡}t d d d gdddgdddgg¡}t j |dddgtd f|¡ t|dddgd,dd-gd.d/dgg t d¡ ddd¡}t d ddg¡}t j |td dddgtd f|¡ t|dddgd0dd1gd2d/dggd dd3gd4d d5gd6d7d8ggd#d9d:gd;d'd<gd=d>d?ggg t d ¡ dd¡}t d ddg¡}t j |ddddgf|¡ t|ddd gdddgdd d
|
gg t d¡ ddd¡}t d ddg¡}t j |dddgdtd f|¡ t|dddgdddgdd d
|
ggd@ddAgd dBdCgd dDdEggdFdGdHgd&dIdJgd)dKdLggg t d¡ ddd¡}t d ddg¡}t j |td td td f|¡ t|d ddMgdNdOdPgd2d/dggdQdRdSgdTdUdVgd6d7d8ggdFdGdHgdWdXdYgd=d>d?ggg t d¡}t j |dddg¡ t|ddddddZdd d
|
d g
|
t d¡}t j |d[d¡ t|d t t
|
t jj|dd t t
|
t jj|g d t d¡}t j |ddddgd\¡ t|t ddd]d^dddd d
|
d g
|
¡ t d¡}|d dk}t j ||ddddd
|
g¡ t|dddddddd dd g
|
t jdd_d`}t j |dddg¡ t|ddddddadd d
|
d g
|
t d¡}|d d
d f d d
dd
f }t j |ddgd¡ t|t d¡ t dddgt db¡ ¡ ¡}t ddddgt dc¡ ¡ ¡}t j ||d¡ t|dd
|
ddg t jdddgtd`}t tt jj|ddgd t|t jdddgtd` t tt jjt d¡dg t dddg¡}t j |dgd¡ tt dddg¡| d S )eNr4 r r r r r] rS r r r5 r» r éÊ éi r r¡ éÉ éf i éÍ i_ i4 rò i i éÐ i éÓ rT i éÖ r6 i éÙ r, i£ éÜ é i¦ éß ru i© éâ i i éÎ éÏ éË i] éj r+ éÔ if és éØ i= r8 r éÝ io r éá iF éÑ ic r, ró r7 é év éÛ i@ é é é r i. ég éÌ i1 ém éÒ i7 ép éÕ i: éy éÞ iC éûÿÿÿr rè i> é. Zu4r5 l úÿ rÌ r9 rÖ )r r r" Úatr rV r^ rµ rr r r< rÿ rÖ Úinvertr r+ rò rI r! r ÚmodfrÛ )r% rÖ r× ÚindexÚorigÚvaluesr r r Útest_inplace_fancy_indexing sø
|
|
" "$þþþøÿ " "$þþþøÿ " þþþøÿ "þþþøÿ
|
|
|
|
$
|
|
z%TestUfunc.test_inplace_fancy_indexingc C s^ t d¡}t d¡}ttt jj|dg| t ddgddggg¡}ttt jjj j|dg d S )Nr® ra r r r rS r] )
|
r rH r r! r rj rV ÚlinalgÚ _umath_linalgr rÕ r r r Útest_at_not_none_signature s
|
|
|
z$TestUfunc.test_at_not_none_signaturec C sB t jj}t jdtd}t jd|jd}|d }t||| t||dd| t||d| t||dd d| t||ddd| t||dd| t||dd | t||dd d d| t||dd |d| t||dd || t||dd d | t||dd d d d
|
| t||dd d d | d |j ¡ t||dd d d d| t||dd d d dd | t||dd d d dd | t||dd d d dd d| t||dd d d d| t||dd d d d| t||dd d d d| t||dd d d dd d| t t
|
| t t
|
||dd d d dd d
|
t t
|
||d t t
|
||dd t t
|
||dd d d t t
|
||dd t t
|
||dd t t
|
||dd d t t
|
||dd d t t
|
||dd t t
|
||dd d t t
|
||dddd d t t
|
||dd dd t t
|
||dd t t
|
||dd dd d t t
|
||dd d d dd t t
|
||dd d dd d S ) N)r r r5 rà r r r] rÌ rÐ Fr TrA rN rW )rS r+ rÑ r )r+ rÑ r )rÑ r )rÑ r rO rX r r )rS r+ r ©r+ rÑ ©rÑ r+ )r r+ rÑ )rS r+ r ©r )r r rÑ )rS r+ r rÑ r )rS r+ rÑ r ) r r" r rH r# r+ r r^ r¼ r r! )r% r9 rõ ÚoÚrr r r Útest_reduce_arguments st " ÿÿ
|
ÿÿÿ ÿ ÿz TestUfunc.test_reduce_argumentsc C st G dd dt j}t d¡ dd¡}|jdd ¡ }| |¡}t jdd d d gtd}tt |||d
|
kk¡d d S ) Nc @ s e Zd Zdd ZdS )z,TestUfunc.test_structured_equal.<locals>.MyAc _ s t ||dd |D |S )Nc s s | ]}| tj¡V qd S r> )rì r rë )r Úinputr r r Ú <genexpr>a s ÿzOTestUfunc.test_structured_equal.<locals>.MyA.__array_ufunc__.<locals>.<genexpr>)r^ )r% r ÚmethodÚinputsr
|
r r r Ú__array_ufunc__` s
|
ÿÿz<TestUfunc.test_structured_equal.<locals>.MyA.__array_ufunc__N)r/ r0 r1 r~ r r r r ÚMyA_ s r ç (@r] rS zf8,f8,f8r5 TFr )
|
r rë r r^ rì r¨ rV rÄ r rJ )r% r rÖ ÚraZmrar( r r r Útest_structured_equal\ s
|
z TestUfunc.test_structured_equalc C sL t d¡}t d¡}t||k t||k t||k t||k d S )Nr rÖ )r rV r rÕ r r r Útest_scalar_equalj s
|
|
zTestUfunc.test_scalar_equalc C s¶ t jt jt jt jt jt jt jt jt j t j
|
t j t j t j t jt jt jt jt jt jt jt jt jt jt jt jt jt jg}t d¡}d}t ddg¡}|D ] }t t ||| t t ||| qd S )NÚ1r rô r ) r rÿ r" rî r2 r r Z floor_divideZ bitwise_andZ
|
bitwise_orZ bitwise_xorZ
|
left_shiftZ right_shiftZfmaxZfminÚfmodÚhypotZ logaddexpZ
|
logaddexp2rÛ rû ÚmodZgreaterZ greater_equalrù Z
|
less_equalrÖ Ú not_equalrV r r! )r% Z binary_funcsrÖ r× rh r9 r r r Ú test_NotImplemented_not_returnedt sD ù
|
z*TestUfunc.test_NotImplemented_not_returnedNc C s4 t jdd dgtd}||||d}|jtks0td S )NTFr5 rÚ )r rV rI r+ rX )r% r r( rÖ r[ r r r Ú%test_logical_ufuncs_object_signatures s z/TestUfunc.test_logical_ufuncs_object_signaturesc C s8 t dd dg¡}t t¡ ||||d W 5 Q R X d S )NTFrÚ )r rV r{ rÜ r! )r% r r( rÖ r r r Ú+test_logical_ufuncs_mixed_object_signatures s z5TestUfunc.test_logical_ufuncs_mixed_object_signaturesc C sâ t jddd}t ddg¡}t||||ddgd | |¡dksHtt jdt jd}|ddgd |j¡}t||||d| t jd t jd}|j||ddks¨tt jd
|
gd d}t jd |jd}|j||dd ksÞtd S ) Nó 1ZV3r5 rô r Tr rÐ r rS rÌ r ) r rV r r rX rÔ r
rs r+ )r% r rÖ rh rÑ r r r r Ú$test_logical_ufuncs_support_anything s z.TestUfunc.test_logical_ufuncs_support_anythingc C sV t jtdd |dgdg W 5 Q R X t jtdd | dddg¡ W 5 Q R X 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 signaturerð r Ú3Ú2Ú0N)r{ rÜ r! r )r% r r r r Ú!test_logical_ufuncs_reject_string° s z+TestUfunc.test_logical_ufuncs_reject_stringc C sJ t d¡}t ddg¡}| ¡ }t t¡ ||||dd W 5 Q R X d S )Nr rô r r r# )r rV r r{ rÜ r! )r% r rÖ rh rÑ r r r Ú"test_logical_ufuncs_out_cast_checkÁ s
|
|
z,TestUfunc.test_logical_ufuncs_out_cast_checkc C sx t jddd}t jddd}t j |¡t j |¡ks8ttt j |¡t j |¡ tt j |dg¡t j |dg¡ d S )Nr4 z>i8r5 z<i8r )r r r" r rX r rá r )r% Zarr_beZarr_ler r r Ú$test_reducelike_byteorder_resolutionË s ÿz.TestUfunc.test_reducelike_byteorder_resolutionc C s² t jdt jd}t jdt jd}t jj||ddks8td|d d
< t jj||ddks\tt j dd t j
|
d}t jdt j d}t jdt j d}t jj||t j d
|
||ks®td S ) Nr4 r5 r rÐ r r4 r; r iÿÿÿru ) r rH rý rÔ r' r" r rX r2 r7 ræ r÷ )r% rZ rÑ r[ Z
|
single_resr r r Ú test_reducelike_out_promotesÖ s z&TestUfunc.test_reducelike_out_promotesc C sÎ t jddd}t jd|j ¡ d}t j |¡}t jj||d t|| t jd|j ¡ d}t j |ddg¡}t jj|ddg|d t|| t j|j |j ¡ d}t j
|
|¡}t jj
|
||d t|| d S ) Nr rÊ r5 r rÐ r r r ) r rH rJ r+ rò r" r r r r¼ rá )r% rZ rÑ r r r r Ú+test_reducelike_output_needs_identical_castí s
|
|
z5TestUfunc.test_reducelike_output_needs_identical_castc C sð t jdt jd ddd¡}|dd
dd d
dd
|
f dd d ¡}t jd t jd dd¡}|d d d
d d
f }| ¡ }t jj|| ¡ d d}t jj||d d}t|| t|dd d
f |dd d
f t|dd d
f |dd d
f d S )NiØ r5 r r, r5 r] r r r+ r r r r7 r rS ) r r ry r^ r r r" r r )r% r: Zy_baser; Z y_base_copyZr0Zr1r r r Ú test_reduce_noncontig_output s &
|
"z&TestUfunc.test_reduce_noncontig_outputÚ with_castc C s t d¡}t d¡}|r$| t j¡}tjtdd t jj|ddg|d W 5 Q R X tjtdd t jj ||d W 5 Q R X d S )Nr rS z (shape|size)rð r rÐ )
|
r r rs rn r{ rÜ r r" r rá )r% r rZ rÑ r r r Ú/test_reduceat_and_accumulate_out_shape_mismatch s
|
|
z9TestUfunc.test_reduceat_and_accumulate_out_shape_mismatchÚ out_shaperA r¶ ©r r )r rS r Úf_reducec C s t d¡ dd¡}t ||j¡}||d|d}||jkr^tt ||d||d W 5 Q R X n&||d||d}t||k t || d S )Nr r] rS r r ©rS rÑ r )
|
r r r^ rJ r+ r¼ r r r r )r% r r r rÖ rÑ Z correct_outr r r r Ú"test_reduce_wrong_dimension_output& s
|
|
z,TestUfunc.test_reduce_wrong_dimension_outputc C s² t d¡}t d¡}t d¡}t jj|d|dd t jj|d|d d
df dd tt t jj|d|dd W 5 Q R X tt& t jj|d|d d
df dd W 5 Q R X d S ) N)r r4 r )rS r r Tr r F)r rH rJ r" r r r )r% rÖ Z out_correctZ out_incorrectr r r Ú+test_reduce_output_does_not_broadcast_input9 s
|
|
|
|
|
z5TestUfunc.test_reduce_output_does_not_broadcast_inputc C sv G dd dt j}t d¡}t jjt d¡|d | |¡}t jjt d¡|d|ksZttt j |¡|ksrtd S )Nc @ s e Zd ZdS )z7TestUfunc.test_reduce_output_subclass_ok.<locals>.MyArrNré r r r r ÚMyArrG s r r r rÐ ) r rë rJ r" r rH rì rX rr )r% r rÑ r r r Ú test_reduce_output_subclass_okF s
|
|
z(TestUfunc.test_reduce_output_subclass_okc C s t dtjjk d S )NÚ
|
)r r Zinner1d_no_docrv r$ r r r Útest_no_doc_stringP s z TestUfunc.test_no_doc_stringc C s" t ttjd ¡}| d¡s td S )Nz loop of ufunc does not support)r{ rÜ r! r rà rñ rX )r% rt r r r Útest_invalid_argsT s zTestUfunc.test_invalid_argsÚnatc C s, zt |¡rtW n tk
|
r& Y nX d S r> )r r% rX r! ©r% r¤ r r r Útest_nat_is_not_finiteZ s z TestUfunc.test_nat_is_not_finitec C s, zt |¡stW n tk
|
r& Y nX d S r> )r r$ rX r! r¥ r r r Útest_nat_is_nana s zTestUfunc.test_nat_is_nanc C s, zt |¡rtW n tk
|
r& Y nX d S r> )r ÚisinfrX r! r¥ r r r Útest_nat_is_not_infh s z TestUfunc.test_nat_is_not_inf)r/ r0 r1 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) r0 Úskipifr rI rM rU r[ r_ ri rk rJ rH rn rp rs rw 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 rV r r r r r r! r$ r) r. r0 r1 r3 rB rp rs ry r r r rÏ rÎ rI r rF rÄ r r r r r r r r r r rû r r r r¢ r£ Z
|
datetime64Z timedelta64r¦ r§ r© r r r r r ½ sx T
|
|
|
|
þ
|
þ
|
|
|
|
W
|
" + ýü
|
g,O
|
G6
|
|
.
|
|
þþ þÿ
|
A
|
@
|
|
ÿÿÿ ÿÿÿ ÿ
|
ÿ
|
ÿ
|
|
ÿ
|
"""r r c C s( g | ] }t tt|tjrtt|qS r ©r r^ r r rÉ r r r r p s ÿc
|
C sÌ | j D ]À}d|ksd|kr q| d¡\}}dd |D }tjdd t d¡ | | }W 5 Q R X t|tr²t|}t|t|kstt ||D ] \}}|j
|
t
|
|¡kstqq|j
|
t
|
|¡kstqd S )
|
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
|
rU ú?ú->c S s g | ]}t d |¡qS )r¥ )r rH ©r r r r r r s z$test_ufunc_types.<locals>.<listcomp>Tr< r> N) r ÚsplitrB rC Úfilterwarningsr r: rF rX r r+ r ) r ÚtypÚinprÑ r r[ Zoutsrx r r r r Útest_ufunc_typesp s
|
|
|
|
r³ c C s( g | ] }t tt|tjrtt|qS r r« rÉ r r r r s ÿc
|
C s" | j D ]}ttdt|@ r"q| d¡\}}dd |D }dd |D }|D ]}tdd|_qP|D ]}tdd|_qftjdd t d
|
¡ | | }| | }W 5 Q R X t |dkrÂ|f}|f}t
|
||D ]N\} }
|
| j } t | t j¡rt | ¡j} d | } t||| | d qÌt| |
|
qÌqd S )zs
|
Check that contiguous and non-contiguous calls to ufuncs
|
have the same results for values in range(9)
|
zO?mMr c S s g | ]}t d |¡qS )r ©r rJ r® r r r r s z,test_ufunc_noncontiguous.<locals>.<listcomp>c S s" g | ]}t d |¡ddd
qS )r, NrS r´ r® r r r r s r r Tr< r> r )r r N)r rÜ Úsetr¯ r ZflatrB rC r° rF r r+ r Z
|
issubdtypeZfloatingr Zepsr
|
r )r r± r² rÑ Zargs_cZargs_nrÖ Zres_cZres_nZc_arZn_arrä Zres_epsZtolr r r Útest_ufunc_noncontiguous s0
|
r¶ c C s\ t dgd¡ d¡}t |¡s"t| jdkr6| | n"| jdkrP| || ¡ ntdd S )Nl r% rÊ r r z ufunc with more than 2 inputs)r rV rì r$ rX r r r )r r× r r r Útest_ufunc_warn_with_nan³ s
|
|
|
r· zPython lacks refcountsr2 c C sö d} t j| gtt jd dg | gtdt j td}t jt|t jd}t | ¡}t
|
t ¡ t j |||dd W 5 Q R X |t | ¡kst|d dks¤tt
|
t ¡ t j |||t jdd W 5 Q R X |t | ¡ksât|d dksòtd S )
|
Né{ rË Ústringr5 rÍ r# r r )rÑ r+ rÎ )r rV r# ÚBUFSIZErI rH rF r ÚsysÚ getrefcountr{ rÜ r r" rX )ÚvaluerZ rÑ Úcountr r r Útest_ufunc_out_casterrorsÄ s$ ÿþþ
|
r¿ Ú
|
bad_offsetrË c C s` d}t j|g| dg |gtdt j td}t t¡ t j||t j dd W 5 Q R X d S )Nr¸ r¹ rË r5 rÍ rþ )
|
r rV r# rº rI r{ rÜ r r" r ©rÀ r½ rZ r r r Útest_ufunc_input_casterrorsä s ÿþþ r r1 c
|
C sv d}t |g| t jg |gtdt j ¡}t jdd0 t t¡ t j ||t j
|
dd W 5 Q R X W 5 Q R X d S )Nr¸ rË rl rv rÍ rþ ) r rV Únanr# rº rp r{ rÜ ÚFloatingPointErrorr" r rÁ r r r Ú$test_ufunc_input_floatingpoint_errorï s ÿþrÅ c C s6 t jtdd tjt dd¡ddd W 5 Q R X d S )Nzcast ufunc 'add' input 0rð r zi,irS zdd->drÚ )r{ rÜ r! r r" rV r r r r Ú test_trivial_loop_invalid_castû s ÿrÆ Úoffsetr c C s¤ d}t j|g| dg |gtdt j td}t jdt jd}t |¡}tj t
|
dd t j j |t j|d W 5 Q R X |t |¡kst |d || k s t d S )
|
Nr¸ r¹ rË r5 r zinvalid literalrð rt r )r rV r# rº rI r r» r¼ r{ rÜ r r" r rX )rÇ r½ rZ rÑ r¾ r r r Útest_reduce_casterrors
|
s ÿþþ
|
rÈ r| c C s t j | dg¡S r7 )r r" r rí r r r rc !
|
rd rc r rî c C s t j | dg¡S )Nr )r Úlogrj rí r r r rc "
|
rd rj c
|
C s t t jdt j g¡}t jdd( tjtdd | | W 5 Q R X W 5 Q R X t t jdt j g¡}t jdd$ t t¡ | | W 5 Q R X W 5 Q R X d S )Nr Úwarnrm z invalid valuerð rl ) r rV r rp r{ rÒ rE rÜ rÄ )r| rZ r r r Ú test_ufunc_methods_floaterrors
|
s rË c C s8 | dkr dS t | j¡s dS | jjdkr4t | j¡S dS )Nr Frh T)r ÚsignbitrK r+ ÚkindrL )r½ r r r Ú_check_neg_zero0
|
s rÎ r+ ZAllFloatc C sl t | ¡} | jdkr&| tdd¡}n
|
| d¡}t |¡}t |¡}t|| sTt||7 }t|shtd S )Nrh ç )r r+ rÍ rr rW rV rÎ rX )r+ Úneg_zerorZ Zarr2r r r Útest_addition_negative_zero9
|
s
|
|
|
|
|
rÑ Ú use_initialTFc C s¼ t | ¡} | jdkr&| tdd¡}n
|
| d¡}i }|rB||d< n
|
t d¡ tddD ]`}t j|g| | d}t j |f|}|dks|rt
|
|s¶t qVt |j ¡r¦t t |j¡rVt qVd S )Nrh rÏ rO z1-0. propagation in sum currently requires initialr é r5 )r r+ rÍ rr rW r{ Zxfailr rV r- rÎ rX rÌ rK rL )r+ rÒ rÐ r
|
rÌ rZ r[ r r r Ú"test_addition_reduce_negative_zeroJ
|
s
|
|
|
|
|
rÔ c
|
@ sÒ e Zd Zdd Zdd Zdd Zej de
|
d¡e
|
d¡fd e
|
d¡e
|
d
|
¡fe
|
d¡d e
|
d
|
¡fd g¡d d Z dd Zdd Z ejj eed dddd Zej ddddg¡dd Zd d Zd d Zd S ) ÚTestLowlevelAPIAccessc C s¾ t d¡}t d¡}t d¡}t j ||d f¡}||||fksBtt jj||d fdd}||||fksjtt j |td f¡}||||fkstt t¡ t jj||d fdd W 5 Q R X d S )NrÏ Úf4rÊ )NNrÖ rÚ r r ) r r+ r" Úresolve_dtypesrX r# r{ rÜ r! )r% rÏ rÖ rÊ rx r r r Útest_resolve_dtypes_basicf
|
s"
|
|
|
ÿÿ z/TestLowlevelAPIAccess.test_resolve_dtypes_basicc C sj t d¡}t j ||d f¡}|||t t¡fks4tt d¡}t t¡ t j ||d f¡ W 5 Q R X d S )NÚS0Z10i) r r+ rÖ r× rÄ rX r{ rÜ ÚNotImplementedError)r% rÙ rx Zdtsr r r Útest_weird_dtypes|
|
s
|
|
z'TestLowlevelAPIAccess.test_weird_dtypesc C s: t d¡}t t¡ t jj|||fdd W 5 Q R X d S )NrÏ T©Z reduction)r r+ r{ rÜ rÚ r" r× )r% rÏ r r r Ú test_resolve_dtypes_reduction
|
s
|
z3TestLowlevelAPIAccess.test_resolve_dtypes_reductionÚdtypesrÌ Nr9 )rÏ rÏ Nc C s& t t¡ tj |¡ W 5 Q R X d S r> )r{ rÜ r! r r" r× )r% rÞ r r r Útest_resolve_dtypes_errors
|
s z0TestLowlevelAPIAccess.test_resolve_dtypes_errorsc C s@ t d¡}t d¡}t jjd |d fdd}||||fks<td S )NÚi2ÚlongTrÜ )r r+ r" r× rX )r% rà Zlong_r[ r r r rÝ
|
s
|
|
c C sX t d¡}t t¡ t j d ||f¡ W 5 Q R X t t¡ t j d¡ W 5 Q R X d S )Nrà )NNrÏ )r r+ r{ rÜ r! r" r× r( )r% rà r r r Ú$test_resolve_dtypes_reduction_errors
|
s
|
|
z:TestLowlevelAPIAccess.test_resolve_dtypes_reduction_errorsÚ pythonapiz2`ctypes.pythonapi` required for capsule unpacking.r2 c
|
s4 t t jd¡}t t jd¡}t t jd¡}t t jt j|||t j¡ G fdddt j}t d¡}tj
|
||f¡\}}|||fkst tj
|
|¡ t jt jj_t j t |¡t d¡¡}t |t |¡¡j}tjd|d} | |j|| jj| jj| jj|| jjd | jjd |j ¡ t | tjd|d d S )
|
Nr r c s6 e Zd Zd fdejfdejfdejfdejfgZdS )z;TestLowlevelAPIAccess.test_loop_access.<locals>.call_info_tÚ strided_loopÚcontextÚauxdataZrequires_pyapiZno_floatingpoint_errorsN)r/ r0 r1 ÚctÚc_void_pÚc_byteÚ_fields_r ©Zstrided_loop_tr r Ú call_info_t°
|
s ûrì rÏ s numpy_1.24_ufunc_call_infor4 r5 r ) rç ÚARRAYÚc_char_pÚ c_ssize_tÚ CFUNCTYPEÚc_intrè Ú Structurer r+ rr Ú_resolve_dtypes_and_contextrX Ú_get_strided_looprã ZPyCapsule_GetPointerÚrestypeÚ py_objectÚcastÚPOINTERÚcontentsr rä rå ÚctypesÚdatar¼ rf ræ r )
|
r% Zdata_tZdim_tZ strides_trì rÏ rä Z call_info_objÚ call_inforZ r rë r Útest_loop_access¦
|
s> ÿ
|
þûz&TestLowlevelAPIAccess.test_loop_accessrf r r )r r c C sL t d¡}t j ||f¡\}}tjtdd t jj||d W 5 Q R X d S )NrÏ z fixed_strides.*tuple.*or Nonerð )Z fixed_strides)r r+ rr ró r{ rÜ r! rô )r% rf rÏ rä rü r r r Ú)test__get_strided_loop_errors_bad_stridesÒ
|
s
|
z?TestLowlevelAPIAccess.test__get_strided_loop_errors_bad_stridesc C s t d¡}t j ||f¡\}}tjtdd t j d¡ W 5 Q R X tjtdd t j |¡ W 5 Q R X t j |¡ t t¡ t j |¡ W 5 Q R X d S )NrÏ Z PyCapsulerð znot the capsule!z.*incompatible context)
|
r r+ rr ró r{ rÜ r rô r! r" )r% rÏ rä rü r r r Ú+test__get_strided_loop_errors_bad_call_infoÚ
|
s
|
zATestLowlevelAPIAccess.test__get_strided_loop_errors_bad_call_infoc C sP t jdt jd}d|d d< d|d d< t |¡}t|d d |d d d S )N)i i r5 r r é i )r rÔ rw r r )r% r r' r r r Útest_long_arraysé
|
s
|
|
z&TestLowlevelAPIAccess.test_long_arrays)r/ r0 r1 rØ rÛ rÝ r{ r| r} r r+ rß râ rª Úhasattrrç rý rþ rÿ r r r r r rÕ e
|
s( ü
|
ÿ
|
*
|
rÕ )IrB r r» rú rç r{ r Únumpyr Znumpy.core._umath_testsr_ Z _umath_testsr Znumpy.linalg._umath_linalgrq rr r Z numpy.core._operand_flag_testsZ_operand_flag_testsr- Znumpy.core._rational_testsr Z numpy.testingr r r r r r r r
|
r r r Z numpy.testing._private.utilsr Z numpy.compatr r` Ú__dict__ro Z UNARY_UFUNCSr r r2 r r r| r} Údirr³ Z_no_nep50_warningr¶ ÚsignrÖ r· rª r¿ r# rº r rÅ rÆ rÈ r" rá r rË rÎ Z typecodesrÑ rÔ rÕ r r r r Ú<module> s 4 D
|
%
|
|
|
|
|
ÿ þÿ
|
|
