U
|
O ±d9] ã @ sP d dl Z d dlZd dlZd dlZd dlm mZ d dl m
|
Z
|
m Z m Z m Z mZ ejdd8 e eejd ¡jd kpe eejd ¡jejkZW 5 Q R X e j d¡ p¨eZejjeddZ G dd d Z G d
|
d d Z G d d d Z G dd dZ G dd dZ!G dd dZ"d ddZ#d ddZ$G dd dZ%G dd d Z&G d d d Z'dS )!é N)Ú assert_raisesÚ assert_equalÚassert_array_equalÚassert_almost_equalÚassert_array_max_ulpÚignore)ÚallÚlinuxz Inadequate C99 complex support©Úreasonc @ s6 e Zd Zdd Zedd Zejjdddd Z d S )
|
ÚTestCexpc C sz t }tj}||ddt d¡dd ||ddt d¡t d¡d t d¡tt d¡t d¡ }||dd|j|jd d S )Né r F)Úcheck_complex_valueÚnpÚexpÚcosÚsinÚcomplexÚrealÚimag)ÚselfÚcheckÚfÚref© r úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/core/tests/test_umath_complex.pyÚ test_simple! s zTestCexp.test_simplec s¼ t }tj | tjdddd | tjdddd | dtjtjtj | dtjtjtj | dtjtjtj | tjdtjd | tj dtjtj | tj dtj tjtj | tjdtjtj | tjdtj tj tj fdd}|d fdd }|d fd
|
d }|d fd d }|d | tjdtjtj | tjdtjtj | tjtjtjtj | tjtj tjtj | tjtjtjtj d S )Nr r Féÿÿÿÿç è?c sb d}t jddH t tt j t j¡}|jdks@|jdkrTt||j|jf W 5 Q R X d S )Nz0cexp(-inf, inf) is (%f, %f), expected (+-0, +-0)r ©Úinvalidr )r ÚerrstateÚarrayr Úinfr r ÚAssertionError©ÚdummyZmsgformÚz©r r r Ú_check_ninf_infH s
|
z5TestCexp.test_special_values.<locals>._check_ninf_infc sd d}t jddJ t tt jt j¡}t |j¡rBt |j¡sVt ||j|jf W 5 Q R X d S )Nz1cexp(inf, inf) is (%f, %f), expected (+-inf, nan)r r )
|
r r! r" r r# Úisinfr Úisnanr r$ r% r( r r Ú_check_inf_infR s
|
z4TestCexp.test_special_values.<locals>._check_inf_infc sb d}t jddH t tt j t j¡}|jdks@|jdkrTt||j|jf W 5 Q R X d S )Nz0cexp(-inf, nan) is (%f, %f), expected (+-0, +-0)r r r ) r r! r" r r# Únanr r r$ r% r( r r Ú_check_ninf_nan\ s
|
z5TestCexp.test_special_values.<locals>._check_ninf_nanc sd d}t jddJ t tt jt j¡}t |j¡rBt |j ¡sVt
|
||j|j f W 5 Q R X d S )Nz2cexp(-inf, nan) is (%f, %f), expected (+-inf, nan)r r ) r r! r" r r# r- r* r r+ r r$ r% r( r r Ú_check_inf_nanf s
|
z4TestCexp.test_special_values.<locals>._check_inf_nan)r r r ÚPZEROÚNZEROr# r- Úpi)r r r) r, r. r/ r r( r Útest_special_values+ s2 z TestCexp.test_special_valuesz)cexp(nan + 0I) is wrong on most platformsr
|
c C s" t }tj}||tjdtjd d S )Nr )r r r r- )r r r r r r Útest_special_values2{ s z TestCexp.test_special_values2N)
|
Ú__name__Ú
|
__module__Ú __qualname__r Ú platform_skipr3 ÚpytestÚmarkÚskipr4 r r r r r s
|
|
|
O r c @ s8 e Zd Zdd Zeejje ¡ dkdddd Z
|
dS ) ÚTestClogc C sD t ddg¡}t t |¡¡dt |¡ }t |¡}t|| d S )Nù ð? ù ð? @ù ð?)r r" ÚlogÚabsZangler ©r ÚxZy_rÚyr r r r
s
|
zTestClog.test_simpleZarmv5telz See gh-413.r
|
c C s g }g }t jdd6 t jt jgtd}tt j t j}ttt j | W 5 Q R X t jdd t
|
t |¡| W 5 Q R X | |¡ | |¡ t jdd2 t jdgtd}tt j d}ttt j | W 5 Q R X t jdd t
|
t |¡| W 5 Q R X | |¡ | |¡ t jtdt jgtd}tt jdt j }t
|
t |¡| | |¡ | |¡ t jtdt jgtd}t
|
t |¡| | |¡ | |¡ t jdd , t jtd
|
t j gtd}tt j t j }W 5 Q R X t jdd t
|
t |¡| W 5 Q R X | |¡ | |¡ t jdd " t jt jd t j gtd}W 5 Q R X t jdd t
|
t |¡| W 5 Q R X | |¡ | |¡ t jt j d gtd}tt jt j}t
|
t |¡| | |¡ | |¡ t jt jd gtd}tt jd}t
|
t |¡| | |¡ | |¡ t jtt j t jgtd}tt jd t j }t
|
t |¡| | |¡ | |¡ t jtt jt jgtd}tt jd t j }t
|
t |¡| | |¡ | |¡ t jtt jt j gtd}tt jt j }t
|
t |¡| | |¡ | |¡ t jtt j t j gtd}t
|
t |¡| | |¡ | |¡ t jtt j dgtd}tt j t j }t
|
t |¡| | |¡ | |¡ t jtt j t jgtd}tt jt j }t
|
t |¡| | |¡ | |¡ t jtt j t j gtd}tt j t j }t
|
t |¡| | |¡ | |¡ t j|td}t j|td}t jdd: t t|D ]&}t
|
t || ¡ ¡|| ¡ qVW 5 Q R X d S )NÚraise)Údivide©Údtyper r r ç à?r r ç ð?r? r ç Ð?)r r! r" r1 r r# r2 r ÚFloatingPointErrorr@ r Úappendr- ÚrangeÚlenÚconj)r ZxlZylrC rD ÚxaZyaÚir r r r3 s²
|
|
|
|
|
|
|
|
|
|
&
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
z TestClog.test_special_valuesN) r5 r6 r7 r r8 r9 r: ÚskipifÚplatformÚmachiner3 r r r r r< s r< c @ s( e Zd Zdd Zdd Zedd ZdS )Ú TestCsqrtc C sN t tjdddd dt d¡ }|}t tjdd||d t tjdddd d S )Nr r rI é Fr )r r Úsqrt)r ZrresZiresr r r r s
|
zTestCsqrt.test_simplec C s8 t t tdd¡¡}dd }t|dd|j|jd d S )Nr c S s t t | ¡¡S ©N)r rX rP )r' r r r r % s z*TestCsqrt.test_simple_conjugate.<locals>.fF)r rP rX r r r r )r r r r r r Útest_simple_conjugate" s z TestCsqrt.test_simple_conjugatec C s¶ t }tj}||tjddd ||tjddd ||dtjtjtj ||dtjtjtj ||tjtjtjtj ||tjtjtjtj ||tjtjtjtj ||tj tjtjtj ||tj tjtjtj ||dtjtjtj ||dtjtjtj ||dtjtjtj ||tj dtjtj ||tjdtjtj dd }|d ||tjtjtjtj ||tjdtjtj ||tjdtjtj ||tjtjtjtj d S )Nr r r c S sh d}t t tt j t j¡¡}t jdd2 t |j¡rFt |j
|
¡sZt ||j|j
|
f W 5 Q R X d S )Nz3csqrt(-inf, nan) is (%f, %f), expected (nan, +-inf)r r ) r rX r" r r# r- r! r+ r r* r r$ r% r r r r. N s
|
z6TestCsqrt.test_special_values.<locals>._check_ninf_nan)r r rX r0 r1 r# r- )r r r r. r r r r3 - s, z TestCsqrt.test_special_valuesN)r5 r6 r7 r rZ r8 r3 r r r r rV s rV c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) ÚTestCpowc C s t jdd| _d S ©Nr r ©r ZseterrZolderr©r r r r Ú setup_methode s zTestCpow.setup_methodc C s t jf | j d S rY r] r^ r r r Úteardown_methodh s zTestCpow.teardown_methodc C s: t dddt jt jg¡}|d }t |d¡}t|| d S )Nù ð? ð?ù @r> rW )r r" r# r- Úpowerr rB r r r r k s zTestCpow.test_simplec s t ddddt jt jg¡ t ddddddg¡ttt }ddd d
|
tt jt jtt jt jg} fd d |D }|D ] }t|| || d | d q|d S )Nr r? rW ù @®Gáz®×?ù à¿ ø?é r= ùàg÷FÌÊ? ù¨
|
h´xëÖ?éÁXÂã?ùSLïJ#4À?éÆ4Cß?c s g | ]} | | qS r r )Ú.0rR ©rC rD r r Ú
|
<listcomp> s z(TestCpow.test_scalar.<locals>.<listcomp>úLoop %d
|
©Úerr_msg© r r" r# r- ÚlistrN rO r r )r ÚlxÚp_rÚn_rrR r rk r Ú test_scalarq s ú zTestCpow.test_scalarc C s t ddddt jt jg¡}t ddddddg¡}ttt|}ddd d
|
tt jt jtt jt jg}|| }|D ] }t|| || d | d qpd S ) Nr r? rW rd re rf r= rg rh ri rm rn rp )r rC rD rr rs rt rR r r r Ú
|
test_array
s ú zTestCpow.test_arrayN)r5 r6 r7 r_ r` r ru rv r r r r r[ d s
|
r[ c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) ÚTestCabsc C s t jdd| _d S r\ r] r^ r r r r_ s zTestCabs.setup_methodc C s t jf | j d S rY r] r^ r r r r` s zTestCabs.teardown_methodc C sT t dddt jt jg¡}t t d¡dt d¡t jt jg¡}t |¡}t|| d S )Nra rb r> g @rW é )r r" r# r- rX rA r rB r r r r s $
|
zTestCabs.test_simplec C s¸ t jdgtd}tt |¡t |¡ t jtdt jgtd}tt |¡t |¡ t jtt jt jgtd}tt |¡t |¡ t jtt jt jgtd}tt |¡t |¡ d S )Nr= rG r ) r r" r r rA r r1 r# r- )r rC r r r Ú test_fabs¦ s zTestCabs.test_fabsc C sD g g }}| tj¡ | tj¡ ttjtjtjtj | tj¡ | tj ¡ ttjtj tjtj | tj¡ | tj¡ ttjtjtjtj | tj ¡ | tj¡ ttjtj tjtj dd }dd }tj|td}t|t| kr t|ksn t t
|
||D ]"\}}|||}t|||| q d S )Nc S s t t | ¡¡S rY )r rA rP )Úar r r r Ë s z%TestCabs.test_cabs_inf_nan.<locals>.fc S s t t| |¡S rY )r rA r )rz Úbr r r ÚgÎ s z%TestCabs.test_cabs_inf_nan.<locals>.grG ) rM r r- Úcheck_real_valuerA r# r" r rO r$ Úzip) r rC rD r r| rQ ÚxiÚyir r r r Útest_cabs_inf_nan´ s(
|
*
|
zTestCabs.test_cabs_inf_nanN)r5 r6 r7 r_ r` r ry r r r r r rw s
|
rw c @ s2 e Zd Zdd Zejjdddd Zdd Zd S )
|
ÚTestCargc C s\ t tjdddd t tjdddtj d t tjdddtj d t tjtjtjtj d S )Nr r FrI rK )r} ÚncuÚ_argr r2 r0 r^ r r r r Ø s zTestCarg.test_simplez;Complex arithmetic with signed zero fails on most platformsr
|
c C s$ t tjtjtjtjd t tjtjtjtj d t tjtjtjtj t tjtjtjtj t tjdtjtjd t tjdtjtjd t tjdtjtjd t tjdtjtj d t tjtjddtj d t tjtjddtj d t tjtjddtj d t tjtjddtj d d S )NFr r rI ç à¿)r} r r r r1 r0 r2 r^ r r r Ú test_zeroà s zTestCarg.test_zeroc C sl t tjtj dtjd t tjtj dtj d t tjtjdtjd t tjtjdtjd t tjdtjdtj d t tjdtj dtj d t tjtj tjdtj d t tjtj tj dtj d t tjtjtjdtj d t tjtjtj d tj d t tjtjd
|
tjd t tjd
|
tjtjd t tjtjtjtjd t tjtjtjtjd d S ) Nr Fr rI r
r g è¿rK g пr ) r} r r r r# r2 r0 r1 r- r^ r r r r3 þ s z TestCarg.test_special_valuesN) r5 r6 r7 r r9 r: r; r r3 r r r r r × s ÿ
|
r Tc C s8 t t||g¡}|r&t| || nt| || d S rY )r r" r r r )r Úx1Úy1rC ÚexactÚz1r r r r} s r} c C sZ t t||g¡}t||}t jdd( |r>t| || nt| || W 5 Q R X d S r\ )r r" r r! r r )r r r Zx2Úy2r r Zz2r r r r # s
|
r c @ sD e Zd Zej dddddddg¡ej dejejg¡d d
|
Z d S ) ÚTestSpecialComplexAVXÚstrideéüÿÿÿéþÿÿÿr r rW é Úastypec C s~ t jtt jt jtt jt jtt jt jtt jt jtdt jtt jdtddtdt jtt jdg |d}t jt jt jt jt jt jt jdt jt jg |jjd}t jtt jt jtt jt jtt jt jtt jt jtt j t jtt jt jtddtt jt jtt jt jg |d}tt |d d |
¡|d d |
t j dd* tt
|
|d d |
¡|d d |
W 5 Q R X d S )Nç rG r r ) r r" r r- r# r rH r rA r! Úsquare)r r r ÚarrÚabs_trueZsq_truer r r rv - s8
|
|
|
|
øø 4 øø $z TestSpecialComplexAVX.test_arrayN©
|
r5 r6 r7 r9 r: Z parametrizer Z complex64Z
|
complex128rv r r r r r , s r c @ sv e Zd Zej ddddddddd d
|
d d d ddddg¡ej dddddddddg¡ej dejejg¡dd Z dS )ÚTestComplexAbsoluteAVXÚ arraysizer rW rf r rx é é é é é
|
é é é é é é r r éýÿÿÿr r r c C sH t j||d}t j||jjd}tt |d d |
¡|d d |
d S )NrG )r Zonesr rH r rA )r r r r r r r r r rv H s z!TestComplexAbsoluteAVX.test_arrayNr r r r r r G s , r c @ s\ e Zd Zej ddddddddd g¡ej d
|
ejejg¡ej d d d dg¡dd Z dS )Ú TestComplexAbsoluteMixedDTypesr r r¤ r r r rW rf r r ÚfuncrA r Ú conjugatec
|
C s¢ ddddddddd |fd
|
g
|
}t jd d d ddddddg |d}tt |}|d }||d d |
}|d ¡ } || d d |
}
|
t|
|
j|jd t|
|
j|jd d S )N)Z template_idú<i8)Z
|
bank_chisqú<f4)Zbank_chisq_dofr¨ )Zchisqr© )Z chisq_dofr¨ )Z
|
cont_chisqr© )Z psd_var_valr© )Zsg_chisqr© Z mycomplex)Z
|
time_indexr¨ )
|
r r r gW@¡ª?ÀéÈ r r rJ y @ @iâZ )
|
r r r g /ݤNp@é* r r rJ y @ (@is )
|
r r r gºÚýeJ@r« r r rJ y @ .@iñN )
|
r r r gó7¡7J@r« r r rJ y @ 8@iTÎ )
|
rW r r g2 {½ûs@é: r r rJ y 4@ 5@i{ )
|
rf r r gÿÏa¾¼¶r@é4 r r rJ y (@ A@i )
|
r r r g=,Ôæ½C@é r r g/EH]"@y "@ D@ih )
|
r r r g+~©5S@r® r r rJ y <@ F@i+ )
|
rx r r gm¨çoèa@é r r gÂ1Ëþ%@y &@ N@ic rG r )r r" ÚgetattrÚcopyr r r ) r r r r¦ rH ZvecZmyfuncrz r| r{ Úhr r r rv S s< ý÷ ÷
|
|
z)TestComplexAbsoluteMixedDTypes.test_arrayNr r r r r r¥ R s r¥ )T)T)(ÚsysrT r9 Únumpyr Z numpy.core._multiarray_umathÚcoreZ_multiarray_umathr Z numpy.testingr r r r r r! r r r# r r@ r1 r2 Zfunctions_seem_flakyÚ
|
startswithZxfail_complex_testsr: rS r8 r r< rV r[ rw r r} r r r r¥ r r r r Ú<module> s2 ÿ ÿd P5>D
|
|
|
