U
|
P ±d$ ã @ sò d Z ddlZddlZddlZddlZddlmZ ddlm Z m
|
Z
|
ddl m Z m Z mZmZ ddlmZmZmZmZmZmZmZmZmZmZmZ ddl 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> G dd d Z?G d
|
d d Z@G d d d ZAG dd dZBG dd dZCG dd dZDG dd dZEG dd dZFG dd dZGG dd dZHG d d d ZIG d d d ZJG d d! d!ZKG d"d# d#ZLG d$d% d%ZMdS )&zæTests suite for MaskedArray.
|
Adapted from the original test_ma by Pierre Gerard-Marchant
|
|
:author: Pierre Gerard-Marchant
|
:contact: pierregm_at_uga_dot_edu
|
:version: $Id: test_extras.py 3473 2007-10-29 15:18:13Z jarrod.millman $
|
|
é N)Únormalize_axis_tuple)Ú assert_warnsÚsuppress_warnings)Úassert_Úassert_array_equalÚ assert_equalÚassert_almost_equal) ÚarrayÚarangeÚmaskedÚ MaskedArrayÚ masked_arrayÚ getmaskarrayÚshapeÚnomaskÚonesÚzerosÚcount)"Ú
|
atleast_1dÚ
|
atleast_2dÚ
|
atleast_3dÚmr_ÚdotÚpolyfitÚcovÚcorrcoefÚmedianÚaverageÚuniqueÚsetxor1dÚ setdiff1dÚunion1dÚ intersect1dÚin1dÚediff1dÚapply_over_axesÚapply_along_axisÚ compress_ndÚcompress_rowcolsÚ mask_rowcolsÚ clump_maskedÚclump_unmaskedÚflatnotmasked_contiguousÚnotmasked_contiguousÚnotmasked_edgesÚ
|
masked_allÚmasked_all_likeÚisinÚdiagflatÚ ndenumerateÚstackÚvstackc @ sL e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dS )Ú TestGenericc C sJ t dtd}tddgddgtd}t|| t ddgddgd¡}t d|d}td d gd
|
d
|
g|d}t|| t d |d}td d gd d ggd
|
d
|
gd
|
d
|
gg|d}t|| t d dd dgfg¡}t d|d}tddgddg|d}t|| t d|d}tddgddg|d}t|| t d
|
|d}tdggdgg|d}t|| d S )N©é ©Údtypeé ©Úmaskr: ÚaÚbÚf©ÚnamesÚformats©r r ©r; r; ©r8 r8 ©r> r@ ©Úbar@ ©Zbbr@ ©r; rE )r/ Úfloatr r Únpr: )ÚselfÚtestÚcontrolÚdt© rR úQd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/ma/tests/test_extras.pyÚtest_masked_all& s:
|
|
þ
|
ÿ
|
ÿ
|
zTestGeneric.test_masked_allc C s t ddtfgdffg¡}t j d|¡}tt|d t jjj tt|d d t jjj tt |d d d t|d d j
|
d t|d d j j
|
d d S )Nr? Úc©r; r; rE rR © rM r: ÚobjectÚmar/ r ÚtypeÚcorer Úlenr Z _fill_value©rN Zmy_dtypeÚ
|
masked_arrrR rR rS Ú"test_masked_all_with_object_nestedD s z.TestGeneric.test_masked_all_with_object_nestedc C sr t dtdffg¡}t j d|¡}tt|d t jjj tt |d d t|d j
|
d t|d j j
|
d d S )Nr? rV r; rE rR rW r] rR rR rS Útest_masked_all_with_objectP s z'TestGeneric.test_masked_all_with_objectc C sÐ t ddgtd}t|}t ddgddgtd}t|| t ddgddgd¡}t d d gd
|
d
|
g|d}t|}t d d gd
|
d
|
g|d}t|| t d dd dgfg¡}t ddgddg|d}t|}t|| d S )Nr; r8 r9 r< r> r? r@ rA rD rE )é
|
ra rG rH rJ rK )r rL r0 r rM r: )rN ÚbaserO rP rQ rR rR rS Útest_masked_all_likeZ s
|
|
ÿz TestGeneric.test_masked_all_likec C s¾ t ddD ]®}t d| D ]}tj|td}tj||td}td| }|d| @ dk|_d}||D ]}||j| ¡ 7 }qf|t krt
|
| ¡ ¡ | q|j |_t
|
| ¡ ¡ | qq
|
d S )Nr; é r8 r9 r ) ÚrangerM r
|
ÚintÚfullr r= ÚdataÚsumr+ r Ú
|
compressed) rN r@ ÚiÚjÚkÚjar> ÚsÚslrR rR rS Ú check_clumpn s
|
zTestGeneric.check_clumpc C s^ t t d¡}t|ddddddg< t|}tddtdd tddg}t|| | t¡ d S ©
|
Nra r r; r8 é é é é rd )r rM r
|
r r* Úslicer rq ©rN r> rO rP rR rR rS Útest_clump_masked~ s
|
z TestGeneric.test_clump_maskedc C sV t t d¡}t|ddddddg< t|}tddtd dg}t|| | t¡ d S rr )r rM r
|
r r+ rw r rq rx rR rR rS Útest_clump_unmasked s
|
z TestGeneric.test_clump_unmaskedc C s¨ t d}t|}t|td|jg tjdtd|_t|td|jg t ||dk |dkB |dkB < t|}t|tddtddg t |d d
< t|}t|g d S ) Nra r r9 rv rt é rs ru )
|
r
|
r, r rw ÚsizerM r Úboolr= r )rN r> rO rR rR rS Ú test_flatnotmasked_contiguous s z)TestGeneric.test_flatnotmasked_contiguousN) Ú__name__Ú
|
__module__Ú __qualname__rT r_ r` rc rq ry rz r~ rR rR rR rS r6 $ s
|
|
r6 c @ s´ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Ze j
|
d dddgddgdddgdgdgfdddgdddggddddggddgdd d ggdddggfg¡d d Z d d! Z dS )"Ú TestAveragec C s< t ddddgddddgd}tdt|dd tdt|ddddgd
|
t|ddddgdd \}}td| t|d k t|d d
< tt|dd jdg t ddddgddddgd}| d d ¡}t|d d
df< tt|dd ddg tt|dd jd dg tddgt|dd t|ddd\}}t|ddg d S )Nç ç ð?ç @ç @TF©r= r ©Úaxis©Úweights©r Úreturnedç @r8 r; ©r r )r r r r r r= Úreshape)rN ZottÚresultÚwtsrR rR rS Útest_testAverage1¨ s
|
z TestAverage.test_testAverage1c C s ddddddg}ddddddgddddddgg}t dtjd}tt|ddd tt|d|dd tt dtjddt d g}tt|d tj t d¡¡d d
|
tt|ddt d¡d d tt|ddt|ddt|ddd g tt|d |d d tt|d|dd ddd ddg tt|ddt|ddt|ddd g td}ddddddg}ddddddgddddddgg}t d}ddddddg} ttt
|
||ddd ttt
|
||ddd ttt
|
||ddj dg ttt
|
|| ddd tt tt
|
||ddd t
|
||}
|
tt|
|
d d tt|
|
ddd dddddg tt|
|
ddddg tt|
|
d|dd dddddg d S )Nr r; rs r9 r ç @)r r r
r ç (@r g«ªªªªª
|
@r r r g $@Tg ÀX@g @ç @) r
|
rM Zfloat_r r r ÚaddÚreducer r r r= r ) rN Úw1Úw2ÚxÚyÚm1Úm2Zm3Zm4Zm5ÚzrR rR rS Útest_testAverage2» sF & ÿÿ ÿ
|
ÿz TestAverage.test_testAverage2c C sf t d}t dd }t||g||ggddd\}}tt|t| t|j|j ttddddgdd\}}tt|t| ttddd \}}tt|t| ttdtddd
|
\}}tt|t| tdd gdd ggt}t|d d gdd gg}t|dd} t| ddg t|dd}
|
t|
|
ddg t|d d}
|
t|
|
d t|dd}
|
t|
|
ddg d S )Nrs rv r; Tr )r8 r8 rv r ©r r r )r r r8 é Fr ç à?r r g«ªªªªª@ç ø?r )r
|
r r r r r rL r ) rN r> r? Zr1r Zr2r Za2dZa2dmZa2daZa2dmarR rR rS Útest_testAverage3Ý s*
|
z TestAverage.test_testAverage3c C s t dddg¡ dd¡}t jj|dgdgdggd}t ddd g¡ dd¡}t||ddd
|
}td gd gd ggdgdgdgg}t|| d S )Nr8 rv r¢ r; FTr r{ rs )r r Úkeepdimsr
r r )rM r r rY r r r )rN r r? ÚwÚactualZdesiredrR rR rS Útest_testAverage4õ s "z TestAverage.test_testAverage4c C sH t tddg}t|d t tddddgddddgd}t|d d S ) Nr; r8 r¤ rv r¢ FTr )r r r ©rN r> rR rR rS Útest_onintegers_with_maskþ s
|
z%TestAverage.test_onintegers_with_maskc C s t jdddddgdddddggtd}tdddddgdd d
|
d d gg|d }t|}t | ¡ ¡}t|j|j t|j|j t|dd}t|jddt|jddd }t|j|j t|j|j t|dd}t|jddt|jddd }t|j|j t|j|j t dddddgdddddgg¡} t|| d}
|
t j| ¡ | | d}t|
|
j|j t|
|
j|j t|| dd} t|j| ddt|j| ddd }t| j|j t| j|j t|| dd} t|j| ddt|j| ddd }t| j|j t| j|j d S )Nr r; r9 y ð? @y @ @y @ @y @ @y "@y ð?y @ @y @ @y @ @r r r£ r r
r ©r r ) rM r r} r r rj r ÚrealÚimag) rN r= r> ÚavÚexpectedZav0Z expected0Zav1Z expected1r ZwavZwav0Zwav1rR rR rS Ú test_complex sR ÿÿ ÿþ ÿ ÿÿzTestAverage.test_complexz<x, axis, expected_avg, weights, expected_wavg, expected_wsumr; r8 rv Nr
r¢ g ü?ç @r{ rs é r r r r g #@c
|
C s¼ t jj||dd}|jt |¡ks&tt|| t jj|||dd}|jt |¡ksXtt|| t jj|||ddd\}} |jt |¡kstt|| | jt |¡ks®tt| | d S )NT)r r¦ )r r r¦ )r r r r¦ )rM rY r r ÚAssertionErrorr )
|
rN r r Z expected_avgr Z expected_wavgZ expected_wsumÚavgZwavgZwsumrR rR rS Útest_basic_keepdims1 s
|
|
ÿ
|
|
z TestAverage.test_basic_keepdimsc C s t jjt d¡ dd¡dddgdddgdddggd}tdddgd d}tdddgdddgd}t|d|d d
|
}t d d d g¡}t|| t|d|d d
|
}t d ddg¡}t|| t jjddddgd d ddgd}t|dddt jgd}t|d t jjddddgdd ddgddddggd ddd gdd ddgdd dd ggd}t jj|dt jdgdd}t jjdt jt jd gd ddd gd}t|| t |j
|
|j
|
d S ) Nru rv r; r r r{ é é Fr¡ ç @g à@g à@g0N@g0N@r r
r r Tr r¤ r ç @r² g "@r¬ ç @) rM rY r r
|
r r r r Únanr r= ) rN r> Zweights_unmaskedZweights_maskedZ avg_unmaskedZexpected_unmaskedZ
|
avg_maskedZexpected_maskedZ avg_expectedrR rR rS Útest_masked_weightsH sF ÿ ÿ
|
|
|
ÿ
|
|
|
|
ý
|
|
|
ýü
|
|
ÿ
|
z TestAverage.test_masked_weights)r r r r r r¥ r© r« r± ÚpytestÚmarkÚ parametrizer¶ r½ rR rR rR rS r ¦ s$ " ,
|
|
ÿÿþ
|
r c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestConcatenatorc C s t td tddddddg td}dddddg}t||d }t|dd|f }tt|t t |ddddddddddddg t |jt|dd|f d S )
|
N)r; r8 rv r¢ r{ rs r; r8 rv r¢ r{ rs r r ) r r r r r r Ú
|
isinstancer r= )rN r? ÚmÚdrU rR rR rS Útest_1dw s "zTestConcatenator.test_1dc C s0 t j dd¡}t j dd¡}t t j dd¡d¡}t t j dd¡d¡}t||d}t||d}td||f }t|jdk t|d d
d d
f | t|d d
dd
f | t|j t j
|
d||f t||f }t|jdk t|d d
d d
f | t|dd
d d
f | t|j t j
|
||f d S )Nr{ r r Ú1)r{ ra )ra r{ ) rM ÚrandomÚrandZround_r r r r r r= Zr_)rN Za_1Za_2Zm_1Zm_2Zb_1Zb_2rÄ rR rR rS Útest_2d s zTestConcatenator.test_2dc C sr t tjjdf }t|jddg t|jd d t ddgtjjf }t|jdddg t|jd d
ddg d S )Nr; TFr8 )r rM rY r r r= rh )rN r¨ rR rR rS Útest_masked_constant s z%TestConcatenator.test_masked_constantN)r r r rÅ rÉ rÊ rR rR rR rS rÁ t s rÁ c @ s e Zd Zdd Zdd ZdS )Ú TestNotMaskedc C s¬ t t d¡ dd¡dddddgdddddgdddddgdddddgdddddggd}t|d }t|ddg t|d}t|d ddg t|d d dg t|d}t|d dd
|
g t|d dd g t|jd }t|ddg t|jd}t|d d dg t|d d dg t|jd}t|d dd g t|d dd g t|d< t|d}t|d ddg t|d ddg t|d}t|d ddg t|d ddg d S )Né r{ r r; r é )r r r; r r )r r; r8 rv r¢ )rv rv rv r¢ r¢ )r r r8 r rv )r¢ r8 r¢ r¢ r¢ )r r r r r )r¢ r¢ r¢ r¢ r¢ éÿÿÿÿéþÿÿÿ)r; r; r8 r¢ r¢ )r r; r8 r¢ )r r r8 rv )r¢ r8 r¢ r¢ )r rM r
|
r r. r rh r ©rN rh rO rR rR rS Ú
|
test_edges£ s> üÿ
|
|
|
|
|
zTestNotMasked.test_edgesc C sP t t d¡ dd¡ddddddddgddddddddgddddddddggd}t|d }t|tddd tdd d td
|
dd g t|d}t|tddd td dd gtddd td dd gtddd td dd gtddd td dd gtd dd gtd dd gg td dd gg t|d}t|tddd gg tdd d td dd gg d S )NrÍ rv rt r r; r r¢ é é é r8 rs rd )r rM r
|
r r- r rw )rN r> ÚtmprR rR rS Útest_contiguousÅ s: þÿ
|
|
|
|
ý
|
ø
|
ýz TestNotMasked.test_contiguousN)r r r rÑ rÖ rR rR rR rS rË s "rË c @ s~ e Zd Zdd Zdd Zdd Zej ddd d
|
g¡ej d d ge j
|
j d fe j
|
j d
|
fg¡d d Z dd Zdd Zdd ZdS )ÚTestCompressFunctionsc C s> t ttd¡ ddd¡}t d¡ t¡}d|d< t||d}t|}t |d d
|
ddgd d d dgddddggddddgddddgdd d d ggg t|d }t |d d d
|
ddgdd d!d"d#gd d$d d dgdd%dddggdd&dddgd'd(d)d*d+gdd,dddgdd-d d d ggg t|d }t |d d d
|
ddgd d$d d dgdd%dddggd.d/d0d1d2gd3d4d5d6d7gd8d9d:d;d<ggdd&dddgdd,dddgdd-d d d ggg t|d=}t|d>}t|d?}t || t || t || t|d
|
}t |d d
|
ddgdd!d"d#gd d d dgddddggd.d0d1d2gd@dAdBdCgd3d5d6d7gd8d:d;d<ggddddgd'd)d*d+gddddgdd d d ggg t|dD}t|dE}t|dF}t || t || t || t|dG}t |d d d
|
ddgd d$d d dgdd%dddggdd&dddgdd,dddgdd-d d d ggg t|dH}t || t|dI}t |d d
|
ddgd d d dgddddggd.d0d1d2gd3d5d6d7gd8d:d;d<ggddddgddddgdd d d ggg t|dJ}t|dK}t|dL}t || t || t || t|dM}t |d d
|
ddgdd!d"d#gd d d dgddddggddddgd'd)d*d+gddddgdd d d ggg t|dN}t || d S )ONé< rv r¢ r{ )rv r¢ r{ T©r; r; r; r r r8 ra é é é é é é é é( é* é+ é, é2 é4 é5 é6 é7 é9 é: é; r; rs rd rt ru r³ rÒ é) é- é. é/ é0 é1 é3 é8 é é rÓ rÔ rÍ é r¸ é é! é" é# é$ é% é& é' rV rÏ )rÏ rÌ é r· é r7 rÎ )rÎ ©r r; )r rÏ ©r; r8 )rÏ r8 )r; rÎ )rÏ rÎ ©r r8 )r rÎ )
|
rM r Úlistre r r Úastyper} r' r )rN r rà r> Úa2Úa3Za4rR rR rS Útest_compress_ndè sæ
|
|
þ
|
|
|
þý
|
ý ýü
|
|
þ þ þú
|
|
|
|
|
|
|
|
|
|
|
ý
|
|
|
|
ý
|
|
|
|
ýø
|
|
|
|
|
|
|
þ þý
|
|
|
|
|
þ
|
|
|
þ
|
|
|
þú
|
|
|
|
|
|
|
|
|
|
|
ý
|
|
|
|
ýü
|
z&TestCompressFunctions.test_compress_ndc C sÚ t t d¡ dd¡dddgdddgdddggd}tt|ddgdd gg tt|ddddgd
|
dd gg tt|ddd gddgdd gg t |jdddgdddgdddggd}tt|dd gd
|
d gg tt|dddd gd
|
dd gg tt|ddd gddgd
|
d gg t |jdddgdddgdddggd}tt|d gg tt|dd
|
dd gg tt|dd gdgd gg t |jdddgdddgdddggd}tt|jd tt|djd tt|djd d S ) Nru rv r; r r r¢ r{ rd rt rs r8 )r rM r
|
r r r( Ú_datar| ©rN r rR rR rS Útest_compress_rowcolsX s$ ÿ "& "& &z+TestCompressFunctions.test_compress_rowcolsc C s t t d¡ dd¡dddgdddgdddggd}tt|jdddgdddgdddgg tt|djdddgdddgdddgg tt|djdddgdddgdddgg t |jdddgdddgdddggd}tt|jdddgdddgdddgg tt|djdddgdddgdddgg tt|djdddgdddgdddgg t |jdddgdddgdddggd}tt|jdddgdddgdddgg tt|djdddgdddgdddgg tt|djdddgdddgdddgg t |jdddgdddgdddggd}tt| ¡ t
|
k tt|d ¡ t
|
k tt|d ¡ t
|
k tt|j ¡ tt|dj ¡ tt|dj ¡ d S )Nru rv r; r r ) r rM r
|
r r r) r= r
|
r Úallr r rR rR rS Útest_mask_rowcolsl sN ÿ
|
ÿ ÿ ÿ&
|
ÿ ÿ ÿ&
|
ÿ ÿ ÿ&z'TestCompressFunctions.test_mask_rowcolsr Nr r; ÚfuncÚ rowcols_axisc C sf t t d¡ dd¡dddgdddgdddggd}tt" |||d}t|t|| W 5 Q R X d S )Nru rv r; r r r )r rM r
|
r r ÚDeprecationWarningr r) )rN r r r r ÚresrR rR rS Ú#test_mask_row_cols_axis_deprecation s ÿ
|
z9TestCompressFunctions.test_mask_row_cols_axis_deprecationc C s t dd¡}ddddddg}t||d dd¡}t||d dd¡}t||dd}t|jddgddgg t||dd}t|jdddgdddgdddgg t||d d}t|t | d¡| d¡¡ t||d d}t|t | d¡| d¡¡ ddddddg}t||d dd¡}t||d dd¡}t||dd}t|jddgddgg t||dd}t|jdddgdddgdddgg t||d d}t|t | d¡| d¡¡ t|t|| t||d d}t|t | d¡| d¡¡ ddddddg}t||d dd¡}t||d dd¡}t||}t|jt t||}t|jt t|ddddddgd dd¡}t|ddddddgd dd¡}t||dd}t|jddgddgg t||d d}t|t | d¡| d¡¡ t||dd}t|jdddgdddgdddgg t||d d}t|t | d¡| d¡¡ t|ddddddgd dd¡}t|ddddddgd dd¡}t||dd}t|jddgddgg t||}t|t | d¡| d¡¡ t||dd}t|jdddgdddgdddgg t||d d}t|t | d¡| d¡¡ t|ddddddgd dd¡}t|ddddddgd dd¡}t||dd}t|jddgddgg t||d d}t|t | d¡| d¡¡ t||dd}t|jdddgdddgdddgg t||d d}t|t | d¡| d¡¡ d S )
|
Nr; rd r r r8 rv T)ÚstrictF) rM r
|
r r r r r= Úfilledr )rN Únrà r> r? rU rR rR rS Útest_dot sz $ $
|
|
$
|
$ $z TestCompressFunctions.test_dotc C sn t d¡}t|}ttt||tk ttt||tk ttt||tk ttt||tk d S )Nrv )rM Úeyer r rZ r r ©rN r> r? rR rR rS Ú test_dot_returns_maskedarrayÞ s
|
z2TestCompressFunctions.test_dot_returns_maskedarrayc C sD t t d¡}t t d¡}t|||d}t||k t|| d S )Nrv )rv rv ©Úout)r rM r r r r r )rN r> r r rR rR rS Ú test_dot_outç s
|
z"TestCompressFunctions.test_dot_out)r r r r r r r¾ r¿ rÀ rM rY Z mask_rowsZ mask_colsr r r r rR rR rR rS r× æ s p ÿ F r× c @ s e Zd Zdd Zdd ZdS )ÚTestApplyAlongAxisc C s@ t d ddd¡}dd }t|d|}t|ddgdd gg d S )
|
Nr r8 rv c S s | d S ©Nr; rR )r? rR rR rS Úmyfuncô s z*TestApplyAlongAxis.test_3d.<locals>.myfuncr; r¢ rd ra ©r
|
r r& r ©rN r> r ZxarR rR rS Útest_3dñ s zTestApplyAlongAxis.test_3dc C sF t d ddd¡}d dd}t|d|dd}t|dd gd
|
d gg d S ) NrÚ r8 rv r c S s | d| S r rR )r? ÚoffsetrR rR rS r þ s z1TestApplyAlongAxis.test_3d_kwargs.<locals>.myfuncr; )r$ r{ rt r³ )r r! r" rR rR rS Útest_3d_kwargsû s
|
z!TestApplyAlongAxis.test_3d_kwargsN)r r r r# r% rR rR rR rS r ï s
|
r c @ s e Zd Zdd ZdS )ÚTestApplyOverAxesc C s t d ddd¡}ttj|ddg}t dgdgdggg¡}t|| t||d t ¡< ttj|ddg}t d gd
|
gdggg¡}t|| d S ) NrÍ r8 rv r¢ r rØ é\ é| r· rä )
|
r
|
r r% rM ri r r r r r} )rN r> rO ZctrlrR rR rS Ú
|
test_basic s
|
z TestApplyOverAxes.test_basicN)r r r r) rR rR rR rS r& s r& c @ sÞ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zejjd d d d d!d"gd#d$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zd.d/ Zd0d1 Zd2d3 Zd4d5 Z d S )6Ú
|
TestMedianc C s4 t jjt jt jgt jt jggdd}t|t j d S )NrÎ r )rM rY r Úinfr ©rN ÚrrR rR rS Ú test_pytype s $zTestMedian.test_pytypec C sì t jjt j t jt jgt jt jgg¡dd}t|t j t jjt j t jt jgt jt jgg¡d d}t|t j t jjt jjt jt jgt jt jggdddd}t|jd t jjt jjt jt jgt jt jggddd d}t|jd d S )NrÎ r Tr )rM rY r r r+ r r= r, rR rR rS Útest_inf s8
|
ÿÿ
|
ÿÿ
|
ÿÿþ
|
ÿÿþzTestMedian.test_infc C sl t d¡}tt j |¡d ttt j |¡tk td}tt j |¡d ttt j |¡tk d}tt j |¡d ttt j |¡tk t d¡ dd¡}tt jj|dd t j|dd tt jj|d
|
d t j|d
|
d tt jj|d
|
d tk t d ¡ dd¡}tt jj|dd t j|dd tt jj|d
|
d t j|d
|
d tt jj|d
|
d tk d S ) Nru r rt r» r{ r éH r r r; g R@)
|
rM r
|
r rY r r rZ r re r r rR rR rS Útest_non_masked) s"
|
zTestMedian.test_non_maskedc C s t t d¡dgd dgd d}ttj |¡d ttj |¡jdd tttj |¡t k t t d ¡
|
d
|
d ¡dgd dgd d}ttj |¡d ttj |¡jdd tttj |¡t k tjj|ddd}t|ddg t|jdd tt|t k dS )z5test the examples given in the docstring of ma.medianrt r r¢ r; r r¤ rR úshape mismatchra r8 r{ rs r rÎ T©r Zoverwrite_inputr
r r7 N) r rM r
|
r rY r r r rZ r r )rN r Zma_xrR rR rS Útest_docstring_examples> s "*z"TestMedian.test_docstring_examplesc C sþ d}t dD ]ì}dD ]â}td||d}t t | |ddg¡}|D ]H\}}ztjj|||d W qB tk
|
r t|||||f Y qBX qBt |d |gddg¡}|D ]J\}}ztjj|||d W n tj k
|
rÞ Y qªX t|||||f qªqq d S ) Nz5mask = %s, ndim = %s, axis = %s, overwrite_input = %sr{ )FTr; )Úndminr= FTr3 )
|
re r Ú itertoolsÚproductrM rY r Ú Exceptionr´ Z AxisError)rN Úmsgr5 r= r Úargsr ZoverrR rR rS Útest_axis_argument_errorsM s" z$TestMedian.test_axis_argument_errorsc C sD t ddd}ttj |¡d t ddd}ttj |¡tjj d S )Nr; Fr T)r r rM rY r r r rR rR rS Útest_masked_0de s zTestMedian.test_masked_0dc C sÞ t t d¡dd}ttj |¡tjj ttj |¡jdd tt tj |¡tjj
|
j k t t d¡dd}ttj |¡d ttj |¡jdd tt tj |¡t k t t d¡dd dddgd}ttj |¡d
|
ttj |¡jdd tt tj |¡t k t t d¡dd d d d gd}ttj |¡d ttj |¡jdd tt tj |¡t k t t d¡dd d ddgd}ttj |¡d ttj |¡jdd tt tj |¡t k t t d ¡dd d ddgd}ttj |¡d ttj |¡jdd tt tj |¡t k t t d¡dd d d d dgd}ttj |¡d
|
ttj |¡jdd tt tj |¡t k t t d¡dd d d d dgd}ttj |¡d
|
ttj |¡jdd tt tj |¡t k d S )Nr{ Tr rR r2 Fr
r r; r r r r rs r¹ ) r rM r
|
r rY r r r r rZ r[ ZMaskedConstantr r rR rR rS Útest_masked_1dl s@ zTestMedian.test_masked_1dc C sJ t tj tdddgdddgd¡jtj tdddgdddgd¡j d S )Nr; r8 rv r r )r rM rY r r r ©rN rR rR rS Útest_1d_shape_consistency s " ÿz$TestMedian.test_1d_shape_consistencyc C sü d\}}t t dd|¡}t |d d
< |dd
< t tj||ftd}|d d
|d d
df< t t|¡}td|D ]$}tj
|
|¡ || |d d
|f< qxt t |d d
df d t t |d t t |dd t |¡ t t |jdd t |¡ d S )
|
N)ée r÷ ç ð¿r ra éöÿÿÿr9 r r; r )r rM Zlinspacer ÚemptyrL r
|
r\ re rÇ Úshuffler r r ÚT)rN r Úpr r Úidxrk rR rR rS rÉ s zTestMedian.test_2dc C sö t t d¡ dd¡}t |d d
< |dd
< tt|d tttj |¡t
|
k tt|ddddd g tttj j|ddt
|
k tt|d
|
dddddd d d dddg
|
tttj j|d
|
dt
|
k tt|d
|
dj d
|
d
|
d
|
ddddd
|
d
|
d
|
g
|
d S )Nr÷ ra rv éýÿÿÿç -@r r ç +@g /@r; rÛ rÒ rà ) r rM r
|
r r r r r rZ rY r r= r rR rR rS Ú test_2d_waxis¦ s & zTestMedian.test_2d_waxisc C s t j d¡ ddd¡}t||d dk< tt|dddgdd gddgd
|
d gg d |_tt|dd d gdd gddgg t j d¡ ddd¡}t||d dk< tt|ddd gddgddgg d S )NrÍ rv r¢ r8 r rÚ ru rs rÝ rß )r¢ rv r8 éc ra r³ rÛ rÜ r{ rt rÒ rÞ )rM rY r
|
r r r r r r rR rR rS r# ² s ("zTestMedian.test_3dc C sL t t d¡ dd¡}t |d d
< |dd
< tt|ddt|dd d S )Nr÷ ra rv rH rÎ r r; )r rM r
|
r r r r r rR rR rS Ú test_neg_axis½ s zTestMedian.test_neg_axisc C s dD ]~}t t |¡}t |d d
< |dd
< t t d¡}t||d}|dkr\t|d n
|
t|d t||k tt|t k qd S ) N)r÷ ç >@r¸ g ?@rv rH rR r r÷ rI g .@)
|
r rM r
|
r r r r r rZ r )rN Úvr r r- rR rR rS Ú test_out_1d s
|
zTestMedian.test_out_1dc C s dD ]}t t |¡ dd¡}t |d d
< |dd
< t t d¡}t|d|d}|dkr t d gd dd
|
d d g d gd d gd dgd d gd d}nBt d gd ddddg d gd d gd dgd d gd d}t|| t||k tt |t
|
k qd S )N)rá g D@r÷ rN ra rÎ rv rH r; )r r r÷ r rÛ rÒ rà TFr¢ r rJ g 1@g 5@g 9@) r rM r
|
r r r r r r rZ r )rN rO r r r- ÚerR rR rS Útest_outÐ s
|
ÿ ÿ
|
zTestMedian.test_outr Nr; rV r )rH rÎ )ÚargnamesZ argvaluesc sØ t jdtd}t j d¡t |j¡d d
d f }| t j¡}t j|t |< t
|
t |j¡|d|d krtdj }n*t |j t fddtj D }t
|
t |¡}t|d|d }||ksÈtt|j| d S )
|
N)rv r{ rd r³ r9 )r¢ éÈ r rV c 3 s$ | ] }| krd nj | V qdS )r; N)r )Ú.0rk ©Z axis_normrÄ rR rS Ú <genexpr>ö s z/TestMedian.test_keepdims_out.<locals>.<genexpr>T)r r¦ r )rM r r} rÇ r r r Zintpr¼ Útupler r Úndimr re rC r r´ r )rN r r= r§ Z shape_outr r rR rV rS Útest_keepdims_outá s $ ÿ z TestMedian.test_keepdims_outc C sF ddgddgddgg}t j |d¡}ddg}tt jj|dd| d S )Nr r r r r )rM rY Z masked_equalr r )rN rh r^ r° rR rR rS Ú$test_single_non_masked_value_on_axisý s þÿz/TestMedian.test_single_non_masked_value_on_axisc C sª dt jdtdfD ] }t j dt jdgdddgg¡}||_t jj|d d}tt |¡ t
|
|t j t jj| ¡ dd}tt |¡ t
|
|t j t jj|dd}t t |t t
|
|dt jdg t jj|dd}t t |t t
|
|t jdg t jj|d d}t t |t t
|
|t jdg qt j dt jdgdddgg¡}t jj|d d
df< t
|
t jj|d dt j t
|
t jj|dddt jdg t
|
t jj|ddt jd
|
g d S ) NFrs r9 r; rv r8 r r rÎ r¤ )rM r r} rY r r¼ r= r r Úisscalarr Zravelr rZ r r )rN r= Údmr- rR rR rS Útest_nan s. zTestMedian.test_nanc C s¨ t j t d¡¡}t j t d¡¡}t j|d< t jj|d< tt jj|d|d| t j t d¡¡}tt jj|d|d| t j t d ¡¡}tt jj||d| d S )
|
N)r¢ ©rv r¢ ©r8 r; rF r r ©rv r; rR ) rM rY r r r r¼ r r r )rN ÚorÄ rR rR rS Ú test_out_nan# s
|
zTestMedian.test_out_nanc C s t j t jdtd¡}t jj|d d d
< t j|d< tt j |¡t j tt jj|ddt j t j t jdtd ddd¡¡}t |j
|
¡d dk|_ | ¡ }t j|d < t j|d
|
< tt j |¡t j t t t j |¡¡ t jj|dd}t j|d < t j|d < tt j |d¡| t jj|dd}t j|d < t j|d < tt j |d¡| t jj|dd}t j|d< t j|d< tt j |d¡| d S )NrÍ r9 rv r8 r r r¢ r; )r; r8 rv )r; r; r8 ©r8 rv r ©r; rv r )rM rY r r
|
rL r r¼ r r r r| r= Úcopyr r\ r )rN r> Zaorigr? rR rR rS Útest_nan_behavior. s0
|
|
|
|
|
|
|
|
|
z TestMedian.test_nan_behaviorc C s t jdddgdddggt jd}t jj||dkd}tt jj|ddd tt jj|ddjd tt jj|dd|d tt j |¡d d S ) Nrv éÿ r9 r r; r Fr )rM r Zuint8rY r r r r= rª rR rR rS Útest_ambigous_fillQ s z TestMedian.test_ambigous_fillc C s t jt j fD ]ì}t |t jgt jt jgg¡}t jj|t |¡d}tt jj|dd|t jg tt jj|dd|t jg tt j |¡| t t jt j|gt jt j|gg¡}t jj|t |¡d}t t jj|dd| t t jj|ddj
|
d t t jj|dd|d t t j |¡| t ||g||gg¡}tt j |¡| tt jj|dd| tt jj|dd| t j|d| dgdt jt jd gd
|
t jt j|ggt j d }t jj|t |¡d}|dkrtt jj|ddd d | dg tt j |¡d n2tt jj|dddd | dg tt j |¡d tt jj|dddd|g t ddD ] }t ddD ]}t t jg| |g| gd ¡}t jj|t |¡d}tt j |¡| tt jj|dd| tt jj|ddt jg| |g| qhqZqd S )Nr r r r; Frd i÷ÿÿÿrB r{ r¢ r9 r rº r g @g $Àg "Àg ÀrA ra r8 ) rM r+ r r¼ rY r Úisnanr r r r= Úfloat32re )rN r+ r> rk rl rR rR rS Ú test_specialZ sL "þý
|
"ÿzTestMedian.test_specialc C st t j t jg td¡}t 8}| t¡ tt j |¡t j
|
t |j d j tk W 5 Q R X t j t jg tdd¡}t F}| t¡ t ddt¡ tt j |¡t j
|
t |j d j tk W 5 Q R X t j t jg tdd¡}tt jj |dd| tt jj |d d| t j t jt j
|
tdd¡}tjd
|
d <}t ddt¡ tt jj |dd| t |d j tk W 5 Q R X d S ) Nr9 r rv )r: r5 ÚalwaysÚ r8 r r; T)Úrecord)rM rY r r rL r ro ÚRuntimeWarningr r r¼ r ÚlogÚcategoryÚwarningsÚfilterwarningsr Úcatch_warnings)rN r> r§ r? rR rR rS Ú
|
test_empty
s&
|
|
zTestMedian.test_emptyc C sX t j t d¡¡}ttt j | t¡¡t t j
|
|d< ttt j | t¡¡t d S )Nrº r8 ) rM rY r r
|
r rZ r r rX rL r¼ )rN rb rR rR rS Ú test_object¢ s
|
zTestMedian.test_object) r r r r. r/ r1 r4 r; r< r= r? rÉ rK r# rM rP rR r¾ r¿ rÀ rZ r[ r^ rc rg ri rl rv rw rR rR rR rS r* s@ & ûþ
|
|
# + r* c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) ÚTestCovc C s t tj d¡| _d S ©NrÚ )r rM rÇ rÈ rh r> rR rR rS Ú setup_method« s zTestCov.setup_methodc C sZ | j }tt |¡t| ttj|ddt|dd ttj|dddt|ddd d S )NF©ÚrowvarT©r| Úbias)rh r rM r r rR rR rS Útest_1d_without_missing® s ÿz TestCov.test_1d_without_missingc C sb | j dd¡}tt |¡t| ttj|ddt|dd ttj|dddt|ddd d S )Nrv r¢ Fr{ Tr} )rh r r rM r r rR rR rS Útest_2d_without_missing¶ s ÿz TestCov.test_2d_without_missingc C s@ | j }t|d< || ¡ 8 }| ¡ }tt |¡t| ttj|ddt|dd ttj|dddt|ddd zt|dd W n tk
|
r Y nX |dd
}tt ||d d d
¡t||d d d
ttj||d d d
ddt||d d d
dd ttj||d d d
dddt||d d d
ddd d S )NrÎ Fr{ Tr} ©Z allow_maskedr; )rh r Úmeanrj r rM r Ú
|
ValueError)rN r ÚnxrR rR rS Útest_1d_with_missing¾ s* ÿ ,ÿ ÿz TestCov.test_1d_with_missingc C s6 | j }t|d< | dd¡}t t|¡ t¡}t ||j ¡}||
|
d¡d d
d f d¡}t t |t |¡|jd d |d t t |ddtj |dd|jd | t |j |¡}||
|
d¡ d¡}t t |d d
|
tj |d d
|
|jd d |d t t |d dd tj |d dd |jd | d S ) NrÎ rv r¢ r; r r T©r~ Fr{ r} )rh r r rM Z logical_notr r rf r rE r r r r r )rN r ZvalidÚfracZxfrR rR rS Útest_2d_with_missingÖ s8 ÿ ÿ ÿÿÿÿÿÿz TestCov.test_2d_with_missingN)r r r rz r r r
r rR rR rR rS rx © s
|
rx c @ sD e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dS )Ú TestCorrcoefc C s( t tj d¡| _t tj d¡| _d S ry )r rM rÇ rÈ rh Údata2r> rR rR rS rz î s zTestCorrcoef.setup_methodc C sÈ | j | j }}t |¡}t ||¡}t }t d¡ ttt|dd | td¡ t
|
tj|ddt|dd t
|
t|dd| t
|
t||dd| t
|
t|dd| t
|
t||dd| W 5 Q R X d S )Nrm rÎ ©Zddofú bias and ddof have no effectr rv © rh r rM r r rs Ú simplefilterr r Úfilterr )rN r r r° Z expected2ÚsuprR rR rS Ú test_ddofò s
|
|
zTestCorrcoef.test_ddofc C s | j | j }}t |¡}t b}t d¡ ttt||dd ttt||dd ttt|dd | td¡ t
|
t|dd| W 5 Q R X d S )Nrm TFr r r; r )rN r r r° r rR rR rS Ú test_bias s
|
|
zTestCorrcoef.test_biasc C sx | j }tt |¡t| ttj|ddt|dd t 2}| td¡ ttj|dddt|ddd W 5 Q R X d S )NFr{ r Tr} )rh r rM r r r r ©rN r r rR rR rS r s
|
ÿ ÿz$TestCorrcoef.test_1d_without_missingc C s | j dd¡}tt |¡t| ttj|ddt|dd t 2}| td¡ ttj|dddt|ddd W 5 Q R X d S )Nrv r¢ Fr{ r Tr} )rh r r rM r r r r r rR rR rS r s
|
ÿ ÿz$TestCorrcoef.test_2d_without_missingc C s¤ | j }t|d< || ¡ 8 }| ¡ }tt |¡t| ttj|ddt|dd t 2}| t d¡ ttj|dddt|ddd W 5 Q R X zt|dd W n t
|
k
|
r´ Y nX |dd
}tt ||d d d
¡t||d d d
ttj||d d d
ddt||d d d
dd t r}| t d¡ tt ||d d d
¡t||d d d
dd tt ||d d d
¡t||d d d
d
|
d W 5 Q R X d S ) NrÎ Fr{ r Tr} r r; r r8 r ) rh r r rj r rM r r r r r )rN r r r rR rR rS r
% s<
|
ÿ ÿ ,ÿ ÿÿz!TestCorrcoef.test_1d_with_missingc C s | j }t|d< | dd¡}t|}t |¡}t|d d
d d
f |d d
d d
f t ¨}| td¡ tt|ddd d
d d
f |d d
d d
f tt|ddd d
d d
f |d d
d d
f tt|ddd d
d d
f |d d
d d
f W 5 Q R X d S ) NrÎ rv r¢ r rÏ r r; r ) rh r r r rM r r r r )rN r rO rP r rR rR rS r C s"
|
* ÿ ÿ ÿz!TestCorrcoef.test_2d_with_missingN)
|
r r r rz r r r r r
r rR rR rR rS r ì s r c @ s e Zd Zdd Zdd ZdS )ÚTestPolynomialc C s¸ t j d¡}t j d¡ dd¡}tt||dt ||d¡ | t¡}t|d< | t¡}t |d< |d< t||d d
df dd d
|
\}}}}}t j|d d
|d d
df ¡ dd d
|
\}} }
|
} } t
|
|||||f|| |
|
| | fD ]\} }t| | qàt||d d
df dd d
|
\}}}}}t j|d d
|d d
df dd d
|
\}} }
|
} } t
|
|||||f|| |
|
| | fD ]\} }t| | qht||dd d
|
\}}}}}t j|d d
|d d
d d
f dd d
|
\}} }
|
} } t
|
|||||f|| |
|
| | fD ]\} }t| | qêt j d¡d }| ¡ }|d d
}|d d
}|d d
}t||dd |d \}}}}}t j||dd |d \}} }
|
} } t || t
|
|||||f|| |
|
| | fD ]\} }t| | qd S ) Nra rõ rÎ r8 rv r rD ©rÎ rÎ T©rg r; )rg r§ ) rM rÇ rÈ r r r Úviewr r rj Úziprf r )rN r r ÚCÚRÚKÚSÚDrU r- rm ro rÄ r> Úa_r§ ZwoÚxsZysÚwsrR rR rS Ú test_polyfitY s>
|
|
&"ÿ& &0&4&
|
&zTestPolynomial.test_polyfitc C sÜ t j d¡}t j d¡ dd¡}t j|d< t j|d< | t¡}| t¡}t|d< t|d< t||ddd \}}}}}t j|d
|
d
|d
|
d
d d
f ddd \}} }
|
} } t |||||f|| |
|
| | fD ]\} }t
|
| | qÄd S ) Nra rõ rÎ r8 r r rv Tr r; ) rM rÇ rÈ r r¼ r r r r r r )rN r r r r r r r rU r- rm ro rÄ r> r rR rR rS Ú test_polyfit_with_masked_NaNs s
|
|
|
|
4&z,TestPolynomial.test_polyfit_with_masked_NaNsN)r r r r¡ r¢ rR rR rR rS r W s 'r c @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zd d Zd d Zd!S )"ÚTestArraySetOpsc C s ddddddg}t |ddd}tt|d t t|d tdddgdddgd t|d dddg t|d ddddddg d S ) Nr; r8 rv T©Z return_indexZreturn_inverser r r{ )r r r r r r rÐ rR rR rS Útest_unique_onlist s "z"TestArraySetOps.test_unique_onlistc C s t ddddddgddddddgd}t|ddd}t|d t ddddgddddgd t|d ddd dg t|d ddddddg d|_t ddddddgddddddgdd
|
}t|ddd}t|d t ddddgddddgd t|d ddd dg t|d ddddddg d S ) Nr; r8 rv r r Tr¤ rÎ r{ )rh r= Ú
|
fill_value)r r r r¦ rÐ rR rR rS Útest_unique_onmaskedarray s $& ÿ&z)TestArraySetOps.test_unique_onmaskedarrayc C s¦ t dddgdd}t|ddd}t|d t dgdgd t|d dg t|d dddg t}t|ddd}t|d t t t|d dg t|d dg d S )Nr; Tr r¤ r r8 )r r r r rÐ rR rR rS Útest_unique_allmasked¬ s z%TestArraySetOps.test_unique_allmaskedc C sr t t d¡dddddgd}tddddgddddgd}t|}t|| t| d¡| d¡ t|j|j d S )Nr{ r; r r r¢ )r rM r
|
r r$ r r r= )rN r rP rO rR rR rS Ú test_ediff1d» s
|
z TestArraySetOps.test_ediff1dc C sâ t t d¡dddddgd}t|td}tdddddgdddddgd}t|| t| d¡| d¡ t|j|j t|dddgd}tdddddddgdddddddgd}t|| t| d¡| d¡ t|j|j d S ) Nr{ r; r r )Úto_beginr¢ r8 rv © r rM r
|
r$ r r r r r= ©rN r rO rP rR rR rS Útest_ediff1d_tobeginÄ s
|
(
|
z$TestArraySetOps.test_ediff1d_tobeginc C sâ t t d¡dddddgd}t|td}tdddddgdddddgd}t|| t| d¡| d¡ t|j|j t|dddgd}tdddddddgdddddddgd}t|| t| d¡| d¡ t|j|j d S ) Nr{ r; r r )Úto_endr¢ r8 rv r« r¬ rR rR rS Útest_ediff1d_toendÓ s
|
(
|
z"TestArraySetOps.test_ediff1d_toendc
|
C sî t t d¡dddddgd}t|ttd}tddddddgddddddgd}t|| t| d¡| d¡ t|j|j t|dddgtd}tddddddddgddddddddgd}t|| t| d¡| d¡ t|j|j d S ) Nr{ r; r r ©r® rª r¢ r8 rv r« r¬ rR rR rS Útest_ediff1d_tobegin_toendâ s $
|
ÿ
|
z*TestArraySetOps.test_ediff1d_tobegin_toendc C sÒ t d¡}t|}tddddgddddgd}t|| tt|t t| d¡| d¡ t|j |j t|t
|
t
|
d}tddddddgddddddgd}tt|t t| d¡| d¡ t|j |j d S )Nr{ r; r r r° ) rM r
|
r$ r r r r r r r= r r¬ rR rR rS Útest_ediff1d_ndarrayò s
|
|
$z$TestArraySetOps.test_ediff1d_ndarrayc C sh t ddddgddddgd}t ddddgddddgd}t||}t dddgdddgd}t|| d S )Nr; rv r r rÎ )r r" r )rN r r rO rP rR rR rS Útest_intersect1d s
|
|
z TestArraySetOps.test_intersect1dc C s t dddddgdddddgd}t dddd ddgddddddgd}t||}t|t dd dg t dddddgdddddgd}dddd dg}t||}t|t dd ddgddddgd t dddg}t d
|
dd g}t||}tt|t t|dddd dd
|
g t dd ddgddddgd}t d
|
dd d gddddgd}t||}tt|t t|dddd dd
|
g tg tg g d S ) Nr; r8 r{ rd rÎ r r rv r¢ rs rt )r r r r r r r ©rN r> r? rO rR rR rS Ú test_setxor1d
|
s& $
|
|
"
|
|
z TestArraySetOps.test_setxor1dc C sÚ t d¡ dddg¡}t dddg¡}d|d< t||d}tdd d
|
d ddd d dg dddddddddg d}tdtd}d|d< d|d< d|d< t||}tt|t t
|
|| t |||j ¡|j @ }t
|
|| d S )NrÍ r8 rv r¢ r; )r; r8 r r r ra rõ r÷ r³ rÓ rù )rh r= ©r8 rv r¢ r9 T)r r r )r r r; )r r8 rv ) rM r
|
r r r r} r1 r r r r r= )rN r> r= r? ZecrU rÄ rR rR rS Ú test_isin$ s ÿ
|
|
zTestArraySetOps.test_isinc C sÌ t dddddgdddddgd}t dddd ddgddddddgd}t||}t|d
|
d
|
d
|
d d
|
g t dddddgdddddgd}t dddgdddgd}t||}t|d
|
d
|
d d
|
d
|
g tg tg g d S ) Nr; r8 r{ rd rÎ r r rv r¢ TF)r r# r r r´ rR rR rS Ú test_in1d9 s $
|
|
zTestArraySetOps.test_in1dc C sÔ t dddddgdddddgd}t dddd ddgddddddgd}tt t||¡t||d
|
d t dddddgdddddgd}t dddgdddgd}tt t||¡t||d
|
d tg tg g d
|
d d S ) Nr; r8 r{ rd rÎ r r rv r¢ T)Úinvert)r r rM r¹ r# r r rR rR rS Útest_in1d_invertG s $ z TestArraySetOps.test_in1d_invertc C s t ddddddgddddddgd}t dddd ddgddddddgd}t||}t dddd dddgdddddddgd}t|| t dddgdd dggdddgdddggd}t ddddd gdddddgd}t ddddd dgddddddgd}t||}t|| tg tg g d S )
|
Nr; r8 r{ rd rÎ r r rv r¢ )r r! r r ) rN r> r? rO rP r r Zezr rR rR rS Ú test_union1dS s $$
|
(
|
, $
|
|
z TestArraySetOps.test_union1dc
|
C sº t ddddddddgddddddddgd}t ddd d dddg}t||}t|t ddd
|
gdddgd td }td }tt||t d d g t g tjg d}tt|g jtj d S )Nrs r{ r¢ rd r; r8 r r rv rÎ ra rt ru )r r r r
|
rM Zuint32r: r´ rR rR rS Útest_setdiff1de s ,
|
z TestArraySetOps.test_setdiff1dc C s< t dddg¡}t dddg¡}tt||t dg¡ d S )Nr> r? rU ro )rM r r r r rR rR rS Útest_setdiff1d_char_arrayr s z)TestArraySetOps.test_setdiff1d_char_arrayN)r r r r¥ r§ r¨ r© r r¯ r± r² r³ rµ r· r¸ rº r» r¼ r½ rR rR rR rS r£ s r£ c @ s e Zd Zdd Zdd ZdS )Ú TestShapeBasec C sr t dddgdddgd}t|}t|jd t|jj|jj t|jd t|jj|jj t|jj|jj d S )Nr r; r8 r re ra )r r r r r= rh r rR rR rS Útest_atleast_2d{ s z TestShapeBase.test_atleast_2dc C s t d}t|jd t|jj|j t|jj|j t dd}|D ]0}t|jd t|jj|j t|jj|j qBtd}t|jd t|jj|j t|jj|j tdd}|D ]0}t|jd t|jj|j t|jj|j q¶td}t|jd t|jj|j t|jj|j tdd}|D ]2}t|jd t|jj|j t|jj|j q*td}t|jd t|jj|jj d S )Nr rV r
rE rÙ )r r r r= rh r r r2 )rN r? r> rR rR rS Útest_shape_scalar
s<
|
|
|
z TestShapeBase.test_shape_scalarN)r r r r¿ rÀ rR rR rR rS r¾ y s
|
r¾ c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestNDEnumeratec C s t d¡ d¡}t j|td}t||d}ttt |¡tt| ttt|tt| ttt|tt|dd d S )Nr¹ )r; rv r8 r9 r F©rj ) rM r
|
r Z
|
zeros_liker} r r r r3 )rN ZordinaryZ
|
empty_maskZ with_maskrR rR rS Útest_ndenumerate_nomasked² s
|
ÿ
|
ÿ ÿz)TestNDEnumerate.test_ndenumerate_nomaskedc C sª t d}t d}t d}ttt|g ttt|g ttt|ddttt d¡dtg ttt|g ttt|ddttt d¡dtg d S )NrR )éd r¶ Fr rÄ rÍ )r/ r r r3 r rM Zndindexr )rN r> r? rU rR rR rS Útest_ndenumerate_allmasked½ s ÿÿz*TestNDEnumerate.test_ndenumerate_allmaskedc C s t t d¡ d¡ddddgddddgddddggd}dddd d
|
g}ttt|| tttt|d d |j t|d d D ]\}}t|| | q~d S ) NrÚ r_ r; r r )r rs ))r8 r rt )r` ru )rF ra )rd r³ Fr ) r rM r
|
r r r r3 r\ r| )rN r> ÚitemsZ
|
coordinateÚvaluerR rR rS Ú test_ndenumerate_mixedmaskedÉ s
|
|
|
þÿ ÿ z,TestNDEnumerate.test_ndenumerate_mixedmaskedN)r r r rà rÅ rÈ rR rR rR rS rÁ ° s rÁ c @ s$ e Zd Zdd Zdd Zdd ZdS )Ú TestStackc C sì t dddgdddgd}t dddgdddgd}t||gdd}t|jd t|j|d j t|j|d j t||g}t|j|j t|j|j t||gdd}t|jd
|
t|j|d d
df j t|j|d d
df j d S ) Nr r; r8 r ru rt rd r rd ©rv r8 ©r r4 r r r r= r5 rh ©rN r> r? rU rÄ rR rR rS Ú test_stack_1dØ s zTestStack.test_stack_1dc C sà t dddgdd}t dddgd d}t||gdd
|
}t|jd t|j|d j t|j|d j t||g}t|j|j t|j|j t||gdd
|
}t|jd t|j|d d
df j t|j|d d
df j d S ) Nr r; r8 Tr ru rt rd Fr rd rÊ rË rÌ rR rR rS Útest_stack_masksê s zTestStack.test_stack_masksc
|
C sô d}t j dd|¡}t j dd|¡}t j dd|¡ t¡}t j dd|¡ t¡}t||d}t||d}t||gdd}d| } t|j| t |j
|
|d j
|
t |j
|
|d j
|
t||gd d}|d } t|j| t |j
|
|d
|
j
|
t |j
|
|d j
|
d }t j dd|¡}t j dd|¡}t j dd|¡ t¡}t j dd|¡ t¡}t||d}t||d}t||gdd}d| } t|j| t |j
|
|d j
|
t |j
|
|d j
|
t||gd d}|d } t|j| t |j
|
|d
|
j
|
t |j
|
|d j
|
d S ) NrÊ r ra r8 r r r7 r; rÎ ).r ).r; )rv r8 r¢ r{ ) rM rÇ Úrandintr r} r r4 r r r r= )
|
rN ZshpÚd1Zd2r r Za1r rU Zc_shprR rR rS Ú test_stack_ndü sD zTestStack.test_stack_ndN)r r r rÍ rÎ rÑ rR rR rR rS rÉ Ö s rÉ )NÚ__doc__rs r6 r¾ ÚnumpyrM Znumpy.core.numericr Z numpy.testingr r Znumpy.ma.testutilsr r r r Z numpy.ma.corer r
|
r r r r r r r r r Znumpy.ma.extrasr 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 r rÁ rË r× r r& r* rx r r r£ r¾ rÁ rÉ rR rR rR rS Ú<module> s@ 4
|
O,F Ck: i7&
|
