U
|
P ±d( ã @ sÜ d dl mZ d dlZd dlZd dlm mZ d dlm m Z d dl
|
m Z m Z m Z d dlmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m Z m Z m Z m Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZY d dlZm[Z[ ej\Z\dfddZ]G d d
|
d
|
Z^G d d d Z_G d d dZ`dd ZadS )é )ÚreduceN)Úassert_Ú assert_raisesÚ assert_equal)KÚMaskTypeÚ MaskedArrayÚabsoluteÚaddÚallÚallcloseÚallequalÚalltrueÚarangeÚarccosÚarcsinÚarctanÚarctan2ÚarrayÚaverageÚchooseÚ concatenateÚ conjugateÚcosÚcoshÚcountÚdivideÚequalÚexpÚfilledÚgetmaskÚgreaterÚ greater_equalÚinnerÚ isMaskedArrayÚlessÚ
|
less_equalÚlogÚlog10Ú make_maskÚmaskedÚ masked_arrayÚ masked_equalÚmasked_greaterÚmasked_greater_equalÚ masked_insideÚ masked_lessÚmasked_less_equalÚmasked_not_equalÚmasked_outsideÚmasked_print_optionÚ masked_valuesÚ masked_whereÚmaximumÚminimumÚmultiplyÚnomaskÚnonzeroÚ not_equalÚonesÚouterÚproductÚputÚravelÚrepeatÚresizeÚshapeÚsinÚsinhÚsometrueÚsortÚsqrtÚsubtractÚsumÚtakeÚtanÚtanhÚ transposeÚwhereÚzeros)ÚpickleÚ c C s, t | |}|s(td| d| d| |S )NzNot eq:Ú
|
z
|
----)r Úprint)ÚvÚwÚmsgÚresult© rY úQd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/ma/tests/test_old_ma.pyÚeq s
|
r[ c @ s e Zd Zdd Zdd Zej dddg¡dd Zd
|
d Z d d Z
|
dd Z dd Z dd Z dd Zdd Zdd Zdd Zd d Zd d Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zd.d/ Zd0d1 Z d2d3 Z d4d5 Z d6d7 Z d8d9 Z d:d; Z!d<d= Z"d>d? Z#d@S )AÚTestMac C s t ddddtd dddddddg ¡}t dd ddd
|
d d dddd dg ¡}d}d d d d d d d d d d d d g }d d d d d d d d d d d d g }t||d}t||d}t dd ddg¡}t|d d d d gd} t |d|¡}
|
|j} | d¡ ||||||||| |
|
| f | _d S )Nç ð?g Àç @ç @ç @g $Àç $@ç @ç g ð¿ç Àé r ©Úmaskg à¿ç à?gé?ç@µx ¯D)Únpr ÚpirO rC Zset_fill_valueÚd© ÚselfÚxÚyÚa10Úm1Úm2ÚxmÚymÚzÚzmÚxfÚsrY rY rZ Ú setup_method& s &"
|
zTestMa.setup_methodc C sÈ | j \ }}}}}}}}} }
|
} tt| tt| tt|| t|j| t|j|j t|jtdd | tt|t |tdd | tt
|
||
|
tt
|
t |d|
|
tt
|
|| d S )Nc S s | | S ©NrY ©ro rp rY rY rZ Ú<lambda>= ó z)TestMa.test_testBasic1d.<locals>.<lambda>c S s | | S r{ rY r| rY rY rZ r} > r~ ri ) rl r r# r rC ÚdtypeÚsizer r Úlenr[ r rm rY rY rZ Útest_testBasic1d5 s zTestMa.test_testBasic1dry ©é é )é é c C sØ | j \ }}}}}}}} }
|
} }||_||_||_||_|| _tt| tt| tt|| t|j| t|jtdd | tt|t|tdd | tt || tt t
|
|d| tt || d S )Nc S s | | S r{ rY r| rY rY rZ r} Q r~ z)TestMa.test_testBasic2d.<locals>.<lambda>c S s | | S r{ rY r| rY rY rZ r} R r~ ri ) rl rC r r# r r r r r r[ r ) rn ry ro rp rq rr rs rt ru rv rw rx rY rY rZ Útest_testBasic2dC s zTestMa.test_testBasic2dc C sì | j \ }}}}}}}}} }
|
} tddgddgg} t| ddgddgg} tt| | | | tt| | | | tt| | | | dD ]X} | | ¡}| | ¡}| | ¡}| | ¡}|
|
| ¡}
|
tt| | tt|| || tt|| || tt|| || tjddd tt|| || W 5 Q R X tt|| || tt|| || tt|| || tjddd tt|| || W 5 Q R X tt|| || tt|| || tt|| || tt|| || tt|d |d ttt|d t|d tt|| || ttt ||¡t || ttt
|
||¡t
|
|| ttt ||¡t || tjddd" ttt ||¡t || W 5 Q R X qd S ) Nre r r r ))é r )r r Úignore©r Úinvalidç @) rl r r* r r[ Úreshaperj ÚerrstateÚabsr rI r8 r )rn ro rp rq rr rs rt ru rv rw rx ry Úa2dÚa2dmrY rY rZ Útest_testArithmeticW sF
|
|
|
|
|
|
zTestMa.test_testArithmeticc C s> t dg¡}tdg}tt|| t tt|| t d S )Nre )rj r r Ú
|
isinstancer )rn ÚnaÚmarY rY rZ Útest_testMixedArithmetic} s
|
z TestMa.test_testMixedArithmeticc C sö | j \ }}}}}}}}} }
|
} ttt |¡t| ttt |¡t| ttt |¡t| ttt |¡t| ttt |¡t| ttt |¡t | tj
|
dddZ ttt t |¡t | ttt t |¡t | ttt t |¡t| W 5 Q R X ttt |¡t| ttt |¡t| ttt |¡t| ttt |¡t| ttt ||¡t|| ttt |¡t| ttt ||¡t|| ttt ||¡t|| ttt ||¡t|| ttt ||¡t|| ttt ||¡t|| ttt ||¡t|| ttt |¡t| ttt ||f¡t ||f ttt ||f¡t ||f ttt ||f¡t ||f ttt |||f¡t |||f d S )Nr r ) rl r r[ rj r r rD rE rL rM r rH r r& r' r r r r r r r r; r$ r r% r! r r rm rY rY rZ Útest_testUfuncs1 s8 & zTestMa.test_testUfuncs1c C sö t ddddgddddgd}tt|jjtjk tdt| tdtd ttdt ddgd | d ¡}tt|jjtjk tt
|
t|dtj tt|jjtjk ttdt| tt t|dt k ttdd
|
gt|d d S ) Nrc r] r^ rb re r rf r
)r r r )r r r r Útyperj Zintpr r[ r r Úndarrayr r9 )rn ÚottrY rY rZ Útest_xtestCount¢ s
|
zTestMa.test_xtestCountc C sb | j \ }}}}}}}}} }
|
} t |¡} t|} ttt| t | ¡ ttt| t | ¡ d S r{ )
|
rl rj r@ r r[ Úmaxr6 r Úminr7 )rn ro rp rq rr rs rt ru rv rw rx ry ZxrZxmrrY rY rZ Útest_testMinMax± s
|
|
zTestMa.test_testMinMaxc C sÚ | j \ }}}}}}}}} }
|
} tttj |¡t |¡ tttj |¡t |¡ ttdttddd ttdttddd tttj|ddt|dd tttjt |dddt|dd ttt |d¡t|d tttj
|
|ddt
|
|dd ttt
|
|d¡t
|
|d tttj
|
t |dddt
|
|dd t | dkrÖttt ||fd¡t ||fd tttj |d¡t |d¡ ttt |d¡t|d ttt
|
|d¡t
|
|d d S )Nr r ©Úaxisre ) rl r r[ rj r r Ú
|
accumulaterJ r r r> r r rm rY rY rZ Útest_testAddSumProd» s( &
|
ÿ ÿ zTestMa.test_testAddSumProdc C s* t ddddg¡}t|ddddgd}t|ddddgd}t|}t| t| ttt |¡t|dd tt|d t|d k t|d |d k t|d tk tt|d |d tt|dd
|dd
tt|d d
|d d
tt|dd
|dd
d |d< d |d< tt|| d
|
|dd
< d
|
|dd
< tt|| t|d< tt|| t|dd
< tt|| ||d d
< t|d< tt t
|
|tddddg t ddddgddddg|d d
< tt t
|
|tddddg t ddddgddddg|d d
< tt t
|
|tddddg tt |tddddg t d¡d }t |d }tt|| tt tdddddgt|j ttd |j tdd ddgt}t dd ddgt¡}|d }|d }tt|t tt|t t|| t|dd
jdk d S )Nre r r r
r rf )Ú
|
fill_valueé é éc r] rb Zhello)r )rj r ÚstrÚreprr r[ rG r r) r r r* r r4 r rg r¤ Úobjectr rC )rn Úx1Zx2Zx3Úx4Ús1Ús2rY rY rZ Ú test_testCIÐ sZ " "
|
|
zTestMa.test_testCIc C s. dddddg}t |}t |}t||k t |dd}t||k t d¡}t||d}t|j|k tt||j t|j|k t|dd}t|jj|jjk t||dd}t|j|k t|d t k d |d< t|d t k t|j|k tt|j
|
d t||dd} t| j|k t| d t k d | d< t| d t k t| j|k tt| j
|
d t|d
|
|d}
|
tt |
|
j |d
|
j k td } t | d< t | d } ttt| | g| ttt| ddddddddg t| d dd} tt| ddddddddg t| ddd}tt| | d S )Nr re T)Úcopyr¥ rf ©rg r° r r¦ r] r )é )r r r r r r
)r( r rj r r Ú_datar Ú_maskZ__array_interface__r) rg r r rB r[ r r rA )rn ÚnÚmrs Úm3r« Úy1Zy1aÚy2Zy2aZy3r¬ Zy4Zy5Zy6rY rY rZ Útest_testCopySize sP
|
ÿ
|
" zTestMa.test_testCopySizec C s t d}dddddg}t|}| ¡ }t||d}t|d tk t|d tk ddg|ddg< t|j|k t|d tk t|d tk tt|ddd d
|
dg t||d d }| ddd gd
|
d dg¡ t|j|k t|d tk t|d tk tt|d
|
d dddg d S )Nr¥ r re rf r
r é
|
é( r éÿÿÿÿTr± éd éÈ ) r r( r° r r r) r´ r[ r? )rn rl rµ r¶ rs ro rY rY rZ Ú test_testPut5 s$ zTestMa.test_testPutc C sn t d}t|dddddgd}tddgddgd}t|d tk t|d tk ||dd
< t|d tk t|d tk tt|ddddd g t d}t|dddddgd}|dd
}tddgddgd}t|d tk t|d tk ||d d
< t|d tk t|d tk tt|ddg t|d tk t|d tk tt|ddddd g d S )
|
Nr¥ r rf r» r¼ re r r
r )r r r r) r[ )rn rl ro rv rp rY rY rZ Ú test_testPut2K s, zTestMa.test_testPut2c C sp | j \ }}}}}}}}} }
|
} ddddddddddddg } t | ¡d } t|| | ttt|| dd| k d S )Nre r r )rl rj r: r? r r
|
rK )rn ro rp rq rr rs rt ru rv rw rx ry r¶ ÚirY rY rZ Útest_testMaPute s
|
zTestMa.test_testMaPutc
|
|
C sÄ t d}| dd¡}d|jd< t|d dk |d| }tt|j| tt|jd| tt|t| jd| | d |jd
|
< t d}t|d < tt |d t tk |d k}tt
|
t |ttd k tt t |tt|j k t ||t}t|j |j k t|d tk t|d tk t|d tk t|d tk t|d tk tt|| t |t|}t|j |j k t|d tk t|d tk t|d tk t|d tk t|d tk t||}t|j |j k t|d tk t|d tk t|d tk t|d tk t|d tk tt|| tdddddg}tdddd d g}t|d< t ||| }tt|ddd ddg t|d < t ||| }tt|ddd ddg t|d tk t|d tk t|d tk tttt|d|t|d tttt|d|t|d tttt|d|t|d tttt|d|t|d tttt|d|t|d tttt|d|t|d tttt|d|t|d ttt t t ddd d ddddg ttt t t ddd dddd dg ttt tt t ddd d d d gddd j ddddd g ttt tt t dd dd d d gddd j ddd d dg ttttt t ddd d d d gddj dd dd d g ttttdddddgdd d d d gddj dd dd dg tttddd d d gddd ddgddd ddg t!d t"j#d }t$|j t%}t||}tt|| t&|| |f}tt|ddd ddg t|d tk t|d tk t|d tk t d }t|d< t d d }t|d< tdddd d d gdd d d d d gd}| 'd¡}t |||}t |||} tt|| tt(| t)k tt| d ddd d d!g t |td}tt|ddddddg t |dt}tt|ddddddg d S )"Né r r¥ r )re r y $@r» ée rc .r
r² r é r¦ r] r^ rb r_ r` re r rd éûÿÿÿéÇ rf r§ )r» r» r» )r r é r¼ é2 )*r r Zflatr r[ ÚrealÚimagr r) r¨ r rO rC r r5 r r r, r! r- r$ r/ r% r0 r; r1 r r+ r. ÚlistÚranger2 rg r< rj Úfloat32rP r r r r r9 )
|
rn ro rv ÚcZatestZbtestZctestrp Úcmrw rY rY rZ Útest_testOddFeaturesl sê
|
|
|
ÿ (( ÿ ÿ þ ÿ ÿ þ ÿÿ þ ÿÿ þ ÿ
|
$
|
zTestMa.test_testOddFeaturesc C sÜ t ttdddgdddgdddg t ttdddgdddgdddg td}tdd }t|d< t|d< t tt||tt|||| t tt||tt|||| t t |¡dk t t |¡dk d S )Nre r r
r r r¦ r¥ )
|
r r[ r7 r6 r r) rO r$ r r )rn ro rp rY rY rZ Útest_testMinMax2Ñ s && ""zTestMa.test_testMinMax2c C s& t d}t d¡}t|dd
< | ddd¡}| ddd¡}ttt |d¡t|d ttt |dd¡t|dd ttt t |d t |d ¡t|| ttt
|
t |d t |d ¡t
|
|| t d
|
dd ddgt }t|d< t|d ddg}t|d d
|
k t|d dk t|d dk d S ) Né r¥ r r r
r )r r re re r ÚabcÚdef) r rj r) r r r[ rN rK r" r r= r rª )rn ro rp ÚtrY rY rZ Ú test_testTakeTransposeInnerOuterÞ s&
|
ÿÿz'TestMa.test_testTakeTransposeInnerOuterc C sÒ t d}t d}t d}t|d< |d7 }tt||d |d7 }tt||d t d}t d}t|d< |d8 }tt||d |d8 }tt||d t dd }t dd }t|d< |d9 }tt||d |d9 }tt||d t dd }t d}t|d< |d }tt|| |d }tt|| t dd }t dd }t|d< |d }tt||d |t d }tt|td t d tj¡}t d}t|d< |d7 }tt||d d S )Nr» r re r] r^ )r» )r r) r r[ r< Zastyperj rÏ )rn rp ro rt rY rY rZ Útest_testInplaceò sR zTestMa.test_testInplacec C sb t d}t|ddd
< | dd¡}tdtjd D ]*}tj||d}t |¡}tt || q2d S )Nr r r» r r
re )Úprotocol)
|
r r) r rÎ rQ ÚHIGHEST_PROTOCOLÚdumpsÚloadsr r[ )rn ro Úprotory rp rY rY rZ Útest_testPickle$ s
|
zTestMa.test_testPicklec C sH t d}t|d< tttdk t|d tk tt|d dd d S )Nr re z--r )r r) r r¨ r r )rn ZxxrY rY rZ Útest_testMasked. s
|
zTestMa.test_testMaskedc C sP t ddddgddddgd}ttdt|dd ttdt|ddddgd t|ddddgd
|
d \}}ttd| t|d k t|d d
< tt|ddtk t ddddgddddgd}| d d ¡}t|d d
df< ttt|ddddg tt|ddd tk ttddgt|dd t|dd
|
d\}}tt|ddg d S )Nrc r] r^ rb re r rf r ©ÚweightsT©râ Úreturnedr_ r ©r¡ rä )r r r[ r r) r )rn r rX ZwtsrY rY rZ Útest_testAverage16 s zTestMa.test_testAverage1c C s0 ddddddg}ddddddgddddddgg}t d}ttt|ddd ttt|d|dd tt ddt d g}ttt|d tj t d¡¡d d ttt|ddt d¡d d ttt|ddt|ddt|ddd g ttt|d |d
|
d ttt|d|dd d ddddg ttt|ddt|ddt|ddd g td}ddddddg}ddddddgddddddgg}t d}ddddddg} tttt
|
||ddd tttt
|
||ddd ttt
|
||ddt k t tt
|
|| ddd t t tt
|
||ddd t
|
||}
|
ttt|
|
d d ttt|
|
ddd d ddddg ttt|
|
ddddg ttt|
|
d|dd d ddddg t d} t dd } t| | g| | ggddd\} }t t| t| t | j|j tt ddddgdd\}}t t|t| tt ddd\}}t t|t| tt dt ddd\}}tt|t|k tddgddggt}t
|
|ddgddgg}t|dd}tt|d dg t|dd}tt|d dg t|d d}tt|d t|dd}tt|d dg d S ) Nr re r r r )r¡ râ r^ rb g (@rá g«ªªªªª
|
@rc r] r_ ra g ÀX@g @r` r
Trå )r r r
)r¡ râ rä )rä rã r r rh g«ªªªªª@g ø?)r r r r r rj r r rP r< r* r) r r rC Úfloatr[ )rn Zw1Zw2ro rp rr rs r· Zm4Zm5rv ÚaÚbZr1Zr2r r Za2daZa2dmarY rY rZ Útest_testAverage2I sx ÿ$ÿÿÿ
|
ÿÿ zTestMa.test_testAverage2c C s t dttd t dttd t dttdggg t dttdgg ttttddg ttttddg ttttddgddgd d S )Nre r] r rf )r Úintr rç r Ú TypeErrorÚ
|
ValueErrorÚbool©rn rY rY rZ Útest_testToPython s zTestMa.test_testToPythonc C sÆ t ddd}tjdd tdt d j W 5 Q R X td| j t| j t| j tt||j tt||j t| ¡ j|j jk t ddd}t| ¡ |j k t
|
t |t t d S )Nr re rf r ©r ) r rj r r rg r6 r7 r r r³ r r¨ r3 )rn rt ro rY rY rZ Útest_testScalarArithmetic s z TestMa.test_testScalarArithmeticc
|
C sv t dddg}tt| ¡ |j ¡ tt| ¡ |j ¡ tt| ¡ |j ¡ tt| ¡ |j ¡ tt| ddddd¡|j ddddd¡ tt| dddg¡|j dddg¡ tt|
|
¡ |j
|
¡ tt| ¡ |j ¡ t ddgddgg}tt| ¡ |j ¡ tt| ¡ |j ¡ tt| ddg¡|j ddg¡ tt| ¡ |j ¡ d S )Nre r
r r r )r r r[ Úanyr³ r
|
ZargmaxZargminr ÚcompressZconjr ÚdiagonalrJ rK rN )rn rè r¶ rY rY rZ Útest_testArrayMethods s ÿ($z TestMa.test_testArrayMethodsc C s t dddg}t|jd d S )Nre r
r )r r Úndim©rn rè rY rY rZ Útest_testArrayAttributes¬ s z TestMa.test_testArrayAttributesc C s t dd ttjD d S )Nc S s& g | ] }|t tkr| d ¡s|qS )Ú_)Údirr Ú
|
startswith)Ú.0r¶ rY rY rZ Ú
|
<listcomp>± s
|
þz'TestMa.test_testAPI.<locals>.<listcomp>)r rû rj r rï rY rY rZ Ú test_testAPI° s zTestMa.test_testAPIc C sZ t dddg}t dddgdddgd}t|d jd t|d jd t|d jd d S )Nre r
r r rf rY )r r rC )rn rè ré rY rY rZ Ú test_testSingleElementSubscriptµ s
|
z&TestMa.test_testSingleElementSubscriptc C s@ t ddddgddddgd}|dk}d||< t|d tk d S )Nre r r
r r rf r¥ )r r r) )rn rè rÐ rY rY rZ Ú test_assignment_by_condition¼ s z#TestMa.test_assignment_by_conditionc C sT t ddgddgd}t ddgddgd}|dk }|| ||< ddg}t|j| d S )Nr re Frf T)r* r rg )rn rè ré rg Z expected_maskrY rY rZ Ú test_assignment_by_condition_2Ã s z%TestMa.test_assignment_by_condition_2N)$Ú__name__Ú
|
__module__Ú __qualname__rz r ÚpytestÚmarkZ parametrizer 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 rY rY rY rZ r\ $ s>
|
&
|
23e 2
|
: r\ c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) Ú
|
TestUfuncsc C sZ t dddtd gd ddgdgd dt dddtd gd ddgdgd df| _d S )Nr] r r½ r re r rf )r rk rl rï rY rY rZ rz Î s ((ÿzTestUfuncs.setup_methodc C sð dddddddddd d
|
d d g }d D ]È}zt t|}W n tk
|
rR t t|}Y nX t tj|}| jd |j
}t ¡ > ||krtj dd |dkr¢tj dd || }|| }W 5 Q R X t
|
t | d¡| d¡| t
|
t |j|j q"d S )NrH Úarctanhr r Úarccoshr& r' r Ú true_divideÚ floor_divideÚ remainderÚfmod)*rH r& r' r r rD r rL r r r rE r rM Zarcsinhr
|
r r ÚfabsÚnegativeÚfloorÚceilZ logical_notr rI r8 r r r r r Úhypotr r r; r% r! r$ r Ú logical_andÚ
|
logical_orÚ logical_xorr )r )r r& r' rñ r )ÚgetattrÚumathÚAttributeErrorÚ fromnumericrj r rl Zninr Zseterrr r[ r Úeqmaskrg )rn Zf_invalid_ignoreÚfZufZmfÚargsÚurÚmrrY rY rZ Útest_testUfuncRegressionÒ s< ý
|
z#TestUfuncs.test_testUfuncRegressionc C s\ | j d }tt|dd tt|dd tt|d d
ddd tt|ddd d S )Nr r r
)rl r r rF r rJ r> rø rY rY rZ Ú test_reduce÷ s
|
|
zTestUfuncs.test_reducec C s¦ t dd dd¡}t|dk |}t| ¡ | ¡ t| ¡ d t| d¡| d¡k ¡ t| d¡dddd gk ¡ t| d¡d j t| d¡d j d S )
|
Nre é r
r r¥ r r rÆ r² ) r r r5 r r r r r
|
rg )rn rè ZamaskrY rY rZ Ú test_minmaxþ s zTestUfuncs.test_minmaxc C s> dD ]4}t ddddgddddgd}ttt|dg qd S )Nz?bhilqpBHILQPfdgFDGOre r r rf )r r r[ r: )rn r× ro rY rY rZ Ú test_nonzero s zTestUfuncs.test_nonzeroN)r r r rz r r! r# r$ rY rY rY rZ r Í s
|
%
|
r c @ sL e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dS )ÚTestArrayMethodsc & C s t ddddddddd d
|
d d d ddddddddddddddd d d d d d!d"d#d$g$¡}| d%d%¡}| d&d'd'd&¡}t d(d)d(d)d(d(d)d(d)d)d(d)d(d(d(d)d(d)d(d(d(d)d)d)d)d(d(d)d(d(d(d(d)d(d)d(g$¡}t||d*}t|| |j¡d*}t|| |j¡d*}|||||||f| _d S )+Ng À @g®Gáz. @g-²ï§!@g !@gé&1¬ ü?gÀÊ¡E¶@g\Âõ(Ü @g
ëQ¸ @g{®Gáº#@gPn@ghí|?õ!@g!°rhí@gj¼t@gJ +@gáz®Ga@g!°rhí@gÁÊ¡E¶ó@g-²ï§@g)\Âõ(@gÃõ(\B#@g +Ù @guV @gÛù~j¼ô@gÑ"Ûù~ê@g¨ÆK7Á @g
|
×£p=J#@g®Gáz@gòÒMbØ@g}?5^ºÉ#@g¾/Ý$@g!°rhm @g-²ï§#@gJ +@gªñÒMb@gÀÊ¡Å@g-²ï§Æï?r r
r r re )Údatarg )rj r r rC rl )rn ro ÚXÚXXr¶ ÚmxÚmXÚmXXrY rY rZ rz s û û z TestArrayMethods.setup_methodc C sf | j \}}}}}}}| ¡ }t| ¡ | ¡ ¡ ¡ tt| ¡ | ¡ t|j| ¡ dd d S )Nr r ) rl rõ r ÚtraceÚ
|
compressedrJ r r[ rg ) rn ro r' r( r¶ r) r* r+ ZmXdiagrY rY rZ Ú
|
test_trace& s
|
ÿÿzTestArrayMethods.test_tracec C sh | j \}}}}}}}| dd¡}tt|j|j tt|j| dd¡ tt|j|j dd¡ d S )Nr r² )rl Zclipr r[ rg r³ ) rn ro r' r( r¶ r) r* r+ ZclippedrY rY rZ Ú test_clip. s
|
zTestArrayMethods.test_clipc C sÈ | j \}}}}}}}|j\}}t| ¡ | ¡ ¡ t |tj¡} t |tj¡}
|
t|D ] } |d d
| f ¡ ¡ |
|
| < qXt|D ]} || ¡ ¡ | | < qt t
|
| d¡|
|
t t
|
| d¡| d S ©Nr re ) rl rC r Zptpr- rj rP Zfloat_rÎ r r[ ) rn ro r' r( r¶ r) r* r+ rµ ÚrowsÚcolsÚkrY rY rZ Útest_ptp5 s
|
zTestArrayMethods.test_ptpc
|
C sZ | j \}}}}}}}| dd¡}tt|d |d d
df | dd¡} t| jd d S )Nr re r½ r )r r r
r
)rl Zswapaxesr r[ r rC )
|
rn ro r' r( r¶ r) r* r+ Z mXswappedZ
|
mXXswappedrY rY rZ Ú test_swapaxesB s
|
z TestArrayMethods.test_swapaxesc C sd | j \}}}}}}}| d¡}tt|j| d¡ d¡ | d¡}tt|j| d¡ d¡ d S r0 )rl Zcumprodr r[ r³ r © rn ro r' r( r¶ r) r* r+ ZmXcprY rY rZ Ú test_cumprodI s
|
|
|
z TestArrayMethods.test_cumprodc C sd | j \}}}}}}}| d¡}tt|j| d¡ d¡ | d¡}tt|j| d¡ d¡ d S r0 )rl Zcumsumr r[ r³ r r6 rY rY rZ Ú test_cumsumP s
|
|
|
z TestArrayMethods.test_cumsumc C s( | j \}}}}}}}tt|jd d| ¡ ¡ tt|jd d| ¡ ¡ tt|jddj|jddj tt| ¡ j| ¡ j |jdd|jdd }} tdD ]t}
|
tt| |
|
||
|
¡ ¡ tt||
|
|d d
|
|
f ¡ ¡ ttt ||
|
¡|d d
|
|
f ¡ ¡ q®d S )Nr r
r re r )
|
rl r r[ Úvarr- ZstdrC rÎ rj rH ) rn ro r' r( r¶ r) r* r+ ZmXvar0ZmXvar1r3 rY rY rZ Ú test_varstdW s " &ÿz TestArrayMethods.test_varstdN) r r r rz r. r/ r4 r5 r7 r8 r: rY rY rY rZ r% s r% c C s, | t kr|t kS |t kr | t kS | |k ¡ S r{ )r9 r
|
)rr rs rY rY rZ r e s
|
r )bÚ functoolsr r Únumpyrj Znumpy.core.umathÚcorer Znumpy.core.fromnumericr Z numpy.testingr r r Znumpy.mar r r r r
|
r r r r r r r r r r r r r r r r r r r r r r r! r" r# r$ r% r& r' r( r) r* r+ r, r- r. r/ r0 r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? r@ rA rB rC rD rE rF rG rH rI rJ rK rL rM rN rO rP Z numpy.compatrQ rk r[ r\ r r% r rY rY rY rZ Ú<module> s& ÿ 5 .AW
|
