U
|
O ±d ³ ã
|
@ s¬ d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlm Z d dl
|
m Z d dl m Z d dlm mZ d dlmZ d dlZd dlmZmZmZmZm Z m Z m Z m Z m Z m!Z!m"Z"m#Z#m$Z$m%Z% d dl&m'Z' dd ejjj( )¡ D Z*d
|
d e*D Z+d d e+D Z,d d e*D Z-d d e-D Z.dd Z/dd Z0dd Z1G dd dZ2G dd dZ3G dd dZ4G dd dZ5G d d d Z6G d d d Z7d d! Z8d"d# Z9G d$d% d%Z:G d&d' d'Z;G d(d) d)Z<G d*d+ d+Z=G d,d- d-Z>G d.d/ d/Z?G d0d1 d1Z@G d2d3 d3e2ZAG d4d5 d5ZBG d6d7 d7ZCG d8d9 d9ZDG d:d; d;ZEG d<d= d=ZFG d>d? d?ZGd@dAdBgd dCdBgd@dDdBgd@dCdBgd dCdBgd dCdBgd dCdBgd dCdBgdEZHG dFdG dGZIG dHdI dIZJG dJdK dKe2ZKG dLdM dMZLG dNdO dOZMG dPdQ dQZNdRdS ZOdTdU ZPG dVdW dWZQdXdY ZRdZd[ ZSd\d] ZTd^d_ ZUd`da ZVG dbdc dcZWG ddde deZXG dfdg dge2ZYG dhdi die2ZZG djdk dke2Z[G dldm dme2Z\G dndo doZ]G dpdq dqZ^G drds dsZ_G dtdu duZ`G dvdw dwZaG dxdy dyZbG dzd{ d{ZcG d|d} d}ZdG d~d dZeG dd dZfG dd dZgG dd
d
ZhG dd dZiG dd dZjG dd dZkG dd dZlG dd dZmG dd dZnG dd dZod@ddepfddZqdd Zrdd Zsdd Ztdd Zuejvjwe xejy¡e xejz¡kd d¡ejvj{e |¡ }d¢¡d£d¤d¥d¦ Z~d§d¨ Zd©dª Zd«d¬ Zdd® Zejvjwe xejy¡e xejz¡kd d¡ejvj{e |¡ }d¢¡d£d¤d¯d° Zd±d² Zd³d´ Z
dµd¶ Zd·d¸ Zd¹dº Zd»d¼ Zd½d¾ Zejv d¿ejejg¡dÀdÁ Zejv dÂdÃd e.D ¡dÄdÅ Zejv dÂdÆd e.D ¡dÇdÈ ZdÉdÊ Zejv dËe dÌ¡e d d@g¡e d d@gdÌdÍgg¡g¡dÎdÏ ZdÐdÑ ZdÒdÓ ZdÔdÕ Zejv dÖe dסdAdDdØgg¡dÙdÚ ZdS )Ûé N)ÚFraction©Úreduce)Ú
|
namedtuple)Ú _umath_tests)Úassert_Ú assert_equalÚ assert_raisesÚassert_raises_regexÚassert_array_equalÚassert_almost_equalÚassert_array_almost_equalÚassert_array_max_ulpÚassert_allcloseÚassert_no_warningsÚsuppress_warningsÚ_gen_alignment_dataÚ assert_array_almost_equal_nulpÚIS_WASM)Ú_glibc_older_thanc 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_umath.pyÚ
|
<listcomp> s ÿr c C s g | ]}|j d kr|qS ©é ©Únin©r Zufr r r r s
|
c C s g | ]}d |j kr|qS )zf->f©Útypesr" r r r r s
|
c C s g | ]}|j d kr|qS )é r r" r r r r " s
|
c C s$ g | ] }t |d r|jdkr|qS )Ú
|
accumulater )ÚhasattrÚnoutr" r r r r % s
|
|
c
|
# sJ d}t jd||d}t jd||d}| |d< ||d< dd }|d |d |dfV dd }|d |d |d
|
fV ||d< ||d< d D ]^ | | < || < fd d}|||d fV || |d dfV || < || < qdD ]\}|dd|
}|dd|
} | |d< || d< dd }|| |d| fV ||d< || d< qèdS )a
|
Helper to create "interesting" operands to cover common code paths:
|
* scalar inputs
|
* only first "values" is an array (e.g. scalar division fast-paths)
|
* Longer array (SIMD) placing the value of interest at different positions
|
* Oddly strided arrays which may not be SIMD compatible
|
|
It does not attempt to cover unaligned access or mixed dtypes.
|
These are normally handled by the casting/buffering machinery.
|
|
This is not a fixture (currently), since I believe a fixture normally
|
only yields once?
|
r i' )ÚdtypeÚ
|
fill_valuer c S s | S ©Nr ©Úresr r r Ú<lambda>? ó z,interesting_binop_operands.<locals>.<lambda>Zscalarsc S s | S r+ r r, r r r r. B r/ )r .z scalar-arrays)
|
r r r% é é é éÿÿÿÿéþÿÿÿéýÿÿÿéüÿÿÿc s | S r+ r r, ©Úposr r r. M r/ zoff-z -with-scalar)r3 éq Né
|
c S s | d S )Nr: r r, r r r r. Z r/ zstride-)r Úfull)
|
Úval1Úval2r) r* Úarr1Úarr2Ú extractorÚstrideÚop1Úop2r r7 r Úinteresting_binop_operands) s8
|
rD c C s t ¡ dkpt ¡ d¡S )z/ True if we are running on a Power PC platform.ZpowerpcÚppc)ÚplatformÚ processorÚmachineÚ
|
startswithr r r r Ú
|
on_powerpca s ÿrJ c C sX t ¡ dkrd} nt r d} ndS t t | ¡¡}t t | ¡¡j}t|| d dkS )zôThe blocklisted trig functions are not accurate on aarch64/PPC for
|
complex256. Rather than dig through the actual problem skip the
|
test. This should be fixed when we can move past glibc2.17
|
which is the version in manylinux2014
|
Úaarch64gÙ£wÑvè=gýzý Ó¯í=Fç ð?gQ²@³-(;) rF rH rJ r ÚarcsinhZfloat128Z
|
complex256ÚrealÚabs)ÚxÚv1Úv2r r r Ú bad_arcsinhg s rS c @ s e Zd Zdd Zdd ZdS )Ú_FilterInvalidsc C s t jdd| _d S )NÚignore©Úinvalid©r ZseterrZolderr©Úselfr r r Ú setup_methodz s z _FilterInvalids.setup_methodc C s t jf | j d S r+ rX rY r r r Úteardown_method} s z _FilterInvalids.teardown_methodN)Ú__name__Ú
|
__module__Ú __qualname__r[ r\ r r r r rT y s rT c @ s$ e Zd Zdd Zdd Zdd ZdS )Ú TestConstantsc C s t tjdd d S )Ng-DTû! @çVç¯Ò<)r ÚncuÚpirY r r r Útest_pi s zTestConstants.test_pic C s t tjdd d S )NgiW
|
¿@ra )r rb ÚerY r r r Útest_e
s zTestConstants.test_ec C s t tjdd d S )Ng¶oüxâ?ra )r rb Z euler_gammarY r r r Útest_euler_gamma s z TestConstants.test_euler_gammaN)r] r^ r_ rd rf rg r r r r r` s r` c @ s e Zd Zdd Zdd ZdS )ÚTestOutc
|
|
C s¾ dD ]²}t d¡}t d¡}t j|d||d}t||k t j|d||d}t||k t j|d|f|d}t||k t d¡}t d¡}t jdt jd}t j||d |d\}} t||k t j|d ||d\}} t| |k t j||||d\}} t||k t| |k t j||d f|d\}} t||k t j|d |f|d\}} t| |k t j|||f|d\}} t||k t| |k tt t j|||d\}} W 5 Q R X ttt j|d|||d ttt j|d|||d ttt j|dd ||d tt t j|d||f|d tt t j|dd|d ttt j|dg |d ttt j|dg |d ttt j|dg f|d d |j
|
_ tt t j|d||d tt t j|d||d tt t j|d|f|d qd S )
|
N©TFç à?r r% ©Úsubok©Úoutrl çÍÌÌÌÌÌ@©r) F) r ÚarrayÚemptyÚaddr Úint32Úfrexpr Ú TypeErrorÚ
|
ValueErrorÚflagsZ writeable)
|
rZ rl ÚaÚoÚrÚdÚo1Úo2Úr1Úr2r r r Útest_out_subok sP
|
|
|
|
|
|
zTestOut.test_out_subokc
|
|
C s G dd dt j}dD ]p}|dg}t j|d|d}|rJtt|| ntt|t jk t j|dd |d}|rtt|| ntt|t jk t j|dd |d}|rºtt|| ntt|t jk t j|dd|d}|ròtt|| ntt|t jk |d g}t d
|
¡}t jd
|
t jd }t j|||d\}} |rRtt| | ntt| t jk t j||d |d\}} |rtt| | ntt| t jk t j|d ||d\}} |rÎtt|| ntt|t jk t j||d f|d\}} |rtt| | ntt| t jk t j|d |f|d\}} |rNtt|| ntt|t jk t t
|
t j|||d\}} W 5 Q R X qd S ) Nc @ s e Zd ZdZdd Zdd ZdS )z.TestOut.test_out_wrap_subok.<locals>.ArrayWrapr: c S s t |¡ | ¡ ¡ S r+ ©r ÚasarrayÚviewÚcopy)ÚclsÚarrr r r Ú__new__Â s z6TestOut.test_out_wrap_subok.<locals>.ArrayWrap.__new__c S s | t| ¡S r+ ©r Útype©rZ r Úcontextr r r Ú__array_wrap__Å s z=TestOut.test_out_wrap_subok.<locals>.ArrayWrap.__array_wrap__N©r] r^ r_ Ú__array_priority__r r r r r r Ú ArrayWrap¿ s r ri rj r% rk rm r+ ro r rp ) r Úndarrayrs r r r rr rt ru r rv )
|
rZ r rl ry r{ r| r} r~ r r r r r Útest_out_wrap_subok¾ sX
|
|
|
|
|
zTestOut.test_out_wrap_subokN)r] r^ r_ r r r r r r rh s 1rh c @ s¾ e Zd ZddlZej dejd ejd ejd ej g ¡ej dej
|
ej fej ej fejejfejejfejejfejejfg¡dd Zd
|
d Zd d Zdd Zdd Zdd ZdS )ÚTestComparisonsr Nr) ÚuintÚintÚfloatzpy_comp,np_compc sZ |t jkr8t jjddgdd}t jjddgdd}dn4t jjdddd |¡}t jjdddd |¡}dt |¡ ¡}| ¡ }| ¡ }||| t j
|
¡} fd d
|
t ||D }
|
||| t j
|
¡} fd d
|
|D } ||| t j
|
¡} fd d
|
|D }t | ¡ |
|
kd j d t | ¡ | kd j d t | ¡ |kd j d d S )NFTéè )ry Úsizer r: ©ÚlowÚhighr r2 c s g | ]\}}t ||qS r ©r ©r rP Úy)Úpy_compr r r ' s z=TestComparisons.test_comparison_functions.<locals>.<listcomp>c s g | ]}t |qS r r ©r rP ©r Zscalarr r r + s c s g | ]}t |qS r r r r¡ r r r / s zFailed comparison (ú))r Úbool_ÚrandomÚchoiceÚrandintÚastyper) r Útolistr Úuint8Úzipr r] )rZ r) r Znp_compry ÚbZ np_scalarÚa_lstÚb_lstZcomp_bZ comp_b_listZcomp_s1Z comp_s1_listZcomp_s2Z comp_s2_listr r¡ r Útest_comparison_functions s2
|
ÿ ÿ ÿz)TestComparisons.test_comparison_functionsc C s t jt dddg¡d gtd}ttt j|| G dd d}t | g¡}ttt j|| t jt jgtd}tt ||¡dg d S )Nr r% r0 rp c @ s e Zd Zdd ZdS )zGTestComparisons.test_ignore_object_identity_in_equal.<locals>.FunkyTypec S s t dd S ©NzI won't compare©rv ©rZ Úotherr r r Ú__eq__A s zNTestComparisons.test_ignore_object_identity_in_equal.<locals>.FunkyType.__eq__N)r] r^ r_ r³ r r r r Ú FunkyType@ s r´ F) r rq Úobjectr rw Úequalrv Únanr ©rZ ry r´ r r r Ú$test_ignore_object_identity_in_equal9 s z4TestComparisons.test_ignore_object_identity_in_equalc C s t jt dddg¡d gtd}ttt j|| G dd d}t | g¡}ttt j|| t jt jgtd}tt ||¡dg d S )Nr r% r0 rp c @ s e Zd Zdd ZdS )zKTestComparisons.test_ignore_object_identity_in_not_equal.<locals>.FunkyTypec S s t dd S r¯ r° r± r r r Ú__ne__S s zRTestComparisons.test_ignore_object_identity_in_not_equal.<locals>.FunkyType.__ne__N)r] r^ r_ rº r r r r r´ R s r´ T) r rq rµ r rw Ú not_equalrv r· r r¸ r r r Ú(test_ignore_object_identity_in_not_equalK s z8TestComparisons.test_ignore_object_identity_in_not_equalc C s8 t ddg¡}tt jj|tdd ttt jj| d S )Nr rp T)r rq r r¶ r Úboolr rv ©rZ ry r r r Útest_error_in_equal_reduce] s z*TestComparisons.test_error_in_equal_reducec C s< t jddgtdjtks tt jddgddjtks8td S )Nr rp )NNÚO)Ú signature)r r¶ rµ r) ÚAssertionErrorrY r r r Útest_object_dtypee s z!TestComparisons.test_object_dtypec C sv t jddgtdjtks ttjtdd t jddt jd W 5 Q R X tjtdd t jdddd W 5 Q R X d S )Nr rp zNo loop matching©Úmatch)NNÚl)Úsig) r r¶ r½ r) r ÚpytestÚraisesrv Úint64rY r r r Ú test_object_nonbool_dtype_errori s
|
z/TestComparisons.test_object_nonbool_dtype_error) r] r^ r_ ÚoperatorrÈ ÚmarkÚ parametrizer Úsctypesr£ ÚltZlessÚleZ
|
less_equalÚgtZgreaterÚgeZ greater_equalÚeqr¶ Úner» r® r¹ r¼ r¿ rà rË r r r r r
|
s( ÿÿ
|
|
|
|
|
|
ú "r c @ s e Zd Zdd ZdS )ÚTestAddc C s< t jddt jfdt jfgd}d|d< t|d ¡ d d S )Nr% ry r« rp r3 r )r Úzerosrt Úfloat64r Úsumr¾ r r r Útest_reduce_alignmentv s z TestAdd.test_reduce_alignmentN)r] r^ r_ rÚ r r r r rÖ u s rÖ c @ sF e Zd Zdd Zejjeddej de
|
e j d e j d d¡¡d d
|
Z ejjeddej de
|
e j d e j d d ¡¡d d Zej de dd¡e dd¡dfe dd¡e dd¡dfe dd¡e dd¡dfe dd¡e dd¡dfe dd¡e dd¡dfe dd¡e dd¡dfe dd¡de dd¡fe dd¡de dd¡fe dd¡de dd¡fe dd¡de dd¡fe dd¡de dd¡fe dd¡de dd¡fe dd¡de dd¡fg ¡dd Zd d Zd d Zd d! Zd"d# Zejjeddej d$e jd% ¡d&d' Zej d$e jd% ¡d(d) Zd*S )+Ú TestDivisionc C sª t ddddddddd g ¡}d
|
d
|
krFt|d d d d ddddddg n t|d dddddddddg t|d dddddddddg t|d dddddddddg d S )Nr2 r: éZ éd éûÿÿÿéöÿÿÿi¦ÿÿÿéÿÿÿiÿÿÿrj g©?ç¹?gÍÌÌÌÌÌì?r g©¿ç¹¿gÍÌÌÌÌÌì¿r3 g333333ó¿r r4 é_ éP )r rq r ©rZ rP r r r Útest_division_int s ÿ z TestDivision.test_division_intú fp errors don't work in wasm©Úreasonz dtype,ex_valr r )z\np.array(range(fo.max-lsize, fo.max)).astype(dtype),np.arange(lsize).astype(dtype),range(15)zunp.arange(fo.min, fo.min+lsize).astype(dtype),np.arange(lsize//-2, lsize//2).astype(dtype),range(fo.min, fo.min + 15)znp.array(range(fo.max-lsize, fo.max)).astype(dtype),np.arange(lsize).astype(dtype),[1,3,9,13,neg, fo.min+1, fo.min//2, fo.max//3, fo.max//4]c
|
s8 t |¡jdk rdnd}d}t|\}}}| ¡ | ¡ }} fdd t jdd | ¡ }
|
|
|
| }
|
|| } W 5 Q R X fd d
|
t|| D } d } t| | ks¬t | d }t|
|
| ksÄt ||D ]t| ¡ }
|
t jddd | }|
|
}
|
W 5 Q R X fdd
|
|D } t|| ks(t | t|
|
| ksÈt |qÈt jddd â d|kr|t
|
j t dd || W 5 Q R X n|| jr¼j|kr¼t
|
j t dd |d W 5 Q R X njrÌ|d t
|
j t dd |d W 5 Q R X t
|
j t dd | ¡ }
|
|
|
d }
|
W 5 Q R X t j g |dd W 5 Q R X d S )Nr r3 r i c s0 |dkr dS | r(| j kr(|dkr( j S | | S ©Nr r3 ©Úmin©Únr| ©Úfor r r. ° s z9TestDivision.test_division_int_boundary.<locals>.<lambda>rU ©Údividec s g | ]\}} ||qS r r r )Úc_divr r r ¹ s z;TestDivision.test_division_int_boundary.<locals>.<listcomp>z(Integer arrays floor division check (//)z)Integer arrays floor division check (//=)©rò Úoverc s g | ]} |qS r r ©r Úi)ró Údivisorr r r Å s Úraisez*divide by zero encountered in floor_dividerÄ z$overflow encountered in floor_dividerp )r Úiinforì Úevalr¨ Úerrstater
rª Úallr rÈ rÉ ÚFloatingPointErrorrq )rZ r) Úex_valÚnegZlsizery r« Zdivisorsr¬ r ÚacZdiv_abÚdiv_lstÚmsgZmsg_eqÚdiv_ar )ró rø rð r Útest_division_int_boundary s^
|
|
ÿÿÿÿz'TestDivision.test_division_int_boundary)z1np.array([fo.max, 1, 2, 1, 1, 2, 3], dtype=dtype)z4np.array([fo.min, 1, -2, 1, 1, 2, -3]).astype(dtype)z3np.arange(fo.min, fo.min+(100*10), 10, dtype=dtype)z8np.array(range(fo.max-(100*7), fo.max, 7)).astype(dtype)c
|
sø t |¡ t|}| ¡ } fdd}t jdd t j |¡}W 5 Q R X t||}d}||ksht|t jdddx tj t
|
dd t j t d
|
d ¡ |¡¡ W 5 Q R X j rêtj t
|
d d $ t j t j j d dg|d¡ W 5 Q R X W 5 Q R X d S )Nc s* |dks | r"| j kr"|dkr"dS | | S rê rë rí rï r r r. ï s z7TestDivision.test_division_int_reduce.<locals>.<lambda>rU rñ z#Reduce floor integer division checkrù rô z$divide by zero encountered in reducerÄ rà rÝ z overflow encountered in reducer r3 rp )r rú rû r¨ rü Ú floor_divider r rÈ rÉ rþ Úaranger§ rì rq ) rZ r) rÿ ry Úlstró r r r r rï r Útest_division_int_reduceâ s*
|
|
|
ÿ$ÿÿz%TestDivision.test_division_int_reducezdividend,divisor,quotientr% ÚYÚMé r4 éôÿÿÿr3 r r r5 ZNatc
|
C s¸ |rnt |tst |¡snd}|| |ks0t|d}t |gd ¡}t |gd ¡}t|| |ks´t|nFtr|t d¡ tj
|
ddd$ t t ¡ || W 5 Q R X W 5 Q R X d S )Nz Timedelta floor division checkz%Timedelta arrays floor division checkr2 rç rù ©rò rW ) r r r Zisnatr rq rý r rÈ Úskiprü rÉ rþ )rZ Údividendrø Zquotientr Zdividend_arrayZquotient_arrayr r r Útest_division_int_timedelta s
|
z(TestDivision.test_division_int_timedeltac C sn d}t jdddgt jd}t|d | ||d d}t jd d
|
gt jd}|d | }t|| d d g|d d S ) Nz%Complex division implementation checkù ð? ð?y ð? à?ù ð? @rp r% ©Úerr_msgz)Complex division overflow/underflow checkgþÆG0JÅVgòys? )r )r rq Ú
|
complex128r )rZ r rP r r r r Útest_division_complex& s z"TestDivision.test_division_complexc C sÐ t jddd¸ t jdgt jd}d| }tt |¡d tt jt j| }tt |¡d tt jt j| }tt |¡d tt jt j| }tt |¡d d| }tt |¡d W 5 Q R X d S )NrU ©rW rò ç rp rL r )
|
r rü rq r r ÚisinfÚcomplexÚinfr· Úisnan)rZ rP r r r r Útest_zero_division_complex1 s z'TestDivision.test_zero_division_complexc C s~ t jddddgt jd}t t¡ |d W 5 Q R X t t¡ t |d¡ W 5 Q R X t t¡ t |d¡ W 5 Q R X d S )NyÍÌÌÌÌÌì? ð?y¹¿ ð?yÍÌÌÌÌÌì? à?yÍÌÌÌÌÌì? @rp é )r rq r rÈ rÉ rv ÚdivmodÚ remainderrå r r r Útest_floor_division_complex? s z(TestDivision.test_floor_division_complexc C s8 t d¡}tt |d ¡d tt | d ¡d d S )Nr: r r )r r× r Úsignbitrå r r r Ú test_floor_division_signed_zeroI s
|
z,TestDivision.test_floor_division_signed_zeror) ÚFloatc C sØ t jt j|d}t jd|d}t jd|d}t jt j|d}t jddd ttt j|| W 5 Q R X t jddd t ||¡ W 5 Q R X t jdd6 t ||¡ t ||¡ t ||¡ t ||¡ W 5 Q R X d S )Nrp rL r rù rU r ©rý )r rq r· r rü r rþ r )rZ r) ÚfnanÚfoneÚfzerÚfinfr r r Útest_floor_division_errorsP s z'TestDivision.test_floor_division_errorsc C s" t jd|d}t jd|d}t jt j|d}t jd|d}t jd|d}t jt j|d}t }| td¡ t ||¡} t
|
| ¡st dt | f t ||¡} t
|
| ¡s¸t dt | f t ||¡} t
|
| ¡sÞt dt | f W 5 Q R X t j dd$ t ||¡}
|
tt |
|
¡ ¡ W 5 Q R X d S ) Nr: rp rL r z)invalid value encountered in floor_divideúdt: %s, div: %srU rñ )r r× Úonesrq r· r r ÚfilterÚRuntimeWarningr r r Údtrü r r rý ) rZ r) rP r r' r( r) r* ÚsupÚdivÚzr r r Ú test_floor_division_corner_casesd s" $ z-TestDivision.test_floor_division_corner_casesN)r] r^ r_ ræ rÈ rÍ Úskipifr rÎ Ú itertoolsÚproductr rÏ r r Z timedelta64r r r r" r$ Ú typecodesr+ r4 r r r r rÛ sP ÿ : ÿ ôþ
|
|
rÛ c C s t | |¡t | |¡fS r+ )r r r! ©rP r r r r Úfloor_divide_and_remainderz s r: c C s | t jd krdS dS d S )NÚUnsignedIntegerr ©r r3 )r r8 )r0 r r r Ú_signs~ s r= c @ sê e Zd Zdd Zdd Zdd Zejje ddejj
|
e j d ¡d
|
dej d ejd ¡d d Zejje ddejj
|
e j d ¡d
|
dej d ejd ¡ej dejejg¡dd Zejje dddd Zdd Zdd ZdS )Ú TestRemainderc C s t jd t jd }tt jfD ]Þ}t ||¡D ]Ì\}}t t|t|¡D ]®\}}d}||j||||f }t j|d |d} t j|d |d}
|
|| |
|
\} } t | |
|
| | |d |dkrÚt
|
|
|
| k oÎd kn | qJt
|
|
|
| koîd kn | qJq.q d S )
|
NÚ
|
AllIntegerr% ú*op: %s, dt1: %s, dt2: %s, sg1: %s, sg2: %séG rp é r r3 r ) r r8 r: r r6 r7 r= r] rq r r © rZ r0 ÚopZdt1Zdt2Zsg1Zsg2Úfmtr ry r« r2 Úremr r r Útest_remainder_basic s z"TestRemainder.test_remainder_basicc C s( t tdd}t tdd}|dg | }|| }t t ||¡}t dd |D }tj|tdj\}}tj|tdj\} }
|
t | dk|dk |dk A @ d | ¡} t |
|
dk|dk @ d |
|
¡}
|
t
|
tj fD ]`} tj d
|
D ]P} d | j | f } | | ¡}| | ¡}| ||\}}t|| | d t||
|
| d qÐqÂd S ) Niÿÿÿr r é c s s | ]}t | V qd S r+ )r )r Útr r r Ú <genexpr> s z;TestRemainder.test_float_remainder_exact.<locals>.<genexpr>rp r ç r% zop: %s, dtype: %sr )ÚlistÚranger6 r7 r rq r ÚTr Úwherer: r r8 r] r§ r )rZ ZnlstZplstr rø ÚargÚtgtry r« ZtgtdivZtgtremrD r0 r ÚfaZfbr2 rF r r r Útest_float_remainder_exact s$ "
|
|
z(TestRemainder.test_float_remainder_exactc C sô t jd }tt jfD ]Ú}t ||¡D ]È\}}t dd¡D ]²\}}d}||j||||f }t j|d d |d} t j|d |d}
|
|| |
|
\} } t| |
|
| | |d |dkrÌt |
|
| k oÀd kn | q8t |
|
| koàd kn | q8q$qd S )
|
Nr% r< r@ éN g+i¤)+p>rp r r3 r )
|
r r8 r: r r6 r7 r] rq r r rC r r r Ú test_float_remainder_roundoff± s
|
z+TestRemainder.test_float_remainder_roundoffrç rè Údarwinz_MacOS seems to not give the correct 'invalid' warning for `fmod`. Hopefully, others always do.r) r% c C s4 t jd|d}t jd|d}t jt j|d}t jt j|d}t jddd ttt j|| W 5 Q R X t jddd ttt j|| W 5 Q R X t jdd ttt j|| W 5 Q R X t jdd ttt j|| W 5 Q R X t jddd ttt j|| W 5 Q R X t jddd t ||¡ W 5 Q R X d S )Nr rp rL rù rU r rV )r rq r r· rü r rþ r )rZ r) Úfzeror( r* r' r r r Útest_float_divmod_errorsà s z&TestRemainder.test_float_divmod_errorsÚfnc
|
C s¤ t jd|d}t jd|d}t jt j|d}t jt j|d}t jddR tjtdd ||| W 5 Q R X ||| ||| ||| ||| W 5 Q R X d S )Nr rp rL rù r& z invalid valuerÄ )r rq r r· rü rÈ rÉ rþ )rZ r) rY rW r( r* r' r r r Útest_float_remainder_errorsÞ s ÿ
|
|
|
z)TestRemainder.test_float_remainder_errorsc C s¦ t t j¡j}t jddd, t d|¡\}}t |¡ t|dk W 5 Q R X t jddd tt t jd| W 5 Q R X t jddd tt t jd| W 5 Q R X d S )NrU )rõ rW r1 r rù )rW rõ )
|
r ÚfinforØ Útinyrü r r r r rþ )rZ ry r2 Úmodr r r Ú test_float_remainder_overflowô s
|
z+TestRemainder.test_float_remainder_overflowc
|
C sn t jd D ]\}t jt j|d}t jd|d}t jd|d}t jt j|d}t }| td¡ | td¡ t ||¡\}}t |¡st
|
d||f t |¡s²t
|
d||f t ||¡\}}t |¡sÜt
|
d||f t t |¡d||f f t ||¡\}}t |¡s"t
|
d||f t |¡s>t
|
d||f t ||¡\}}t |¡sjt
|
d||f t |¡st
|
d||f t ||¡\}}t |¡s²t
|
d||f t |¡sÎt
|
d||f t ||¡\}}t |¡sút
|
d||f t |¡st
|
d||f t ||¡\}}t |¡sBt
|
d||f t |¡s^t
|
d||f W 5 Q R X q
|
d S ) Nr% rp rL r z#invalid value encountered in divmodz$divide by zero encountered in divmodr, údt: %s, rem: %s) r r8 rq r· r r r. r/ r r r r r ) rZ r0 r' r( r) r* r1 r2 rF r r r Ú test_float_divmod_corner_cases s:
|
z,TestRemainder.test_float_divmod_corner_casesc C sö t jd D ]}t jd|d}t jd|d}t jt j|d}t jd|d}t t jd|d| ¡}t ||¡}t||kd| t | | ¡}t|| kd| q
|
t @}| t d¡ | t d¡ t jd D ]}t jd|d}t jd|d}t jt j
|
|d} t jt j|d}t ||¡}tt |¡d||f t | |¡}t | |¡}
|
tt |
|
¡d ||
|
f tt |¡d||f t | | ¡}t | |¡}
|
tt |¡d||f tt |
|
¡d ||
|
f t | |¡}t | |¡}
|
tt |¡d||f tt |
|
¡d ||
|
f t ||¡}t ||¡}
|
tt |¡d||f tt |
|
¡d ||
|
f t ||¡}t ||¡}
|
tt |¡d||f tt |
|
¡d ||f t ||¡}t ||¡}
|
tt |¡d||f tt |
|
¡d ||f qÐW 5 Q R X d S )
|
Nr% rL rp r zdt: %sz&invalid value encountered in remainderz!invalid value encountered in fmodr_ zdt: %s, fmod: %s) r r8 rq r· Ú nextafterr! r r r. r/ r r Úfmod) rZ r0 r( r) r' r« ry rF r1 r* rb r r r Ú!test_float_remainder_corner_cases sX
|
z/TestRemainder.test_float_remainder_corner_casesN)r] r^ r_ rG rS rU rÈ rÍ r5 r ÚxfailÚsysrF rI rÎ r r8 rX rb r! rZ r^ r` rc r r r r r>
s( ÿÿ
|
r> c @ s e Zd ZedddgZdd dd dd dd d d d
|
Zejeed ed ejeed ed e j
|
eed ed e j eed ed ej eed ed ej eed ed iZejjeddej dejd ¡dd Zejjeddej dejd ¡dd Zejjeddej dejd ¡ej dejd ¡ej d ejejej ej e j
|
e j g¡ejd d d d d Zd!S )"Ú+TestDivisionIntegerOverflowsAndDivideByZeroÚ result_typeÚnocastÚcastedc C s dS ©Nr r rp r r r r. Y r/ z4TestDivisionIntegerOverflowsAndDivideByZero.<lambda>c C s t | ¡jS r+ ©r rú rì rp r r r r. Z r/ c C s t | ¡j S r+ rk rp r r r r. [ r/ c C s t | ¡jdfS rj rk rp r r r r. \ r/ c C s t | ¡j dfS rj rk rp r r r r. ] r/ )Úzerorì Úneg_minúmin-zeroú neg_min-zerorl rì rm rn ro rç rè r) ÚIntegerc
|
|
C s4 t t |¡jd|}|D ]\}}}}tjtdd || }W 5 Q R X |j|jksXt||t |j¡jksrt|| }|j|jkst||dkstt ||¡}||dks¶ttjtdd t
|
||¡\}} W 5 Q R X |j| j krü|jksn t||t |j¡jks t|| dkstqd S )Nr3 zoverflow encounteredrÄ r ) rD r rú rì rÈ Úwarnsr/ r) r rb r ©
|
rZ r) Zto_checkrB rC r@ Zoperand_identifierr- Úres1Úres2r r r Ú test_signed_division_overflown s " zITestDivisionIntegerOverflowsAndDivideByZero.test_signed_division_overflowr? c
|
|
C sÎ t dd|}|D ]¸\}}}}tjtdd || }W 5 Q R X |j|jksNt||dks^ttjtdd t ||¡\}} W 5 Q R X |j| j kr¢|jks¨n t||dks¸t|| dkstqd S )Nr r zdivide by zerorÄ )rD rÈ rq r/ r) r r r rr r r r Útest_divide_by_zero s z?TestDivisionIntegerOverflowsAndDivideByZero.test_divide_by_zeroÚdividend_dtyper Ú divisor_dtypeÚ operationÚwarnrô c
|
s fddt ddD }tjdg|d}t ¡jt |¡jkr|tjtjtjfkrt j
|
t dd8 | t ¡j |d}|| j| ¡kstW 5 Q R X |D ]b}t j
|
t ddJ t |||¡ d ¡}t | j| ¡gt| ¡ ¡ }t|| W 5 Q R X q¨n| t ¡j |d}|| j| ¡ksBt|D ]J}t |||¡ d ¡}t | j| ¡gt| ¡ ¡ }t|| qFd S )
|
Nc s( g | ] }t jt ¡jg| d qS )rp )r rq rú rì rö ©rw r r r « s ÿÿzNTestDivisionIntegerOverflowsAndDivideByZero.test_overflows.<locals>.<listcomp>r é r3 rp zoverflow encountered inrÄ Úf)rM r rq r) Úitemsizer r rÌ ÚfloordivrÈ rq r/ rú rì Úoverflow_resultsrh r ÚflattenÚlenr ri ) rZ rw rx ry Zarraysrø Úresultry Zexpected_arrayr r{ r Útest_overflows sn
|
ÿÿ
|
ÿÿþþ ÿþ
|
ÿÿÿþ ÿ
|
ÿÿÿz:TestDivisionIntegerOverflowsAndDivideByZero.test_overflowsN)r] r^ r_ r rg Zhelper_lambdasr r! rb rÌ r] r r r r rÈ rÍ r5 r rÎ r8 ru rv rÏ rü r r r r r rf U sz ÿû ÿ ÿ ÿ ÿ ÿ ÿõ ÿÿ ÿÿ rf c @ s e Zd Zdd Zdd ZdS )ÚTestCbrtc C s t t t d¡d ¡d d S )Nç Àr0 )r r ÚcbrtÚfloat32rY r r r Útest_cbrt_scalarß s zTestCbrt.test_cbrt_scalarc C st t dddt jt j g¡}tt |d ¡| tt t t j¡¡ tt t j¡t j tt t j ¡t j d S )NrL ç @ç Àr0 ) r rq r r r r r r· r rå r r r Ú test_cbrtâ s
|
zTestCbrt.test_cbrtN)r] r^ r_ r r r r r r r
Þ s r
c @ s\ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dS )Ú TestPowerc C sP t dddg¡}t|d dddg t|d | t|d dddg | ¡ }|dC }t|dddg t|d dd
|
d g t|d
|
dt d¡t d ¡g tt jd ddD ]F\}}}dd |D }t|d
|
||d t j||d t|||d qªtt j d ddD ]H\}}}dd |D }t|d
|
||d t j||d t|||d qd S )NrL r ç @r r r% ç @ç "@r3 rj gUUUUUUÕ?r0 Úunaryé ©r) r Úmax_sizec S s g | ]}t |¡qS r ©rb Úsqrtrö r r r r ú s z.TestPower.test_power_float.<locals>.<listcomp>r ©rn r c S s g | ]}t |¡qS r r rö r r r r s )
|
r rq r r
r rb r r r rØ )rZ rP r rn Úinpr Úexpr r r Útest_power_floatì s2 þþzTestPower.test_power_floatc
|
s t dddg¡}t|d dddg t|d | t|d dd d
|
g t|d d d dg t|d dddg t|d dddg t|d dddg t|d d d d g t|d t d¡t d¡t d¡g d|d d t|d fd!d"d#D d$d% }tdt jtdt jfD ]f}t j|gt jd&}t j d'd(< ||d | ||d || ||d || | W 5 Q R X q d S ))Nr y @ @y @ @r rL r r% y À @y À (@y À 8@r0 y &À Ày GÀ "@y @]À F@r1 y À 8Ày À]À ^Ày xÀ uÀr3 yÉ?Ù¿y;±;±Ã?ØØÍ¿y¸
ëQ¸¾?{®GázÄ¿r4 y¸
ëQ¸¾¿{®GázÄ¿yn_Y½K¿&q-²¿yǺ¸ð¿a2U0*©£¿r5 yºI +¶¿ü©ñÒMb?y:бbp¿Ô|Çp¿yF̱¼«~¿¢&ú|g¿rj é c s g | ] }| qS r r rö ©Znormr r r s z0TestPower.test_power_complex.<locals>.<listcomp>)y °©òÀ ~Ï@y 0vA ÕÎAy ãÌôA ¥Þ]âAc S s t | j|j t | j|j d S r+ )r rN Úimagr9 r r r Úassert_complex_equal s z:TestPower.test_power_complex.<locals>.assert_complex_equalrp rU rV )
|
r rq r r rb r r r Úcomplex_rü )rZ rP r r3 r r r Útest_power_complex s2 ÿÿ ÿ z TestPower.test_power_complexc C s t dg¡}t dg¡}t tt jt jg¡}dd }dD ]}|t ||¡| q:|t |d¡| t jddH |t |d ¡| dD ]}|t || ¡| q|t |d
|
¡| W 5 Q R X d S ) Ny y ð? c S s6 t | ¡t |¡ } }t| j|j t| j|j d S r+ )r r r rN r r9 r r r r - s z7TestPower.test_power_zero.<locals>.assert_complex_equal) g
ëQ¸ Õ?rj r ç ø?r% r0 r1 r2 gffffff@r rU rV ù ð?y ð¿É?)r rq r r· Úpowerrü )rZ rl ÚoneÚcnanr Úpr r r Útest_power_zero% s zTestPower.test_power_zeroc C s t dddgt j¡}|d }t|d j|jk t|dddg tt ||¡ t|dddg |t dggg¡ }t|jd d S ) Nr r% r0 r g9b-> @r1 é )r r r0 ) r rq Úint16r r) r Zmay_share_memoryr Úshape)rZ rP r- r r r Útest_fast_power@ s zTestPower.test_fast_powerc C s. t ddgd¡}t ||¡}t|ddg d S )Né Úi8l ïî2±X0)r rq r£ r ©rZ ry r« r r r Útest_integer_powerM s z TestPower.test_integer_powerc C sz t jd }|D ],}t jdd|d}tt |d¡t |¡ qt jd }|D ]*}t jd|d}tt |d¡t |¡ qJd S )Nrp rß r: rp r r; ©r r8 r r r£ Ú ones_like©rZ Údtypesr0 r r r r Ú-test_integer_power_with_integer_zero_exponentR s
|
|
z7TestPower.test_integer_power_with_integer_zero_exponentc C s> t jd }|D ]*}t jd|d}tt d|¡t |¡ qd S )Nr? r: rp r r° r² r r r Útest_integer_power_of_1] s
|
z!TestPower.test_integer_power_of_1c C s@ t jd }|D ],}t jdd|d}tt d|¡t |¡ qd S )Nr? r r: rp r )r r8 r r r£ Ú
|
zeros_liker² r r r Útest_integer_power_of_zeroc s
|
z$TestPower.test_integer_power_of_zeroc C s t jd }|D ]}t jddddg|d}t jddddg|d}t jd|d}t jd|d}ttt j|| ttt j|| ttt j|| ttt j|| qd S ) Nrp r r r% r0 rp r5 r3 )r r8 rq r rw r£ )rZ r³ r0 ry r« r¤ Zminusoner r r Ú test_integer_to_negative_poweri s
|
z(TestPower.test_integer_to_negative_powerc C s t jt jfD ]}t ddddddt jt j g|¡}t t jt j t jt j t jt j t jt j g|¡}t ddt jdt jdt jdg|¡}tt ||¡| q d S )Nr r% r4 r )r r rØ rq r r r£ )rZ r0 ry r« r{ r r r Útest_float_to_inf_poweru s " ÿÿ"z!TestPower.test_float_to_inf_powerN) r] r^ r_ r r r§ r« r¯ r´ rµ r· r¸ r¹ r r r r r ë s r c @ s e Zd Zdd ZdS )ÚTestFloat_powerc C s` d}d}t ||D ]H\}}d||f }tjd|d}t ||¡}t|jjt |¡jk| qd S )Nz?bhilBHILefdgFDGZddddddddddddgDDGzdtin: %s, dtout: %sr rp )rª r r- Z float_powerr r) Úname)rZ Zarg_typeZres_typeZdtinZdtoutr rP r- r r r Útest_type_conversion s z$TestFloat_power.test_type_conversionN)r] r^ r_ r¼ r r r r rº ~ s rº c @ s\ e Zd Zej ddddg¡dd Zej dedd ¡d
|
d Zejj e
|
d d dd Z dS )ÚTestLog2r0 r} r| Úgc C sd ddddddddd d
|
d g }d ddd dddddddg }t j||d}t j||d}tt |¡| d S )Nr r% r1 é é é é@ rH é é é r r0 r2 é r r¨ r: rp )r rq r Úlog2©rZ r0 rP r ÚxfÚyfr r r Útest_log2_values s
|
zTestLog2.test_log2_valuesr÷ r éA c C s( t d| ¡}t|t|d| d d S )Nr zat exponent %dr )r rÇ r r )rZ r÷ Úvr r r Útest_log2_ints s zTestLog2.test_log2_intsrç rè c C sÚ t t d¡d t t tj¡tj tt t tj¡¡ tjdd}t ddt
|
¡ tt t d¡¡ tt t tj ¡¡ t t d¡tj t|d j t
|
k t|d j t
|
k t|d
|
j t
|
k W 5 Q R X d S ) NrL r T)ÚrecordÚalwaysÚ ç ð¿r r r% ) r r rÇ r r r r· ÚwarningsÚcatch_warningsÚfilterwarningsr/ Úcategory)rZ Úwr r r Útest_log2_special s zTestLog2.test_log2_specialN) r] r^ r_ rÈ rÍ rÎ rË rM rÎ r5 r rØ r r r r r½ s
|
|
r½ c @ s e Zd Zdd ZdS )ÚTestExp2c C sn ddddddddd d
|
d g }d ddd dddddddg }dD ]0}t j||d}t j||d}tt |¡| q8d S )Nr r% r1 r¿ rÀ rÁ r rH rà rÄ rÅ r r0 r2 rÆ r r¨ r: ©r} r| r¾ rp )r rq r Úexp2©rZ rP r r0 rÉ rÊ r r r Útest_exp2_valuesª s zTestExp2.test_exp2_valuesN)r] r^ r_ rÝ r r r r rÙ © s rÙ c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) ÚTestLogAddExp2c C s dddddg}dddddg}dddddg}t ddd gdd
|
d
|
gD ]Z\}}t tj||d ¡}t tj||d ¡}t tj||d ¡}tt ||¡||d q@d S © Nr r% r0 r1 r2 rÆ r} r| r¾ r¬ rp ©Údecimal)rª r rÇ rq r Ú
|
logaddexp2© rZ rP r r3 r0 Údec_rÉ rÊ Úzfr r r Útest_logaddexp2_valuesµ s z%TestLogAddExp2.test_logaddexp2_valuesc C sn ddddg}ddddg}ddddg}dD ]@}t j||d}t j||d}t j||d}tt ||¡| q(d S ©Né@B iÀ½ðÿiC iø¼ðÿrÚ rp )r rq r râ ©rZ rP r r3 r0 ÚlogxfÚlogyfÚlogzfr r r Útest_logaddexp2_range¿ s z$TestLogAddExp2.test_logaddexp2_rangec C s² t j}|| || |d| dg}||| | d|d| g}|||| ||ddg}t jddL dD ]@}t j||d}t j||d}t j||d}tt ||¡| qbW 5 Q R X d S ©Nr rù rV rÚ rp )r r rü rq r râ © rZ r rP r r3 r0 rê rë rì r r r Útest_infÉ s zTestLogAddExp2.test_infc C s t t t tjtj¡¡ t t t tjtj¡¡ t t t tjd¡¡ t t t dtj¡¡ t t t tjtj¡¡ d S rj )r r r râ r· r rY r r r Útest_nanÕ s
|
zTestLogAddExp2.test_nanc C sb t tjjtj t tj g ¡tj t tj tj g¡tj t tj tj dg¡d d S rj )r r râ Úidentityr r rY r r r Ú test_reduceÜ s zTestLogAddExp2.test_reduceN)r] r^ r_ ræ rí rð rñ ró r r r r rÞ ³ s
|
|
|
rÞ c @ s e Zd Zdd Zdd ZdS )ÚTestLogc C sà ddddddddd d
|
d g }d ddd dddddddg }dD ]8}d}t j||d}t j||d| }tt |¡| q8t ddd ddg¡}t |¡}tt j||d| dD ]2}t jdd t |¡j}t |¡ W 5 Q R X q¨d S ) Nr r% r1 r¿ rÀ rÁ r rH rà rÄ rÅ r r0 r2 rÆ r r¨ r: rÚ çï9úþB.æ?rp g î?gR¸
ëQî?gl#ö àð?r rù r& )r rq r Úlogrü r[ Úmax©rZ rP r r0 Zlog2_rÉ rÊ r r r Útest_log_valuesä s
|
zTestLog.test_log_valuesc
|
C sÜ t j d¡ t dddddddd g¡}t dd
|
¡}|D ] }t t jjd d |d ¡}| ¡ }d|ddd
< t |¡}t |¡}|D ]T}t t |d d |
¡|d d |
dd t t |d d |
¡|d d |
dd qq6d S )Né* r6 r5 r4 r3 r r% r0 r1 rÝ ç{®Gáz?ç Y@r rL ©Znulp)
|
r r¤ Úseedrq r rØ Úuniformr
rö r ) rZ ÚstridesÚsizesÚiiÚx_f64Z x_specialÚy_trueZ y_specialÚjjr r r Útest_log_stridesø s
|
|
(zTestLog.test_log_stridesN)r] r^ r_ rù r r r r r rô ã s rô c @ s e Zd Zdd Zdd ZdS )ÚTestExpc C sv ddddddddd d
|
d g }d ddd dddddddg }dD ]8}d}t j||d}t j||d| }tt |¡| q8d S )Nr r% r1 r¿ rÀ rÁ r rH rà rÄ rÅ r r0 r2 rÆ r r¨ r: rÚ rõ rp )r rq r r rø r r r Útest_exp_values s zTestExp.test_exp_valuesc
|
C s t j d¡ t dddddddd g¡}t dd
|
¡}|D ]X}t t jjd d |d ¡}t |¡}|D ],}tt |d d |
¡|d d |
dd q`q6d S )Nrú r6 r5 r4 r3 r r% r0 r1 rÝ rû gÍÌÌÌÌ(@r rý ) r r¤ rþ rq r rØ rÿ r r )rZ r r r r r r r r r Útest_exp_strides s
|
zTestExp.test_exp_stridesN)r] r^ r_ r r r r r r r s r c # @ sä e Zd Zdd Zejjeddddd Zejj e
|
ddd d
|
Z ejj e
|
ddd d Z ej d ddddg¡dd Zdd Zejj e
|
dddd Zejj e
|
dddd Zejj e
|
dddd Zejj e
|
ddd d Zd d Zejj e
|
ddd d! Zejj e
|
ddd"d# Zd$d% Zd&d' Zejj e
|
ddd(d) Zejj e
|
ddd*d+ Zejjedddd,d- Zejj e
|
ddd.d/ Z e j e j e j e j!e j"e j#e j$gZ%e j&e j'e j(e j)e j*e j+e j$gZ,e j$gZ-e j+gZ.e j&e j'e j(e j/e j+gZ0ej d0e1¡ej d1d2¡ej d3d4ge.fd4gd5 e.fd6ge,fd6gd5 e,fd7ge-fd7gd5 e-fd8ge0fd8gd5 e0fd9ge0fd9gd5 e0fd:d:d:e j2ge.fd:d:d:e j2gd5 e.fe j2d7d7d7ge-fe j2d7d7d7gd5 e-fe j2gg fe j2gd5 g fd:d:d:e j3ge%e. fd:d:d:e j3gd5 e%e. fe j3d7d7d7ge%fe j3d7d7d7gd5 e%fe j3ge%fe j3gd5 e%fd:d:d:e j3 ge,e% e. fd:d:d:e j3 gd5 e,e% e. fe j3 d7d7d7ge,e% fe j3 d7d7d7gd5 e,e% fe j3 ge,e% fe j3 gd5 e,e% ff ¡d;d< Z4d=S )>ÚTestSpecialFloatsc C s~ t jdddf t jt jt jdg}t jt j t jt j g}dD ]0}t j||d}t j||d}tt |¡| q>W 5 Q R X d S )Nrù )Úunderrõ r ©re r} r| r¾ rp )r rü r· r rq r r rÜ r r r r s z!TestSpecialFloats.test_exp_valuesú2.17z<Older glibc versions may not raise appropriate FP exceptionsrè c C sü t jddj ttt jt d¡ ttt jt d¡ ttt jt d¡ ttt jt d¡ ttt jt d¡ W 5 Q R X t jddj ttt jt d¡ ttt jt d ¡ ttt jt d
|
¡ ttt jt d ¡ ttt jt d
|
¡ W 5 Q R X d S ) Nrù ©rõ g Òo_.&@rü g =`äXáCg @)r g 1Àg @Àg =`äXáÃ)r rü r rþ r Úfloat16r rØ rY r r r Útest_exp_exceptions% s z%TestSpecialFloats.test_exp_exceptionsrç c C s t jddÐ t jt jt jt jt j t jg}t jt j t jt j ddg}t jt j t jt j ddg}dD ]n}t j||d}t j||d}t j||d}tt |¡| tt |¡| tt |¡| tt |¡| qhW 5 Q R X t jdd p d
|
D ]d}t
|
t t jt jd|d t
|
t t jt jd|d t
|
t t jt jd|d t
|
t t j t jd|d qôW 5 Q R X t jdd â d
|
D ]Ö}t
|
t t jt jt j |d t
|
t t jt jd|d t
|
t t jt jt j |d t
|
t t jt jd|d t
|
t t jt jt j |d t
|
t t jt jd|d t
|
t t j t jt j |d t
|
t t j t jd|d qvW 5 Q R X t t jd d d}t |¡ W 5 Q R X d S )NrU r& r rÒ ç Àr rp rù rñ ©re r} r| rV g eÍÍAr ) r rü r· r rq r rö rÇ Úlog10Úlog1pr rþ r ) rZ rP r Zy1pr0 rÉ rÊ Zyf1pry r r r rù 8 sn ÿ ÿ ÿ ÿÿ ÿÿ ÿÿ ÿÿ ÿz!TestSpecialFloats.test_log_valuesc C sî t jddx t jt jt jt jg}t jt j t jt j g}dD ]@}t j||d}t j||d}tt |¡| tt |¡| q>W 5 Q R X t jddN t jt jfD ]:}t jt j fD ]&}dD ] }tt |t j|g|d q¾q¶q¤W 5 Q R X d S )NrU r& r rp rù rV r )
|
r rü r· r rq r ÚsinÚcosr rþ )rZ rP r r0 rÉ rÊ ÚcallableÚvaluer r r Útest_sincos_valuesj s ÿz$TestSpecialFloats.test_sincos_valuesr0 re r} r| r¾ c C sx t jddb t jt jt jt jdg}t jt j t jt j dg}t j||d}t j||d}tt |¡| W 5 Q R X d S )NrU r& r rp )r rü r· r rq r r rÈ r r r Útest_sqrt_values} s z"TestSpecialFloats.test_sqrt_valuesc C sv t jt jt jt jddddg}t jt j t jt j ddddg}dD ]0}t j||d}t j||d}tt |¡| q@d S )Nr rL rK rÒ r rp )r r· r rq r rO rÜ r r r Útest_abs_values s z!TestSpecialFloats.test_abs_valuesc C sÞ t jt jt jt jg}t jt j t jt j g}t jdd< dD ]0}t j||d}t j||d}tt |¡| q>W 5 Q R X t jddN ttt jt jddd ttt jt jd d
|
d ttt jt jd d d W 5 Q R X d S ) NrU r& r rp rù r ç @@re gnµµ¸Fr} gZb××çtir| ) r r· r rü rq r Úsquarer rþ rÜ r r r Útest_square_values s" ÿ ÿ ÿz$TestSpecialFloats.test_square_valuesc C s t jddr t jt jddt jt j g}t jt j t jt j ddg}dD ]0}t j||d}t j||d}tt |¡| qHW 5 Q R X t jdd( dD ] }ttt jt jd|d qW 5 Q R X d S ) NrU r& r rK r rp rù rñ ) r rü r· r rq r Ú
|
reciprocalr rþ rÜ r r r Útest_reciprocal_values¥ s ÿz(TestSpecialFloats.test_reciprocal_valuesc C sÞ t jddp t jt j ddt jt j g}t jt jddt jt jg}dD ]0}t j||d}t j||d}tt |¡| qFW 5 Q R X t jddF dD ]:}ttt jt jt j|d ttt jt jt j |d qW 5 Q R X d S ) NrU r& r rK r rp rù rV ) r rü r· r rq r Útanr rþ ©rZ Úin_rn r0 Úin_arrÚout_arrr r r Útest_tan´ s ÿÿzTestSpecialFloats.test_tanc C sð t jddx t jt j t jt j g}t jt jt jt jg}dD ]@}t j||d}t j||d}tt |¡| tt |¡| q>W 5 Q R X t jt jfD ]T}t jt j ddfD ]<}dD ]2}t jdd tt |t j||d W 5 Q R X q´q¬qd S ) NrU r& r rp r r rù rV )
|
r rü r· r rq r ÚarcsinÚarccosr rþ )rZ r# rn r0 r$ r% r r r r r Útest_arcsincosÅ s ÿz TestSpecialFloats.test_arcsincosc C sl t jddV t jt j g}t jt jg}dD ]0}t j||d}t j||d}tt |¡| q,W 5 Q R X d S )NrU r& r rp )r rü r· rq r Úarctanr" r r r Ú test_arctan× s z TestSpecialFloats.test_arctanc C sÈ t jt j t jt j g}t jt jt jt j g}dD ]0}t j||d}t j||d}tt |¡| q2t jddN ttt jt jddd ttt jt jddd ttt jt jd d
|
d W 5 Q R X d S © Nr rp rù r ç (@re g ^@r} g À@r| ) r r· r rq r Úsinhrü r rþ r" r r r Ú test_sinhà s ÿ ÿ ÿzTestSpecialFloats.test_sinhc C sÆ t jt j t jt j g}t jt jt jt jg}dD ]0}t j||d}t j||d}tt |¡| q0t jddN ttt jt jddd ttt jt jddd ttt jt jd d
|
d W 5 Q R X d S r, ) r r· r rq r Úcoshrü r rþ r" r r r Ú test_coshñ s ÿ ÿ ÿzTestSpecialFloats.test_coshc C sb t jt j t jt j g}t jt jddg}dD ]0}t j||d}t j||d}tt |¡| q,d S )NrL rÒ r rp )r r· r rq r Útanhr" r r r Ú test_tanh s zTestSpecialFloats.test_tanhc C sh t jt j t jt j g}t jt jt jt j g}dD ]0}t j||d}t j||d}tt |¡| q2d S )Nr rp )r r· r rq r rM r" r r r Ú test_arcsinh
|
s z TestSpecialFloats.test_arcsinhc
|
C sÔ t jddr t jt j t jt j ddg}t jt jt jt jdt jg}dD ]0}t j||d}t j||d}tt |¡| qHW 5 Q R X dt j fD ]>}t jdd( dD ] }ttt jt j||d q¦W 5 Q R X qd S ) NrU r& rL r r rp rù rV ) r rü r· r rq r Úarccoshr rþ ©rZ r# rn r0 r$ r% r r r r Ú test_arccosh s ÿz TestSpecialFloats.test_arccoshc
|
C sè t jdd| t jt j t jt j dddg}t jt jt jt jt jt j t jg}dD ]0}t j||d}t j||d}tt |¡| qRW 5 Q R X dt jt j ddfD ]@}t jd d d
|
( dD ] }ttt jt j||d qºW 5 Q R X q¢d S ) NrU r& rL rÒ r r rp g)\Âõ(ð?rù r ) r rü r· r rq r Úarctanhr rþ r6 r r r Ú test_arctanh" s " ÿz TestSpecialFloats.test_arctanhc
|
C s t jddf t jt j t jt j g}t jt jt jdg}dD ]0}t j||d}t j||d}tt |¡| q<W 5 Q R X dD ]@}t jddd( dD ] }ttt jt j||d qW 5 Q R X q|d S ) NrU r& r r rp )ç @@g @Àrù )rõ r ) r rü r· r rq r rÛ r rþ r6 r r r Ú test_exp23 s ÿzTestSpecialFloats.test_exp2c
|
C sÀ t jddf t jt j t jt j g}t jt jt jdg}dD ]0}t j||d}t j||d}tt |¡| q<W 5 Q R X dD ]>}t jdd( d D ] }ttt jt j||d qW 5 Q R X q|d S )
|
NrU r& rÒ r rp )g i@r: rù r )re r} ) r rü r· r rq r Úexpm1r rþ r6 r r r Ú
|
test_expm1F s ÿz TestSpecialFloats.test_expm1r r) r z data, escapeg¸
ëQ¸?rÁ rÒ rL r rK rj c C sX |r||krd S |t jt jfkr,|dkr,d S t j||d}t || W 5 Q R X d S )Nre rp )r ÚspacingÚceilrq r )rZ r r) ÚdataÚescaperq r r r Ú test_unary_spurious_fpexceptionh s ) z1TestSpecialFloats.test_unary_spurious_fpexceptionN)5r] r^ r_ r rÈ rÍ rd r r r5 r rù r rÎ r r r r r& r) r+ r/ r1 r3 r4 r7 r9 r; r= r r r r! r( r' r> r8 ZINF_INVALID_ERRrö rÇ r r r r5 ZNEG_INVALID_ERRZONE_INVALID_ERRZLTONE_INVALID_ERRr Z
|
BYZERO_ERRÚUFUNCS_UNARY_FPr· r rB r r r r r
|
sÌ
|
þ
|
|
1
|
|
|
|
|
|
|
|
|
|
þ
|
|
ÿ þÿÿ ÿ
|
|
|
ÿ
|
ÿÜ&r
|
c @ s. e Zd Zej dddddddg¡dd Zd
|
S ) Ú TestFPClassrA r6 r4 r3 r r% r1 c C sø t jt jt j t jt j ddddddg
|
dd}t jt jt j t jt j ddddd d
|
g
|
d d}t d d d d d d d d d d g
|
¡}t d d d d d d d d d d g
|
¡}t d d d d d d d d d d g
|
¡}t d d d d d d d d d d g
|
¡}tt |d d |
¡|d d |
tt |d d |
¡|d d |
tt |d d |
¡|d d |
tt |d d |
¡|d d |
tt |d d |
¡|d d |
tt |d d |
¡|d d |
tt |d d |
¡|d d |
tt |d d |
¡|d d |
d S )NrÒ rL rK r giV½Q giV½Q r| rp g¶J-& 6g¶J-& ¶r} TF) r rq r· r r r r r# Úisfinite)rZ rA Zarr_f64Zarr_f32r· r ÚsignZfiniter r r Ú test_fpclass s .. $$$$$$$zTestFPClass.test_fpclassN)r] r^ r_ rÈ rÍ rÎ rG r r r r rD s rD c @ s@ e Zd Zej dddddddg¡ej dd d
|
g¡d d Zd S )Ú TestLDExprA r6 r4 r3 r r% r1 r) r} r| c C s¾ t jddddddddg|d}t jd d
|
d d d d ddgdd}t jd|d}tt j|d d |
|d d |
|d d |
dt jd|dd d |
t|d d |
t jd|dd d |
d S )Ng À?ç Ð?rj rL r r g @rp r0 r% r r r3 r4 r5 r÷ r¿ r )r rq r× r Úldexpr- )rZ r) rA Úmantr rn r r r Ú
|
test_ldexp¯ s
|
HzTestLDExp.test_ldexpN)r] r^ r_ rÈ rÍ rÎ rL r r r r rH ® s rH c @ sZ e Zd Zej dddddddg¡ej dd d
|
g¡ejjej d ¡ d d dd Z
|
dS )Ú TestFRExprA r6 r4 r3 r r% r1 r) r} r| ÚlinuxzAnp.frexp gives different answers for NAN/INF on windows and linuxrè c
|
C s* t jt jt jt jt j ddddg|d}t jt jt jt jt j ddddg|d}t jddddddd d gd
|
d}t jd |d}d t jd d
|
d }t j|d d |
|d d |
|d d |
fd \}} t|d d |
| t|d d |
| t|d d |
|d d |
t|d d |
|d d |
d S )Nr rK rL rÒ rp rj ç à¿r r r÷ r¿ r% r )r rq r· r r- ru r )
|
rZ r) rA r Z mant_trueÚexp_trueZout_mantZout_exprK r r r r Ú
|
test_frexp¹ s (( 4 zTestFRExp.test_frexpN) r] r^ r_ rÈ rÍ rÎ r5 re rF rI rQ r r r r rM ¸ s ÿrM r r rü ç YÀrL )r Úabsoluter r ÚrintÚfloorr? Útruncc @ s e Zd Zdd ZdS )Ú TestAVXUfuncsc
|
C sp t ddddddddg¡}t j d ¡ t ¡ D ]:\}}|d
|
}|d }|d }tdd D ] }tt |}t t jj |||d ¡} t
|
| ¡}
|
t | ¡} || } |d
|
krÐt || t | ¡ t ||
|
t
|
| ¡ n0t || t | ¡|d t ||
|
t
|
| ¡|d |dkrZ|| } ||
|
}|D ]J}t ||
|
d d |
|d d |
t || d d |
| d d |
q qZq.d S )Nr6 r5 r4 r3 r r% r0 r1 rú r rÁ r ©Zmaxulp)r rq r¤ rþ Ú
|
avx_ufuncsÚitemsrM Úgetattrr rÿ rØ Ú
|
longdoubler r )rZ r ÚfuncÚpropZ maxulperrÚminvalÚmaxvalr ZmyfuncÚx_f32r Zx_f128Z y_true128Zy_true32Zy_true64r r r r Útest_avx_based_ufuncÔ s:
|
ÿ
|
|
ÿÿ"z"TestAVXUfuncs.test_avx_based_ufuncN)r] r^ r_ rb r r r r rW Ó s rW c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
Ú TestAVXFloat32Transcendentalc C sR t j d¡ t t jjdddd¡}t |¡}tt |¡t t |¡¡dd d S )Nrú r çfffffV@rè r r0 rX )r r¤ rþ r rÿ rØ r r ©rZ ra r r r r Útest_exp_float32ô s
|
z-TestAVXFloat32Transcendental.test_exp_float32c C sR t j d¡ t t jjdddd¡}t |¡}tt |¡t t |¡¡dd d S )Nrú r r rè r r1 rX )r r¤ rþ r rÿ rØ r rö re r r r Útest_log_float32ú s
|
z-TestAVXFloat32Transcendental.test_log_float32c C s t j d¡ d}t |d ¡}t jjd||d}t t jjdd|d¡}tdsjt d t j |¡ ¡||< t |¡}t
|
t |¡t t |¡¡d
|
d t
|
t |¡t t |¡¡d
|
d | ¡ }t
|
t j ||d t t |¡¡d
|
d t
|
t j ||d t t |¡¡d
|
d d S ) Nrú rè é r r rR rü r g èvH7Br% rX r )r r¤ rþ Úint_r¦ r rÿ r ZrandrØ r r r r
)rZ ÚNr Úindexra r Ztx_f32r r r Útest_sincos_float32 s
|
$z0TestAVXFloat32Transcendental.test_sincos_float32c
|
C sB t j d¡ t dddddddd g¡}t dd
|
¡}|D ]}t t jjd d |d ¡}| ¡ }d|ddd
< t |¡}t |¡}t
|
|¡}t |¡} |D ]¤}
|
t t |d d |
|
¡|d d |
|
dd t t |d d |
|
¡|d d |
|
dd t t
|
|d d |
|
¡|d d |
|
dd t t |d d |
|
¡| d d |
|
dd qq6d S )Nrú r6 r5 r4 r3 r r% r0 r1 rÝ rû rd r g Lý@rý ) r r¤ rþ rq r r rÿ r
r rö r r r ) rZ r r r ra Z x_f32_largerP Zlog_trueZsin_trueZcos_truer r r r Útest_strided_float32 s
|
|
|
|
|
(((z1TestAVXFloat32Transcendental.test_strided_float32N)r] r^ r_ rf rg rl rm r r r r rc ó s rc c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) Ú TestLogAddExpc C s dddddg}dddddg}dddddg}t ddd gdd
|
d
|
gD ]Z\}}t tj||d ¡}t tj||d ¡}t tj||d ¡}tt ||¡||d q@d S rß )rª r rö rq r Ú logaddexprã r r r Útest_logaddexp_values% s z#TestLogAddExp.test_logaddexp_valuesc C sn ddddg}ddddg}ddddg}dD ]@}t j||d}t j||d}t j||d}tt ||¡| q(d S rç )r rq r ro ré r r r Útest_logaddexp_range/ s z"TestLogAddExp.test_logaddexp_rangec C s² t j}|| || |d| dg}||| | d|d| g}|||| ||ddg}t jddL dD ]@}t j||d}t j||d}t j||d}tt ||¡| qbW 5 Q R X d S rî )r r rü rq r ro rï r r r rð 9 s zTestLogAddExp.test_infc C s t t t tjtj¡¡ t t t tjtj¡¡ t t t tjd¡¡ t t t dtj¡¡ t t t tjtj¡¡ d S rj )r r r ro r· r rY r r r rñ E s
|
zTestLogAddExp.test_nanc C s, t tjjtj t tj g ¡tj d S r+ )r r ro rò r r rY r r r ró L s zTestLogAddExp.test_reduceN)r] r^ r_ rp rq rð rñ ró r r r r rn $ s
|
|
|
rn c @ s e Zd Zdd Zdd ZdS )Ú TestLog1pc C s0 t t d¡t d¡ t t d¡t d¡ d S )NçÉ?g333333ó?çíµ ÷Æ°>g zo ð?)r rb r rö rY r r r Ú
|
test_log1pR s zTestLog1p.test_log1pc C s t jdddj tt t j¡t j tt t j¡t j tt d¡t j tt d¡t j tt t j ¡t j W 5 Q R X d S )NrU r rÒ r )r rü r rb r r· r rY r r r Ú test_specialV s zTestLog1p.test_specialN)r] r^ r_ ru rv r r r r rr Q s rr c @ s$ e Zd Zdd Zdd Zdd ZdS )Ú TestExpm1c C s8 t t d¡t d¡d t t d¡t d¡d d S )Nrs r rt )r rb r< r rY r r r r= ` s zTestExpm1.test_expm1c C s` t t tj¡tj t t d¡d t t d¡d t t tj¡tj t t tj ¡d d S )Nr rK rÒ )r rb r< r r rY r r r rv d s
|
zTestExpm1.test_specialc C s: t d¡}t|t |¡ | t j¡}t|t |¡ d S )Ngê-q=)r r r rb r< r§ r rå r r r Ú test_complexk s
|
zTestExpm1.test_complexN)r] r^ r_ r= rv rx r r r r rw _ s rw c @ s e Zd Zdd Zdd ZdS )Ú TestHypotc C s. t t dd¡t d¡ t t dd¡d d S ©Nr r% r )r rb Úhypotr rY r r r Ú test_simples s zTestHypot.test_simplec C s\ t tj ddg¡d t tj dddg¡d t tj dddg¡d ttj g ¡d d S )
|
Nr r ç @r r r- ç 4@g 9@r )r rb r{ r r rY r r r ró w s zTestHypot.test_reduceN)r] r^ r_ r| ró r r r r ry r s ry c
|
C sF t jdd0 tt t | |¡¡d| |t | |¡f W 5 Q R X d S )NrU rV z hypot(%s, %s) is %s, not nan)r rü r r rb r{ r9 r r r Úassert_hypot_isnan~ s ÿr c
|
C sF t jdd0 tt t | |¡¡d| |t | |¡f W 5 Q R X d S )NrU rV z hypot(%s, %s) is %s, not inf)r rü r r rb r{ r9 r r r Úassert_hypot_isinf s ÿr c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestHypotSpecialValuesc C s t tjtj t tjd d S ©Nr )r r r· rY r r r Útest_nan_outputs s z'TestHypotSpecialValues.test_nan_outputsc C sR t tjtj t tjtj t tjd t dtj t tjtj t tjd d S )Nr g 7@)r r r· r rY r r r Útest_nan_outputs2 s z(TestHypotSpecialValues.test_nan_outputs2c C s t tjtjd d S rj )r rb r{ r r rY r r r Ú test_no_fpe s z"TestHypotSpecialValues.test_no_fpeN)r] r^ r_ r r r
r r r r r s r c C s. t t t | |¡¡d| |t | |¡f d S )Nz arctan(%s, %s) is %s, not nan)r r r rb Úarctan2r9 r r r Úassert_arctan2_isnan s r c C s> t t t | |¡¡o"t | |¡dkd| |t | |¡f d S )Nr z arctan(%s, %s) is %s, not +inf©r r r rb r r9 r r r Úassert_arctan2_ispinf s r c C s> t t t | |¡¡o"t | |¡dk d| |t | |¡f d S )Nr z arctan(%s, %s) is %s, not -infr r9 r r r Úassert_arctan2_isninf£ s r c C s@ t t | |¡dko$t t | |¡¡ d| |t | |¡f d S )Nr z arctan(%s, %s) is %s, not +0©r rb r r r# r9 r r r Úassert_arctan2_ispzero§ s r c C s> t t | |¡dko"t t | |¡¡d| |t | |¡f d S )Nr z arctan(%s, %s) is %s, not -0r r9 r r r Úassert_arctan2_isnzero« s r c @ st e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd ZdS ) ÚTestArctan2SpecialValuesc C sL t t dd¡dtj t t dd¡dtj t t dd¡dtj d S )Nr rI r3 ç пç è?)r rb r r rc rY r r r Ú test_one_one° s z%TestArctan2SpecialValues.test_one_onec C s6 t t tjtj¡tj t t tjtj¡tj d S r+ )r rb r r ÚPZEROÚNZEROrc rY r r r Útest_zero_nzero¶ s z(TestArctan2SpecialValues.test_zero_nzeroc C s t tjtj ttjtj d S r+ ©r r r r r rY r r r Útest_zero_pzero» s z(TestArctan2SpecialValues.test_zero_pzeroc C s2 t t tjd¡tj t t tjd¡tj d S )Nr3 ©r rb r r r rc r rY r r r Útest_zero_negativeÀ s z+TestArctan2SpecialValues.test_zero_negativec C s t tjd ttjd d S r r rY r r r Útest_zero_positiveÅ s z+TestArctan2SpecialValues.test_zero_positivec C s8 t t dtj¡dtj t t dtj¡dtj d S )Nr rj r rY r r r Útest_positive_zeroÊ s z+TestArctan2SpecialValues.test_positive_zeroc C s8 t t dtj¡dtj t t dtj¡dtj d S )Nr3 rO r rY r r r Útest_negative_zeroÏ s z+TestArctan2SpecialValues.test_negative_zeroc C s2 t t dtj¡tj t t dtj¡tj d S ©Nr r3 )r rb r r ZNINFrc rY r r r Ú test_any_ninfÔ s z&TestArctan2SpecialValues.test_any_ninfc C s t dtj tdtj d S r )r r r r rY r r r Ú test_any_pinfÙ s z&TestArctan2SpecialValues.test_any_pinfc C s: t t tjd¡dtj t t tj d¡dtj d S )Nr rj rO ©r rb r r r rc rY r r r Ú test_inf_anyÞ s z%TestArctan2SpecialValues.test_inf_anyc C sB t t tjtj ¡dtj t t tj tj ¡dtj d S )Nr g è¿r rY r r r Ú test_inf_ninfã s z&TestArctan2SpecialValues.test_inf_ninfc C s> t t tjtj¡dtj t t tj tj¡dtj d S )NrI r r rY r r r Ú test_inf_pinfè s z&TestArctan2SpecialValues.test_inf_pinfc C s. t tjtj t tjtj t tjtj d S r+ )r r r· r rY r r r Ú test_nan_anyí s z%TestArctan2SpecialValues.test_nan_anyN)r] r^ r_ r r r r r r r r r r r¡ r¢ r£ r r r r r ¯ s r c @ s$ e Zd Zdd Zdd Zdd ZdS )Ú TestLdexpc C sp t t t dtj¡t d|¡¡d t t t dtj¡t d|¡¡d t t t dtj¡t d|¡¡d d S )Nr r0 ç 0@)r rb rJ r rq r rØ r\ )rZ Útpr r r Ú _check_ldexpõ s
|
ÿÿ
|
ÿÿ
|
ÿÿzTestLdexp._check_ldexpc C sN t t dd¡d | tj¡ | tj¡ | tj¡ | d¡ | d¡ d S )Nr r0 r¥ r÷ rÆ )r rb rJ r§ r Úint8r© rt rY r r r rL ý s
|
zTestLdexp.test_ldexpc C sf t jddP t t d¡¡j}t t d¡¡j}tt d|¡t j tt d|¡d W 5 Q R X d S )NrU r rÆ r r )
|
r rü rú r) r÷ rì r rb rJ r )rZ ZimaxZiminr r r Útest_ldexp_overflow s
|
z TestLdexp.test_ldexp_overflowN)r] r^ r_ r§ rL r© r r r r r¤ ô s
|
r¤ c @ sL e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dS )Ú TestMaximumc C sì t jd }t jd }t d¡}|d d d
}t jj}|D ]4}| |¡}| |¡}t||d t||d q8|D ]t}| |¡}| |¡}t||d t||d t j|d d d
< t j|d d d
< t||t j t||t j qrd S )NÚAllFloatr? r r3 r: r% )r r8 r Úmaximumr r§ r r· © rZ ZdfltZdintZseq1Zseq2r] r0 Ztmp1Ztmp2r r r ró s&
|
|
|
|
|
|
|
zTestMaximum.test_reducec C s0 t tj ddg¡d t tj ddg¡d d S ©Nr ù @y ð? @)r r r¬ r rY r r r Útest_reduce_complex& s z TestMaximum.test_reduce_complexc C sL t j}t d||g¡}t |d|g¡}t |||g¡}tt ||¡| d S rj )r r· rq r r¬ ©rZ r· Úarg1Úarg2rn r r r Útest_float_nans* s
|
zTestMaximum.test_float_nansc C s^ t dD ]P}t tdt¡}d}t tdt¡}tt ||¡dk tt ||¡dk qd S ©Nr r· rL )rM r rq r rµ r r¬ ©rZ r÷ rP r r3 r r r Útest_object_nans1 s z TestMaximum.test_object_nansc C sz t j}t|dtd|t||fD ]R}t jd||gtd}t j|d|gtd}t j|||gtd}tt ||¡| q"d S ©Nr rp )r r· r rq r r¬ ©rZ r· r¥ r² r³ rn r r r Útest_complex_nans< s z TestMaximum.test_complex_nansc C s, t jdtd}|d }tt ||¡| d S ©Nr2 rp r )r r rµ r r¬ ©rZ r² r³ r r r Útest_object_arrayD s z TestMaximum.test_object_arrayc
|
C s t ddddt jt j t jt j g¡}t ddt jddt jddg¡}t ddt jdt jt jt jdg¡}t d¡}t ddddddt jdg¡}tt ||¡| tt |d d d
|d d d
¡|d d d
tt |d d
|
|d d d
¡t dt jddg¡ tt |d d d
|d d
¡t ddt jg¡ tt j|d d d
|d d d
|d d d
d t ddt jg¡ t|| d S ©Ng ÀrL g $@r r rÒ r r¿ r% r1 r0 rÆ r )r rq r· r r- r r¬ )rZ r> r? Zmaxtruern Z out_maxtruer r r Útest_strided_arrayI s & "
|
042Bz TestMaximum.test_strided_arrayc C s¾ t jt jt jt jg}|D ] }t |¡j}t |¡j}|d}t |t j ¡}|t j |f|t j |f|||f|t j
|
t j
|
fg}|D ]8\}} }
|
t t |g| g¡|
|
g t t j || g¡|
|
q~qd S ©Nrá )r r r rØ r\ r[ rì r÷ ra r r· r r¬ r © rZ r³ r0 ZdtminZdtmaxÚd1Zd1_nextZ
|
test_casesrQ rR Úexpectedr r r Útest_precisionV s ûzTestMaximum.test_precisionN© r] r^ r_ ró r° r´ r· rº r½ r¿ rÄ r r r r rª s rª c @ sL e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dS )Ú TestMinimumc C sì t jd }t jd }t d¡}|d d d
}t jj}|D ]4}| |¡}| |¡}t||d t||d q8|D ]t}| |¡}| |¡}t||d t||d t j|d d d
< t j|d d d
< t||t j t||t j qrd S )Nr« r? r r3 r r% )r r8 r Úminimumr r§ r r· r r r r ró m s&
|
|
|
|
|
|
|
zTestMinimum.test_reducec C s0 t tj ddg¡d t tj ddg¡d d S r® )r r rÇ r rY r r r r° s z TestMinimum.test_reduce_complexc C sL t j}t d||g¡}t |d|g¡}t |||g¡}tt ||¡| d S rj )r r· rq r rÇ r± r r r r´ s
|
zTestMinimum.test_float_nansc C s^ t dD ]P}t tdt¡}d}t tdt¡}tt ||¡dk tt ||¡dk qd S rµ )rM r rq r rµ r rÇ r¶ r r r r· s z TestMinimum.test_object_nansc C sz t j}t|dtd|t||fD ]R}t jd||gtd}t j|d|gtd}t j|||gtd}tt ||¡| q"d S r¸ )r r· r rq r rÇ r¹ r r r rº s z TestMinimum.test_complex_nansc C s, t jdtd}|d }tt ||¡| d S r» )r r rµ r rÇ r¼ r r r r½ s z TestMinimum.test_object_arrayc
|
C s t ddddt jt j t jt j g¡}t ddt jddt jddg¡}t ddt jdt jt jdt j g¡}t d¡}t ddddddt jdg¡}tt ||¡| tt |d d d
|d d d
¡|d d d
tt |d d
|
|d d d
¡t dt jddg¡ tt |d d d
|d d
¡t ddt jg¡ tt j|d d d
|d d d
|d d d
d t ddt jg¡ t|| d S r¾ )r rq r· r r- r rÇ )rZ r> r? Zmintruern Z out_mintruer r r r¿ ¥ s & $
|
042Bz TestMinimum.test_strided_arrayc C sº t jt jt jt jg}|D ]}t |¡j}t |¡j}|d}t |t j ¡}|t j |f|t j |f|||f|t j
|
t j
|
fg}|D ]8\}} }
|
t t |g| g¡|
|
g t t j || g¡|
|
qzqd S rÀ )r r r rØ r\ r[ rì r÷ ra r r· r rÇ r rÁ r r r rÄ ² s
|
|
ûzTestMinimum.test_precisionNrÅ r r r r rÆ l s rÆ c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) ÚTestFmaxc C sè t jd }t jd }t d¡}|d d d
}t jj}|D ]4}| |¡}| |¡}t||d t||d q8|D ]p}| |¡}| |¡}t||d t||d t j|d d d
< t j|d d d
< t||d t||d qrd S )Nr« r? r r3 r: r% r¨ )r r8 r Úfmaxr r§ r r· r r r r ró É s&
|
|
|
|
|
|
|
zTestFmax.test_reducec C s0 t tj ddg¡d t tj ddg¡d d S r® )r r rÉ r rY r r r r° Þ s z TestFmax.test_reduce_complexc C sL t j}t d||g¡}t |d|g¡}t dd|g¡}tt ||¡| d S rj )r r· rq r rÉ r± r r r r´ â s
|
zTestFmax.test_float_nansc C sz t j}t|dtd|t||fD ]R}t jd||gtd}t j|d|gtd}t jdd|gtd}tt ||¡| q"d S r¸ )r r· r rq r rÉ r¹ r r r rº é s zTestFmax.test_complex_nansc C s¼ t jt jt jt jg}|D ]}t |¡j}t |¡j}|d}t |t j ¡}|t j |f|t j |f|||f|t j
|
|fg}|D ]8\}} }
|
t t |g| g¡|
|
g t t j || g¡|
|
q|qd S rÀ )r r r rØ r\ r[ rì r÷ ra r r· r rÉ r rÁ r r r rÄ ñ s
|
ûzTestFmax.test_precisionN©r] r^ r_ ró r° r´ rº rÄ r r r r rÈ È s
|
rÈ c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) ÚTestFminc C sè t jd }t jd }t d¡}|d d d
}t jj}|D ]4}| |¡}| |¡}t||d t||d q8|D ]p}| |¡}| |¡}t||d t||d t j|d d d
< t j|d d d
< t||d t||d qrd S )Nr« r? r r3 r r% r )r r8 r Úfminr r§ r r· r r r r ró s&
|
|
|
|
|
|
|
zTestFmin.test_reducec C s0 t tj ddg¡d t tj ddg¡d d S r® )r r rÌ r rY r r r r° s z TestFmin.test_reduce_complexc C sL t j}t d||g¡}t |d|g¡}t dd|g¡}tt ||¡| d S rj )r r· rq r rÌ r± r r r r´ ! s
|
zTestFmin.test_float_nansc C sz t j}t|dtd|t||fD ]R}t jd||gtd}t j|d|gtd}t jdd|gtd}tt ||¡| q"d S r¸ )r r· r rq r rÌ r¹ r r r rº ( s zTestFmin.test_complex_nansc C s¸ t jt jt jt jg}|D ]}t |¡j}t |¡j}|d}t |t j ¡}|t j |f|t j |f|||f|t j
|
|fg}|D ]8\}} }
|
t t |g| g¡|
|
g t t j || g¡|
|
qxqd S rÀ )r r r rØ r\ r[ rì r÷ ra r r· r rÌ r rÁ r r r rÄ 0 s
|
|
|
ûzTestFmin.test_precisionNrÊ r r r r rË s
|
rË c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
ÚTestBoolc C s@ t jdt jd}ttt j| ttt j| ttt j|| d S )Nr rp )r r- r£ r rv ÚnegativeÚpositiveÚsubtractr¾ r r r Útest_exceptionsG s zTestBool.test_exceptionsc C sþ ddddg}ddddg}t jd t jd d }tt j|D ]¼}t j||d}t j||d}d d
|
d
|
d
|
g}t jt jfD ]}t||| t ¡| qtd d d d
|
g}t j
|
t j fD ]}t||| t ¡| q¨d d
|
d
|
d g}t j t j fD ]}t||| t ¡| qÜq<d S ) Nr r0 r% r1 r« r? ú?rp FT)r r8 Úmapr) r Ú
|
logical_orr¬ r r§ r½ Ú logical_andrÇ Ú logical_xorr» ) rZ Zinput1Zinput2r8 r) r² r³ rn r] r r r Útest_truth_table_logicalM s& ÿþ z!TestBool.test_truth_table_logicalc C sv ddddg}ddddg}ddddg}t t ||¡| ddddg}t t ||¡| ddddg}t t ||¡| d S )NFT)r r Ú
|
bitwise_orÚ bitwise_andÚ bitwise_xor)rZ r² r³ rn r r r Útest_truth_table_bitwisef s z!TestBool.test_truth_table_bitwisec C sÀ t ddddgt¡}t ddddgt¡}t ddddgt¡}t g t¡}||||g}|D ]}tt j |¡t| qX|D ]}tt j |¡t| qx|D ]"}tt j |¡|
|
¡ d dk qd S )Nr r r% ) r rq r½ r rÕ r rý rÔ ÚanyrÖ rÙ )rZ ÚnoneZsomeZeveryrr Zarrsr r r r ró s s zTestBool.test_reduceN)r] r^ r_ rÑ r× rÛ ró r r r r rÍ F s rÍ c @ s: e Zd Zdd dD Zdd Zdd Zdd Zd
|
d Zd S ) ÚTestBitwiseUFuncsc C s g | ]}t |¡qS r )r r) )r Úcr r r r s z TestBitwiseUFuncs.<listcomp>z ?bBhHiIlLqQOc C sn | j D ]`}tjdg|d}t dg¡ |¡}d|j }tt |¡||d tt |¡||d tt ||¡||d tt ||¡||d tt ||¡||d tt ||¡||d tt ||¡||d tt ||¡||d tt ||¡||d tt ||¡||d tt ||¡||d tt ||¡||d tt ||¡||d tt ||¡||d qd S )Nr rp r3 ú dt = '%s'r )
|
Ú bitwise_typesr rq r§ Úcharr Ú bitwise_notrØ rÚ rÙ ©rZ r0 r× r- r r r r Ú test_values s$
|
z TestBitwiseUFuncs.test_valuesc C s | j D ]}tjdg|d}t dg¡ |¡}d|j }tt |¡j|k| tt ||¡j|k| tt ||¡j|k| tt
|
||¡j|k| qd S )Nr rp r3 rà ) rá r rq r§ râ r rã r) rØ rÚ rÙ rä r r r Ú
|
test_types¡ s
|
|
z TestBitwiseUFuncs.test_typesc C s: t tjjdkd t tjjdkd t tjjdkd d S )Nr rØ rÚ r3 rÙ )r r rØ rò rÚ rÙ rY r r r Ú test_identity¬ s z TestBitwiseUFuncs.test_identityc C sr t jt jt jf}| jD ]d}t jdg|d}t dg¡ |¡}|D ]8}d||f }t| |¡||d t| |¡||d q@q| jd d
D ]d}t jg |d}|D ]L}d||f }t |j ¡ |¡}| |¡} t| ||d t
|
| j |j k| q q|D ]:}d|f }t jg t d}|j }| |¡} t| ||d qô|D ]8}d|f }t jdgt d}
|
t
|
t | |
|
¡tk| q4d S )Nr rp r3 zdt: '%s', f: '%s'r zdt: '%s'T)r rØ rÚ rÙ rá rq r§ r r rò r r) rµ r r½ ) rZ Z binary_funcsr0 r× r- r} r rr rQ r- Zbtyper r r Útest_reduction± s4
|
|
|
|
|
z TestBitwiseUFuncs.test_reductionN)r] r^ r_ rá rå ræ rç rè r r r r rÞ
s
|
rÞ c @ s e Zd Zdd ZdS )ÚTestIntc C sd t jdt jd}t jdtd}| ¡ }d|d d d
< |d d d
}tt j||dd t|| d S )Nr: rp rh Fr% r )r r- r© r½ r
r Z logical_not)rZ rP rz rQ Úosr r r Útest_logical_notÚ s zTestInt.test_logical_notN)r] r^ r_ rë r r r r ré Ù s ré c @ s e Zd Zdd ZdS )ÚTestFloatingPointc C s t tjd d S r )r rb ZFLOATING_POINT_SUPPORTrY r r r Útest_floating_pointå s z%TestFloatingPoint.test_floating_pointN)r] r^ r_ rí r r r r rì ä s rì c @ s e Zd Zdd ZdS )Ú TestDegreesc C s, t t tj¡d t t dtj ¡d d S )Nç f@rO ç VÀ)r rb Údegreesr rc rY r r r Ú test_degreesê s zTestDegrees.test_degreesN)r] r^ r_ rò r r r r rî é s rî c @ s e Zd Zdd ZdS )Ú TestRadiansc C s, t t d¡tj t t d¡dtj d S )Nrï rð rO )r rb Úradiansr rc rY r r r Ú test_radiansð s zTestRadians.test_radiansN)r] r^ r_ rõ r r r r ró ï s ró c @ s e Zd Zdd ZdS )Ú TestHeavsidec C sÚ t ddddgdt jt jt j gg¡}t ddddgdt jddgg¡}| ¡ }d|d < t |d¡}t|| t |d¡}t|| | t j ¡}t |t d¡¡}t|| t j ¡ t |t d¡¡}t|| t j ¡ d S )
|
Ng >Àrâ r rs g @rj rL r ©r r% )
|
r rq r· r r
rb Z heavisider r§ r )rZ rP Z expectedhalfZ expected1Úhr r r Útest_heavisideö s &
|
|
zTestHeavside.test_heavisideN)r] r^ r_ rù r r r r rö õ s rö c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestSignc C s t t jt j t jdddg¡}t |j¡}t ddt jdddg¡}t jdd: t |¡}t || t ||¡}t || t || W 5 Q R X d S )Nr r r rL rÒ rU rV )
|
r rq r r· r× rª rü rb rF r )rZ ry rn rQ r- r r r Ú test_sign
|
s
|
|
|
zTestSign.test_signc C s8 t dddg¡}t | t¡¡}t |¡}t|| d S )Nrâ r rá )r rq rF r§ rµ r )rZ Úfoory r« r r r Útest_sign_dtype_object
|
s
|
z TestSign.test_sign_dtype_objectc C s dd }t t| d S )Nc S s" t t jg¡} t | t¡¡}d S r+ )r rq r· rF r§ rµ )rü ry r r r rñ #
|
s z5TestSign.test_sign_dtype_nan_object.<locals>.test_nan)r rv )rZ rñ r r r Útest_sign_dtype_nan_object!
|
s z#TestSign.test_sign_dtype_nan_objectN)r] r^ r_ rû rý rþ r r r r rú
|
s rú c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
Ú
|
TestMinMaxc sú t jdft jdffD ]à\}}t|d|dD ]È\} t jD ]²}t j j|d d d
< t j |< fdd}t >}| t
|
d¡ t t ¡ ¡|d t t ¡ ¡|d W 5 Q R X d
|
|< t ¡ d
|
d d |< t ¡ d d q>q*qd S ) Nr¬ r r r rp c s d f S )Nz%r
|
%sr r ©r r r r r. 4
|
r/ z0TestMinMax.test_minmax_blocked.<locals>.<lambda>z#invalid value encountered in reduce)r g _ Br g _ Â)r r rØ r rM r r r· r r. r/ r r r÷ rì r )rZ r0 Úszrn r÷ Úemsgr1 r r r Útest_minmax_blocked+
|
s$ ÿ
|
ÿ z TestMinMax.test_minmax_blockedc C sH t jdt jddd
t j¡}t| ¡ |d t| ¡ |d d S )Né¸ rp r1 r6 r )r r× r¨ r rØ r r÷ rì ©rZ r| r r r Útest_lower_align@
|
s zTestMinMax.test_lower_alignc C sX dD ]N}t jt jt jfD ]8}t t jt jg| |d¡D ]}tt |¡t j q8qqd S )N)r% r1 r¿ rÀ rÁ rp ) r r r Ú complex64Zdiagflatrq r· r rì )rZ rî r0 r{ r r r Útest_reduce_reorderG
|
s z TestMinMax.test_reduce_reorderc C s t t jd¡}t|t j d S r )r rÇ r· r r¾ r r r Útest_minimize_no_warnsP
|
s z!TestMinMax.test_minimize_no_warnsN)r] r^ r_ r r r r r r r r rÿ *
|
s rÿ c @ s e Zd Zdd Zdd ZdS )ÚTestAbsoluteNegativec
|
C st t jdft jdffD ]X\}}t|d|dD ]>\}}}dd |D }t j||d t|||d t|d k ¡ d
|
d |D }t j||d t|||d t j t j
|
t j
|
fD ]¼}t |j D ]¬}t j |j |d } | |d d
< |||< |t j
|
krø| n|| |< tt |¡| |d t j||d t|| |d t| d | |d d | } t j||d t|| |d q¼q®q,qd S ) Nr r2 r r c S s g | ]}t |¡qS r )rb rS rö r r r r [
|
s z=TestAbsoluteNegative.test_abs_neg_blocked.<locals>.<listcomp>r r r c S s g | ] }d | qS )r3 r rö r r r r `
|
s rp r3 )r r rØ r rS r r rý rÎ r· r rM r r r rO )
|
rZ r0 r rn r r rQ rÍ r÷ r| r r r Útest_abs_neg_blockedV
|
s0 ÿz)TestAbsoluteNegative.test_abs_neg_blockedc C s t jdt jddd
t j¡}tt |¡| tt |¡| t j||d t jt |¡|d t j||d t jt |¡|d d S )Nr rp r1 r6 r ) r r× r¨ r rØ r rO rÎ r± r r r r r s
|
s z%TestAbsoluteNegative.test_lower_alignN)r] r^ r_ r r r r r r r
|
U
|
s r
|
c @ s e Zd Zdd Zdd ZdS )Ú TestPositivec C sD t tttg}|D ].}tjd|d}t |¡}t||t|d qd S )Nr2 rp r ) r r r rµ r r rÏ r Ústr)rZ Z valid_dtypesr) rP r r r r Ú
|
test_valid
|
s
|
|
zTestPositive.test_validc C s t t t d¡ W 5 Q R X t t t t d¡¡ W 5 Q R X t t t tjdgtd¡ W 5 Q R X t t t tjdgtd¡ W 5 Q R X d S )NTz
|
2000-01-01rü rp Úbar)r rv r rÏ Z
|
datetime64rq r rµ rY r r r Ú test_invalid
|
s
|
|
|
|
zTestPositive.test_invalidN)r] r^ r_ r r r r r r r
|
s r c @ sè e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z e j dddg¡dd Ze j dddg¡dd Zd d Zd d Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zd.d/ Zd0d1 Z d2S )3ÚTestSpecialMethodsc s G fddd }t ||¡}t|jt d¡ |j\}}}t|t jk tt|d t|d | t|d | t|d d S )Nc s e Zd Zdd Z fddZdS )z/TestSpecialMethods.test_wrap.<locals>.with_wrapc S s
|
t d¡S r ©r r× rY r r r Ú __array__
|
s z9TestSpecialMethods.test_wrap.<locals>.with_wrap.__array__c s }||_ ||_|S r+ )r r ©rZ r r r{ ©Ú with_wrapr r r
|
s z>TestSpecialMethods.test_wrap.<locals>.with_wrap.__array_wrap__N©r] r^ r_ r r r r r r r
|
s r r r% r ) rb rÇ r r r r× r r r )rZ ry rP r] Úargsr÷ r r r Ú test_wrap
|
s
|
z TestSpecialMethods.test_wrapc sÄ G dd dt j fdd}|dd dd |dd d d |d
|
d d d |d d d d |dd dd |dd dd |dd dd |dd dd |dd dd d S )Nc @ s@ e Zd ZdZdZdd Zdd Zdd Zedd Z d
|
d Z
|
dS ) zKTestSpecialMethods.test_wrap_and_prepare_out.<locals>.StoreArrayPrepareWrapNc S s t d¡ | ¡S )Nr )r r× r ©r r r r r °
|
s zSTestSpecialMethods.test_wrap_and_prepare_out.<locals>.StoreArrayPrepareWrap.__new__c S s |d | _ |S r )Ú
|
_wrap_args©rZ r r r r r r ²
|
s
|
zZTestSpecialMethods.test_wrap_and_prepare_out.<locals>.StoreArrayPrepareWrap.__array_wrap__c S s |d | _ |S r )Ú _prepare_argsr r r r Ú__array_prepare__µ
|
s
|
z]TestSpecialMethods.test_wrap_and_prepare_out.<locals>.StoreArrayPrepareWrap.__array_prepare__c S s | j | jfS r+ )r r rY r r r r ¸
|
s zPTestSpecialMethods.test_wrap_and_prepare_out.<locals>.StoreArrayPrepareWrap.argsc S s dS )Nry r rY r r r Ú__repr__½
|
s zTTestSpecialMethods.test_wrap_and_prepare_out.<locals>.StoreArrayPrepareWrap.__repr__) r] r^ r_ r r r r r Úpropertyr r r r r r ÚStoreArrayPrepareWrap
|
s
|
r! c s }| | |j \}}||}zt|| t|| W nJ tk
|
r } z,td dd |¡d |¡d |¡g¡W 5 d }~X Y nX d S )NÚ
|
z"Bad arguments passed in ufunc callz expected: {}z __array_prepare__ got: {}z __array_wrap__ got: {})r r r ÚjoinÚformat)Zf_callZ
|
f_expectedry r¦ r× rà re ©r! r r Údo_testÀ
|
s
|
|
üz=TestSpecialMethods.test_wrap_and_prepare_out.<locals>.do_testc S s t | d¡S rj ©r rs ©ry r r r r. Ò
|
r/ z>TestSpecialMethods.test_wrap_and_prepare_out.<locals>.<lambda>c S s | dfS rj r r( r r r r. Ò
|
r/ c S s t | dd ¡S rj r' r( r r r r. Ó
|
r/ c S s | dfS rj r r( r r r r. Ó
|
r/ c S s t j| dd dS ©Nr r r' r( r r r r. Ô
|
r/ c S s | dfS rj r r( r r r r. Ô
|
r/ c S s t j| dddS )Nr r+ r r' r( r r r r. Õ
|
r/ c S s | dfS rj r r( r r r r. Õ
|
r/ c S s t dd| ¡S rj r' r( r r r r. Ø
|
r/ c S s
|
dd| fS rj r r( r r r r. Ø
|
r/ c S s t jdd| dS r) r' r( r r r r. Ù
|
r/ c S s
|
dd| fS rj r r( r r r r. Ù
|
r/ c S s t jdd| fdS r) r' r( r r r r. Ú
|
r/ c S s
|
dd| fS rj r r( r r r r. Ú
|
r/ c S s t j| dddS ©Nr F©rO r' r( r r r r. Ý
|
r/ c S s | dfS rj r r( r r r r. Ý
|
r/ c S s t jdd| ddS r* r' r( r r r r. Þ
|
r/ c S s
|
dd| fS rj r r( r r r r. Þ
|
r/ )r r )rZ r& r r% r Útest_wrap_and_prepare_out©
|
s z,TestSpecialMethods.test_wrap_and_prepare_outc C sF G dd dt j}| }t |d¡}tt|| t|t d¡ d S )Nc @ s e Zd ZdZdd Zdd ZdS )z=TestSpecialMethods.test_wrap_with_iterable.<locals>.with_wrapr: c S s t d¡ | ¡ ¡ S r r r r r r r æ
|
s zETestSpecialMethods.test_wrap_with_iterable.<locals>.with_wrap.__new__c S s | t| ¡S r+ r r r r r r é
|
s zLTestSpecialMethods.test_wrap_with_iterable.<locals>.with_wrap.__array_wrap__Nr r r r r r ã
|
s r )r r% r0 )r r rb Úmultiplyr r r rq ©rZ r ry rP r r r Útest_wrap_with_iterableà
|
s
|
z*TestSpecialMethods.test_wrap_with_iterablec C sH G dd dt j}| }t d¡| }tt|| t|t d¡ d S )Nc @ s e Zd ZdZdd ZdS )z7TestSpecialMethods.test_priority_with_scalar.<locals>.Ar: c S s t dd¡ | ¡ ¡ S )NrL rØ r r r r r r ÷
|
s z?TestSpecialMethods.test_priority_with_scalar.<locals>.A.__new__N)r] r^ r_ r r r r r r ÚAô
|
s r0 r )r r rØ r r r rq )rZ r0 ry rP r r r Útest_priority_with_scalarñ
|
s
|
z,TestSpecialMethods.test_priority_with_scalarc s: G fddd }t ||¡}t|jt d¡ d S )Nc s e Zd Zdd Z fddZdS )z3TestSpecialMethods.test_old_wrap.<locals>.with_wrapc S s
|
t d¡S r r rY r r r r s z=TestSpecialMethods.test_old_wrap.<locals>.with_wrap.__array__c s }||_ |S r+ )r )rZ r r{ r r r r s zBTestSpecialMethods.test_old_wrap.<locals>.with_wrap.__array_wrap__Nr r r r r r s r r )rb rÇ r r r r× )rZ ry rP r r r Ú test_old_wrapÿ
|
s z TestSpecialMethods.test_old_wrapc C sÌ G dd d}G dd d|}G dd d|}t d¡}| }| }| }tj}tt|||t jk tt||||k tt||||k tt||||k tt||||k tt||||k tt||||k tt||||k tt||||k tt||||k tt||||k tt||||k tt||||k tt||||k ttt |¡|k ttt |¡|k ttt |¡|k d S )Nc @ s e Zd Zdd Zdd ZdS )z+TestSpecialMethods.test_priority.<locals>.Ac S s
|
t d¡S r r rY r r r r s z5TestSpecialMethods.test_priority.<locals>.A.__array__c S s t | }||_||_|S r+ )r r r r r r r r s
|
z:TestSpecialMethods.test_priority.<locals>.A.__array_wrap__Nr r r r r r0 s r0 c @ s e Zd ZdZdS )z+TestSpecialMethods.test_priority.<locals>.Br~ N©r] r^ r_ r r r r r ÚB s r4 c @ s e Zd ZdZdS )z+TestSpecialMethods.test_priority.<locals>.Cg D@Nr3 r r r r ÚC s r5 r )r r× rb rÇ r r r r ) rZ r0 r4 r5 rP ry r« rß r} r r r Ú test_priority s2
|
|
z TestSpecialMethods.test_priorityc C s8 G dd d}| }t ttj|| t ttjj| d S )Nc @ s e Zd Zdd Zdd ZdS )z/TestSpecialMethods.test_failing_wrap.<locals>.Ac S s
|
t d¡S ©Nr% r rY r r r r < s z9TestSpecialMethods.test_failing_wrap.<locals>.A.__array__c S s t d S r+ ©Ú RuntimeErrorr r r r r ? s z>TestSpecialMethods.test_failing_wrap.<locals>.A.__array_wrap__Nr r r r r r0 ; s r0 )r r9 rb r¬ r ©rZ r0 ry r r r Útest_failing_wrap9 s z$TestSpecialMethods.test_failing_wrapc sx t dg¡ G fdddt j}G dd dt j}t d¡ |¡}t d¡ |¡}tdD ]}tttj d|| q\d S )NrL c s e Zd Z fddZdS )z4TestSpecialMethods.test_failing_out_wrap.<locals>.Okc s S r+ r ©rZ r ©Z singletonr r r K s zCTestSpecialMethods.test_failing_out_wrap.<locals>.Ok.__array_wrap__N©r] r^ r_ r r r= r r ÚOkJ s r? c @ s e Zd Zdd ZdS )z5TestSpecialMethods.test_failing_out_wrap.<locals>.Badc S s t d S r+ r8 r< r r r r O s zDTestSpecialMethods.test_failing_out_wrap.<locals>.Bad.__array_wrap__Nr> r r r r ÚBadN s r@ r r: )
|
r rq r rr r rM r r9 rb ru )rZ r? r@ ÚokÚbadr÷ r r= r Útest_failing_out_wrapF s z(TestSpecialMethods.test_failing_out_wrapc C s* G dd d}| }t t ||¡d d S )Nc @ s e Zd Zdd ZdddZdS )z,TestSpecialMethods.test_none_wrap.<locals>.Ac S s
|
t d¡S r r rY r r r r \ s z6TestSpecialMethods.test_none_wrap.<locals>.A.__array__Nc S s d S r+ r r r r r r _ s z;TestSpecialMethods.test_none_wrap.<locals>.A.__array_wrap__)Nr r r r r r0 [ s r0 )r rb r¬ r: r r r Útest_none_wrapX s z!TestSpecialMethods.test_none_wrapc C sD G dd d}| }t ||¡}t|t d¡ tt|tj d S )Nc @ s e Zd ZdZdd Zdd ZdS )z:TestSpecialMethods.test_default_prepare.<locals>.with_wrapr: c S s
|
t d¡S r r rY r r r r j s zDTestSpecialMethods.test_default_prepare.<locals>.with_wrap.__array__c S s |S r+ r r r r r r m s zITestSpecialMethods.test_default_prepare.<locals>.with_wrap.__array_wrap__N)r] r^ r_ r r r r r r r r g s r r )rb rÇ r r r× r r r. r r r Útest_default_preparee s
|
z'TestSpecialMethods.test_default_prepareÚ use_whereTFc sr G fdddt j t d¡j d}|rDt j||t d¡d}n t ||¡}t|t d¡ tt| d S )Nc s e Zd ZdZ fddZdS )z5TestSpecialMethods.test_prepare.<locals>.with_preparer: c s t |¡j dS ©N©r ©r rq r r ©Ú with_preparer r r { s zGTestSpecialMethods.test_prepare.<locals>.with_prepare.__array_prepare__N©r] r^ r_ r r r rJ r r rK x s rK r rH Tr+ r% )r r rq r rs r r ©rZ rF ry rP r rJ r Ú test_prepareu s z TestSpecialMethods.test_preparec s G fdddt j t dg¡j d}|rDt j|||dgd}nt |||¡}tt ||¡ t|t dg¡ tt| d S )Nc s e Zd ZdZ fddZdS )z9TestSpecialMethods.test_prepare_out.<locals>.with_preparer: c s t |¡j dS rG rI r rJ r r r s zKTestSpecialMethods.test_prepare_out.<locals>.with_prepare.__array_prepare__NrL r rJ r r rK s rK r rH Tr+ r% ) r r rq r rs r Z shares_memoryr r rM r rJ r Útest_prepare_out s z#TestSpecialMethods.test_prepare_outc C s< G dd d}| }t ttj|| t ttj||dd d S )Nc @ s e Zd Zdd ZdddZdS )z2TestSpecialMethods.test_failing_prepare.<locals>.Ac S s
|
t d¡S r r rY r r r r s z<TestSpecialMethods.test_failing_prepare.<locals>.A.__array__Nc S s t d S r+ r8 r r r r r ¡ s zDTestSpecialMethods.test_failing_prepare.<locals>.A.__array_prepare__)N)r] r^ r_ r r r r r r r0 s r0 Fr+ )r r9 rb r¬ r: r r r Útest_failing_prepare s z'TestSpecialMethods.test_failing_preparec C s( G dd d}| }t tdtj| d S )Nc @ s e Zd Zdd ZdS )z6TestSpecialMethods.test_array_too_many_args.<locals>.Ac S s
|
t d¡S r r )rZ r) r r r r r « s z@TestSpecialMethods.test_array_too_many_args.<locals>.A.__array__N)r] r^ r_ r r r r r r0 ª s r0 z2 required positional)r
|
rv r rÙ r: r r r Útest_array_too_many_args¨ s z+TestSpecialMethods.test_array_too_many_argsc C sö G dd d}G dd dt j}| }t dg¡ |¡}t ||¡}t j|||d}t|d | t|d | t|d t j t|d t j t|d d t|d d t|d
|
||f t|d
|
||f t|d i t|d d |fi d S ) Nc @ s e Zd Zdd ZdS )z1TestSpecialMethods.test_ufunc_override.<locals>.Ac _ s | ||||fS r+ r ©rZ r] ÚmethodÚinputsÚkwargsr r r Ú__array_ufunc__´ s zATestSpecialMethods.test_ufunc_override.<locals>.A.__array_ufunc__N©r] r^ r_ rV r r r r r0 ³ s r0 c @ s e Zd ZdZdS )z9TestSpecialMethods.test_ufunc_override.<locals>.MyNDArrayrÝ Nr3 r r r r Ú MyNDArray· s rX r r r r% Ú__call__r0 r1 rn )r r rq r r- r )rZ r0 rX ry r« Úres0rs r r r Útest_ufunc_override± s z&TestSpecialMethods.test_ufunc_overridec C s¤ dd }dd }t |dd¡}t |dd¡}G dd d }G d
|
d d |}G d d d }G dd d}G dd d|} | }
|
| } | } | } t |
|
| ¡}t|d t | | ¡}t|d t | |
|
¡}t|d t| jd t | |
|
¡}t| jd | } | }ttt j| | t| jd t|jd d | _|_ttt j|| t| jd t|jd d| _ttt j| | t| jd d| _ttt jd| t| jd t||
|
ddd t|d|
|
dd t|dd|
|
d t||
|
|
|
dd t||
|
d|
|
d t||
|
d| d t||
|
d| d t||
|
| dd d| _t|| | dd t| jd d| _t|d| | d t| jd d| _t||
|
| | d t| jd d|_t||
|
| |d t|jd t|dd| d tt|dd| tt||d| tt||dd t||
|
dddd t|d|
|
ddd t|dd|
|
dd t|ddd|
|
d t||
|
| ddd t|d|
|
d| d t|| d|
|
dd t|| dd|
|
d t||
|
dd| d | } | }tt|ddd| t| jd d| _tt|dd|| t|jd t| jd | }d | _|_tt|d| || t|jd t| jd t|jd d | _ |_|_tt||| || t|jd t| jd t|jd d S )Nc S s | | | S r+ r )ry r« rß r r r Útres_mulÎ s z<TestSpecialMethods.test_ufunc_override_mro.<locals>.tres_mulc S s | | | | S r+ r )ry r« rß r| r r r Ú
|
quatro_mulÑ s z>TestSpecialMethods.test_ufunc_override_mro.<locals>.quatro_mulr0 r r1 c @ s e Zd Zdd ZdS )z5TestSpecialMethods.test_ufunc_override_mro.<locals>.Ac _ s dS )Nr0 r rR r r r rV Ù s zETestSpecialMethods.test_ufunc_override_mro.<locals>.A.__array_ufunc__NrW r r r r r0 Ø s r0 c @ s e Zd Zdd ZdS )z8TestSpecialMethods.test_ufunc_override_mro.<locals>.ASubc _ s dS )NÚASubr rR r r r rV Ý s zHTestSpecialMethods.test_ufunc_override_mro.<locals>.ASub.__array_ufunc__NrW r r r r r^ Ü s r^ c @ s e Zd Zdd ZdS )z5TestSpecialMethods.test_ufunc_override_mro.<locals>.Bc _ s dS )Nr4 r rR r r r rV á s zETestSpecialMethods.test_ufunc_override_mro.<locals>.B.__array_ufunc__NrW r r r r r4 à s r4 c @ s e Zd Zdd Zdd ZdS )z5TestSpecialMethods.test_ufunc_override_mro.<locals>.Cc S s
|
d| _ d S rj )ÚcountrY r r r Ú__init__å s z>TestSpecialMethods.test_ufunc_override_mro.<locals>.C.__init__c _ s | j d7 _ tS r ©r_ ÚNotImplementedrR r r r rV è s zETestSpecialMethods.test_ufunc_override_mro.<locals>.C.__array_ufunc__N)r] r^ r_ r` rV r r r r r5 ä s r5 c @ s e Zd Zdd ZdS )z8TestSpecialMethods.test_ufunc_override_mro.<locals>.CSubc _ s | j d7 _ tS r ra rR r r r rV í s zHTestSpecialMethods.test_ufunc_override_mro.<locals>.CSub.__array_ufunc__NrW r r r r ÚCSubì s rc r% r rÆ )r Ú
|
frompyfuncr- r r_ r rv )rZ r\ r] Zthree_mul_ufuncZfour_mul_ufuncr0 r^ r4 r5 rc ry Za_subr« rß r- Zc_subÚc2r r r Útest_ufunc_override_mroË s²
|
|
|
z*TestSpecialMethods.test_ufunc_override_mroc C sn G dd d}| }t t tjjd|ddd W 5 Q R X tjjd|ddd}t|d | t|d tj t|d d
|
t|d d|f t|d ddd t ttj| t ttj|||| t ttj||d d d t ttj||dddgd tj |dddd¡}t|d | t|d tj t|d d t|d |f t|d ddddd tjj|ddddddd}t|d | t|d tj t|d d t|d |f t|d ddddddd tj |dd d d¡}t|d dd dd tjj|d dd d }t|d dd d tjj|d d d d!}t|d d d d" tj |dd d dd d ¡}t|d dd dd d d# tj |dd d dtj d ¡}t|d dd dd d$ tj |dd d dd d ¡}t|d dd dd d d# t t
|
tjj|d%d& t t
|
tjj|d'd& t ttjj|ddd( tj |ddd¡}t|d | t|d tj t|d d) t|d |f t|d dddd* tjj |dddd+}t|d | t|d tj t|d d) t|d |f t|d dddd* tj |dd d ¡}t|d dd d" tjj |d dd,d-}t|d dd,d" tjj |d d d d!}t|d d d d" t t
|
tjj |d%d& t t
|
tjj |d'd& t ttjj |ddd( tj |d d gddd¡}t|d | t|d tj t|d d. t|d |d d gf t|d dddd* tjj |d d gdddd+}t|d | t|d tj t|d d. t|d |d d gf t|d dddd* tj |d d gdd d ¡}t|d dd d" tjj |d d gd d d/d0}t|d d d/d" tjj |d d gd d d d&}t|d d d d" t t
|
tjj|d d gd%d& t t
|
tjj|d d gd'd& t ttjj|d d gddd( tj |d¡}t|d | t|d tj t|d d1 t|d |df t|d i t ttjj | t ttjj |||| t ttjj ||d d d tj |d d gd2¡}t|d | t|d tj t|d d3 t|d |d d gd2f t ttjj| t ttjj|||| d S )4Nc @ s e Zd Zdd ZdS )z9TestSpecialMethods.test_ufunc_override_methods.<locals>.Ac _ s | ||||fS r+ r ©rZ r rS rT rU r r r rV U s zITestSpecialMethods.test_ufunc_override_methods.<locals>.A.__array_ufunc__NrW r r r r r0 T s r0 r r rú )rü Zanswer)rl rO r r% rY r0 r1 ry )rÇ rÁ )ÚaxisZaxesZaxis0Zdtype0Úout0Zkeep0r )ri )r) rn Úkeepdimsrh Zinit0Zwhere0)rh r) rn rj ÚinitialrO )r) rn rj rh rk rO F)rh r) rj T)rn rh rj )rh rj r+ )rn r) )rh r) )rh r) rj rk rO )rh r) rj rO r r )ri Úout1©rh r& )r) rn rh )rh r) rn Zdtype1)rn rh r) Úreduceatr0 )rh rn r) ÚouterZb0Úat)r rv r r- rY r Ú ncu_testsÚinner1dr Z_NoValuerw r& rn ro rp )rZ r0 ry r- r r r Útest_ufunc_override_methodsR s,
|
|
ý þ
|
û ÿ ÿÿ ÿ
|
þ ÿ
|
þ ÿ ÿ
|
þÿ
|
þÿ ÿz.TestSpecialMethods.test_ufunc_override_methodsc C s G dd d}G dd d}| }| }t ||d¡}t j||dd}t d|d¡}t jd|dd}t |d d¡} t j|d
|
dd}
|
t|d d d t|d d d t|d d d t|d d d t| d d d t|
|
d d d t |d d¡} t |d d¡} t| d d d t| d d d t| d d d t| d d d tt |d ¡i k tt j|d di k tt j|ddi k tt |d ¡i k tt |d d ¡i k tt j|ddi k tt t j|d d W 5 Q R X ttt j||ddd ttt j||dd tt t j||dd ttt j|dd ttt j|ddd ttt j|ddd tt t j|dd tt t j|dd d S )Nc @ s e Zd Zdd ZdS )z5TestSpecialMethods.test_ufunc_override_out.<locals>.Ac _ s |S r+ r rg r r r rV s zETestSpecialMethods.test_ufunc_override_out.<locals>.A.__array_ufunc__NrW r r r r r0 ÿ s r0 c @ s e Zd Zdd ZdS )z5TestSpecialMethods.test_ufunc_override_out.<locals>.Bc _ s |S r+ r rg r r r rV s zETestSpecialMethods.test_ufunc_override_out.<locals>.B.__array_ufunc__NrW r r r r r4 s r4 Zout_argr r% r0 r1 r2 rn r ri rl r r+ )NNr¤ Útwo©r¤ rt r )rt Úthreerv )r¤ rt rv )r¤ )
|
r r- r Úmodfru r r r rv rw ) rZ r0 r4 ry r« rZ rs rt Zres3Zres4Zres5Zres6Zres7r r r Útest_ufunc_override_outý sL
|
z*TestSpecialMethods.test_ufunc_override_outc C sH G dd d}| }t ttjd|d t ttj| t ttjd| d S )Nc @ s e Zd Zdd ZdS )z;TestSpecialMethods.test_ufunc_override_exception.<locals>.Ac _ s t dd S )NZoops)rw )rZ ry rU r r r rV 8 s zKTestSpecialMethods.test_ufunc_override_exception.<locals>.A.__array_ufunc__NrW r r r r r0 7 s r0 r r rL )r rw r rÎ rò r: r r r Ú test_ufunc_override_exception5 s
|
z0TestSpecialMethods.test_ufunc_override_exceptionc C sr G dd d}d}t tt |¡ t | ¡ W 5 Q R X d}t tt |¡ tj| t dd W 5 Q R X d S )Nc @ s e Zd Zdd ZdS )zATestSpecialMethods.test_ufunc_override_not_implemented.<locals>.Ac _ s t S r+ )rb ©rZ r rU r r r rV C s zQTestSpecialMethods.test_ufunc_override_not_implemented.<locals>.A.__array_ufunc__NrW r r r r r0 B s r0 zjoperand type(s) all returned NotImplemented from __array_ufunc__(<ufunc 'negative'>, '__call__', <*>): 'A'zoperand type(s) all returned NotImplemented from __array_ufunc__(<ufunc 'add'>, '__call__', <*>, <object *>, out=(1,)): 'A', 'object', 'int'r r )r
|
rv ÚfnmatchÚ translater rÎ rs rµ )rZ r0 r r r r Ú#test_ufunc_override_not_implemented@ s z6TestSpecialMethods.test_ufunc_override_not_implementedc C sî G dd d}| }d}t t| t |d¡ W 5 Q R X t t| t d|¡ W 5 Q R X t t| t |¡ W 5 Q R X G dd d}| }tt |¡|k t t| t ||¡ W 5 Q R X t t| tj|d|d W 5 Q R X d S )Nc @ s e Zd ZdZdS )z?TestSpecialMethods.test_ufunc_override_disabled.<locals>.OptOutNrW r r r r ÚOptOutS s r~ z(operand 'OptOut' does not support ufuncsr c @ s e Zd Zdd ZdS )zDTestSpecialMethods.test_ufunc_override_disabled.<locals>.GreedyArrayc _ s | S r+ r rz r r r rV e s zTTestSpecialMethods.test_ufunc_override_disabled.<locals>.GreedyArray.__array_ufunc__NrW r r r r Ú GreedyArrayd s r r )r
|
rv r rs rÎ r )rZ r~ Zopt_outr r Zgreedyr r r Ú test_ufunc_override_disabledQ s z/TestSpecialMethods.test_ufunc_override_disabledc C s& G dd d}t j}| }|||}t|d | t|d | t|d d t|d ||f t|d i |dd|d }t|d | t|d | t|d d t|d d
|
t|d d |fi tt||d d tt|||d d d tt|||d d tt|||dd tt|||dd d S )Nc @ s e Zd Zdd ZdS )z2TestSpecialMethods.test_gufunc_override.<locals>.Ac _ s | ||||fS r+ r rg r r r rV s s zBTestSpecialMethods.test_gufunc_override.<locals>.A.__array_ufunc__NrW r r r r r0 r s r0 r r r% rY r0 r1 r )r r rn rt r¤ ru r )rq rr r r rv rw )rZ r0 rr ry r- r r r Útest_gufunc_overrideo s(
|
z'TestSpecialMethods.test_gufunc_overridec sZ G fdddt j G fddd}t d¡}t d¡ ¡}t |¡}t |¡}tt ||k¡ t|jddgi t j||fd}tt ||k¡ t|jd dgi t||k t d¡ ¡}t j||d}tt ||k¡ t|jdgdgd
|
t d¡ ¡}t |¡\}}t|jddgi t j |d |fd\}}t||k t|jd d gi t d¡ ¡}t d¡ ¡}t j |||fd\}} t||k t| |k t|jdgdd gd
|
t d¡ ¡}t d¡ ¡}t j
|
|||d}
|
t|
|
|k t|
|
jdd gdgd
|
t d¡}| }t| t j
|
d ||¡t k t| t j
|
d ||¡t k t tt j
|
|| | ¡}t| t j
|
d ||¡t k t| t j
|
d ||¡d k tt
|
||¡d k t d ddgd ddgg¡}| ¡}| ¡ }
|
| ¡ }t|
|
| t|
|
jddgi | ¡ }
|
| ¡ }t|
|
| t|
|
jddgi t d¡ ¡}|j|d}
|
t|
|
| t|
|
|k t|
|
jdgdgd
|
|jdd}t |¡ ¡}|jd|d}
|
t|
|
| t|
|
|k t|
|
jdgdgd
|
t j
|
j|d d}t j
|
j|d d}
|
t|
|
| t|
|
jddgi t |
|
¡}t j
|
|d d |¡}
|
t|
|
| t|
|
|k t|
|
jdgdgd
|
t j
|
j|dd}t j
|
j|dd}
|
t|
|
| t|
|
jddgi t |
|
¡}t j
|
|dd |¡}
|
t|
|
| t|
|
|k t|
|
jdgdgd
|
ddd g} t j
|
j|| d d}t j
|
j|| d d}
|
t|
|
| t|
|
jddgi t |
|
¡}t j
|
|| d d |¡}
|
t|
|
| t|
|
|k t|
|
jdgdgd
|
t d ddgd ddgg¡}| ¡ }| ¡ ¡}t j
|
|dd gddgfd¡ t j
|
|dd gddgfd¡ t|| t|jddgi t d¡ ¡}| ¡ ¡}t j
|
|dd gddgf|¡ t|| t|jdddgi d S )Nc s$ e Zd Zdd fdd
|
Z ZS )z<TestSpecialMethods.test_ufunc_override_with_super.<locals>.ANr c s g }g }t |D ]:\}} t| r@| |¡ | | tj¡¡ q| | ¡ q|}
|
g } |
|
r®g } t |
|
D ]:\} }t| r| | ¡ | | tj¡¡ qd| |¡ qdt| |d< n
|
d|j }
|
i }|rÈ||d< | rÔ| |d< t j ||f||}|t
|
kröt
|
S |dkr t|d r||d _ d S |jdkr0|f}t fdd t ||
|
D }|rlt|d rl||d _ t |dkr|d S |S )
|
Nrn r+ rT Úoutputsrp r r c 3 s. | ]&\}}|d kr"t |¡ ¡n|V qd S r+ )r r r )r r Úoutput©r0 r r rJ ¼ s ÿÿz_TestSpecialMethods.test_ufunc_override_with_super.<locals>.A.__array_ufunc__.<locals>.<genexpr>)Ú enumerater Úappendr r r Útupler( ÚsuperrV rb Úinforª r )rZ r rS rn rT rU r Zin_nor÷ Úinput_r Zout_noZout_argsÚjr r Úresults)r0 Ú __class__r r rV sT
|
|
|
|
|
ÿÿ
|
|
þ
|
zLTestSpecialMethods.test_ufunc_override_with_super.<locals>.A.__array_ufunc__)r] r^ r_ rV Ú __classcell__r r )r r r0 s r0 c s e Zd Z fddZdS )z<TestSpecialMethods.test_ufunc_override_with_super.<locals>.Bc s" t fdd|D rdS tS d S )Nc 3 s | ]}t | V qd S r+ )r )r r r r r rJ Æ s z_TestSpecialMethods.test_ufunc_override_with_super.<locals>.B.__array_ufunc__.<locals>.<genexpr>úA!)rÜ rb rg r r r rV Å s zLTestSpecialMethods.test_ufunc_override_with_super.<locals>.B.__array_ufunc__NrW r r r r r4 Ä s r4 r} rT r r r )rT r r rY r r% r0 rm )rh rn rL )r r r r r r rý r r rw rs rV rb r rv rq rÜ r÷ r¶ r r& rn r
rp ) rZ r4 r| ry r« ÚcheckZb1Zb2Zc1re rß Úindicesr r r Ú test_ufunc_override_with_super sÎ 4
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
z1TestSpecialMethods.test_ufunc_override_with_superN) r] r^ r_ r r, r/ r1 r2 r6 r; rC rD rE rÈ rÍ rÎ rN rO rP rQ r[ rf rs rx ry r} r r r r r r r r
|
s6 7+
|
|
,8 r c @ s e Zd Zdd ZdS )Ú
|
TestChoosec C s@ t ddg¡}t ddg¡}tt ||df¡t ddg¡ d S )NTr )r rq r Úchoose)rZ rß ry r r r Ú
|
test_mixed< s zTestChoose.test_mixedN)r] r^ r_ r r r r r r ; s r c @ sd e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dS )ÚTestRationalFunctionsc C s | tj¡ | tj¡ d S r+ )Ú_test_lcm_innerr r© Úuint16rY r r r Útest_lcmC s z TestRationalFunctions.test_lcmc C s | tj¡ d S r+ )r r Úobject_rY r r r Útest_lcm_objectG s z%TestRationalFunctions.test_lcm_objectc C s | tj¡ | tj¡ d S r+ )Ú_test_gcd_innerr r© r r rY r r r Útest_gcdJ s z TestRationalFunctions.test_gcdc C s | tj¡ d S r+ )r r r rY r r r Útest_gcd_objectN s z%TestRationalFunctions.test_gcd_objectc C sì t jddg|d}t jddg|d}tt ||¡ddg t|t jst jddddg|d}t jddd d g|d}tt ||¡dgd
|
t jd ddg|d}tt j d ddg¡d t d ¡ |¡}d}tt ||¡d dddddg d S )Nr éx rp rh éÈ é< iX r éìÿÿÿr1 r0 rÆ r ) r rq r ÚlcmÚ
|
issubclassÚunsignedintegerr r r§ ©rZ r) ry r« r r r r Q s z%TestRationalFunctions._test_lcm_innerc C sæ t jddg|d}t jddg|d}tt ||¡ddg t|t jst jddddg|d}t jddd d g|d}tt ||¡dgd t jd
|
d d g|d}tt j |¡d t d¡ |¡}d}tt ||¡dddddd g d S )Nr r rp rh r r1 é( r r¢ r¬ é é# r2 rÆ r r% ) r rq r Úgcdr¤ r¥ r r r§ r¦ r r r r f s z%TestRationalFunctions._test_gcd_innerc C sB t t t j¡jd ¡}d| }d| }tt ||¡d| d S )Nr r% r2 r: )r rt rú r÷ r r£ )rZ Úbigry r« r r r Útest_lcm_overflow{ s z'TestRationalFunctions.test_lcm_overflowc C s^ t jt jfD ]L}|t |¡j}|d }tt ||d ¡| tt || d ¡| q d S )Nr1 r0 )r rt rÊ rú rì r rª )rZ r) ry Úqr r r Útest_gcd_overflow s
|
|
z'TestRationalFunctions.test_gcd_overflowc C s| ddl m} t ddddg¡|d }t ddddg¡|d }tt ||¡d|dg tt ||¡d|d g d S )
|
Nr )ÚDecimalr r3 z0.20z0.12r1 z0.04z0.60)rá r¯ r rq r rª r£ )rZ r¯ ry r« r r r Ú test_decimal s
|
z"TestRationalFunctions.test_decimalc C s$ t ttjdd t ttjdd d S )Ng333333Ó?gÙ?)r rv r rª r£ rY r r r Ú
|
test_float s z TestRationalFunctions.test_floatc C s¦ t t dd ¡ ¡ dd t dd d ¡}t dd d dg¡}t t ||¡dd dg t t ||¡dd d d dd d g t t dd d d ¡d
|
d S ) Nr% r rÝ éó i-1 l U9 l ` i©æ r0 r )r r rq Úitemrª r£ r® r r r Útest_builtin_long s *z'TestRationalFunctions.test_builtin_longN)r] r^ r_ r r r r r r r¬ r® r° r± r´ r r r r r B s r c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestRoundingFunctionsc C s` G dd d}t | | g¡}tt |¡ddg tt |¡ddg tt |¡ddg dS )z3 test direct implementation of these magic methods c @ s$ e Zd Zdd Zdd Zdd ZdS )z3TestRoundingFunctions.test_object_direct.<locals>.Cc S s dS r r rY r r r Ú __floor__ª s z=TestRoundingFunctions.test_object_direct.<locals>.C.__floor__c S s dS r7 r rY r r r Ú__ceil__¬ s z<TestRoundingFunctions.test_object_direct.<locals>.C.__ceil__c S s dS )Nr0 r rY r r r Ú __trunc__® s z=TestRoundingFunctions.test_object_direct.<locals>.C.__trunc__N)r] r^ r_ r¶ r· r¸ r r r r r5 © s r5 r r% r0 N)r rq r rU r? rV ©rZ r5 r r r r Útest_object_direct§ s
|
z(TestRoundingFunctions.test_object_directc C sl G dd d}t | | g¡}tt |¡ddg tt |¡ddg t t¡ t |¡ W 5 Q R X dS )z$ test implementations via __float__ c @ s e Zd Zdd ZdS )z5TestRoundingFunctions.test_object_indirect.<locals>.Cc S s dS )Nr r rY r r r Ú __float__¹ s z?TestRoundingFunctions.test_object_indirect.<locals>.C.__float__N)r] r^ r_ r» r r r r r5 ¸ s r5 r5 r4 N) r rq r rU r? rÈ rÉ rv rV r¹ r r r Útest_object_indirect¶ s z*TestRoundingFunctions.test_object_indirectc C s> t dd}tt |¡d tt |¡d tt |¡d d S )Nr6 r0 r4 r3 )r r r rU r? rV )rZ r} r r r Ú test_fraction s
|
z#TestRoundingFunctions.test_fractionN)r] r^ r_ rº r¼ r½ r r r r rµ ¥ s rµ c @ sî e Zd Zejejejejejej ej
|
ej ej ej ejejejejejejgZdd Zejjedddd Zejjedddd Zejjeddd d
|
Z d d Z ejjeddej d ej ej ej!g¡dd Z"ej#dddd Z$dS )ÚTestComplexFunctionsc C s` | j D ]T}|tjkrd}nd}||}|t|}t|j|d| d t|jdd| d qd S )Nr¡ rj z real part %sr r z imag part %s)Úfuncsr r5 r r rN r )rZ r} rP ÚfrZfzr r r Útest_itÏ s
|
|
z TestComplexFunctions.test_itz doesn't workrè c C sf d}| j D ]V}|t |¡}|t |¡}|t |¡}t||dd| d t||dd| d q
|
d S )Nr rÆ z
|
fch-fcd %s)rá r r¬ z
|
fch-fcl %s)r¿ r ZcsingleZcdoubleZ clongdoubler )rZ r3 r} ZfcfZfcdZfclr r r Útest_precisions_consistentÚ s
|
z/TestComplexFunctions.test_precisions_consistentc C s¸ t tjddddd t tjddddd t tjddddd t tjddddd t tjddddd t tjddgddgddd t tjddgddgddd t tj d d
|
gddgddd t tj
|
d d
|
gddgddd t tj dd gddgddd t tj ddgddgddd t tjd d
|
gddgdd t tjd d
|
gddgdd t tj ddgddgdd t tj
|
ddd gdddgdd t tj d d
|
dgdddgdd t tj d d
|
d gdddgdd d S ) NrO r¢ r r3 Tç ø¿r4 r% ù Àr¯ rj r ) Ú_check_branch_cutr rö rÇ r r r r' r( r* rM r5 r8 rY r r r Útest_branch_cutsä s" z%TestComplexFunctions.test_branch_cutsc C s t tjdddddtj t tjdddddtj t tjdddddtj t tjdddddtj t tjdddddtj t tjddgddgdddtj t tj ddgddgdddtj t tj
|
d d
|
gddgdddtj t tj d d
|
gddgdddtj t tj dd gddgdddtj t tj ddgddgdddtj t tjd d
|
gddgddd tj t tj d d
|
gddgddd tj t tj
|
ddgddgddd tj t tj ddd gdddgddd tj t tj d d
|
dgdddgddd tj t tj d d
|
d gdddgddd tj d S )NrO r¢ r r3 Trà r4 r% rÄ r¯ rj Fr )rÅ r rö r rÇ r r r r' r( r* rM r5 r8 rY r r r Útest_branch_cuts_complex64þ s" $$z/TestComplexFunctions.test_branch_cuts_complex64c C sÐ dd l }ddddg}dddd d
|
d d }d t t¡j }| jD ]}|j d¡d }| ||¡}zt ||}W n t
|
k
|
r Y q<Y nX |D ]@} t|t | ¡}
|
|| } t t |
|
| |k d|| |
|
| f qq<d S )Nr y ð¿ ð¿y ð¿ ð?y ð? ð¿r ÚasinÚacosÚatanÚasinhÚacoshÚatanh)r' r( r* rM r5 r8 r1 Ú.r3 z%s %s: %s; cmath: %s)Úcmathr r[ r Úepsr¿ r] ÚsplitÚgetr[ ÚAttributeErrorr r rO ) rZ rÏ ZpointsZname_mapÚatolr] ÚfnameÚcnameZcfuncr¦ ry r« r r r Útest_against_cmath s& ÿ
|
|
z'TestComplexFunctions.test_against_cmathr) c s t ¡} djj|j fdd}t ddd¡}t jddd d
|
d } t jkrtt rdt d ¡ ||d n||d ||d d t j
|
dg d}d}t dt |¡| ¡}t t |dk ¡ d}t dt |¡| ¡}t t |dk ¡ d}t dt |¡| ¡}t t |dk ¡ d}t dt |¡| ¡}t t |dk ¡ d fdd }t jt jt jt j t jfD ]8} dd d D }
|
|| |
|
d || |
|
d || |
|
d qd S )!z1Check loss of precision in complex arc* functionsr c s~ | ¡} | ¡}t t | ¡t |¡j d ¡}tt ||k ¡t |¡| t |¡ | ¡ df d| ¡}t t | ¡t |¡j
|
d ¡}tt ||k ¡t |¡| t |¡ | ¡ df | ¡}t t | ¡t |¡j d ¡}tt ||k ¡t |¡| t |¡ | ¡ df d| ¡}t t | ¡t |¡j
|
d ¡}tt ||k ¡t |¡| t |¡ | ¡ df d S )Nr rM r¢ r' r8 r* ) r§ r rS rM rN r rý Zargmaxr÷ r' r r8 r* )rP Zrtolr3 r| )r) Ú
|
real_dtyper r r 6 s*
|
|
(ÿ (ÿ
|
(ÿ (ÿz:TestComplexFunctions.test_loss_of_precision.<locals>.checkr¢ g5^ºI Àr gË¡E¶óýÀr r: F)ZendpointzvTrig functions of np.longcomplex values known to be inaccurate on aarch64 and PPC for some compilation configurations.g I@çÍÌÌÌÌÌ @r çü©ñÒMbP?yñhãµøä>ñhãµøä>rp y¶gݵøä>+jéµøä>r ra yiæµøä>Shàµøä>y+jéµøä>¶gݵøä>yShàµøä>iæµøä>c s t j| d}|t|| d }|t|| d }tt ||k¡||f t| || | d k }tt |¡| || f d S )Nrp r% )r r rO r rý )r] Zz0r| ZzpZzmZgood)r) rÐ r r r u s c S s2 g | ]*}d D ] }|dks |dkr |d| q qS )©gü©ñÒMbP¿r rÚ r r¢ r )r ÚrpÚipr r r r s ÿz?TestComplexFunctions.test_loss_of_precision.<locals>.<listcomp>rÛ r¢ r N)r )r r[ rN r) rÐ ZlogspaceÚ longcomplexrS rÈ r rq rS r8 r rý rM r* r' ) rZ r) r r Zx_seriesZx_basicr3 r¦ r| r] Zptsr )r) rÐ rØ r Útest_loss_of_precision+ s>
|
|
|
|
z+TestComplexFunctions.test_loss_of_precisionrU r& c C s^ | j D ]R}|t d¡jtjks$t|t d¡jtjks>t|t d¡jtjkstqd S r )r¿ r r r) r r© r© r )rZ r] r r r Útest_promotion_corner_cases s
|
z0TestComplexFunctions.test_promotion_corner_casesN)%r] r^ r_ r r' r( r* rM r5 r8 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¾ É s4 ý
|
|
|
Y
|
r¾ c @ s e Zd Zdd Zdd ZdS )ÚTestAttributesc C sV t j}t|jd t|jdk td|jk t|jd t|jd t|j d d S )Nrs é zii->ir% r r )
|
rb rs r r] r Zntypesr$ r! r( rò )rZ rs r r r Útest_attributes s z TestAttributes.test_attributesc C s( t tjj d¡ t tjj d¡ d S )Nz&add(x1, x2, /, out=None, *, where=Truez<frexp(x[, out1, out2], / [, out=(None, None)], *, where=True)r rb rs Ú__doc__rI ru rY r r r Útest_doc s
|
ÿ
|
ÿzTestAttributes.test_docN)r] r^ r_ rã rå r r r r rá s rá c @ s e Zd Zdd ZdS )Ú TestSubclassc C s, G dd dt j}|d}t|| | d S )Nc @ s e Zd Zdd ZdS )z-TestSubclass.test_subclass_op.<locals>.simplec S s t jj| |td}| d¡ |S )Nrp r )r r r rµ Úfill)Úsubtyperª rZ r r r r § s
|
z5TestSubclass.test_subclass_op.<locals>.simple.__new__N)r] r^ r_ r r r r r Úsimple¦ s ré )r0 r1 )r r r )rZ ré ry r r r Útest_subclass_op¤ s z TestSubclass.test_subclass_opN)r] r^ r_ rê r r r r ræ ¢ s ræ c @ s e Zd Zdd ZdS )ÚTestFrompyfuncc sü dd }t j|dddd t dddg¡d t jt d ¡d
|
d d t g ¡d t j|ddd d t dddg¡d t jt d ¡d
|
d d tt fd d t j|ddd t dddg¡d tt fdd tt fdd d S )Nc S s | | S r+ r )ry r« r r r Úmul³ s z)TestFrompyfunc.test_identity.<locals>.mulr% r )r! r( rò r0 r1 é ©r% r% ©r r rm c s
|
g ¡S r+ r r ©Z mul_ufuncr r r. À r/ z.TestFrompyfunc.test_identity.<locals>.<lambda>)r! r( c s j t d¡ddS )Nrî rï rm )r r r- r rð r r r. Å r/ c s
|
g ¡S r+ r r rð r r r. Æ r/ )r rd r r r- r rw )rZ rì r rð r rç ² s z TestFrompyfunc.test_identityN)r] r^ r_ rç r r r r rë ° s rë r3 Fc C s t |¡ |¡}t |¡ |¡}t |¡jdkrLt |¡jd }t d¡}nt |¡jd }d}| |} | ||| t |¡ t |¡ }
|
| ||| t |¡ t |¡ } t t
|
t | j |
|
j ¡|k ¡| |
|
f t t
|
t | j |
|
j ¡|k ¡| |
|
f t t
|
t | j | j | ¡|k ¡| | f t t
|
t | j | j | ¡|k ¡| | f |r|j dk|j dk@ } |j dk|j dk@ } t | ¡r|| }t j|_ | |} t t
|
t | | j | j | ¡|k ¡| | | f t t
|
t | | j | j | ¡|k ¡| | | f t | ¡r|| }t j|_ | |} t t
|
t | | j | j | ¡|k ¡| | | f t t
|
t | | j | j | ¡|k ¡| | | f dS )aR
|
Check for a branch cut in a function.
|
|
Assert that `x0` lies on a branch cut of function `f` and `f` is
|
continuous from the direction `dx`.
|
|
Parameters
|
----------
|
f : func
|
Function to check
|
x0 : array-like
|
Point on branch cut
|
dx : array-like
|
Direction to check continuity in
|
re_sign, im_sign : {1, -1}
|
Change of sign of the real or imaginary part expected
|
sig_zero_ok : bool
|
Whether to check if the branch cut respects signed zero (if applicable)
|
dtype : dtype
|
Dtype to check (should be complex)
|
|
ÚFrü rû r g-C ëâ6?r N)r Z
|
atleast_1dr§ r) râ r[ rÐ r rS r rý rN r rÜ r )r} Zx0ZdxZre_signZim_signZ sig_zero_okr) ZscalerÔ Zy0ZypZymZjrZjirP r r r rÅ É s: $$&&** 22 2rÅ c C s t t dd¡dk tjdd6 t dt dd¡ dk t dt dd¡ dk W 5 Q R X t t t tjd¡¡ t t t tjd¡¡ d S )Nr r3 rU rñ r )r r Úcopysignrü r# r· r r r r Ú test_copysign s "ró c C s | d}| d}| d}t | ¡j}tt ||¡| |k tt ||¡| dk tt t t j|¡¡ tt t |t j¡¡ tt ||¡|k d S rz )r r[ rÐ r ra r r· )rI r¤ rt rl rÐ r r r Ú_test_nextafter s rô c C s
|
t tjS r+ )rô r rØ r r r r Útest_nextafter s rõ c C s
|
t tjS r+ )rô r r r r r r Útest_nextafterf s rö z long double is same as doublerè Úppc64zIBM double double)Ú conditionré c C s
|
t tjS r+ )rô r r\ r r r r Útest_nextafterl! s rù c
|
C s° t tjd d¡D ]\} }t ^}| t¡ t t | ¡j ¡szt | ¡j }t
|
d|t | d| |¡ k or|k n W 5 Q R X t t | d| |¡| d |d qd S )Nr r< r r rÙ ) r6 r7 r rÏ r r. Ú UserWarningr r[ r\ r ra r )rI Ú directionr1 r\ r r r Útest_nextafter_0) s
|
*ÿrü c C sª | d}t | ¡j}| t j}| t j}t jddl tt |¡|k tt t |¡¡ tt t |¡¡ tt t | ¡¡ tt | d¡dk W 5 Q R X d S )Nr rU rV gê 9Y>)Fr ) r r[ rÐ r· r rü r r> r )rI r¤ rÐ r· r r r r Ú _test_spacing5 s
|
|
rý c C s
|
t tjS r+ )rý r rØ r r r r Ú test_spacingA s rþ c C s
|
t tjS r+ )rý r r r r r r Ú test_spacingfD s rÿ c C s
|
t tjS r+ )rý r r\ r r r r Ú test_spacinglH s r c C sn t jddddgt jddddgi} tt jt jgd D ]6\}}t jd
|
d d d g|d}tt |¡| | |d q2d S )Ng ;g °<g @=g =g5( p=gýô >g
K8 ?g P?)r: rh çñhãµøä>r r i) rp rà )r rØ r rª rq r r> )Úrefr0 rä rP r r r Útest_spacing_gfortranO s ý ýü r c C sR t jt jfD ]@} dD ]6}| |}| |d }tt ||¡| t |¡k qq d S )N)r r r r )r r rØ r ra r> )rI Z_fr} Úf1r r r Útest_nextafter_vs_spacingm s
|
r c C s t t tj¡dk dS )z Check np.nan is a positive nan.r N)r r r# r· r r r r Ú test_pos_nanu s r c C s
|
t ddt jfdt jfg¡} t jdg| d}d|d< d|d< d|d< d d
|
d d g}g }|d }|d d
D ]&}| t j |d ||
¡¡ |}qh| t j |d |d
¡¡ t |¡}t j |d |¡}t
|
|| t d¡ t j |d |¡}t t j ¡ t
|
|| dS )z;Test bug in reduceat when structured arrays are not copied.)r» ZS11Útimer rÝ rp ZSimpler» r: r r r¬ r¨ r NrÁ ) r r) rÊ r rr r rs r rq rn r Z
|
setbufsizeZUFUNC_BUFSIZE_DEFAULT)Údbry ZindxÚh2r< r= Úh1r r r Ú test_reduceaty s&
|
|
|
r c C s¦ t g d¡} t g d¡}t j || ¡}t|j|j t|jd t d¡}t jj|g dd}t|j|j t|jd t jj|g dd}t|j|j t|jd d
|
S ) z&Reduceat should work with empty arraysÚi4Zf8)r )r2 r% r rm r÷ r )r2 r N)r rq rs rn r r) rª r- )r rP r r r r Útest_reduceat_empty s
|
r c C sH t tjdt dtjt tjtjg} t ddt ddt ddt ddt ddt ddt ddg}tjddÒ | | D ]Â}t |g¡}| | D ]¨}t |g¡}t |¡r´t |¡r´qt||k dd||f d t||kdd ||f d t||kdd
|
||f d t||kdd ||f d t||kdd ||f d qqvW 5 Q R X d S ) Nr r r3 rU rV Fz%r < %rr z%r > %rz%r <= %rz%r >= %rz%r == %r)r r r· rü rq rE r )ZnansZfinsrP r r r r Ú test_complex_nan_comparisons¥ s$ $ ÿ r c C s* d} t | tt| t | t | ¡ d S )Nl *°n¶ )r r r r rT )Úvalr r r Útest_rint_big_intº s r Úftypec C sp t jdddddg| d}t jdddddg| d}t jdddddg| d}tt j |¡| tt j |¡| d S )Ng
ëQ¸
ã?g333333ã?g¤p=
|
×£è?g=
|
×£p=Ú?gR¸
ëQÈ?rp )r rq r r¬ r& rÇ )r r Zout_maxZout_minr r r Útest_memoverlap_accumulateÄ s
|
r z ufunc, dtypec C s< g | ]4}|j D ](}|d dkr|d dkr||d fqqS )r3 rÒ r ZDFGMmOr# ©r r rI r r r r Í s ýc C sv | j rt d¡ dD ]\}tjdddg| |d}| j|dd}| tj¡}tjtt || ¡tjd}t
|
|| qd S )NúFor generic signatures only©r% r¿ rÁ r rH rà r r rp rÒ ) rÁ rÈ r r rq r& r r© rL r6 r )r r) r r ÚaccZacc_u8r r r r Ú test_memoverlap_accumulate_cmpÍ s
|
r c C sP g | ]H}|j D ]<}|d |d kr|d |d kr|d dkr||d fqqS )r r r3 zDFGMmO?r# r r r r r Ý s ýc C s | j rt d¡ tjdd\ dD ]P}t dddg| ¡ |¡}| j||d}tjtt || ¡|d}t
|
|| q"W 5 Q R X d S ) Nr rU r& r r r r% rp ) rÁ rÈ r r rü rq r§ r& rL r6 r )r r) r r r r r r r Ú$test_memoverlap_accumulate_symmetricÝ s
|
r c C s0 t tjdddd} t | ¡ W 5 Q R X d S )Nr r s à¿ÿ)rª r) Úbuffer)r r r r r( r r r Ú test_signaling_nan_exceptionsí s r r r% r2 c C s@ G dd dt j}t j | |¡| |¡¡}|jjdks<td S )Nc @ s e Zd ZdS )z)test_outer_subclass_preserve.<locals>.fooN)r] r^ r_ r r r r rü ù s rü )r r r- ro r r r] r )r rü Úactualr r r Ú test_outer_subclass_preserveò s r c
|
s G dd dt j} G fdddt j | fD ]h}t d¡ |¡}tt}t j |ddg¡ W 5 Q R X t d¡ |¡}tt j ddg|¡|ks0t q0d S )Nc @ s e Zd Zdd ZdddZdS )z(test_outer_bad_subclass.<locals>.BadArr1c S s | j dkr| jd | _d S )Nr0 r )Úndimrª r< r r r Ú__array_finalize__ÿ s
|
z;test_outer_bad_subclass.<locals>.BadArr1.__array_finalize__Nc S s |S r+ r r r r r r s z:test_outer_bad_subclass.<locals>.BadArr1.__array_prepare__)N©r] r^ r_ r r r r r r ÚBadArr1þ s r c s" e Zd Z fddZdddZdS )z(test_outer_bad_subclass.<locals>.BadArr2c s. t | r*| jd dkr*| jd d d
| _d S )Nr3 r )r rª r< ©ÚBadArr2r r r s
|
z;test_outer_bad_subclass.<locals>.BadArr2.__array_finalize__Nc S s |S r+ r r r r r r s z:test_outer_bad_subclass.<locals>.BadArr2.__array_prepare__)Nr r r! r r r" s r" )r% r0 r r% )
|
r r r- r r rv rs ro r r )r r r ry r r! r Útest_outer_bad_subclassý s
|
|
r# c C s0 t d¡} tt t j | | ¡ W 5 Q R X d S )N)r r r r r r r r r r r r r r r r r )r r- r rw rs ro )Údeepr r r Útest_outer_exceeds_maxdims s
|
|
r% c C sV t d¡ t j¡} tjtdd t | | ¡ W 5 Q R X tjtdd6 | d¡
|
dd¡d d
d d d
f }t || ¡ W 5 Q R X tjtdd tj | | ¡ W 5 Q R X tjtdd tj | ¡ W 5 Q R X tjtdd tj | dd g¡ W 5 Q R X tjtdd tj | ¡ W 5 Q R X tjtdd tj | dd dg| ¡ W 5 Q R X d S )
|
Nr0 úHow unexpected :\)!rÄ rh r3 rÆ r% r r )r r r§ rØ rÈ rÉ r9 rq Z always_errorÚrepeatZreshapero r rn r& rp )r Z
|
non_contigr r r Ú#test_bad_legacy_ufunc_silent_errors s $ r( Úx1r r c C s* t jtdd t | d¡ W 5 Q R X d S )Nr& rÄ r )rÈ rÉ r9 rq Zalways_error_gufunc)r) r r r Ú$test_bad_legacy_gufunc_silent_errors> s r* )rF rÓ r{ r6 rÈ re rê rÌ Z fractionsr Ú functoolsr Ú collectionsr Znumpy.core.umathÚcoreZumathrb Z
|
numpy.corer rq Únumpyr Z numpy.testingr r r r
|
r r r r r r r r r r Z numpy.testing._private.utilsr Ú__dict__ÚvaluesZUFUNCSZ UFUNCS_UNARYrC Z UFUNCS_BINARYZUFUNCS_BINARY_ACCrD rJ rS rT r` rh r rÖ rÛ r: r= r> rf r
r rº r½ rÙ rÞ rô r r
|
rD rH rM rY rW rc rn rr rw ry 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Í r5 r[ Údoubler\ rd rH rI rù rü rý rþ rÿ r r r r r r r r rÎ r rØ r r r r r Zmatrixr r# r% r( r* r r r r Ú<module> sT @ ÿÿÿÿ8 ~k { Q
|
|
0#
|
ù 1- E \\???T +* 0c$ Gÿ
|
= ÿÿ ÿÿ
|
|
þ
|
|
þ
|
|
ý
|
|
