U
|
P ±d=M ã @ sZ d Z dZddlZddlZddlZddlZddlZddlZddlm Z ddl
|
Z ddl Z
|
ddl m mZ ddlm mZ ddlmZmZmZmZ ddlmZ ddl
|
mZ ddlmZ dd l m Z m Z m Z m Z m!Z!m"Z"m#Z#m$Z$ dd
|
l 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@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYmZZZm[Z[m\Z\m]Z]m^Z^m_Z_m`Z`maZambZbmcZcmdZdmeZemfZfmgZgmhZhmiZimjZjmkZkmlZlmmZmmnZnmoZompZpmqZqmrZrmsZsmtZtmuZumvZvmwZwmxZxmyZymzZzm{Z{m|Z|m}Z}m~Z~mZmZmZmZmZmZm
Z
mZmZmZmZmZmZmZmZ dd lmZ e jZe Ze e
|
jjjd ¡ d d dD Zdd eD ZG dd dZG dd dZG dd dZG dd dZG dd dZG dd d ZG d d d ZG d d d ZG d!d" d"ZG d#d$ d$ZG 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/d0 d0Z¥d1d2 Z¦d3d4 Z§d5d6 Z¨d7d8 Z©d9d: Zªd;d< Z«d=d> Z¬d?d@ Zej®j¯dAeedBej®j¯dCeedBej® °dD¡dEdF Z±dGdH Z²dIdJ Z³ej®j´ejµj¶dKkdLdMdNdO Z·dS )PznTests suite for MaskedArray & subclassing.
|
|
:author: Pierre Gerard-Marchant
|
:contact: pierregm_at_uga_dot_edu
|
zPierre GF Gerard-Marchanté N)Úreduce)Ú assert_raisesÚ assert_warnsÚsuppress_warningsÚIS_WASM)Úrequires_memory)Úndarray)Úasbytes)Úassert_Úassert_array_equalÚ assert_equalÚassert_almost_equalÚassert_equal_recordsÚ fail_if_equalÚassert_not_equalÚassert_mask_equal)iÚMAErrorÚ MaskErrorÚMaskTypeÚ MaskedArrayÚabsÚabsoluteÚaddÚallÚallcloseÚallequalÚalltrueÚangleÚanomÚarangeÚarccosÚarccoshÚarctan2ÚarcsinÚarctanÚargsortÚarrayÚasarrayÚchooseÚ concatenateÚ conjugateÚcosÚcoshÚcountÚdefault_fill_valueÚdiagÚdivideÚdoc_noteÚemptyÚ
|
empty_likeÚequalÚexpÚ flatten_maskÚfilledÚ fix_invalidÚflatten_structured_arrayÚfromflexÚgetmaskÚ getmaskarrayÚgreaterÚ greater_equalÚidentityÚinnerÚ isMaskedArrayÚlessÚ
|
less_equalÚlogÚlog10Ú make_maskÚmake_mask_descrÚmask_orÚ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ÚmaxÚmaximumÚmaximum_fill_valueÚminÚminimumÚminimum_fill_valueÚmodÚmultiplyÚmvoidÚnomaskÚ not_equalÚonesÚ ones_likeÚouterÚpowerÚproductÚputÚputmaskÚravelÚrepeatÚreshapeÚresizeÚshapeÚsinÚsinhÚsometrueÚsortÚsqrtÚsubtractÚsumÚtakeÚtanÚtanhÚ transposeÚwhereÚzerosÚ
|
zeros_like)ÚpicklezGsetting an item on a masked array which has a shared mask will not copyc C s g | ]}t |¡qS © )ÚnpÚdtype©Ú.0Zdt_r| r| úOd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/ma/tests/test_core.pyÚ
|
<listcomp>? s r z?bhilqBHILQefdgFDc C s g | ]
|
}|j qS r| )Úcharr r| r| r r @ s 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!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zed1d2 Z d3d4 Z d5d6 Z d7d8 Z d9d: Z d;d< Z!d=d> Z"d?d@ Z#dAdB Z$dCdD Z%dEdF Z&dGdH Z'dIdJ Z(dKdL Z)dMdN Z*dOdP Z+dQdR Z,dSdT Z-edUdV Z.dWdX Z/dYdZ Z0d[d\ Z1d]d^ Z2d_d` Z3dadb Z4dcdd Z5dedf Z6dgdh Z7didj Z8dkdl Z9dmdn Z:dodp Z;dqdr Z<dsdt Z=duS )vÚTestMaskedArrayc 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|¡}
|
| d¡ ||||||||| |
|
f
|
| _d S )Nç ð?ç Àç @ç @ç @ç $Àç $@ç @ç ç ð¿ç Àé r ©Úmaskç à¿ç à?çé?ç@µx ¯D)r} r& ÚpirJ rx Úset_fill_valueÚd© ÚselfÚxÚyÚa10Úm1Úm2ÚxmÚymÚzÚzmÚxfr| r| r Ú setup_methodF s &"
|
z TestMaskedArray.setup_methodc C sr t dddg}t dddgdddgd}t|jd t|jd t|jd t|jd t|jd t|jd d S )Nr é é r r ©r§ )r& r ÚndimÚsizerl ©r ÚaÚbr| r| r Útest_basicattributesU s z$TestMaskedArray.test_basicattributesc C st t d}tt|d t ddd}tt|tt t ddd}tt|d tddd}t| ¡ j|jjk d S )Nr Ú0Tr Fr ) rJ r ÚstrrS r& r
|
r7 r~ Ú_data©r r r| r| r Ú test_basic0d` s z TestMaskedArray.test_basic0dc
|
C s | j \
|
}}}}}}}}} }
|
tt| tt| t|| d¡ ¡ t|j t¡|j t¡ |j } t
|
t |¡| t
|
|j | t
|
|j |j t
|
| j |j t
|
|j tdd | t
|
t|t|tdd | t||
|
tt|d|
|
t|| d S )Nr c S s | | S ©Nr| ©r r r| r| r Ú<lambda>w ó z.TestMaskedArray.test_basic1d.<locals>.<lambda>c S s | | S rµ r| r¶ r| r| r r· x r¸ r )r r
|
rA r7 Úanyr r ÚastypeÚintrl r r} r~ r« r r- Úlenr © r r r r r r r¡ r¢ r£ r¤ r¥ Úsr| r| r Ú test_basic1dk s
|
z TestMaskedArray.test_basic1dc
|
C sÔ | j \
|
}}}}}}}}} }
|
dD ]°} | |_| |_| |_| |_| |
|
_tt| tt| tt|| t|j| t|jtdd | tt|t|tdd | t||
|
tt |d|
|
t|| q d S )N)©é r§ )é r¨ c S s | | S rµ r| r¶ r| r| r r· r¸ z.TestMaskedArray.test_basic2d.<locals>.<lambda>c S s | | S rµ r| r¶ r| r| r r· r¸ r )
|
r rl r
|
rA r r« r r- r¼ r7 r½ r| r| r Ú test_basic2d} s
|
z TestMaskedArray.test_basic2dc
|
C s | j \
|
}}}}}}}}} }
|
tt ||f¡t||f tt ||f¡t||f tt ||f¡t||f tt |||f¡t|||f d S rµ )r r r} r) r r| r| r Útest_concatenate_basic s
|
z&TestMaskedArray.test_concatenate_basicc
|
C s" | j \
|
}}}}}}}}} }
|
d} | |_ |_ |_|_t|jt || ¡ t|jt || ¡ t||fd} tt ||fd¡| tt |j|jfd¡| j td}t t
|
dddgd}t||f}t |ddddg t |jddddg t||f}t |ddddg t |jddddg d S )N©r§ rÁ r r¨ FTr r ) r rl r r r} rj r) Ú_maskry r& ra r ) r r r r r r r¡ r¢ r£ r¤ r¥ r¾ Zxmymr| r| r Útest_concatenate_alongaxis s z*TestMaskedArray.test_concatenate_alongaxisc C sZ t tttj d¡t d¡dtfdtfgd}t |d d
|dd
g}t
|
|| d S )Né
|
r r® ©r~ é ) rJ ÚlistÚzipr} ÚrandomÚrandr Úfloatr» r) r )r ÚdataÚtestr| r| r Útest_concatenate_flexible¯ s ÿþ z)TestMaskedArray.test_concatenate_flexiblec C sR t dddgdddgdd}t|jd t|jdddgg t|jdddgg d S )Nr r¨ r§ r )r Úndmin©r r§ )r& r rl r² rÆ r³ r| r| r Útest_creation_ndmin¸ s z#TestMaskedArray.test_creation_ndminc C sH t dddg}t|d< t |dtd}t|j|jj t|j|jj d S )Nr r¨ r§ éÿÿÿÿ)rÓ r~ )r& rI rÏ r rl rÆ ©r r Úxxr| r| r Ú$test_creation_ndmin_from_maskedarray¿ s
|
z4TestMaskedArray.test_creation_ndmin_from_maskedarrayc C s t dtd}t|dddg< t|}t|j|j t|}t|j|j t|ddddgd d}t|j|j tdd dgd
|
d}t|jd
|
d
|
d
|
g tdd dgd d}t|jd d d g tdd dg|jd d }t t
|
|j|j¡ tdd dg|jd
|
d }t t
|
|j|j¡ d S ) Né rÉ r§ r é r r r r¨ TF©r Úcopy) r rÏ rI r r r r r& rÆ r
|
r} Úmay_share_memory)r rÐ Zdma_1Zdma_2Zdma_3r r r| r| r Útest_creation_maskcreationÇ s z*TestMaskedArray.test_creation_maskcreationc C s: t t jj¡ t t¡ t dt jjg¡ W 5 Q R X d S )Nr )r} r& ÚmarI ÚpytestÚwarnsÚ UserWarning©r r| r| r Ú*test_masked_singleton_array_creation_warnsÛ s z:TestMaskedArray.test_masked_singleton_array_creation_warnsc C sÈ t t d¡dddddgd}t ||d d d
f}t|dddddgdddddgg t|jdddddgdddddgg t|_t ||d d d
f}t|dddddgdddddgg t|jtk d S ) NrÊ r r r rÖ r¨ r§ rÁ )r& r} r r rÆ r_ r r
|
)r r rÐ r| r| r Ú'test_creation_with_list_of_maskedarraysæ s "$"z7TestMaskedArray.test_creation_with_list_of_maskedarraysc C sÌ t jjddgddgd}t d¡}t jj||gdd}t|jddgddgg G d d
|
d
|
}t jj| dgddgd}tjt d d t j|gt
|
d W 5 Q R X t j ||g¡}t|jddgddgg d S ) Nr r® TFr r¨ ZU21rÉ c @ s e Zd Zdd ZdS )zUTestMaskedArray.test_creation_with_list_of_maskedarrays_no_bool_cast.<locals>.NotBoolc S s t dd S )Nú not a bool!)Ú
|
ValueErrorrä r| r| r Ú__bool__ü s z^TestMaskedArray.test_creation_with_list_of_maskedarrays_no_bool_cast.<locals>.NotBool.__bool__N)Ú__name__Ú
|
__module__Ú __qualname__ré r| r| r| r ÚNotBoolû s rí rç ©Úmatch) r} rà rJ r r' r r rá Úraisesrè Úbool)r Z
|
masked_strZ
|
normal_intÚresrí Z
|
masked_objr| r| r Ú4test_creation_with_list_of_maskedarrays_no_bool_castò s
|
zDTestMaskedArray.test_creation_with_list_of_maskedarrays_no_bool_castc C s0 t jdgddgddgddgdd }t| d S )
|
N)ÚAr Úf0Úf1ZS4Úi8r é )ÚnamesÚformatsÚoffsetsrÉ )r} r& r³ r| r| r Ú'test_creation_from_ndarray_with_padding s
|
þ
|
z7TestMaskedArray.test_creation_from_ndarray_with_paddingc C s6 t jtdd tdddgdddgd W 5 Q R X d S )Nzunexpected keyword argumentrî r r¨ r§ r )Zmaks)rá rð Ú TypeErrorr rä r| r| r Ú test_unknown_keyword_parameter s z.TestMaskedArray.test_unknown_keyword_parameterc
|
C sj | j \
|
}}}}}}}}} }
|
d|_d|_t|} t| j|j t| j|j t| j|j t| j|j d S )NéñØÿÿT)r Ú
|
fill_valueÚ _hardmaskr' r r² rÆ ) r r r r r r r¡ r¢ r£ r¤ r¥ Zxmmr| r| r Ú test_asarray s z TestMaskedArray.test_asarrayc C s2 t d¡j}t|jj t|}t|jj d S )Nr§ ©r} ÚeyeÚTr
|
ÚflagsÚ c_contiguousr' ©r ÚmZnew_mr| r| r Útest_asarray_default_order s z*TestMaskedArray.test_asarray_default_orderc C s6 t d¡j}t|jj t|dd}t|jj d S )Nr§ ÚC©Úorderr r r| r| r Útest_asarray_enforce_order" s z*TestMaskedArray.test_asarray_enforce_orderc C sd t jddN tt jddgdddgd}t|}t|j|jddg t|jdddg W 5 Q R X d S )NÚignore©Úinvalidr r
r r r ) r} ÚerrstaterJ Únanr8 r r² r rÆ )r rÐ Z
|
data_fixedr| r| r Útest_fix_invalid* s
|
z TestMaskedArray.test_fix_invalidc C sH t d}t|d< tttdk t|d tk tt|d dd d S )Nr r ú--r )r rI r
|
r± r r7 r³ r| r| r Útest_maskedelement2 s
|
z"TestMaskedArray.test_maskedelementc C sd t dtd}d}||d< t|d | t|d |k dd l}|j ¡ }||d< t|d |k d S )Nr rÉ )r r¨ r§ rÁ rÊ r )r2 Úobjectr r
|
ÚdatetimeÚnow)r r r r Údtr| r| r Útest_set_element_as_object: s
|
z*TestMaskedArray.test_set_element_as_objectc C sþ t ddddg¡}t|ddddgd}t|ddddgd}t|}t| t| tt |¡t|dd tt|d t|d k t|d |d k t|d tk t|d |d t|dd
|dd
t|d d
|d d
t|dd
|dd
d
|
|d< d
|
|d< t|| d |dd
< d |dd
< t|| t|d< t|| t|dd
< t|| ||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 }t|| tt tdddddgt|j td |j tddddgt}t ddddgt¡}|d }|d }tt|t tt|t t|| t|dd
jdk d S )Nr r¨ rÁ r§ r r F©ÚendwithrÊ é éc r
r Zhello©r )r} r& r± Úreprr rp r
|
ÚtyperI r r; rJ r rT r r r r rl )r Úx1Zx2Zx3Úx4Ús1Ús2r| r| r Ú test_indexingG sZ
|
|
|
|
" "
|
|
|
z TestMaskedArray.test_indexingc C s t j d¡ d¡}tdddddgf}t j d¡}t j|jd< |||< t jj|d< t jjd t j d
|
}t jj
|
d ddgd d d gd}||d< ||dddg< d S )Né< )r rÈ r rÊ r¨ é ©r¨ r¨ .r rÉ gCDë pTFr r ) r} rà r rj ÚsliceÚ
|
masked_allÚinfr² rI Úfloat32r& )r r ÚindexÚvaluer| r| r Útest_setitem_no_warningy s z'TestMaskedArray.test_setitem_no_warningc
|
C s: dddddg}t |}t |}t||k t |dd}t||k t d¡}t||d}t|jj|j tt||j t|j
|
j|j t|}t|jj|jjk t|j
|
j|j
|
jk t||d}t|jj|jk t|j
|
j|jk t|d t k d|d< t|d t k t|j
|
j|jk tt|j d t||dd } t| jj|jk t| j
|
j|jk t| d t k d| d< t| d t k t| j
|
j|jk tt| j d t|d
|
|d}
|
tt |
|
j|d
|
jk td } t | d< t| d } tt| | g| tt| ddddddddg t| d dd} t| ddddddddg t| ddd}t| | | jd dd}t| | | dd¡}t| | | ¡ }t|j| j t|j
|
| j
|
tdddgdddgd}t|}t|jjj |jjj t|j
|
jj |j
|
jj t|dd}t|jjj |jjj t|j
|
jj |j
|
jj d S )Nr r T©rÝ rÊ r r¨ r rÜ r
rÁ )rø )r¨ r¨ r¨ r¨ ©Úaxisr§ )rF r
|
r} r r& r r² Z__array_interface__r rÐ rÆ rI r r7 r~ rk r) r; ri rÝ rJ Úctypesr )r Únr r Zm3r# Úy1Zy1aÚy2Zy2aZy3r$ Zy4Zy5Zy6Zy7Zy8Zy9r r r| r| r Ú test_copy sv
|
ÿÿ
|
|
|
|
zTestMaskedArray.test_copyc C s( t jjddd}| ¡ }t|jd d S )Né+ Tr )r} rà r& rÝ r r )r r Zxcr| r| r Ú test_copy_0dÔ s z TestMaskedArray.test_copy_0dc C s2 t ttj dddg¡ t ttj d¡ d S )Nr r¨ r§ )r r¨ r§ )r
|
rA r} rà rÝ rä r| r| r Ú test_copy_on_python_builtinsÚ s z,TestMaskedArray.test_copy_on_python_builtinsc C sF t j dddg¡}t j dddg¡}|j}|j t| dddg d S )Nr r¨ r§ rÁ rÊ r )r} rà r& rÝ r )r r r® Z a_copy_methodr| r| r Útest_copy_immutableß s
|
z#TestMaskedArray.test_copy_immutablec C sÄ ddl m} tdddgdddgd}||}t|j|j tt|jt|j d|d< t|jdddg t|jdddg ||}t|j|j d|jd< t|jdddg t|jdddg d S )Nr )Údeepcopyr r¨ FTr )rÝ r> r& r r r ÚidrÆ )r r> r Úcopiedr| r| r Ú test_deepcopyç s
|
z TestMaskedArray.test_deepcopyc C sp t dddgdddgd}tt|d tttd tttd d tt ttd
|
}W 5 Q R X t|d d S ) Nr r r¨ FTr ú[0 -- 2]r Ú z >5)r& r ÚformatrI r Ú FutureWarning)r r Zwith_format_stringr| r| r Ú test_formatø s
|
zTestMaskedArray.test_formatc C s t dddgdddgd}tt|d tt|t d¡ tj d ¡}tjj |dd
|
< tt|t d ¡ tj d ¡}tt|t d ¡ t dddgdddggtj
|
d}tjj |d< tt|t d¡ tt|d d
t d¡ tt| t ¡t d¡ d S )Nr r r¨ FTr rB z masked_array(data=[0, --, 2],
|
mask=[False, True, False],
|
fill_value=999999)éÐ é2 z¶ masked_array(data=[0, --, --, ..., 1997, 1998, 1999],
|
mask=[False, True, True, ..., False, False, False],
|
fill_value=999999)é zÙ masked_array(data=[ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13,
|
14, 15, 16, 17, 18, 19],
|
mask=False,
|
fill_value=999999)r§ rÁ rÊ r rÉ ©r r zë masked_array(
|
data=[[1, 2, 3],
|
[4, --, 6]],
|
mask=[[False, False, False],
|
[False, True, False]],
|
fill_value=999999,
|
dtype=int8)z« masked_array(data=[[1, 2, 3]],
|
mask=[[False, False, False]],
|
fill_value=999999,
|
dtype=int8)zÑ masked_array(
|
data=[[1, 2, 3],
|
[4, --, 6]],
|
mask=[[False, False, False],
|
[False, True, False]],
|
fill_value=999999)) r& r r± r! ÚtextwrapÚdedentr} rà r rI Úint8rº r» ©r r r| r| r Ú test_str_repr s> þ þ þ
|
þ þ
|
þz TestMaskedArray.test_str_reprc C s t ¡ }t jdd zbtdddgdddgd}tt|d tt|d
|
t j d ¡}t jj |dd
< tt|d W 5 t jf | X d S )Nz1.13)Úlegacyr r r¨ FTr rB zcmasked_array(data = [0 -- 2],
|
mask = [False True False],
|
fill_value = 999999)
|
rG rH zmasked_array(data = [0 -- -- ..., 1997 1998 1999],
|
mask = [False True True ..., False False False],
|
fill_value = 999999)
|
)
|
r} Zget_printoptionsZset_printoptionsr& r r± r! rà r rI )r Zoldoptsr r| r| r Útest_str_repr_legacyM s þz$TestMaskedArray.test_str_repr_legacyc C sH d}t |}tj |¡}tjj|dd}t||| t||d d S )Nu caféTr r )r" r} rà r& r )r ÚuZutypeZ
|
arr_nomaskZ
|
arr_maskedr| r| r Útest_0d_unicodeb s z TestMaskedArray.test_0d_unicodec C sÔ t tttfD ]Â}td |¡}d|_ddddddddddg
|
ddf}tdtj d D ]z}|D ]p}||_
|
t tj ||d¡}t |j|j t |j|j |tt fkr®t |jd nt |j|d t|j
|
| qZqRq d S ) NrÈ éç r r TFr¨ ©Úprotocol)r» rÏ r± r r rº r Úranger{ ÚHIGHEST_PROTOCOLr ÚloadsÚdumpsr rÆ r² r )r r~ r ÚmasksÚprotor Ú a_pickledr| r| r Ú test_picklingl s" þ z TestMaskedArray.test_picklingc C s t jddgdtfdtfgd t j¡}t|ddgd}td tj d
|
D ]B}t
|
tj ||d ¡}t |j |j t || tt|jt j qFd S ) N)r
r¨ )r rÁ r r rÉ ©TF©FTr r¨ r rU )r} r& rÏ r» ÚviewÚrecarrayrJ rW r{ rX rY rZ r rÆ r
|
Ú
|
isinstancer² )r r r r\ r] r| r| r Útest_pickling_subbaseclass s
|
ÿÿ
|
z*TestMaskedArray.test_pickling_subbaseclassc C s` t jj}tdtjd D ]B}t tj||d¡}t|j |j t|j
|
|j
|
t|j |j qd S )Nr¨ r rU ) r} rà rI rW r{ rX rY rZ r Z
|
_baseclassrÆ r² )r Zmcr\ Z
|
mc_pickledr| r| r Ú test_pickling_maskedconstant s z,TestMaskedArray.test_pickling_maskedconstantc C sh t ddgddgdtfdtfgd}tdtjd D ]0}t tj||d
|
¡}t|j |j t|| q2d S ) N©r r
©r¨ r ©r r ©r r r r® ©r r~ r¨ r rU )
|
r& r» rÏ rW r{ rX rY rZ r rÆ )r r r\ r] r| r| r Útest_pickling_wstructured s ÿz)TestMaskedArray.test_pickling_wstructuredc C sL t d}d|_|j}tdtjd D ]"}t tj||d¡}t|| q$d S )NrÈ )rÖ r¨ r¨ r rU ) r rl r rW r{ rX rY rZ r )r r r® r\ rÑ r| r| r Útest_pickling_keepalignment s z+TestMaskedArray.test_pickling_keepalignmentc C sZ t dddg}t dddgdddgd}t|d jd t|d jd t|d jd d S )Nr r§ r¨ r r r| )r& r rl r¬ r| r| r Ú test_single_element_subscript¨ s
|
z-TestMaskedArray.test_single_element_subscriptc s( t dttd t dttd t dttdggg t dttdgg ttttddg t }| td¡ t t
|
ttdgdgd¡ tdddgdddgd tt fdd t t d
|
d t t
|
t d ¡ W 5 Q R X ttt t t d
|
d tt fd d d S ) Nr r
z$Warning: converting a masked elementr r¨ r§ r c s t S rµ )rÏ r| ©r r| r r· ½ r¸ z/TestMaskedArray.test_topython.<locals>.<lambda>rÖ r c s t d S ©Nr )r» r| rn r| r r· Â r¸ ) r r» r& rÏ r rý r Úfilterrã r
|
r} Úisnanr )r Úsupr| rn r Ú test_topython° s z TestMaskedArray.test_topythonc C sV t d}| dd¡}d|jd< t|d dk |d| }t|j| t|jd| t|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|j|jk t|d tk t|d tk t|d tk t|d tk t|d tk t|| d S )NrI rÁ rÊ é ©r r y $@rÈ ée r .r§ rø r r) r )r rj Úflatr
|
r ÚrealÚimagr* rI r± r- rx rl rU r~ )r r r£ Úcr| r| r Útest_oddfeatures_1Ä s.
|
|
|
z"TestMaskedArray.test_oddfeatures_1c C s¬ t dddddg}t dddddg}t|d< t||| }t|ddd d
|
d g t|d< t||| }t|ddd d
|
d g t|d tk t|d tk t|d tk d S ) Nr
r r r r r r r¨ r r éûÿÿÿ)r& rI rx r r
|
©r r rz r£ r| r| r Útest_oddfeatures_2à s z"TestMaskedArray.test_oddfeatures_2c C s: t dgdd}t dg}|j}|| ||< t|dg d S )NrÈ Tr rI )r& r r )r ÚatestÚbtestÚidxr| r| r Útest_oddfeatures_3î s
|
|
z"TestMaskedArray.test_oddfeatures_3c C s( t jjddd}t| d¡d d d S )Nr ÚOrÉ r r )r} rà r, r r7 rN r| r| r Ú test_filled_with_object_dtype÷ s z-TestMaskedArray.test_filled_with_object_dtypec C s| t dgdtfddtfgd}t|d< t| ¡ tj tdtdtdfg|jd t|d< t| d ¡tj d
|
g|jd d S ) N©r r r Úi©r¾ ú|S8ÚfrÉ r r° r r )r Ú1r
) r& r» rÏ rI r r7 r} r. r~ )r Zflexir| r| r Ú test_filled_with_flexible_dtypeû s ÿ
|
þþÿ
|
ÿz/TestMaskedArray.test_filled_with_flexible_dtypec C sr dt fdtfg}tddg|d}| ¡ }tt|dtdf | d¡}tt|d d
|
|_tt| ¡ d d S ) Nr r® ©r r ri rj r r
©rÖ rÖ )r rÖ ©éüÿÿr )r r )r» rÏ r^ r7 r Útupler. r )r Úndtyper rÑ r| r| r Útest_filled_with_mvoid s
|
z&TestMaskedArray.test_filled_with_mvoidc C sâ dt fddt fdt fgfg}tddgddg|d }| d
|
¡}tjd d g|d }t|| |d d
|
¡}tjddg|d jd }t|| tj dt dg¡¡}t|j jt dddgdfg¡ t|j
|
jt dddgdfg¡ d S )Nrô ÚBÚBAÚBB©r rJ ©r¨ r* ©r ru ©r ri rj r ©r ri )r¨ ©r¨ r rÉ ri r r¨ )rô z(2,2)i1,(2,2)i1r* )rõ Úi1r* )rö r r* r* )rõ ú?r* )rö r r* ) r» r& r7 r} r r~ Únumpyrà ry rÐ r )r r r rÑ ÚcontrolÚZr| r| r Ú test_filled_with_nested_dtype s, ÿ
|
|
|
ÿÿ
|
ÿÿz-TestMaskedArray.test_filled_with_nested_dtypec C sP t tj ddgddtj ddgdddd}t|jd t| d ¡jd d S )
|
N)r r r¨ )rÁ rÊ r ÚFr )r r r )r r r )r r Z F_CONTIGUOUSr )r& r} r
|
r r7 rN r| r| r Útest_filled_with_f_order* s þz(TestMaskedArray.test_filled_with_f_orderc C sR t dddgtd}d|jd< | ¡ }t|jd d d|jd< t|jd d d S )Nr r¨ r§ rÉ ú???Úinfoz!!!)r& rÏ Ú_optinforÝ r ©r r r r| r| r Útest_optinfo_propagation2 s
|
|
z(TestMaskedArray.test_optinfo_propagationc C s¸ t ddddg}d|jd< t|jd |dkjd t|jd |dkjd t|jd |dkjd t|jd |dkjd t|jd |dkjd t|jd |d jd t|jd |d jd t|jd |d jd t|jd |d jd t|jd |d d
jd t|jd |dddg jd t|jd t |¡jd t|jd t |¡jd t|jd t |ddjd t|jd t |¡jd d S ) Nr r¨ rÁ r0 Úkeyr Tr2 )r& r¦ r r} r5 r rz rN r| r| r Ú test_optinfo_forward_propagation; s"
|
z0TestMaskedArray.test_optinfo_forward_propagationc C s t dtfddtfdtfgfg¡}tddgddg|d }d
|
}tt|| td d d d gd d d ggd fd dd dgd d dggd fdd}d}tt|| d S )Nr r Útr¾ )r )r¨ r )rÁ )rÊ ç @r r rj z [(--, (2, --)) (4, (--, 6.0))]r r FTúint, (2,3)float, float©rÐ r r~ z)(0, [[--, 0.0, --], [0.0, 0.0, --]], 0.0))r} r~ r» rÏ r& r r± rJ )r Z
|
fancydtyperÑ r Zt_2d0r| r| r Útest_fancy_printoptionsN s* "þ ÿþ
|
ÿþúz'TestMaskedArray.test_fancy_printoptionsc C s¼ dt fdtfg}tjddg|d}t|}tjddgddggtd}t|| t|j|j tddgdd g|d
|
}t|}tddgddggd d gd d ggtd
|
}t|| t|j|j t|j|j dt fdd t fdtfgfg}tddgddg|d
|
}t|}tdddgdddggd d d gd d d ggtd
|
}t|| t|j|j t|j|j dt fdtfg}tjdgdgg|d}t|}tjddggddgggtd}t|| t|j|j d S )Nr r® rJ r* rÉ r
r ri ru rj r r ÚbaÚbb)r )r çñ?)r¨ )r¨ ç@r r r² r³ )r» rÏ r} r& r9 r r~ r )r r r rÑ r r| r| r Ú test_flatten_structured_arrayb sD
|
ÿ
|
ÿ ÿ
|
|
z-TestMaskedArray.test_flatten_structured_arrayc C s dt fdt fg}tjdg|dd }t|}tt|t tdgdg|dd }tt|t tddgddg|d}t|jd |jd }tt|t d S ) Nr r® ©r r¨ rÉ r ru rj rh ) r» r} r& r^ r
|
rc rJ r² rÆ )r r r r r| r| r Ú test_void0d
s zTestMaskedArray.test_void0dc C s dt fdt fg}tddgddg|d}|d }tt|t t|d |d fd t|d d
|
|d }tt|t t|d tk t|d tk t|d d tdd gfgd dgfgdgd}t|d d |d d t|d d tdd gd dgdd d S )Nr r® rµ rÅ rh ru rj r rJ r¨ r rÁ TF)rô ú>i2©r¨ r® rô r· )r» rJ r
|
rc r^ r rI )r r r r rô r| r| r Útest_mvoid_getitem s, ÿ
|
þ ÿz"TestMaskedArray.test_mvoid_getitemc C sV dt fdt fg}tddgddg|d}tt|d d d
|
g tt|d td g d S ) Nr r® rµ rÅ rh ru rj r r r¨ rÁ )r» rJ r rË rI )r r r r| r| r Útest_mvoid_iter¬ s ÿz TestMaskedArray.test_mvoid_iterc C s¨ t ddgdtfdtfgd}tt|d d t|d d< tj}t d¡ z(tt|d d tt|d d W 5 t |¡ X t d
|
d gd gd}tt|d d d S )NrJ r* r r® rÉ r z(1, 1)z-X-z(1, -X-)©r r¸ )r r z(1,)) r& r» r r± rI rS Ú_displayZ set_displayr! )r ÚmxZ ini_displayr| r| r Útest_mvoid_print¶ s
|
z TestMaskedArray.test_mvoid_printc C s t dddgfgdddgfgdddgfdgd}tt|d d
|
k tt|d d
|
k t ddgdd ggfgddgddggfgd gd }tt|d dk tt|d dk t dgdgddgd }tt|d dk tt|d dk t ddgdd ggdfgddgddggdfgd dtfgd }tt|d dk tt|d dk t dgdgddgd }tt|d dk tt|d dk d S )Nr r¨ r§ FTé?B )r ú<i4r© )rÐ r r r~ r z ([1, --, 3],)rÁ )r rÀ r* r® z([[1, --], [--, 4]],)rµ r_ )r rÀ )r® rÀ z(--, 2)r® z([[1, --], [--, 4]], 1.0)r )Tr_ )r® úi4,i4z (--, (--, 1)))rJ r
|
r± r! rÏ )r Zt_maZt_2dZt_0dZt_ner| r| r Útest_mvoid_multidim_printÇ s>
|
ýþþ
|
þþz)TestMaskedArray.test_mvoid_multidim_printc C s¢ t dgdgd}t ddg}t ||gddgtd}t|d |k t|d |k tt |d j|jk¡ tt |d j¡ d |d jd< t|d d k d S )
|
Nr
Tr r Frj r r r )rJ r r
|
r} r rÐ r )r Zmx1Zmx2r½ r| r| r Útest_object_with_arrayë s z&TestMaskedArray.test_object_with_arrayN)>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' r1 Ú suppress_copy_mask_on_assignmentr9 r; r< r= rA rF rO rQ rS r^ rd re rk rl rm rs r{ r~ r r r r r¡ r£ r¨ rª r¯ r´ r¶ r¹ rº r¾ r rà r| r| r| r r C sx
|
2
|
HF
|
|
#
|
$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!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Z d5d6 Z d7d8 Z d9d: Z d;d< Z d=d> Z!d?d@ Z"dAdB Z#e$j% &dCdDdEg¡e$j% &dFdGdHg¡dIdJ Z'e$j% &dCdDdEg¡e$j% &dFdGdHg¡dKdL Z(e$j%j&dMe)e*dNe$j%j&dOe)e*dNe$j% &dFdGdPg¡dQdR Z+e$j%j&dMe)e*dNe$j%j&dOe)e*dNe$j% &dFdGdPg¡dSdT Z,e$j%j&dMe)e*dNe$j%j&dOe)e*dNe$j% &dFdGdPg¡e$j% &dUe-j.e-j/e-j0e-j1g¡dVdW Z2e$j% &dUe-j.e-j/e-j0e-j1g¡e$j% &dFdGdXg¡dYdZ Z3d[d\ Z4d]d^ Z5d_d` Z6dadb Z7dGS )cÚTestMaskedArrayArithmeticc 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|¡}
|
| d¡ ||||||||| |
|
f
|
| _t ¡ | _t j ddd d S )Nr
r r r r r r r r r r r r r r r r r r ©r0 r )
|
r} r& r rJ rx r r ÚgeterrÚ
|
err_statusÚseterrr r| r| r r¦ û s &"
|
|
z&TestMaskedArrayArithmetic.setup_methodc C s t jf | j d S rµ ©r} rÉ rÈ rä r| r| r Úteardown_method s z)TestMaskedArrayArithmetic.teardown_methodc
|
C s@ | j \
|
}}}}}}}}} }
|
tddgddgg} t| ddgddgg} t| | | | t| | | | t| | | | dD ]º} | | ¡}| | ¡}| | ¡}| | ¡}|
|
| ¡}
|
t| | t|| || t|| || t|| || t|| || t|| || t|| || t|| || t|| || t|| || t|| || t|| || t|| || t|d |d tt|d t|d t|| || tt ||¡t|| tt ||¡t|| tt ||¡t || tt
|
||¡t
|
|| q~d S )Nr r¨ r rÁ ))rt rÀ )r¨ r ç @) r r& rJ r rj r r} r rr r] r0 )r r r r r r r¡ r¢ r£ r¤ r¥ Za2dZa2dmr¾ r| r| r Útest_basic_arithmetic s@
|
|
|
|
|
|
z/TestMaskedArrayArithmetic.test_basic_arithmeticc C s t dtd}d|_t dtd}|| }t|dddgdddgg t|jd d
|
d
|
gd d
|
d
|
gg ||d d d
f }t|dddgdddgg t|jd d
|
d
|
gd d
|
d
|
gg t d td}||d d
d f }t|dddgd dd gg t|jd d d gd
|
d
|
d
|
gg d S )Nr rÉ ©r¨ r§ r§ r r
r rÌ r r r¨ r r )r rÏ rl r r ©r r r r£ r| r| r Ú test_divide_on_different_shapes2 s z9TestMaskedArrayArithmetic.test_divide_on_different_shapesc C s> t dg¡}tdg}tt|| t tt|| t d S ©Nr )r} r& r
|
rc r )r Únarà r| r| r Útest_mixed_arithmeticD s
|
z/TestMaskedArrayArithmetic.test_mixed_arithmeticc C sR t t¡j}t|d| dg}tt|d dddg ttd| dddg d S )Nr
r r¨ r r )r} ÚfinforÏ Útinyr& r r< )r rÕ r r| r| r Útest_limits_arithmeticK s z0TestMaskedArrayArithmetic.test_limits_arithmeticc C s\ t ddd}tdt d j td| j t| j tt||j tt||j d S )Nr r r )r& r
|
r rW rZ )r r¡ r| r| r Ú test_masked_singleton_arithmeticQ s z:TestMaskedArrayArithmetic.test_masked_singleton_arithmeticc C sh t dddgdddgd}t|d dktk t|d dktk t|d dkd t|d dkd d S ) Nr r¨ r§ r r rÖ FT)r& r
|
rI r rN r| r| r Ú test_masked_singleton_equality[ s
|
z8TestMaskedArrayArithmetic.test_masked_singleton_equalityc C st t ddg}|t }t|j|j t|jddg |d t }t|tk |t }t|j|j t|jddg d S )Nr r¨ Tr )rJ rI r rl rÆ r
|
r§ r| r| r Ú%test_arithmetic_with_masked_singletonc s z?TestMaskedArrayArithmetic.test_arithmetic_with_masked_singletonc C s2 t dg}|t }t|j|j t|jdg d S )Nr T)rJ rI r rl r r§ r| r| r Ú5test_arithmetic_with_masked_singleton_on_1d_singletono s
|
zOTestMaskedArrayArithmetic.test_arithmetic_with_masked_singleton_on_1d_singletonc C sN t ddd}t| ¡ jj|jj t dd }t|jd t|jddg d S )Nr r rh r r¸ r )r& r r7 r5 rÐ rl r )r r r¡ r| r| r Útest_scalar_arithmeticv s
|
z0TestMaskedArrayArithmetic.test_scalar_arithmeticc
|
C sN | j \
|
}}}}}}}}} }
|
tt |¡t| tt |¡t| tt |¡t| tt |¡t| tt |¡t| tt |¡t| tt t
|
|¡t | tt t
|
|¡t | tt t
|
|¡t | tt |¡t | tt |¡t| tt |¡t| tt |¡t| tt ||¡t|| tt |¡t| tt |d| ¡t|d| ttj|d| ddt|d| dd tt ||¡t|| tt ||¡t|| tt ||¡t|| tt ||¡t|| tt ||¡t|| tt ||¡t|| tt |¡t| d S )Nù ð?T)Údeg)r r r} r+ r, rm rn ru rv rq r rD rE r5 r# r r$ r" r r r4 r` rB r= rC r> r* r r| r| r Útest_basic_ufuncs~ s2 $,z+TestMaskedArrayArithmetic.test_basic_ufuncsc C s t dtd t dtddgd tddddgddddgd}t|}t|jjtjk t d| | d ¡}t|}t|jjtjk t d| t|d}tt |t
|
t dd
|
g| tt |t k tddddg}t|d}tt |t
|
t|jjtjk t tj|jdd d S ) Nr r r r r
r r r§ r* r¨ r3 )r r- r& r
|
r~ r" r} Zintprj rc r r; r_ r Ú AxisError)r Zottrò r| r| r Útest_count_func s&
|
|
|
|
|
z)TestMaskedArrayArithmetic.test_count_funcc C s& t dtdddg t dtd d S )Nr§ r r¨ rµ )r r- rä r| r| r Ú test_count_on_python_builtins³ s z7TestMaskedArrayArithmetic.test_count_on_python_builtinsc
|
C sL | j \
|
}}}}}}}}} }
|
t |¡} t|} tt| t | ¡ tt| t | ¡ ttdddgdddgdddg ttdddgdddgdddg t d}t dd }t
|
|d< t
|
|d< tt||t t |||| tt||t t |||| tt |¡dk tt |¡dk t d dd¡}t
|
|d< ttj|d d d d S )
|
Nr r¨ r§ rÁ r r rÊ r r3 )r r} rh r rV rW r rY rZ r rI rx rB r= r
|
rj ) r r r r r r r¡ r¢ r£ r¤ r¥ ZxrZxmrr| r| r Útest_minmax_func¸ s$
|
"" z*TestMaskedArrayArithmetic.test_minmax_funcc C s¾ t d¡}t||}tt|t t|t ||¡ t ||¡}tt|t t|t j ||¡ t||}tt|t t|t ||¡ t ||¡}tt|t t|t j ||¡ d S )Nr* ) r} ra rZ r
|
rc r r rc rW )r r ZaminimumZamaximumr| r| r Útest_minimummaximum_funcÑ s
|
|
|
z2TestMaskedArrayArithmetic.test_minimummaximum_funcc C s2 t dddgdddgd}tj |¡}t|d d S )Nr r¨ r§ Fr )r& r} rW r r r¬ r| r| r Útest_minmax_reduceã s z,TestMaskedArrayArithmetic.test_minmax_reducec C sÖ t j d¡ ¡ }tt j ddd¡|d}d|_dD ]}tt |}ttj j
|
|}t j dt d}z||d|d }W n t k
|
r Y nX t j dtd}||d|d }t||k | d
|
¡ ||d|d }t||k q2d S ) Nrt r rÈ r rÅ ©rY rV ©rÁ rÉ ©r4 Úoutr )r} rÍ rÎ Úroundr& Úuniformrl Úgetattrr rà Úcorer2 r» r rÏ r
|
Úfill)r r r¡ ÚfuncnameÚnpfuncZmafuncÚnoutÚresultr| r| r Ú test_minmax_funcs_with_outputé s"
|
|
z7TestMaskedArrayArithmetic.test_minmax_funcs_with_outputc
|
C sT | j \
|
}}}}}}}}}}|jf|_t| ¡ d t|d ¡ tk t|d d¡tk t|d d¡tk t| ¡ d t|d ¡ tk t|d d¡tk t|d d¡tk t| ¡ d t|d ¡ tk t|d d¡tk t|d d¡tk t dddgd d
|
}t| ¡ tk t| ¡ tk t| ¡ tk d S ) NrÈ r rÖ r ç 4@r r¨ r§ Tr )
|
r r« rl r rV r
|
rI rY Úptpr& )r Ú_r¡ r r| r| r Útest_minmax_methods s$
|
z-TestMaskedArrayArithmetic.test_minmax_methodsc C s@ t ddddtd dddddddg ¡}d}d}d d
|
d
|
d
|
d
|
d
|
d d
|
d
|
d
|
d
|
d
|
g }t||d }| d ¡ t jt jt jt jt j t j
|
t j g}|D ]<}t t|||d ¡ || t t|||d ¡ || qt | ¡ | t | ¡ | |d d
D ]<}t t|||d ¡ || t t|||d ¡ || qê|dd
D ]}tdddgd
|
d d
|
g|d }t | ¡ |d t | ¡ |d tt jd t jd t j d gd
|
d d
|
g|d } t | ¡ |t j d t | ¡ |t jd t jdt t j¡j }
|
t|
|
d
|
gd
|
d gd ¡ |
|
kstt|
|
d
|
gd
|
d gd ¡ |
|
ks4tq4d S )Nr
r r r r r r r r r r r rj rÁ éýÿÿÿy@µx ¯D ð?y@µx ¯D Ày@µx ¯D ð¿y @y @rÜ )r} r& r rJ r ZhalfZsingleÚdoubleZ
|
longdoubleZcfloatZcdoubleZ clongdoubler rV rY r- rÔ Úfloat64ÚAssertionError) r r r Zan10r r¡ Z float_dtypesZ float_dtyper¢ r¤ Zcmaxr| r| r Útest_minmax_dtypes sT &
|
ÿÿÿÿÿÿ&ÿ&z,TestMaskedArrayArithmetic.test_minmax_dtypesc
|
C s¼ | j \
|
}}}}}}}}} }
|
ttj |¡t |¡ ttj |¡t |¡ tdttddd tdttddd ttj|ddt|dd ttjt|dddt|dd tt |d¡t|d ttj |ddt |dd tt |d¡t |d ttj t|dddt |dd d} | |_
|
|_
|
|_
|
|_
|
t | dkr¸tt ||fd¡t ||fd ttj |d¡t |d¡ tt |d¡t|d tt |d¡t |d d S )NrÁ r r3 r rÅ ) r r r} r r Ú
|
accumulaters r& r7 re rl r¼ r) r½ r| r| r Útest_addsumprod@ s$ " " z)TestMaskedArrayArithmetic.test_addsumprodc C sþ t dgdgdggdgdgdggd}t ddgddgd d
|
gg}|| }t ddgddgddggd d gd d gd d ggd}t|| t|j|j t|j|j || }t ddgddgd d
|
ggd d gd d gd d ggd}t|| t|j|j t|j|j t dgdgdgg}t ddgddgd d
|
ggd d gd d gd d ggd}|| }t d dgddgddggd d gd d gd d ggd}t|| t|j|j t|j|j || }t d dgddgddggd d gd d gd d ggd}t|| t|j|j t|j|j d S )Nr
r r FTr r r r¬ ç @r r r¨ r§ rø rÈ é r) ©r& r rÐ r ©r r r® rÑ r r| r| r Útest_binops_d2DU sD $ÿ
|
ÿ
|
ÿÿ
|
ÿ
|
z)TestMaskedArrayArithmetic.test_binops_d2Dc C sþ t dgdgdggdgdgdggd}t ddgddgd d
|
gg}|| }t d d gddgddggd d gddgddggd}t|| t|j|j t|j|j || }t ddgddgd d
|
ggd d gddgddggd}t|| t|j|j t|j|j t dgdgdgg}t ddgddgd d
|
ggd d gd d gd dggd}|| }t d d gd dgd dggd d gd d gd dggd}t|| t|j|j t|j|j || }t ddgddgddggd d gd d gd dggd}t|| t|j|j t|j|j d S )Nr
r r FTr r r r¬ rþ r gUUUUUUÕ?r r gÙ?r§ rÌ r) r r r| r| r Útest_domained_binops_d2Dy sD $ÿ
|
ÿ
|
ÿÿ
|
ÿ
|
z2TestMaskedArrayArithmetic.test_domained_binops_d2Dc C s t dddgdddgdd}|d }t|jdddg |d7 }t|jdddg |d }t|jdddg |d }t|jdddg d S )Nr
r r F©r Úshrinkr r ©rJ r r r¬ r| r| r Útest_noshrinking s ÿz*TestMaskedArrayArithmetic.test_noshrinkingc C s( t j dg¡}tt |d¡jjd d S )Nr rÊ r| )r} rà r& r Ú true_divider rl ©r r r| r| r Útest_ufunc_nomask® s z+TestMaskedArrayArithmetic.test_ufunc_nomaskc C s. t jjdddgddd}t|jdddg d S )Nr
rÌ çÍÌÌÌÌÌ@ç ø?F©r r )r} rà rT r r rN r| r| r Útest_noshink_on_creation´ s z2TestMaskedArrayArithmetic.test_noshink_on_creationc
|
C s¢ | j \
|
}}}}}}}}} }
|
tt||t|| t||} t| t ||¡ t| jt|j|j t||} t| t ||¡ t| jtt|j|j|dk d S ro )r r r\ r} r rH ) r r r r r r r¡ r¢ r£ r¤ r¥ rÑ r| r| r Útest_mod¹ s
|
|
z"TestMaskedArrayArithmetic.test_modc C s t d}t d¡}t|dd
< | ddd¡}| ddd¡}tt |d¡t|d tt |dd¡t|dd tt t|d t|d ¡t|| tt 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 ) NrÚ rÊ r r¨ r§ rÁ )r¨ r r r r ÚabcÚdef) r r} rI rj r rw rt r@ r7 rc r& r r
|
)r r r r« r| r| r Ú test_TakeTransposeInnerOuterÄ s&
|
ÿÿz6TestMaskedArrayArithmetic.test_TakeTransposeInnerOuterc C s t ddgddgd}t|jddg t|j ¡ ddg t|jj|jjj t|jdd g t|j ¡ dd g t|jj|jjj d S )
|
Ny ð? $@y 4@ @r r r rÈ r¨ r rI )r& r ry r7 r~ r² rx )r rØ r| r| r Útest_imag_realØ s z(TestMaskedArrayArithmetic.test_imag_realc C sÚ t tj ddd¡ dd¡}t |d d
df< |d< |d< d}|D ]}tt|}t||}tjdtd}| d ¡ ||d|d
|
}t
|
||k t ||d|d
|
tdt d}|d|d
|
}t
|
||k t
|
|d tk qDd S ) Nr rÈ rt r§ rÁ r )rs ÚprodÚvarÚstdrV rY rô ÚmeanrÉ rÿ rç © r& r} rÍ rê rj rI rë r2 rÏ rí r
|
r r» )r r¡ Zfunclistrî rï ÚxmmethÚoutputrñ r| r| r Útest_methods_with_outputâ s
|
|
|
z2TestMaskedArrayArithmetic.test_methods_with_outputc C sø dt fdt fg}tddgddg|d}||k}t|jddg t|jd d g t|jdk ||d
|
k}t|jdd g t|jd d g t|jdk tddgd dg|d}||k}t|jd dg t|jdd g t|jdk |d
|
|k}t|jd d g t|jdd g t|jdk tddgdd g|d}||k}t|jddg t|jd d g t|jdk dt fdd t fd t fgfg}tddgddggddgddgg|d}|d |k}t|jdd gd d gg t|jd d gd dgg t|jdk d S )Nrô r rJ r* ri rh rj TFr ru r r r r ©r§ ©r§ r§ ©rÁ ©rÁ rÁ r r ©r rh ©r» r& r rÐ r r
|
r ©r r r rÑ r® r| r| r Útest_eq_on_structuredø sJ ÿÿý z/TestMaskedArrayArithmetic.test_eq_on_structuredc C sø dt fdt fg}tddgddg|d}||k}t|jddg t|jddg t|jd k ||d
|
k}t|jdd g t|jddg t|jd k tddgd dg|d}||k}t|jd dg t|jd dg t|jd k |d
|
|k}t|jd d g t|jd dg t|jd k tddgdd g|d}||k}t|jddg t|jddg t|jd k dt fdd t fd t fgfg}tddgddggddgddgg|d}|d |k}t|jdd gd d gg t|jddgdd gg t|jd k d S )Nrô r rJ r* ri rh rj FTr ru r r r r r r r r r r! r" r| r| r Útest_ne_on_structured# sJ ÿÿý z/TestMaskedArrayArithmetic.test_ne_on_structuredc s¨ t d¡}tdd|dtdd|dtdd|dtdd|dfD ]b t¡}| d¡}t jd|d td|d tdd|dtd
|
d|dfD ] t¡}| d¡}||k ¡ }t k| t k| t|k| t |k| t||k| fd d |jD }tt|t d ¡ | ||k
|
¡ }t k| t k| t|k| t |k| t||k| fd d |jD }tt|t d
|
¡ | qq>d S )NrÁ rµ rh rj ri ru rJ Z2i4rÉ ©r§ r¨ c s g | ]} | | kqS r| r| ©r Úname©r r r| r r e s zITestMaskedArrayArithmetic.test_eq_ne_structured_extra.<locals>.<listcomp>c s g | ]} | | kqS r| r| r& r( r| r r m s ) r} r~ r^ ra r r& r r rù rñ r¹ ) r r Úma1Úr1Úma2Zr2Z eq_expectedZel_by_elZ ne_expectedr| r( r Útest_eq_ne_structured_extraN s>
|
ý
|
|
|
|
ý
|
|
|
z5TestMaskedArrayArithmetic.test_eq_ne_structured_extrar ÚSÚUrí Nrô c C sN t ddg|ddg|d}||k}t|jddg t|jddg t|jdk ||d k}t|jddg t|jddg t|jdk t ddg|ddg|d}||k}t|jddg t|jddg t|jdk |d |k}t|jddg t|jddg t|jdk ||d k}t|jddg t|jddg t|jdk d S )Nr r® r r ©r~ r r TF©r& r rÐ r r
|
r ©r r rí r rÑ r® r| r| r Útest_eq_for_stringsp s, z-TestMaskedArrayArithmetic.test_eq_for_stringsc C sN t ddg|ddg|d}||k}t|jddg t|jddg t|jdk ||d k}t|jddg t|jddg t|jdk t ddg|ddg|d}||k}t|jddg t|jddg t|jdk |d |k}t|jddg t|jddg t|jdk ||d k}t|jddg t|jddg t|jdk d S )Nr r® r r r/ FTr0 r1 r| r| r Útest_ne_for_strings s, z-TestMaskedArrayArithmetic.test_ne_for_stringsÚdt1©ÚidsÚdt2r c C sN t ddg|ddg|d}||k}t|jddg t|jddg t|jdk ||d k}t|jddg t|jddg t|jdk t ddg|ddg|d}||k}t|jddg t|jddg t|jdk |d |k}t|jddg t|jddg t|jdk ||d k}t|jddg t|jddg t|jdk d S )Nr r r/ TFr0 ©r r4 r7 rí r rÑ r® r| r| r Útest_eq_for_numeric° s, z-TestMaskedArrayArithmetic.test_eq_for_numericc C sN t ddg|ddg|d}||k}t|jddg t|jddg t|jdk ||d k}t|jddg t|jddg t|jdk t ddg|ddg|d}||k}t|jddg t|jddg t|jdk |d |k}t|jddg t|jddg t|jdk ||d k}t|jddg t|jddg t|jdk d S ©Nr r r/ FTr0 r8 r| r| r Útest_ne_for_numericÑ s, z-TestMaskedArrayArithmetic.test_ne_for_numericÚopc C s t ddg|ddg|d}|||}t|j||j|j t|jddg t|jdk |||d }t|j||j|jd t|jddg t|jdk t ddg|ddg|d}|||}t|j||j|j t|jddg t|jdk ||d |}t|j||jd |j t|jddg t|jdk |||d }t|j||j|jd t|jddg t|jdk d S r: )r& r rÐ r² r r
|
r )r r< r4 r7 rí r rÑ r® r| r| r Ú test_comparisons_for_numericò s,
|
|
z6TestMaskedArrayArithmetic.test_comparisons_for_numericzN/Ac C sT t dddgdddg|d}t dddgdddg|d}t|||j||j|j d S )Nr r® Zcder r )r r )rJ r r² )r r< rí r) r+ r| r| r Útest_comparisons_strings s z2TestMaskedArrayArithmetic.test_comparisons_stringsc C s t }| td¡ td dgddgd}t|d ktddgddgd t|jd kddg t|d ktddgddgd td dgdd}t|d kddg t|d kddg td dgdd}t|d ktddgdd t|d ktddgdd t}t|d kt W 5 Q R X d S )NzComparison to `None`r r r TFr¨ )r rp rE r& r rÐ rI )r rr r r| r| r Útest_eq_with_None s
|
z+TestMaskedArrayArithmetic.test_eq_with_Nonec C s t d}t|dkd t|dkd t|dkd t|dkd t ddd}t|dkt t|dkt t|dkt t|dkt d S )Nr Tr Fr )r& r rI r¬ r| r| r Útest_eq_with_scalar9 s z-TestMaskedArrayArithmetic.test_eq_with_scalarc C s t ddgddgd}t ddgddggt ddgddgg¡fD ]<}||k}t|jddgddgg t|jddgddgg q>d S )Nr r r r¨ FT)r& r} r rÐ r )r r r rÑ r| r| r Ú test_eq_different_dimensionsE s ÿ ÿ ÿz6TestMaskedArrayArithmetic.test_eq_different_dimensionsc C s¼ t dddddgdddddgd}t tjtjdt d¡dgdddddgd}t|}t|| t|j|j t|jdddddg t |¡}t|| t|j|j t|jdddddg d S ©NrÖ r r r¨ r§ r )rJ r} r rD r r )r r r rÑ r| r| r Útest_numpyarithmeticP s ÿ
|
|
|
z.TestMaskedArrayArithmetic.test_numpyarithmetic)8rê 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á ÚmarkÚ parametrizer2 r3 Únum_dtsÚnum_idsr9 r; ÚoperatorÚleÚltÚgeÚgtr= r> r? r@ rA rC 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 )ÚTestMaskedArrayAttributesc C s¢ t dddgdddgd}t |}t|j|j t |dddgdd}t|jdddg t |dddgdd}t|jdddg t |dddgd}t|jdddg d S ) Nr r¨ r§ r r F)r Z keep_maskTr ©r r r½ r| r| r Ú test_keepmaskc s z'TestMaskedArrayAttributes.test_keepmaskc C sF t d}dddddg}t|}t||dd}t||ddd}dd g|dd
|
g< dd g|dd
|
g< t|jddd d d
|
g t|jddd d d g t|jdddddg t|j t|j dd dg|dd
|
< dd dg|dd
|
< t|jddd d d
|
g t|jddd dd g t|jt t |d< t |d< t|jdddddg t|jdddddg d|d d
< d|d d
< t|jdddd d
|
g t|jdddddg t|jdddddg t|jt |
|
¡ t d|d d
< t|jddd d d
|
g t|jt | ¡ t ||d k < t|jddd d d
|
g t|j dddddg d|t |dkd< t|jddd ddg t|j dddddg tdd gd d
|
ggddgddggdd}d|d< t|jddgd d
|
gg t|j ddgddgg d|d< t|jddgd dgg t|j ddgddgg d |t |dk d< t|jdd gd dgg t|j ddgddgg d S )NrÊ r r T©r Ú hard_maskF©r rQ rÝ rÈ é( rÁ r¨ r§ rI é r )r rF r& r r² r r
|
r r_ rI Ú soften_maskÚ harden_maskrÆ r7 )r r r6 r ÚxhÚxsr| r| r Ú test_hardmaskp s`
|
&z'TestMaskedArrayAttributes.test_hardmaskc C s^ t d}dddddg}t|}t||dd}d|dd
< d|dd
< t|jdddd dg d S )
|
NrÊ r r TrP rT rÁ r¨ r§ )r rF r& r r² ©r r r6 r rW r| r| r Útest_hardmask_again§ s z-TestMaskedArrayAttributes.test_hardmask_againc C s t dddgdddgd}| ¡ }t|| d|d< t|| t|t dddgdddgd | ¡ }d|d< t|| t|t dddgdddgd d S ©Nr r¨ r§ r r )r& rV r rU r¬ r| r| r Útest_hardmask_oncemore_yay± s
|
|
|
z4TestMaskedArrayAttributes.test_hardmask_oncemore_yayc C sV t d}t|d< d|d< t|jt t d}d|_t|d< d|d< t|jtd d S )NrÈ r F)r rI r rÆ r_ Z
|
_smallmaskry rN r| r| r Útest_smallmask¿ s z(TestMaskedArrayAttributes.test_smallmaskc C st t dddgdddgd}| ¡ }t|| t|jt tj dgdtfdtfg¡}| ¡ }| ¡ t|j|j d S ) Nr r¨ r§ r r r r r® )
|
r& Z shrink_maskr r r_ r} rà r» rÏ rÝ r¬ r| r| r Útest_shrink_maskË s
|
z*TestMaskedArrayAttributes.test_shrink_maskc C sT t dddgdddggdtfdtfd gd
|
}t|d d < t|d d < t|d d< t|d< |j}t|d |d t|d |d t|d |d t|d d
|d t|d |d t|d |d t|d |d t|dd
|d t|d |d d}d}|D ]<}t||||f |d7 }||jd krd}|d7 }qd S )N)r r² Úone)r¨ r³ Útwo)r§ çffffff
|
@Úthr)rÁ ç@Zfou)rÊ ç @Zfiv)r gffffff@Úsixr r® ©rz r rÉ ri ru rz )r r¨ r r rh r r¨ r§ rÁ rJ rÊ rµ rÖ )r& r» rÏ rI rw r rl )r r Zxflatr Újr¥ r| r| r Ú test_flatØ s6
|
ÿþ z#TestMaskedArrayAttributes.test_flatc sÆ t jddd t j ¡}t d¡|_t| t|jt d¡ fdd}tt| j dt jj
|
d}t|jt d¡ t jddd t j ¡}t d¡|_t|jt d¡ t|j t jj d S )NrÁ zf4,i4rÉ Úf4c s t j ¡} t d¡| _d S )NÚf8)r} rà r& r~ )r rn r| r Úassign s z;TestMaskedArrayAttributes.test_assign_dtype.<locals>.assign)r~ r" ) r} ry rà r& r~ r! r r rè ra r rÆ r_ )r r rl r® r| rn r Útest_assign_dtypeö s
|
z+TestMaskedArrayAttributes.test_assign_dtypeN) rê rë rì rO rY r[ r] r^ r_ ri rm r| r| r| r rM a s 7
|
rM 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!d" Zd#d$ Zd%S )&ÚTestFillingValuesc C s t jjj}|dt}t|d |d t}t|td |dd}t|d |d d}t|td tt|dt tt|dt d S )Nr ú|S3ó 0s camelot!r Ústuff) r} rà rì Ú_check_fill_valuer» r r. r rý )r rr Úfvalr| r| r Útest_check_on_scalar s
|
|
|
|
|
|
|
z&TestFillingValues.test_check_on_scalarc C s t jjj}dtfdtfdg}|dddg|}tt|t t |
|
¡ dddg |d |}tt|t t |
|
¡ t dt d t t d
|
g t j d |d }|||}tt|t t |
|
¡ dddg t j d d tfdtfdgd }|||}tt|t t |
|
¡ dddg t jdtd}d|d< ||t}tt|t t |
|
¡ dddg dtfg}|d|}tt|t t |
|
¡ d d S )Nr r® )rz ro r çÍÌÌÜ)gÁr¤ ó ???r r r° )r ru r¤ rÉ rô r )r ro r» ©rl r~ )r ru rv é6eÄ)rx )r} rà rì rr r» rÏ r
|
rc r r Úitemr. r r& r )r rr r rs Zfill_valr| r| r Útest_check_on_fields# s<
|
|
|
þ
|
ÿ
|
|
|
|
z&TestFillingValues.test_check_on_fieldsc C sê t dddg}|j ddi¡ t |td}t|jddd g t|jtd
|
t |td}t|jddd g t|jtd | t¡}t|jddd g t|jtd
|
t|jd d | d g¡}t|d j|j t|d j|j d S )Nó 3ó 4ó 5Úcommentzupdated!rÉ r§ rÁ rÊ r r )r ro r )
|
r& r¦ Úupdater» r r² r r. rÏ rº r¬ r| r| r Útest_fillvalue_conversionR s
|
z+TestFillingValues.test_fillvalue_conversionc C s> t d}t t d¡}t t d¡j}t|| t|| d S )Nr
)r. r} r& r~ r )r rö Úf2Zf3r| r| r Útest_default_fill_valuej s
|
|
z)TestFillingValues.test_default_fill_valuec C sh t dgdtfddtfgd}t|}t|j}tj tdtdtdf|jd}t|| t|| d S ) Nr
r r r rÉ r r° r )r& r» rÏ r. r~ r} r )r Úfieldsrö r Úexpectedr| r| r Ú"test_default_fill_value_structuredr s ÿ
|
|
þþ
|
z4TestFillingValues.test_default_fill_value_structuredc C s: t dg¡}t|}t|d t t|d |d ¡ d S )N)ÚvZV7r )r} r~ r. r r& )r r r r| r| r Ú test_default_fill_value_void~ s z.TestFillingValues.test_default_fill_value_voidc C s
|
t dddgdd}|dddg }t|j|j dtfdg}td d
|
td fg|d }d |_t|j ¡ ddg t|d jd t|d jd d|_t|j ¡ d t|d jd t|d jd tdddg}d |_tt |j¡j
|
t t|jd t|jt d¡ d S )Nr r¨ r§ r ©r r r ©r¾ ro )r r )r¨ r® r rÉ rT ç 8@s 999r¾ )r r¤ )r rv r rv r ) rJ r Ú _fill_valuerÏ r& r r ry r} r' r~ )r rÐ ZseriesÚmtyper r| r| r Útest_fillvalue s$ z TestFillingValues.test_fillvaluec C sr t dgdtfddtfgd}t F}| td¡ |ddg }tt|jd |dd
|d d
k W 5 Q R X d S )
|
Nr
r r r rÉ zNumpy has detected)r¿ r r rÖ ) r& r» rÏ r rp rE r r r )r r rr Z subfieldsr| r| r Útest_subarray_fillvalue s ÿ z)TestFillingValues.test_subarray_fillvaluec C s8 t jjj}dtfddtfg}t jtdtdtdf|d}t|d || dtd fg}t jtdfdtfgd |¡}t|d || t jd
|
dtfgd |¡}t|d|| t
|
d ¡}t jtdtdtdfd d |¡}|d |}t|| t jd d d |¡}t|d|| t |}t|d j j d d S )Nr r r r r° r rÉ rõ r* r r zint, float, float)r r r rö )r} rà rì rr r» rÏ r& r. r rº r~ rJ r rª )r rr r r rÑ ÚMr| r| r Útest_fillvalue_exotic_dtype¦ sB
|
|
þý ÿÿ
|
|
þýý
|
|
z-TestFillingValues.test_fillvalue_exotic_dtypec C sn dD ]d}t d|¡}tt d| d ¡}tj ||¡ t d|¡}tt d| d ¡}tj ||¡ qd S )N) ÚasÚfsZpsÚnsÚusÚmsr¾ r ÚhÚDÚWr ÚYZNaTz<M8[ú]z<m8[)r Z
|
datetime64r. r~ r} Ztestingr Ú timedelta64)r Ztimecoder rÑ r| r| r Ú!test_fillvalue_datetime_timedeltaÅ s z3TestFillingValues.test_fillvalue_datetime_timedeltac C s t ddgdtfddtfdtfgfgd}|j}t|j|j t|d t|d t|d d t|d d t|d d t|d d t|}t|j|j t|d t|d t|d d t|d d t|d d t|d d t|d t|d t|}t|j|j t|d t|d t|d d t|d d t|d d t|d d t|d t|d d S )
|
N)r rÎ )rÁ ©rÊ r rô r r r rÉ r r )r& r» r r r~ r. r[ rX ©r r rÑ r| r| r Útest_extremum_fill_valueÔ s( ÿ z*TestFillingValues.test_extremum_fill_valuec C s t dddgfdtjdfgd}t|}t|j|j t|d t dt|d ¡ t|}t|j|j t|d t dt|d ¡ d S )Nr¨ r§ rÁ r0 rÉ r )r& r} rM r[ r r~ ÚfullrX r r| r| r Ú!test_extremum_fill_value_subdtypeì s z3TestFillingValues.test_extremum_fill_value_subdtypec C sä dt fdt fg}tttdddgdddgd |d
|
}|d }| d ¡ t|jt d ¡ tt|j d d |j d< tt|j d tttdddgdddg|d}|d }| d ¡ t|jt d ¡ tt|j d t
|
df d S )Nr 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 Zaar« Úttr| r| r Ú test_fillvalue_individual_fields÷ s ÿ
|
|
"
|
z2TestFillingValues.test_fillvalue_individual_fieldsc C s dt f}dt f}tddgddgtjft |g¡d}t|j||gd}|d |d< |d |d j¡ |j d }t
|
t |d ¡ t |d
|
t d d S ) Nr® r ©r
©r )F)r r r~ rÉ r| r rÖ r
)rÏ r& r} r r~ r2 rl r r r r
|
rq r r. )r r Zadtyper r® r r| r| r Ú(test_fillvalue_implicit_structured_array
|
s
|
ÿ
|
z:TestFillingValues.test_fillvalue_implicit_structured_arrayc C sj t ddd}t|jd tddtd}t|jd tddtd}t|jd tddtd}t|jd d S )Nr§ r r r¢ r )r2 r r ra rÏ ry Úcomplexr? rN r| r| r Útest_fillvalue_as_arguments s z-TestFillingValues.test_fillvalue_as_argumentsc C sJ t dd}t|jd tdtd}t|jd tdtd}t|jd d S )Nr© ©rl rw )r2 r rl ra rÏ ry r© rN r| r| r Útest_shape_argument% s
|
z%TestFillingValues.test_shape_argumentc C sÐ t dddgdtjd}| ¡ }t|jdk | t¡}t|jdk |jtd}t|jdk | tj¡}|jtjd}|jtdd}t|jdk |jtdd}t|jdk |jtjd}t|jd k d S )
|
Nr r¨ r§ r¢ )r" r )r" r rÉ r¿ ) r& r} Úint64ra r
|
r r r Úint32r§ r| r| r Útest_fillvalue_in_view1 s
|
z(TestFillingValues.test_fillvalue_in_viewc C s8 t ddd}t|d jtd t|d jtd d S )Nr© z (2)3S,(2)3Urw rõ s spamrö Úeggs)r2 r r r. rN r| r| r Útest_fillvalue_bytes_or_str[ s z-TestFillingValues.test_fillvalue_bytes_or_strN)rê rë rì rt rz r r r
r r r r r r r¡ r¥ r¨ rª r¬ r¯ r± r| r| r| r rn s$ / *rn c @ sL e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dS )Ú
|
TestUfuncsc C sr t dddtd gd ddgdgd dt dddtd gd ddgdgd df| _t ¡ | _tjddd d S )
|
Nr
r rÖ r¨ r r r r rÆ )r& r r r} rÇ rÈ rÉ rä r| r| r r¦ f s
|
((ÿ
|
zTestUfuncs.setup_methodc C s t jf | j d S rµ rÊ rä r| r| r rË m s zTestUfuncs.teardown_methodc C s dD ]}zt t|}W n tk
|
r4 t t|}Y nX t tjj|}| jd |j
}|| }|| }t |
|
d¡|
|
d¡| t |j |j |d qd S )N)*rq rD rE r5 r* rm r+ ru r# r r$ rn r, rv Zarcsinhr! Zarctanhr ÚfabsÚnegativeÚfloorÚceilZ logical_notr rr r] r0 r Z floor_divideÚ remainderÚfmodÚhypotr" r4 r` rC r> rB r= Ú logical_andÚ
|
logical_orÚ logical_xorr )Úerr_msg) rë ÚumathÚAttributeErrorÚ fromnumericr rà rì r Zninr r7 r r )r r ZufZmfÚargsÚurÚmrr| r| r Útest_testUfuncRegressionp s z#TestUfuncs.test_testUfuncRegressionc C sl | j d }tt|dd tt|dd tt|d d
ddd tt|ddd tt |¡t d S )Nr r3 r§ )
|
r r
|
r ro r rs re r r r rN r| r| r Ú test_reduce s
|
zTestUfuncs.test_reducec C s t dd dd¡}t|dk |}t| ¡ | ¡ t| ¡ d t| d¡| d¡ t| d¡dddd g t| d¡d j t| d¡d j d S )
|
Nr é r§ rÁ rÊ r r r) rø )r rj rU r rV rY r
|
r )r r Úamaskr| r| r Ú test_minmax s zTestUfuncs.test_minmaxc C s~ t dddddgdddddgd}t |¡}t dddt d¡dgdddddgd}t|| t|j|j tt|jt d S rB )rJ r} rq r r r
|
rc r )r r rÑ r r| r| r Útest_ndarray_mask¢ s
|
ÿ
|
z TestUfuncs.test_ndarray_maskc sð t ddgddgd tttj d tttj d G fddd}| }t| d k t | d
|
k G d d d }| }t |¡tk tt t
|
|¡d k t |¡tk t| dk t | d k t | dk d S )Nr
r r r r r c s& e Zd Z jd Zdd Zdd ZdS )z<TestUfuncs.test_treatment_of_NotImplemented.<locals>.MyClassr c S s dS )NúMy mulr| ©r Úotherr| r| r Ú__mul__¶ s zDTestUfuncs.test_treatment_of_NotImplemented.<locals>.MyClass.__mul__c S s dS )NúMy rmulr| rË r| r| r Ú__rmul__¹ s zETestUfuncs.test_treatment_of_NotImplemented.<locals>.MyClass.__rmul__N)rê rë rì Ú__array_priority__rÍ rÏ r| rn r| r ÚMyClass³ s
|
rÑ rÊ rÎ c @ s, e Zd ZdZdd Zdd Zdd ZeZdS ) z=TestUfuncs.test_treatment_of_NotImplemented.<locals>.MyClass2éd c S s dS )NÚMe2mulr| rË r| r| r rÍ Ä s zETestUfuncs.test_treatment_of_NotImplemented.<locals>.MyClass2.__mul__c S s dS )NÚMe2rmulr| rË r| r| r rÏ Ç s zFTestUfuncs.test_treatment_of_NotImplemented.<locals>.MyClass2.__rmul__c S s dS )NÚMe2rdivr| rË r| r| r Ú__rdiv__Ê s zFTestUfuncs.test_treatment_of_NotImplemented.<locals>.MyClass2.__rdiv__N)rê rë rì rÐ rÍ rÏ rÖ Ú __rtruediv__r| r| r| r ÚMyClass2Á s
|
rØ rÔ rÓ rÕ ) rJ r rý rH ÚmulÚtruedivr
|
rÍ ÚNotImplementedr r] rc Ú __truediv__)r rÑ ÚmerØ Zme_toor| rn r Ú test_treatment_of_NotImplemented¬ s z+TestUfuncs.test_treatment_of_NotImplementedc C s t jjdt jgddgd}t ¡ n t d¡ t| t|d |dk t | t
|
| t | t | t | t| t|d t|d W 5 Q R X d S )Nr r r r Úerrorr¨ )r} rà r& r ÚwarningsÚcatch_warningsÚfilterwarningsr5 r rq rD ru r# r r! r0 r r r| r| r Útest_no_masked_nan_warnings× s
|
|
|
|
z&TestUfuncs.test_no_masked_nan_warningsN) rê rë rì r¦ rË rÄ rÅ rÈ rÉ rÞ rã r| r| r| r r² c 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!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Z d5d6 Z d7d8 Z d9d: Z d;d< Z d=S )>Ú TestMaskedArrayInPlaceArithmeticc C s t d}t d}t d}t|d< |||f| _| t¡| t¡| t¡f| _tjd tjd | _dd | jD | _| tj ¡| tj ¡| tj ¡f| _
|
d S )NrÈ r¨ Z
|
AllIntegerZAllFloatc S s g | ]}t |¡jqS r| ©r} r~ r" ©r rõ r| r| r r þ s zATestMaskedArrayInPlaceArithmetic.setup_method.<locals>.<listcomp>) r rI Úintdatarº rÏ Ú floatdatar} Ú typecodesÚ
|
othertypesÚuint8Ú uint8data©r r r r¡ r| r| r r¦ ö s
|
|
|
ýz-TestMaskedArrayInPlaceArithmetic.setup_methodc C s | j \}}}t|d< |d7 }t||d |d7 }t||d | j\}}}|jjj}|d7 }t||jjjk t||d d S )Nr¨ r r
)rç rI r rè rÐ r5 r
|
)r r r r¡ rõ Zid1r| r| r Ú test_inplace_addition_scalar
|
s
|
z=TestMaskedArrayInPlaceArithmetic.test_inplace_addition_scalarc C sl | j \}}}|j}tdtjd}t|d< ||7 }||7 }t||| t||| t|jt||j d S ©NrÈ rÉ rÖ )rç r r r} Úint16rI r rH ©r r r r¡ r r r| r| r Útest_inplace_addition_array
|
s z<TestMaskedArrayInPlaceArithmetic.test_inplace_addition_arrayc C s< | j \}}}|d8 }t||d |d8 }t||d d S rÑ )rç r rí r| r| r Ú test_inplace_subtraction_scalar
|
s
|
z@TestMaskedArrayInPlaceArithmetic.test_inplace_subtraction_scalarc C sj | j \}}}|j}tdtd}t|d< ||8 }||8 }t||| t||| t|jt||j d S rï ©rè r r rÏ rI r rH rñ r| r| r Ú test_inplace_subtraction_array(
|
s z?TestMaskedArrayInPlaceArithmetic.test_inplace_subtraction_arrayc C s< | j \}}}|d9 }t||d |d9 }t||d d S )Nr r¨ )rè r rí r| r| r Ú"test_inplace_multiplication_scalar4
|
s
|
zCTestMaskedArrayInPlaceArithmetic.test_inplace_multiplication_scalarc C sj | j \}}}|j}tdtd}t|d< ||9 }||9 }t||| t||| t|jt||j d S rï rô rñ r| r| r Ú!test_inplace_multiplication_array<
|
s zBTestMaskedArrayInPlaceArithmetic.test_inplace_multiplication_arrayc C sT | j \}}}tdd }tdd }t|d< |d }t|| |d }t|| d S )NrÈ r¨ )rç r rI r rí r| r| r Ú test_inplace_division_scalar_intH
|
s
|
zATestMaskedArrayInPlaceArithmetic.test_inplace_division_scalar_intc C s@ | j \}}}|d }t||d |td }t|td d S )Nr rÈ )rÈ )rè r r ra rí r| r| r Ú"test_inplace_division_scalar_floatS
|
s
|
zCTestMaskedArrayInPlaceArithmetic.test_inplace_division_scalar_floatc C st | j \}}}|j}tdtd}t|d< || }|| }t||| t||| t|jtt||j|dk d S )NrÈ rÉ rÖ r rô rñ r| r| r Ú!test_inplace_division_array_float[
|
s zBTestMaskedArrayInPlaceArithmetic.test_inplace_division_array_floatc C sD ddddt d dddddddg }dd ddd
|
d d dddd dg }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|jd d d d d d d d d d d d g t|jdddd
|
t d dddddddg | ¡ }|| }t|jd d d d d d d d d d d d g t|jdddd
|
t d dddddddg d S )Nr
r r r r r r r r r r r r r )r rJ r rÆ r² rÝ )r r r r r r¡ r¢ r£ r| r| r Útest_inplace_division_miscg
|
s" $ ÿ$ ÿz;TestMaskedArrayInPlaceArithmetic.test_inplace_division_miscc C s$ t dddgdddgd}|d }t|jdddg t|jdddg |d7 }t|jdddg t|jdddg t dddgdddgd}|t dddgdddgd }t|jdddg t|jdddg t dddgdddgd}|t dddgdddgd7 }t|jdddg t|jdddg d S )Nr r¨ r§ r r rÁ r r× r| r| r Útest_datafriendly_add{
|
s z6TestMaskedArrayInPlaceArithmetic.test_datafriendly_addc C s< t dddgdddgd}|d }t|jdddg t|jdddg t dddgdddgd}|d8 }t|jdddg t|jdddg t dddgdddgd}|t dddgdddgd }t|jdddg t|jdddg t dddgdddgd}|t dddgdddgd8 }t|jdddg t|jdddg d S r\ r r× r| r| r Útest_datafriendly_sub
|
s z6TestMaskedArrayInPlaceArithmetic.test_datafriendly_subc C s< t dddgdddgd}|d }t|jdddg t|jdddg t dddgdddgd}|d9 }t|jdddg t|jdddg t dddgdddgd}|t ddd gdddgd }t|jdd
|
dg t|jdddg t dddgdddgd}|t ddd gdddgd9 }t|jdd
|
dg t|jdddg d S ) Nr r¨ r§ r r rÁ rÈ rI rT rS r r× r| r| r Útest_datafriendly_mul¨
|
s z6TestMaskedArrayInPlaceArithmetic.test_datafriendly_mulc C s< t dddgdddgd}|d }t|jdddg t|jdddg t ddd gdddgd}|d }t|jdddg t|jdddg t ddd gdddgd}|t d
|
d d gdddgd }t|jdd d g t|jdddg t ddd gdddgd}|t d
|
d d gdddgd }t|jdd d g t|jdddg d S )Nr r¨ r§ r r r r r
r r ró g >@g¹?r r× r| r| r Útest_datafriendly_div¿
|
s z6TestMaskedArrayInPlaceArithmetic.test_datafriendly_divc C st t dddgdddgd}|d }t|jdddg t|jdddg |dC }t|jddd g t|jdddg d S )
|
Nr
r r r r r rÌ gÍ;f @r§ r r× r| r| r Útest_datafriendly_powÖ
|
s z6TestMaskedArrayInPlaceArithmetic.test_datafriendly_powc C sÊ t ddgddgg}t ddgddgd}||7 }t|ddgddgg |jtk rht|jddgddgg t ddgddgg}t ddgddgd}||7 }t|ddgddgg t|jddgddgg d S )Nr r§ r r r¨ rÁ ©r& r r r_ r¬ r| r| r Ú test_datafriendly_add_arraysâ
|
s
|
z=TestMaskedArrayInPlaceArithmetic.test_datafriendly_add_arraysc C sÊ t ddgddgg}t ddgddgd}||8 }t|ddgddgg |jtk rht|jddgddgg t ddgddgg}t ddgddgd}||8 }t|ddgddgg t|jddgddgg d S )Nr r§ r r r¨ r r¬ r| r| r Ú test_datafriendly_sub_arraysð
|
s
|
z=TestMaskedArrayInPlaceArithmetic.test_datafriendly_sub_arraysc C sÊ t ddgddgg}t ddgddgd}||9 }t|ddgddgg |jtk rht|jddgddgg t ddgddgg}t ddgddgd}||9 }t|ddgddgg t|jddgddgg d S )Nr r§ r r r r¬ r| r| r Ú test_datafriendly_mul_arraysþ
|
s
|
z=TestMaskedArrayInPlaceArithmetic.test_datafriendly_mul_arraysc
|
s | j D ] t ¡ n t d¡ fdd| jD \}}}t|d< | d7 }t|| d | d7 }t|| d W 5 Q R X qd S )Nrß c 3 s | ]}| ¡V qd S rµ ©rº ræ ©r« r| r Ú <genexpr> s zUTestMaskedArrayInPlaceArithmetic.test_inplace_addition_scalar_type.<locals>.<genexpr>r¨ r )rê rà rá râ rì rI r rí r| r r Ú!test_inplace_addition_scalar_type s
|
|
|
zBTestMaskedArrayInPlaceArithmetic.test_inplace_addition_scalar_typec
|
s¢ | j D ] t ¡ t d¡ fdd| jD \}}}|j}td d}t|d< ||7 }||7 }t||| t||| t|jt ||j W 5 Q R X qd S )Nrß c 3 s | ]}| ¡V qd S rµ r ræ r r| r r s zTTestMaskedArrayInPlaceArithmetic.test_inplace_addition_array_type.<locals>.<genexpr>rÈ rÉ rÖ ©
|
rê rà rá râ rì r r rI r rH rñ r| r r Ú test_inplace_addition_array_type s
|
|
|
zATestMaskedArrayInPlaceArithmetic.test_inplace_addition_array_typec
|
s | j D ]x t ¡ f t d¡ fdd| jD \}}}| d8 }t|| d | d8 }t|| d W 5 Q R X qd S )Nrß c 3 s | ]}| ¡V qd S rµ r ræ r r| r r , s zXTestMaskedArrayInPlaceArithmetic.test_inplace_subtraction_scalar_type.<locals>.<genexpr>r ©rê rà rá râ rì r rí r| r r Ú$test_inplace_subtraction_scalar_type' s
|
|
|
zETestMaskedArrayInPlaceArithmetic.test_inplace_subtraction_scalar_typec
|
s¢ | j D ] t ¡ t d¡ fdd| jD \}}}|j}td d}t|d< ||8 }||8 }t||| t||| t|jt ||j W 5 Q R X qd S )Nrß c 3 s | ]}| ¡V qd S rµ r ræ r r| r r 7 s zWTestMaskedArrayInPlaceArithmetic.test_inplace_subtraction_array_type.<locals>.<genexpr>rÈ rÉ rÖ r rñ r| r r Ú#test_inplace_subtraction_array_type2 s
|
|
|
zDTestMaskedArrayInPlaceArithmetic.test_inplace_subtraction_array_typec
|
s | j D ]x t ¡ f t d¡ fdd| jD \}}}| d9 }t|| d | d9 }t|| d W 5 Q R X qd S )Nrß c 3 s | ]}| ¡V qd S rµ r ræ r r| r r F s z[TestMaskedArrayInPlaceArithmetic.test_inplace_multiplication_scalar_type.<locals>.<genexpr>r¨ r rí r| r r Ú'test_inplace_multiplication_scalar_typeA s
|
|
|
zHTestMaskedArrayInPlaceArithmetic.test_inplace_multiplication_scalar_typec
|
s¢ | j D ] t ¡ t d¡ fdd| jD \}}}|j}td d}t|d< ||9 }||9 }t||| t||| t|jt ||j W 5 Q R X qd S )Nrß c 3 s | ]}| ¡V qd S rµ r ræ r r| r r Q s zZTestMaskedArrayInPlaceArithmetic.test_inplace_multiplication_array_type.<locals>.<genexpr>rÈ rÉ rÖ r rñ r| r r Ú&test_inplace_multiplication_array_typeL s
|
|
|
zGTestMaskedArrayInPlaceArithmetic.test_inplace_multiplication_array_typec
|
sî dd t jd D }| jD ]Î t ¡ ¼ t d¡ fdd| jD \}}}td d d }td d d }t|d < z0| d }| d }t || t || W n0 t
|
k
|
rÜ d
|
d } |ksØt |Y nX W 5 Q R X qd S ) Nc S s h | ]}t |¡jqS r| rå ©r r« r| r| r Ú <setcomp>^ s z[TestMaskedArrayInPlaceArithmetic.test_inplace_floor_division_scalar_type.<locals>.<setcomp>ÚComplexrß c 3 s | ]}| ¡V qd S rµ r ræ r r| r r b s z[TestMaskedArrayInPlaceArithmetic.test_inplace_floor_division_scalar_type.<locals>.<genexpr>rÈ rÉ r¨ úSupported type ú throwing TypeError) r} ré rê rà rá râ rì r rI r rý rú )r Ú unsupportedr r r¡ Úmsgr| r r Ú'test_inplace_floor_division_scalar_type[ s
|
|
|
|
zHTestMaskedArrayInPlaceArithmetic.test_inplace_floor_division_scalar_typec
|
sú dd t jd D }| jD ]Ú t ¡ È t d¡ fdd| jD \}}}|j}td d}t |d < zR|| }|| }t
|
||| t
|
||| t
|
|jt t ||j| d
|
k W n0 t k
|
rè d d } |ksät |Y nX W 5 Q R X qd S ) Nc S s h | ]}t |¡jqS r| rå r r| r| r r r s zZTestMaskedArrayInPlaceArithmetic.test_inplace_floor_division_array_type.<locals>.<setcomp>r rß c 3 s | ]}| ¡V qd S rµ r ræ r r| r r v s zZTestMaskedArrayInPlaceArithmetic.test_inplace_floor_division_array_type.<locals>.<genexpr>rÈ rÉ rÖ r r r )r} ré rê rà rá râ rì r r rI r rH rý rú )r r r r r¡ r r r r| r r Ú&test_inplace_floor_division_array_typeo s(
|
|
|
þ zGTestMaskedArrayInPlaceArithmetic.test_inplace_floor_division_array_typec sh | j D ]Z t H}| t¡ fdd| jD \}}}td d d }td d d }t|d< z| d }t|| W n8 tt fk
|
r¼ } zt
|
j t |dd W 5 d }~X Y nX z| d }t|| W n: tt fk
|
r } zt
|
j t |dd W 5 d }~X Y nX t tjr>tt|jdd d ntt|jd
|
d d W 5 Q R X qd S ) Nc 3 s | ]}| ¡V qd S rµ r ræ r r| r r s zUTestMaskedArrayInPlaceArithmetic.test_inplace_division_scalar_type.<locals>.<genexpr>rÈ rÉ r¨ r ©Ú
|
stacklevelúFailed on type=Ú.r )rê r Úrecordrã rì r rI r ÚDeprecationWarningrý rà Úwarnr± Ú
|
issubclassr} Úintegerr¼ rD )r rr r r r¡ Úer| r r Ú!test_inplace_division_scalar_type s(
|
|
$ $ zBTestMaskedArrayInPlaceArithmetic.test_inplace_division_scalar_typec st | j D ]f t T}| t¡ fdd| jD \}}}|j}td d}t|d< z|| }t||| W n8 t t
|
fk
|
r¦ } zt j t |dd W 5 d }~X Y nX z<|| }t||| t|jtt||j| dk W n: t t
|
fk
|
r } zt j t |dd W 5 d }~X Y nX t tjrJtt|jd d
|
d ntt|jdd
|
d W 5 Q R X qd S ) Nc 3 s | ]}| ¡V qd S rµ r ræ r r| r r ° s zTTestMaskedArrayInPlaceArithmetic.test_inplace_division_array_type.<locals>.<genexpr>rÈ rÉ rÖ r r r r¨ r r )rê r r rã rì r r rI r r rý rà r r± rH r r} r! r¼ rD )r rr r r r¡ r r r" r| r r Ú test_inplace_division_array_type« s0
|
|
$þ$ zATestMaskedArrayInPlaceArithmetic.test_inplace_division_array_typec
|
C s² | j D ]¦}t ¡ t d¡ tdddgdddg|d}||d }tdddgdddg|d}t|j|j t|j|j ||dC }t|j|j t|j|j W 5 Q R X qd S )Nrß r r¨ r§ r rj rÁ )rê rà rá râ r& r rÐ r )r r« r rØ Zxx_rr| r| r Útest_inplace_pow_typeÒ s
|
|
|
z6TestMaskedArrayInPlaceArithmetic.test_inplace_pow_typeN)!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ä ó 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 dd Zdd Zdd Zed d Zd d Zd!d" Zd#d$ Zd%d& Zej d'd(¡ej d)d*¡d+d, Zd-d. Zd/d0 Z d1d2 Z d3d4 Z d5d6 Z d7d8 Z d9d: Z!d;d< Z"d=d> Z#d?d@ Z$dAdB Z%dCdD Z&dEdF Z'dGdH Z(dIdJ Z)dKdL Z*dMdN Z+dOdP Z,dQdR Z-dSS )TÚTestMaskedArrayMethodsc & 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*}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 ©+Nç À @ç®Gáz. @ç-²ï§!@ç !@çé&1¬ ü?çÀÊ¡E¶@ç\Âõ(Ü @ç
ëQ¸ @ç{®Gáº#@çPn@çhí|?õ!@ç!°rhí@çj¼t@çJ +@çáz®Ga@ç!°rhí@çÁÊ¡E¶ó@ç-²ï§@ç)\Âõ(@çÃõ(\B#@ç +Ù @çuV @çÛù~j¼ô@çÑ"Ûù~ê@ç¨ÆK7Á @ç
|
×£p=J#@ç®Gáz@çòÒMbØ@ç}?5^ºÉ#@ç¾/Ý$@ç!°rhm @ç-²ï§#@çJ +@çªñÒMb@çÀÊ¡Å@ç-²ï§Æï?r r§ r¨ r r ©rÐ r ©r} r& rj rl r © r r ÚXÚXXr r½ ÚmXÚmXXr Úm2xÚm2XÚm2XXr| r| r r¦ å sÒ û û û z#TestMaskedArrayMethods.setup_methodc C sF t dddg}t| ¡ |j ¡ t| ¡ |j ¡ t| ¡ |j ¡ t| ¡ |j ¡ t| ddddd¡|j ddddd¡ t| dddg¡|j dddg¡ t| ¡ |j ¡ t|
|
¡ |j
|
¡ t ddgddgg}t| ¡ |j ¡ t| ¡ |j ¡ t| ddg¡|j ddg¡ t| ¡ |j ¡ d S )Nr r§ r¨ r rÁ )r& r r¹ r² r ÚargmaxÚargminr( ÚcompressZconjr* Údiagonalrs rt rw ©r r r r| r| r Útest_generic_methods s ($ z+TestMaskedArrayMethods.test_generic_methodsc C sä t j d¡}|t j d¡d }tt|| t j|d< tt|| t j|d< tt|| t|}t|d< tt||dd tt||dd |d9 }d|d< tt|ddd tt t j ¡j
|
gt j d}tt|| d S ) NrÈ g:0âyE>r rÖ T)rK FrÉ ) r} rÍ rÎ r
|
r r- rJ rI ZiinfoÚint_rY r¬ r| r| r Ú test_allclose s
|
|
z$TestMaskedArrayMethods.test_allclosec C sH t jddddggdd}t||dds*tt||t dd ¡dsDtd S )
|
Nr r¨ r§ rÁ zm8[ns]rÉ r )Zatolr )r} r& r rú r rN r| r| r Útest_allclose_timedelta/ s z.TestMaskedArrayMethods.test_allclose_timedeltac C sN t dddgdddgddd gg¡}t jd
|
d d gd d d gd
|
d
|
d ggt jd }t||d }|dk}|dk }t| ¡ t| ¡ t| d¡d d d
|
g t| d¡d d d
|
g t| d¡d d d
|
g t| d¡d
|
d
|
d
|
g t| ¡ t| ¡ t| d¡d
|
d
|
d g t| d¡d d d g t| d¡d
|
d
|
d g t| d¡d
|
d
|
d g d S )Ng¤p=
|
×£À?g¤p=
|
×£Ð?gÍÌÌÌÌÌì?gìQ¸
ëÑ?g
ëQ¸ Õ?g)\Âõ(ä?g×£p=
|
×Ó?g×£p=
|
×ë?gffffffæ?TFrÉ r r r r )r} r& Úbool_rJ r
|
r r¹ r )r r r r½ ZmxbigZmxsmallr| r| r Ú test_allany6 s2 þ þþ z"TestMaskedArrayMethods.test_allanyc C s¨ t dtd}tdddgdd}t| ¡ tk |j|d t| t|jd t|tk t dtd}t| ¡ tk |j|d t| t|jd t|tk d S ) Nr| rÉ r r¨ r§ Tr ©rè )r2 rñ r& r
|
r rI rÆ r¹ )r Ústorer r| r| r Útest_allany_odditiesP s
|
z+TestMaskedArrayMethods.test_allany_odditiesc
|
C sn | j \
|
}}}}}}}}} }
|
t| ¡ d t| ¡ d t| ¡ d t| ¡ d t| ¡ d t| ¡ d t| ¡ d t| ¡ d t| d¡ddddddg t| d¡ddddddg t| d¡ddddddg t| d¡ddddddg t| d¡ddddddg t| d¡dddddd g t| d¡ddddddg t| d¡ddddddg d S )
|
Né# rÁ é é r r¨ rÊ r r§ )r r rW rV ) r r rO rP r r½ rQ rR rS rT rU r| r| r Útest_argmax_argminb s" z)TestMaskedArrayMethods.test_argmax_argminc & C sø t ddddddddd d
|
d d d ddddddddddddddd d d d d d!d"d#d$g$¡}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'}| d(d)¡}t|j|j t|j| d(d)¡ t|j|j d(d)¡ d S )*Nr( 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 r r r r¨ rø )r} r& Úclipr r r² )r r r r½ Zclippedr| r| r Ú test_clipy s~ û þ z TestMaskedArrayMethods.test_clipc C sJ t d¡}t jj|ddgd d}|jdd|d t|jddgd d S )NrÈ r r rÊ r ra )r} r rà r rh r r rZ r| r| r Ú test_clip_out s
|
z$TestMaskedArrayMethods.test_clip_outc C s t dddddgdd}|dk|d k @ }t| |¡ddg t|d
|
d g< | |¡}t|jddg t|jd d g t|jd t||| |dk }| |¡}t|jdddg t|jd d d g t|jd t||| t dddgdddggd d d gd d d ggd}| | ¡ dk¡}t|jddddg t|jd d d d g t d d d
|
g¡}|j|d
|
kd d}t|jddgddgg t|jd d gd d gg d S )Nr
r r r r i' r r ç @r¨ r§ r r rÈ rI rT rS rH r( r é r3 )
|
rJ r rX rI r² rÆ r rh r} r& )r r Ú conditionr® r r| r| r Ú test_compress s2
|
|
ÿz$TestMaskedArrayMethods.test_compressc C sR t ddddgddddgd}| ¡ }t|| t|d< | ¡ }t|dddg d S )Nr r¨ r§ rÁ r r )r& Ú
|
compressedr rI r¬ r| r| r Útest_compressed° s
|
z&TestMaskedArrayMethods.test_compressedc C sä dt fdtfdg}tdddg|d}tt|j ¡ t| t|}t|j|j t|j|j t t||d}t|j|j t|j|j tdd d
|
gd d d gd }t|}t
|
t |j |j ¡ | t¡}t
|
t |j |j ¡ d S ©Nr r® rg )r r² z1.1)r¨ r³ z2.2)r§ rb z3.3rÉ r r¨ r§ FTr )r» rÏ rJ r r¼ r ry r3 rl r2 r
|
r} rÞ r ra ©r Údatatyper r® r| r| r Ú
|
test_empty¹ s
|
ÿ
|
z!TestMaskedArrayMethods.test_emptyc C sâ dt fdtfdg}tdddg|d}tt|j ¡ t| tt||d}t|j|j t|j|j t |}t|j|j t|j|j tdd d
|
gd d d gd }t |}t
|
t |j |j ¡ | ¡ }t
|
t |j |j ¡ d S rq )r» rÏ rJ r r¼ r ry ry rl rz r
|
r} rÞ r ra rr r| r| r Ú
|
test_zerosÏ s
|
ÿz!TestMaskedArrayMethods.test_zerosc C sâ dt fdtfdg}tdddg|d}tt|j ¡ t| tt||d}t|j|j t|j|j t |}t|j|j t|j|j tdd d
|
gd d d gd }t |}t
|
t |j |j ¡ | ¡ }t
|
t |j |j ¡ d S rq )r» rÏ rJ r r¼ r ry ra rl rb r
|
r} rÞ r ra rr r| r| r Ú test_oneså s
|
ÿz TestMaskedArrayMethods.test_onesc C sF t d}dddddg}t|}t||d}t|d tk t|d tk ddg|ddg< t|d tk t|d tk t|ddd d
|
dg tt ddddddgd d}dd dd g}| |d dd dg¡ t|td dddd ddd d dg
|
t|j ddddddddddg
|
| |tdd dd gddddg¡ t
|
|ddd dddd d d dg
|
t|j ddddddddddg
|
tt ddddddgd d}t||d dd dg t|td dddd ddd d dg
|
t|j ddddddddddg
|
t||tdd dd gddddg t
|
|ddd dddd d d dg
|
t|j ddddddddddg
|
d S )NrÊ r r r r§ rÁ rÈ rS r¨ rÖ r r) rø r ) r rF r& r
|
rI r rJ rf r' r r )r r r6 r r r r| r| r Útest_putû s2 " " " " z TestMaskedArrayMethods.test_putc C s t d}tddgddgd}| ddg|¡ t|d tk t|d d t|d tk t|d d
|
t|d tk t|d
|
tk t|d
|
d d S ) NrÈ r r FTr r r¨ r r§ )ry r& rf r
|
rI r )r r r£ r| r| r Útest_put_nomask s z&TestMaskedArrayMethods.test_put_nomaskc C sl t d}dddddg}t|}t|d |ddd}| dddddgdddddg¡ t|jdddddg d S ) NrÊ r r TrR rÁ r¨ r§ )r rF r& rf r r² rZ r| r| r Útest_put_hardmask( s z(TestMaskedArrayMethods.test_put_hardmaskc C s t dd }t|ddddddgd}ddddddg}| ¡ }t||d t|ddddddg | ¡ }t||d t|jddddddg t|jddddddg td d
|
d d d dgddddddgd}| ¡ }t||| t|jddd dddg t|jddddddg | ¡ }t||| t|jddd dddg t|jddddddg | ¡ }| ¡ t||| t|ddd dddg d S )Nr r r r r r¨ rÁ rÊ rÈ rI rT rS rH r( )r r& rÝ rg r r² rÆ rV )r r r½ r rØ ZmxxÚvaluesr| r| r Ú test_putmask1 s. $ z#TestMaskedArrayMethods.test_putmaskc C s t dddddggdddddggd}| ¡ }t|jj|j t ddgddgd}| ¡ }t|jj|j t ddddgddddgdd }t| ¡ jddddg d
|
|_d |_| ¡ }t|jddddg t|jddddg t|jd
|
t|jd d ddddg t|jdd ddddg d S )Nr r¨ r§ rÁ rÊ r r Fr éÿÿÿr* r r r¢ )r& rh r rÆ rl r r² )r r ZaravelÚarr| r| r Ú
|
test_ravelO s $ z!TestMaskedArrayMethods.test_ravelr ZAKCFÚ
|
data_orderZCFc C s t jd|d}d|dd d
f< t jdt|dj}d|dd d
f< t||d}|jjj|jjjksdt |
|
d¡dk ¡ szt | |¡}|
|
d¡dk ¡ st d S )N)rÊ rÈ r r )rÈ rÊ )r~ r Fr ) r} ra rñ r r& r² r ZfncrÆ rú r7 r rh )r r r Úarrr r Zraveledr| r| r Útest_ravel_ordere s
|
z'TestMaskedArrayMethods.test_ravel_orderc C sT t d}t|d< | dd¡}t|jd t|jjd t|jd t|jjd d S )NrÁ r r¨ r* ræ )r rI rj r rl rÆ r§ r| r| r Ú test_reshapeu s z#TestMaskedArrayMethods.test_reshapec C sþ t ddddgddddgtjd}t|}t|jddddg t|jddddg t|dd}t|jddddg t|jddddg | ¡ t|jddddg t|jddddg t ddddgddddgtjd}|jdd t|jddddg t|jddddg ddddg}t|}ttt t
|
t ddd d
|
dgt tj d}t|dd}t|jd
|
d dddg t ddd d
|
dgdddddgtj d}t|dd}t|jddd
|
d dg t|jdddddg t dd gtj d }t|d d }t|t d dgtj d d S )Nr rÁ r¨ r§ r rj Fr rÖ éþÿÿÿrÉ Ústable©Úkind) r& r} rë rp r r² rÆ r
|
rc Úsortedr r_ rM )r r Úsortedxr| r| r Ú test_sort s6 $ z TestMaskedArrayMethods.test_sortc C sH t ddddddgtjd}t ddddddg}t|dd }t|| d S )
|
Nr r¨ r§ rÉ r rÁ rÊ r r
)r& r} rë r% r )r r r Zcomputedr| r| r Útest_stable_sort¤ s z'TestMaskedArrayMethods.test_stable_sortc C s t ddddgddddgtjd}t tddtd dtdd
|
tddd tdd d fD ]<}t|f|}|t|f| }t|j|j t|j|j qTd S ) Nr rÁ r¨ r§ r rj Tr Fr )r r ) r& r} rë Údictrp r% r r² rÆ )r r Úkwargsr Z
|
argsortedxr| r| r Útest_argsort_matches_sortª s
|
|
û z0TestMaskedArrayMethods.test_argsort_matches_sortc C sd t dddgdddgg}| d¡ t|dddgdddgg t dddgdddgg}| d¡ t|dddgdddgg t dddgdddggdddgdddggd}| d¡ t|dddgdddgg t|jdddgdddgg t dddgdddggdddgdddggd}| d¡ t|dddgdddgg t|jdddgdddgg t dddgdd d
|
gddd ggddd gdddgdd d
|
ggdddgddd gdd d
|
ggdd d
|
gddd gdddggg}t||d dk< | ¡ }| d ¡}| d¡ | d¡ t|| | ¡ }| d ¡}| d¡ | d¡ t|| | ¡ }| d ¡}| d¡ | d¡ t|| d S ) Nrø rÁ r r¨ r r r r) rÊ r r§ r )rJ rp r rÆ rI rÝ r7 )r r ÚamZanr| r| r Ú test_sort_2d¸ sF
|
|
,
|
,
|
ý
|
|
|
|
|
|
|
|
|
|
|
z#TestMaskedArrayMethods.test_sort_2dc C s t dddddddgdd dddddgd
|
tfd tfgd }t dddddddgdddddd dgd
|
tfd tfgd }t dddddddgdddddd dgd
|
tfd tfgd }t|}t|| t|j|j t|d d}t|| t|j|j t dtdfg¡}| |¡}t|}t|d d}d S )Nr r% r* )r¨ r ru rJ rµ rh ri rô r r® Fr r r¨ )r& r» rp r r r} r~ ra )r r Z mask_lastZ
|
mask_firstrÑ r r| r| r Útest_sort_flexibleà s2 ýýý
|
|
|
z)TestMaskedArrayMethods.test_sort_flexiblec C s8 t dddddgdddddgd}tt |¡t| d S )Nr rÊ r¨ rÁ r§ r r )r& r r} r% rN r| r| r Ú test_argsort s z#TestMaskedArrayMethods.test_argsortc C sî t dddgg}t| ¡ dddg t dddggdddggd}t| ¡ dddg t| ¡ jdddg t dgg¡}| ¡ }t|d d|d< t|d d t dggdd}| ¡ }t|tjjk t|j d d|d< t|d d d S )Nr r¨ r§ r .rh T)
|
rJ r ÚsqueezerÆ r} r& r
|
rà rI r )r rÐ r Zarr_sqÚm_arrZm_arr_sqr| r| r Ú test_squeeze s
|
z#TestMaskedArrayMethods.test_squeezec & C s
|
t ddddddddd d
|
d d d ddddddddddddddd d d d d d!d"d#d$g$¡}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' d(d(¡}| d)d*d*d)¡}| d%d&¡}t|d+ |d d
d+f | d%d*¡}t|jd, d S )-Nr( 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 r r r r r§ r¨ rÖ )r¨ r¨ r§ r§ )r} r& rj Úswapaxesr rl )r r r rQ rR Z mXswappedZ
|
mXXswappedr| r| r Ú test_swapaxes s û û z$TestMaskedArrayMethods.test_swapaxesc C sf t ddddgddddg}t| dddg¡t dddgdddg t| dddg¡|dddg t| ddgddgg¡t ddgddggddgddgg t|d tjjk t| d¡tjjk tdddgddd ggdddgdddggd
|
}t|jdd gdd tddgdd ggddgddggd
|
tt|dd gdd tddgdd ggddgddggd
|
d S ) NrÈ rI rT rS r r r§ rH r( r r¨ r3 )rJ r rt r
|
r} rà rI r& r³ r| r| r Ú test_take3 s ( ÿ,"ÿ"ÿz TestMaskedArrayMethods.test_takec C s> t d¡}t d¡d d d
f t d¡d d
d f }t||t|kd}t||dd}tddd gdd d
|
gd d
|
d gd
|
d d gd d d gg}t|| t||}tddd gdd d
|
gd d
|
d gd
|
d dgd ddgg}t |d < |d < |d< t|| t|j|j tddd}t||}t |d< |d< t|| t|j|j d S )N)rS rÿ é% r rl r§ rÊ r rh ©ÚmoderS rÿ r r rl r% )rÁ r ©rÁ r¨ )r r r r r ri ru )r} r& r r¼ rt r rI r )r r ÚindicesZmindicesrÑ Úctrlr| r| r Útest_take_masked_indicesE s4
|
,
|
ü
|
|
|
ü
|
|
|
z/TestMaskedArrayMethods.test_take_masked_indicesc C st t t d¡}t|ddg< | ¡ }t|d d k t|d d k d|_| ¡ }dd ddgdd d
|
d gd d d dgg}t|d dd ddg t|d dd d
|
d g t|d d d d dg t|| t tt dddgdddgdddgdt
|
fdt fdgd}t|d< t| ¡ ddd g t d dgd d gdt
|
fdt
|
fgd }| ¡ }t|dd gddgg |d }| ¡ }t|dd g d S )!Nrt r r rÅ r r¨ r§ rÁ rÊ r r) rø r é r² r³ rb r` ra rc r r® rg rÉ rÖ )r r² s one)r¨ r³ ó two)NNNrµ ri rh rj ) r& r} r rI Útolistr
|
rl r rË rÌ r» rÏ )r r Zxlistr r rÑ r| r| r Ú test_tolistc sB "
|
þýþÿÿz"TestMaskedArrayMethods.test_tolistc C s~ t ddgdtfdtfgd}|D ]$}| ¡ D ]}tt|tj q,q d|jd< |D ]$}| ¡ D ]}tt|tj q`qTd S )Nri rÎ r r® rÉ r )r& r» r¡ r
|
rc r} Zgenericr )r r Úentryry r| r| r Útest_tolist_specialcase s
|
z.TestMaskedArrayMethods.test_tolist_specialcasec C sX t d}| ¡ }t|d |j t|d |j t|ddddg< | ¡ }t|d |j t|d |j dtfd d
|
tfg}td d t t
|
d¡d t
|
j d¡D |d}t|ddddg< | ¡ }t|d |j t|d |j t
|
d¡}tdd t t
|
d¡t
|
j d¡t
|
j d¡D |d}t|ddddg< | ¡ }t|d |j t|d |j d S )NrÈ r² rÆ r r r¨ rÖ r r r c S s g | ]\}}}|||fqS r| r| )r r r¾ r r| r| r r ¢ s z6TestMaskedArrayMethods.test_toflex.<locals>.<listcomp>Z ABCDEFGHIJKLMrÉ r c S s g | ]\}}}|||fqS r| r| )r r r Úffr| r| r r ¬ s )r Útoflexr r² rÆ rI r» rÏ r& rÌ r} rÍ rÎ r~ r )r rÐ r r r| r| r Ú test_toflex s<
|
þý
|
|
|
þýz"TestMaskedArrayMethods.test_toflexc C s¼ t dddg}t| ¡ }t|| t|j|j t dddgdddgd}t| ¡ }t|| t|j|j t dddgd d
|
d gd tfd tfgd}t| ¡ }t|| t|j|j d S )Nr r¨ r§ r r rf rg )r§ r ru rh ri rô r rj )r& r: r¦ r r r» rÏ rÐ r r| r| r Ú test_fromflexµ s
|
|
ÿ
|
z$TestMaskedArrayMethods.test_fromflexc C s t dddddggdddddgd}t dgdgdgdgdggdddddgd}t|j| t| ¡ | tt |jd¡| d¡ d S )Nr r¨ r§ rÁ rÊ r r )rJ r r rw r Úcumsum)r Zmarrayr r| r| r Útest_arraymethodÇ s " ÿ z'TestMaskedArrayMethods.test_arraymethodc C s4 t jjddd}t|jj|j t|jj|j d S )Né* Tr )r} rà r& r r r rÐ r³ r| r| r Útest_arraymethod_0dÑ s z*TestMaskedArrayMethods.test_arraymethod_0dc C sf t j dddgdddgg¡}t jj|d< |j}d|d < t jj|d< t|j|jj t|j|jj d S )
|
Nr r¨ r§ rÁ rÊ r ri rÈ ru )r} rà r& rI r r rÐ r )r r Zxtr| r| r Útest_transpose_view× s z*TestMaskedArrayMethods.test_transpose_viewc C s` t j d¡}d|d< t jj|d< d|d< | ¡ }d|d< t|j| ¡ j t|j| ¡ j d S )Nr rÈ rh rJ rI r* rÛ )r} rà ry rI rY r r rÐ )r r Zxdr| r| r Útest_diagonal_viewâ s z)TestMaskedArrayMethods.test_diagonal_viewN).rê rë rì r¦ r[ r] r^ r` rc rg ri rj rn rp rt ru rv rÄ rw rx ry r{ r~ rá rD rE 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& ã sT
|
|
%( $
|
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 Zed d Zd!d" Zd#d$ Zd%d& Zed'd(d)d* Zd+S ),ÚTestMaskedArrayMathMethodsc & 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*}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 r' rM rN r| r| r r¦ ï sÒ û û û z'TestMaskedArrayMathMethods.setup_methodc
|
C s¦ | j \
|
}}}}}}}}} }
|
| d¡} t| j| d¡ d¡ | d¡} t| j| d¡ d¡ | d¡} t| j| d¡ d¡ | d¡} t| j| d¡ d¡ d S ©Nr r )r r© r r² r7 Úcumprod) r r rO rP r r½ rQ rR rS rT rU ZmXcpr| r| r Útest_cumsumprod s
|
|
|
|
z*TestMaskedArrayMathMethods.test_cumsumprodc C sÆ t tj ddd¡ dd¡}t |d d
df< |d< |d< dD ]}tt|}t||}tjdtd }| d
|
¡ ||d|d }t
|
||k t ||d|d tdt d }|d|d }t
|
||k q@d S ) Nr rÈ rt r§ rÁ r )r© r± rÅ rÉ rÿ rç r )r r¡ rî rï r r rñ r| r| r Útest_cumsumprod_with_output s
|
|
|
z6TestMaskedArrayMathMethods.test_cumsumprod_with_outputc
|
C s | j \
|
}}}}}}}}} }
|
|j\} }t| ¡ | ¡ ¡ t | t¡} t |t¡} t|D ] }|d d
|f ¡ ¡ | |< qZt| D ]}|| ¡ ¡ | |< qt| d¡| t| d¡| d S r° ) r rl r rô ro r} ry rÏ rW )r r rO rP r r½ rQ rR rS rT rU r6 ÚrowsÚcolsÚkr| r| r Útest_ptp1 s
|
z#TestMaskedArrayMathMethods.test_ptpc C s> t ddgddgtd}|d }t|d d t|jd d S )Nr r® r r rj r Zbx)rJ r r r
|
r r§ r| r| r Útest_add_object? s z*TestMaskedArrayMathMethods.test_add_objectc C s` t dddgdddgtd}t| ¡ d t dddgdddggtd }t|jdd
|
dd d g d S ) Nr r¨ r§ r rj rÊ rÁ r rÉ r3 r) r )rJ r r rs rN r| r| r Útest_sum_objectE s z*TestMaskedArrayMathMethods.test_sum_objectc C s` t dddgdddgtd}t| ¡ d t dddgdddggtd }t|jdd
|
dd d g d S ) Nr r¨ r§ r rj r rÁ rÊ rÉ r3 rÈ rÿ )rJ r r r rN r| r| r Útest_prod_objectL s z+TestMaskedArrayMathMethods.test_prod_objectc C s8 t dddgtd}t| ¡ d t| ¡ dddg d S )Nr r¨ r§ rÉ rÖ r )rJ r r r r rN r| r| r Útest_meananom_objectS s z/TestMaskedArrayMathMethods.test_meananom_objectc C sP t dddg}t| ¡ j|j d|_t| ¡ j|j ttj | ¡ ¡ d S )Nr r¨ r§ T) rJ r r rl r r
|
r} rà Z is_maskedrN r| r| r Útest_anom_shapeY s
|
z*TestMaskedArrayMathMethods.test_anom_shapec C s t t dd¡ dd¡}t| ¡ dddgdd d
|
gg t|jd d dddgd d d gg t|jdd d ddgd ddgg d d dgd dd gg|_d}t| ¡ |¡dd|gd|dgg t|jd d |¡dd|gd |dgg t|jdd |¡dd|gd |dgg d S )Nr r) r¨ r§ g Àg ø¿r r r rÌ r r3 r r r
r| g Àg ô¿g è?g @)rJ r} r rj r r r r7 )r r Zmvalr| r| r Ú test_anom` s* ÿ ÿ ÿÿÿÿz$TestMaskedArrayMathMethods.test_anomc
|
C s¼ | j \
|
}}}}}}}}} }
|
| ¡ } t| ¡ | ¡ ¡ ¡ t| ¡ | ¡ t| j| ¡ dd tt |¡| ¡ t d¡
|
ddd¡} tj | d¡} t| jddd| jddd d S ) Nr r3 é r¨ rÁ Fr )Zaxis1Zaxis2) r rY r Útracero rs r r r} r rj rà rJ )r r rO rP r r½ rQ rR rS rT rU ZmXdiagr r r| r| r Ú
|
test_traceq s ÿÿz%TestMaskedArrayMathMethods.test_tracec
|
C s | j \
|
}}}}}}}}} }
|
| d¡} | |¡} t| d¡| | ¡ t| jtk | d¡} | |¡} t| d¡| | ¡ t| jd t| }|j||d t| | | dd¡}| d¡| d¡ }}| |¡} t| d¡| |¡ t| }|j||d t| | d S )Nr rÔ ra rÖ r ) r r7 Údotr r
|
r r_ r3 r )r r rO rP r r½ rQ rR rS rT rU ZfxÚrZfXr* ZmYYZfXXZfYYr| r| r Útest_dot s&
|
|
|
|
|
|
z#TestMaskedArrayMathMethods.test_dotc C s¢ t ddgddggddgddggd}t ddgddggddgddggd}t ddgddgg}|j||d t| d¡ddgdd gg t|jddgddgg d S )
|
Nr r¨ r§ rÁ r r ra rÛ é )rJ rÁ r r7 r rÏ r| r| r Útest_dot_shape_mismatch s $$ z2TestMaskedArrayMathMethods.test_dot_shape_mismatchc C sî t ddddgdd}t ddddgdd}tt| ¡ tj tt| ¡ tj | ¡ | ¡ kt d¡ksntt t d¡
|
d ¡dd}t dd
|
d}tt|jdd t tt|jdd t t |jd|d |k t |jd|d |k d S ) Nr r¨ r§ rÁ rk rÉ TrÄ r rj r3 rç )r& r r" r r} rù r r_ rú r rj r2 r r
|
)r ZfooÚbarr| r| r Útest_varmean_nomask¡ s z.TestMaskedArrayMathMethods.test_varmean_nomaskc
|
C s | j \
|
}}}}}}}}} }
|
t|jd d| ¡ ¡ t|jd d| ¡ ¡ t|jd dd| ¡ jdd t|jd dd| ¡ jdd t|jddj|jddj t| ¡ j| ¡ j |jdd|jdd } } t|jd dd| ¡ jdd t|jd dd| ¡ jdd tdD ]j} t| | || ¡ ¡ t| | |d d
| f ¡ ¡ tt | | ¡|d d
| f ¡ ¡ q$d S ) Nr3 r ©r4 Úddof©rÉ r§ r r¨ r ©
|
r r r ro r r rl rW r} rq ©r r rO rP r r½ rQ rR rS rT rU ZmXvar0ZmXvar1r¶ r| r| r Ú test_varstd± s0 ÿÿ ÿÿ "ÿz&TestMaskedArrayMathMethods.test_varstdc C sN t jdtd}tdtd}ttddd}dD ]v}t||}t| tk t|dtk t|dtk ||d t|tk t|jd ||d tt |¡ q.ttddd}d |d< dD ]}t||}t|d
|
d tk t|dd
|
d tk t|dd
|
d tk ||d
|
d t|tk t|jd ||d
|
d tt |¡ qÂd S ) NrÖ rÉ rÈ Tr )r r r ra r r rÊ )rè rÉ )
|
r} r& rÏ r rë r
|
rI r r rq )r rð Zmoutr Ú
|
methodnameÚmethodr| r| r Útest_varstd_specialcasesÇ s2
|
|
|
|
z3TestMaskedArrayMathMethods.test_varstd_specialcasesc C sÒ t dddgdddggdddgdddggd}|jddd}t| d¡dddg t|jdddg |jddd}t| d¡dddg t|jdddg |jddd}t| d¡dddg t|jdddg d S )Nr r r rÈ r¨ )r& r r r7 r r r| r| r Útest_varstd_ddofê s ,z+TestMaskedArrayMathMethods.test_varstd_ddofc C s¬ t d d¡}t|d< t |¡}t|dddg t|}t|dddg t|jdddg t|}tdddgdddgdddggdddgdddgdddggd}t|| d S ) Nr r rJ r rÁ rø r r )r rj rI r} r/ r r r& )r r rè r r| r| r Ú test_diagö s
|
ÿz$TestMaskedArrayMathMethods.test_diagc C s t dddgdddgg}t| d¡ddd g t| d
|
¡dd g t| d¡dd g t| d¡dd d g t| d
|
¡ddg t| d¡ddg t| d¡dddg t| d
|
¡ddg t| d¡ddg t| d¡dddg t| d
|
¡ddg t| d¡ddg d S )Nr r¨ r§ rÁ rÊ r r r) r rÖ rÛ rÈ rÿ éx )r& r rs r rY rV rN r| r| r Útest_axis_methods_nomask s z3TestMaskedArrayMathMethods.test_axis_methods_nomaski Zb)Z
|
free_bytesc C s2 t tjddtjdt d¡d}t| ¡ d d S )N)é' rÕ iÿÿ rÉ r g àÿï@)rJ r} r Zuint16ry r r rN r| r| r Útest_mean_overflow s ÿz-TestMaskedArrayMathMethods.test_mean_overflowN)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¯ í s,
|
" r¯ c @ s e Zd Zdd Zdd ZdS )Ú!TestMaskedArrayMathMethodsComplexc & 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*}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 )+Ny À @y ®Gáz. @y -²ï§!@y !@y é&1¬ ü?r- r. r/ r0 r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> y Ñ"Ûù~ê@y ¨ÆK7Á @rA rB rC rD rE rF rG rH rI rJ y -²ï§Æï?r r§ r¨ r r rL rM rN r| r| r r¦ " sÒ û û û z.TestMaskedArrayMathMethodsComplex.setup_methodc
|
C sR | j \
|
}}}}}}}}} }
|
t|jd d| ¡ ¡ t|jd d| ¡ ¡ t|jddj|jddj t| ¡ j| ¡ j |jdd|jdd } } t|jd dd| ¡ jdd t|jd dd| ¡ jdd tdD ]h} t| | || ¡ ¡ t| | |d d
| f ¡ ¡ tt | | ¡|d d
| f ¡ ¡ qäd S ) Nr3 r§ r r r¨ rÈ rÊ r rË rÌ r| r| r rÍ B s$ ÿÿ "ÿz-TestMaskedArrayMathMethodsComplex.test_varstdN)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 dd Z dd Zdd Zdd Zd d Zd d Zejjed!d"d#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Z d/d0 Z d1d2 Z d3d4 Z d5d6 Z ej !d7d8d9g¡d:d; Z"d<d= Z#d>d? Z$d@dA Z%dBdC Z&dDdE Z'dFdG Z(dHdI Z)dJdK Z*dLdM Z+dNdO Z,dPdQ Z-dRS )SÚTestMaskedArrayFunctionsc C s° t ddddtd dddddddg ¡}t dd ddd
|
d d dddd dg ¡}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}| d¡ ||f| _d S )Nr
r r r r r r r r r r r r r r )r} r& r rJ r r¥ )r r r r r r¡ r¢ r| r| r r¦ X s &"
|
z%TestMaskedArrayFunctions.setup_methodc C s2 ddg}t d|}t|ddg t|d d d S )Nr r¨ F)rU r r§ r| r| r Útest_masked_where_boolb s
|
z/TestMaskedArrayFunctions.test_masked_where_boolc C sZ dddg}t |d}t|| t|jdddg t|d}t|| t|jdddg d S ©Nr r¨ r§ r )rK r rÆ rQ rN r| r| r Útest_masked_equal_wlisth s
|
|
|
|
|
z0TestMaskedArrayFunctions.test_masked_equal_wlistc C s6 dddg}t |d}t|jdddg t|jd d S rÚ )rK r rÆ r rN r| r| r Ú test_masked_equal_fill_valueq s
|
|
z5TestMaskedArrayFunctions.test_masked_equal_fill_valuec C s t dddddg}t|d< ttt|d|t|d ttt|d|t|d ttt|d|t |d ttt
|
|d|t |d ttt |d|t |d ttt|d|t|d ttt |d|t |d ttdddddgddd d
|
d gd d d d
|
d g d S ) Nr
r r r r r¨ r r r§ rÁ rÊ r )r& rI r rU r= rL r> rM rB rO rC rP r` rQ r4 rK r³ r| r| r Útest_masked_where_conditionw s ÿ ÿ ÿz4TestMaskedArrayFunctions.test_masked_where_conditionc C s0 t dtd}t|jt}t||}t|| d S )N)rÈ rÈ rÈ rÉ )ra rÏ ry rl r rU r )r r r Zctestr| r| r Útest_masked_where_oddities s
|
z3TestMaskedArrayFunctions.test_masked_where_odditiesc C sT t d}tt td| W 5 Q R X t|d}t|jddddddddddg
|
d S )NrÈ r r )r r Ú
|
IndexErrorrK r r r r| r| r Ú"test_masked_where_shape_constraint s
|
|
|
z;TestMaskedArrayFunctions.test_masked_where_shape_constraintc C s| t jdddgd}t jdd t j |d dk |¡}W 5 Q R X t|jjj|jj t|d t j t d¡t
|
d¡¡ d S ) NrÈ )rô z<f2)r z<f4rÉ r )Zoverrô rÊ ) r} ry r rà rU r r r~ rù rJ ra )r r r r| r| r Ú test_masked_where_structured s ÿz5TestMaskedArrayFunctions.test_masked_where_structuredc C s. t d¡}t d¡}ttt jj|dk| d S )NrÈ rÊ r )r} r r rß rà rU r§ r| r| r Útest_masked_where_mismatch¢ s
|
|
z3TestMaskedArrayFunctions.test_masked_where_mismatchc C s* t tttddddddddg t tttddddddddg t ttttddddddgdddjdddddg t ttttddddddgdddjdddddg t ttttddddddgddjdddddg t ttdddddgdddddgddjdddddg d S ) NrÊ r r§ r éÇ rÁ r¨ r ) r rN rË rW rR r& r rK rQ rä r| r| r Útest_masked_otherfunctions¨ sD ÿ$ ÿ ÿ þ ÿ ÿ þ ÿÿ þ ÿÿ þz3TestMaskedArrayFunctions.test_masked_otherfunctionsc C s@ t dddddgdddddgd}t| ¡ d d
|
d d d g t| d¡dddddg t| d¡dddddg t|}|j|d t|d d
|
d d d g t d d
|
d dd g}t dddddg}t|d< t||| }t|d d
|
d d d g t|d< t||| }t|d d
|
d d d g t|d tk t|d tk t|d tk d S ) Ng82üÁÀó?g¤ª ¢îÃ@g,¼ËE|§ @gÚæÆôE@gÅ1·@r r r r
r r r r¬ g333333ó?gffffff@rk gffffff@gÍÌÌÌÌÌ@r§ gÃõ(\Âó?gøSã¥Ä@g-²ï§ @gÀÊ¡E@gï§ÆK·@ra r r¨ r r r| )r& r ré r3 rI rx r
|
)r r r® r rz r£ r| r| r Ú
|
test_round¹ s( ÿ z#TestMaskedArrayFunctions.test_roundc C s® t tj ddd¡ dd¡}t |d d
df< |d< |d< tjdtd}| d ¡ tj |d
|
|d }t
|
||k t ||j d
|
|d tdtd}|j d
|
|d }t
|
||k d S ) Nr rÈ rt r§ rÁ r rÅ rÉ rÿ r¨ )Zdecimalsrè ) r& r} rÍ rê rj rI r2 rÏ rí ré r
|
r )r r¡ r rñ r| r| r Útest_round_with_outputÏ s
|
z/TestMaskedArrayFunctions.test_round_with_outputc C sè t ddgd}t| ¡ d t ddgd}t| ¡ tk t ddgd}tjdtd}| d¡ |j|d t|d t ddgd}t d dgd}|j|d t|d
|
d t ddgd}t d dgd}|j|d t|d
|
tk d S ) Nr² Fr r TrÉ rÿ ra g ÃÀr| ) r& r ré r
|
rI r} r2 rÏ rí )r r r r| r| r Útest_round_with_scalará s"
|
|
z/TestMaskedArrayFunctions.test_round_with_scalarc C s* t d}tt|t t|t d¡ d S ©NrÊ )r? r
|
rc r r r} rN r| r| r Ú test_identityú s z&TestMaskedArrayFunctions.test_identityc C s d}t t|dd tt|ttk tdddddg}tdddddgddddd gd
|
}t||}t |ddd ddg t|jd dddd g t|_t||}t|jd dddd g || }t|j|j t || t |j |j ||C }t|j|j t || t |j |j d S ) Ngñ¿r g\Âõ(\ó?r² r r r r r r gKTjÉëÇð?)
|
r rd r
|
rI r& r rÆ r_ r r² )r r r® r r£ r| r| r Ú
|
test_powerÿ s&
|
|
|
|
z#TestMaskedArrayFunctions.test_powerc C sl t dddgdddgg¡}t|dddgdddggd }t d
|
d d g¡}t ||g¡}t|dddgdddggd }tdd dgd ddggdddgdddggd }|| }t|| t|j|j || }t|| t|j|j || }t|| t|j|j td ddgd ddggdddgdddggd }|| }t|| t|j|j || }t|| t|j|j d S )Nr
r r r r r¬ r r r r¨ rÁ r§ rÄ é iq éØ é )r} r& r r )r Za2Za2mÚb1Zb2Zb2mr rÑ r| r| r Ú test_power_with_broadcasting s4 ÿ
|
|
|
ÿ
|
|
z5TestMaskedArrayFunctions.test_power_with_broadcastingz fp errors don't work in wasm©Úreasonc
|
C s t ddddtd dddddddg ¡}t dd ddd
|
d d dddd dg ¡}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}| d¡ t|dk|d}t|ddddddddddddg t|j|j t|dkd|}t|dd ddd
|
dddddd dg t|jd d d d d d d d d d d d g t|dk|t}t|ddddddddddddg |j ¡ }d||dk
|
d¡< t|j| t j dd* t j tdd | t¡} W 5 Q R X W 5 Q R X t| dk| t}t|ddddddddddddg t|j| j d S )Nr
r r r r r r r r r r r r r r r¨ i÷ÿÿÿg "ÀTr r z invalid valuerî rÁ rÈ r§ )r} r& r rJ r rx r rÆ rI rÝ r7 r rá râ ÚRuntimeWarningrº r» r~ )
|
r r r r r r¡ r¢ r ÚtmpZixmr| r| r Ú
|
test_where4 sZ &"
|
ÿ ÿ$ ÿ
|
"z#TestMaskedArrayFunctions.test_wherec C sJ t d ¡}td }| ¡ }tt j d||¡| tt j d||¡| d S )NT)r} r& rJ rÝ r rà rx )r r r® r r| r| r Útest_where_objectV s
|
|
z*TestMaskedArrayFunctions.test_where_objectc C sþ t d}t|d< |dk}t||t}t|j|jk t|d tk t|d tk t|d tk t|d tk t|d tk t|| t|t|}t|j|jk t|d tk t|d tk t|d tk t|d tk t|d tk d S )NrÈ r§ rø rÁ r) r )r rI rx r
|
r~ r r} r| r| r Ú test_where_with_masked_choice] s$
|
z6TestMaskedArrayFunctions.test_where_with_masked_choicec C s t dddddg}t dddddg}t|d< t||| }t|ddd d
|
d g t|d< t||| }t|ddd d
|
d g t|d tk t|d tk t|d tk tdd }t|d < tdd d }t|d< t dddddgdddddgd}| d¡}t|||}t|||}t|| tt|tk t|dddddg t|td}t|dddddg t|dt}t|dddddg d S )Nr
r r r r r r r¨ r r r| r rÖ rÈ r r§ rS rH r ) r& rI rx r r
|
r r7 r; r_ )r r rz r£ r Úcmr¤ r| r| r Ú test_where_with_masked_conditions s4
|
|
|
z9TestMaskedArrayFunctions.test_where_with_masked_conditionc C sX t jdt jd}t jdt jdd }t|dk||j}t t jt jgg ¡}t|| d S )NrÁ rÉ r³ r )r} r r® r. rx r~ Zfind_common_typer )r r r rÑ r r| r| r Útest_where_type s
|
z(TestMaskedArrayFunctions.test_where_typec C s^ t d¡ dd¡}t d¡}t dddg||¡}tdddg||}t|| t|j|j d S )Nr r§ r r )r} r rj ry rx r r~ )r r r rì rà r| r| r Útest_where_broadcast s
|
|
z-TestMaskedArrayFunctions.test_where_broadcastc C s| t dtfdtfg¡}t jdddg|d}t jd|d}t dd d g||¡}t dd d g||¡}t|| t|j|j d S )
|
Nr r® rµ rÅ r rÉ )rÈ rI r r )r} r~ r» r& rx r )r r r r rì rà r| r| r Útest_where_structured¢ s
|
z.TestMaskedArrayFunctions.test_where_structuredc C sv t dtfdtfg¡}t jdddg|d}tdddg|t}tdddg|}t|j|j t|| t|j|j d S ) Nr r® rµ rÅ r rÉ r r ) r} r~ r» r& rx rI rU r r )r r r rà r r| r| r Ú test_where_structured_masked s
|
z5TestMaskedArrayFunctions.test_where_structured_maskedc C sL t jdtd}t j|d< t j|d< tjtdd t j |¡ W 5 Q R X d S )NrÊ rÉ r§ r¨ z!not supported for the input typesrî )
|
r} r r ZPINFÚNaNrá rð rý rà Úmasked_invalidrN r| r| r Útest_masked_invalid_error¸ s
|
|
ÿz2TestMaskedArrayFunctions.test_masked_invalid_errorc C sF G dd d}t j | ¡}t|jt | ¡ t|jdddg d S )Nc @ s e Zd ZdZdd ZdS )zCTestMaskedArrayFunctions.test_masked_invalid_pandas.<locals>.SeriesZnonsensec S s t dt jt jg¡S rè )r} r& r r- rä r| r| r Ú __array__Ç s zMTestMaskedArrayFunctions.test_masked_invalid_pandas.<locals>.Series.__array__N)rê rë rì r² r r| r| r| r ÚSeriesÄ s r FT)r} rà rþ r r² r& rÆ )r r r r| r| r Útest_masked_invalid_pandasÀ s z3TestMaskedArrayFunctions.test_masked_invalid_pandasrÝ TFc C sj t j ddddg¡}|jtks"tt jj||d}|jtk s@t|jtksNtt |j |j ¡|ksftd S )Nr r¨ r§ rÁ r2 )
|
r} rà r& rÆ r_ rú rþ r rÞ r² )r rÝ r rò r| r| r Ú test_masked_invalid_full_maskÎ s z6TestMaskedArrayFunctions.test_masked_invalid_full_maskc C s| ddddgddddgd d
|
d d gd dddgg}t ddddg|}t|td dddg t ddddg|dd}t|td dddg t ddddg|dd}t|td dddg tddddgddddgd}t ||dd}t|tddddg t|jddddg t|ddddgddddgddddgddddggd}ddddg}t ||dd}t|td dddg t|jddddg d S )Nr r r¨ r§ rÈ r rt rÆ rI é rl é rT rf r¾ é! rÁ rh r Úwrapr r )r( r r& r )r ÚchoicesÚchosenÚindices_r| r| r Ú test_chooseÚ s,
|
|
ÿ
|
|
ÿ z$TestMaskedArrayFunctions.test_choosec C sN ddddgddddgd d
|
d d gd dddgg}t dtd}tddddg||d}t|td dddg t||k t dtd}tddddgddddgd}t||d|d}t|tddddg t|jddddg t|ddddgddddgddddgddddggd}ddddg}t dtd t¡}t||d|d}t|tddddg d S )Nr r r¨ r§ rÈ r rt rÆ rI r rl r rT rf r¾ r rÁ rÉ ra r r )r rè r r¿ ) r2 r» r( r r& r
|
r ra r )r r rb r r
|
r| r| r Útest_choose_with_outñ s*
|
|
ÿ
|
|
ÿ z-TestMaskedArrayFunctions.test_choose_with_outc C sô t d}t|d< | d¡}t|jd t|jd | dd¡}t|jd t|jd |jddd}t|jd t|jd |jdddd}t|jd t|jd t |d ¡}tt|t t|jd t|d
|
tk t|jd d S ) NrÈ r )rÊ r¨ r rÊ r¨ r¢ r )r¨ rÊ rh )
|
r rI rj r rl r
|
r r} rc r )r r r® rz r| r| r r s&
|
z%TestMaskedArrayFunctions.test_reshapec C sÚ dt fdt fg}t|}t|dtfdtfg t|t|k t df}t|}t|tdf t|t|k t }t|}t|t t¡ t|t|k dt fddt fdt fgfg}t|}t ddddgfg¡}t|| t|t|k dt dffg}t|}t|t dtdffg¡ t|t|k d t fg}t|}t|t d tfg¡ t|t|k t dtd
|
fg¡}t|}t dtfd|fg¡}t|}t|t dtfddtd
|
fgfg¡ t|jd d |k d S ) Nr r® r¨ r° r± )r rî )r° rî )r± rî )rô r r§ r ) rÏ rG r rñ r
|
r} r~ r» r )r ÚntyperÑ r Z base_typeZ
|
base_mtypeZsub_typer| r| r Útest_make_mask_descr! s>
|
|
$z-TestMaskedArrayFunctions.test_make_mask_descrc C s ddg}t |}t|jt t|ddg tjddgtd}t |}t|jt t|ddg dtfdtfg}tjddg|d}t |}t|jt t|ddg dtfdtfg}tjddg|d}t ||jd}t|j| t|| dtfdtfg}dtfdtfg}tjddg|d}t ||jd}t|j| t|tjddg|d tjdd dd
|
}tt |tj
|
t ||jd}t|| t||k tjdd dd
|
}t ||jd}t|| d d g}ttg}d}t |||¡D ]4\} }
|
} t t | |
|
| d} t| t k|| |
|
| f qÞd S )Nr r rÉ r r® rh ri r` z?,?r| rÁ TFz copy=%s, shrink=%s, dtype=%s)rÝ r r~ )rF r r~ r r} r& rñ rÏ r
|
rc ZvoidÚ itertoolsre r_ ) r r rÑ ÚmdtypeZbdtypeZtest2ZboolsZdtypesZ msgformatÚcpyZshrr rò r| r| r Útest_make_maskO sL
|
|
|
z'TestMaskedArrayFunctions.test_make_maskc C sR dt fdt fg}tjddddg|d}t|t}t|| tt|}t|| t|d}t|| tjddddg|d}t||}tjddddg|d}t|| d t fd
|
t fg}tjddddg|d}zt||}W n tk
|
rê Y nX dt fdd t fd t fgfg}tjd d g|d}tjddg|d} tjdd g|d}
|
tt|| |
|
d S )Nr r® rh ri ru rÉ FrJ rô r r° r± r r ©r rh r )rñ r} r& rH r_ r rè ) r r r rÑ rÌ r Z othertyper~ rÇ ZbmaskZcntrlr| r| r Ú test_mask_or s.
|
|
|
|
|
|
|
|
z%TestMaskedArrayFunctions.test_mask_orc C sÊ t jdddgtd}tt|| t jddgdtfdtfgd}t|}t jddddgtd}t|| dtfddtfd tfgfg}d
|
d g}t j||d}t|}t jddddddgtd}t|| d S ) Nr r rÉ rh ri r r® r° r± r r )r} r& rñ r r6 )r r rÑ r r rÐ r| r| r Útest_flatten_mask s
|
z*TestMaskedArrayFunctions.test_flatten_maskc C sT t ddddg¡}t|dd}t|}t|| ¡ t|d}t|| dd¡ d S )Nr r¨ r§ rÁ Fr r* )r} r& r r rj )r r r rÑ r| r| r Útest_on_ndarray° s
|
z(TestMaskedArrayFunctions.test_on_ndarrayc C sx t d¡}d|_t ddddg¡}|dddg }t jj||dd}t|| t j |¡}t jj||dd}t|| d S ) Nrø r TFr r¨ r§ r3 )r} r rl r& rà rX r )r r Zcondr rÑ Zmarrr| r| r rn ¹ s
|
|
z&TestMaskedArrayFunctions.test_compressc C s t j ddg¡}t j |¡}tt|t jk G dd dt j}t j |dd¡}t j |¡}tt||k t j dgdgg¡}t|jd t j dggggg¡}t|jd G dd dt }t j |g gg gg¡}t|jd G d d dt }t j |g gg gg¡}t|d
|
d S ) Nr r¨ c @ s e Zd ZdS )z3TestMaskedArrayFunctions.test_compressed.<locals>.AN©rê rë rì r| r| r| r rô Î s rô r r« c @ s e Zd ZdS )ú3TestMaskedArrayFunctions.test_compressed.<locals>.MNr r| r| r| r r Ü s r c @ s e Zd Zdd ZdS )r c S s dS )Nr« r| rä r| r| r ro ä s z>TestMaskedArrayFunctions.test_compressed.<locals>.M.compressedN)rê rë rì ro r| r| r| r r ã s r« )
|
r} rà r& ro r
|
r" r r rª r )r r rÑ rô r r| r| r rp Æ s" z(TestMaskedArrayFunctions.test_compressedc C s t t d¡d}t ddg¡}tj ||¡}t|t ddddddgd tjj||dd }t|t dddd
|
ddgd tj ddgdddg¡}t|t ddddgd ddg}t ddddgd}tjj||dd }t|t dddddgd tjj||d d }t|t dddddgd d S ) NrÊ r¨ r r rÖ r) rÁ F)Zpropagate_maskr§ T)rK r} r r& rà Zconvolver )r r r® rÑ r| r| r Ú test_convolveê s z&TestMaskedArrayFunctions.test_convolveN).rê rë rì r¦ rÙ rÛ rÜ rÝ rÞ rà rá râ rä rå ræ rç ré rê rï rá rD Úskipifr rô rõ rö rø rù rú rû rü rÿ r rE r r r r r r r r r rn rp r r| r| r| r rØ U sP
|
|
!
|
|
.0 $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 )ÚTestMaskedFieldsc C s dddddg}dddd d
|
g}d d d ddg}dt fdtfdg}dtfdtfdtfg}dddddg}ttt|||||d}t||||d| _d S )Nr r¨ r§ rÁ rÊ r² r³ rb rd re r` ra ZthreeZfourZfiver r® rg rz r rj )Úbaser Úddtyper )r» rÏ rñ r& rË rÌ r rÐ )r ZilistÚflistÚslistr r r r r| r| r r¦ s z TestMaskedFields.setup_methodc C sÀ | j d }| j d }t|_t|jtj|j|d t|_t|jtj |j|d d|_t|jtj|j|d d|_t|jtj |j|d dddddg|_t|jtj
|
dd d
|
D |d d S ) Nr r rÉ FTr r c S s g | ]}|||fqS r| r| )r r r| r| r r s z;TestMaskedFields.test_set_records_masks.<locals>.<listcomp>)r r r r r ) rÐ r_ r r rÆ r} ry rl rI ra r& )r r r r| r| r Útest_set_records_masks
|
s
|
|
ÿÿz'TestMaskedFields.test_set_records_masksc C s | j d }|d |d |d }}}ttdf|d< t|jt t|jdddd d
|
g t|jt t|jtd d d dg t|jd t|jdddddg d S )Nr r r® rz r r r§ r¨ rÁ rÊ r³ rb rd re r ó pir s threeó fouró five©rÐ r r r~ r» r² rÏ ©r r Zbase_aZbase_bZbase_cr| r| r Útest_set_record_element s
|
ÿz(TestMaskedFields.test_set_record_elementc C s¢ | j d }|d |d |d }}}ttdf|d d
< t|jt t|jdddddg t|jt t|jtttd d
|
g t|jd t|jd d d d dg d S )Nr r r® rz r r§ rÁ rÊ rd re r r! r" r# r$ r% r| r| r Útest_set_record_slice- s
|
ÿz&TestMaskedFields.test_set_record_slicec C sP | j d }t|d< dD ]4}t|| jdddddg t|| j|j| qdS )zCheck record accessr r )r r® rz r N)rÐ rI r r r² )r r r6 r| r| r Útest_mask_element< s
|
|
z"TestMaskedFields.test_mask_elementc C sp dt fdtfg}td|d}tt|tjdddgddgd t|d d
< tt|tjdddgddgd d S ) Nr r® r§ rÉ rh )r ú|b1)r® r) rJ )r» rÏ r2 r r< r} r& rI )r r rÑ r| r| r Útest_getmaskarrayE s ÿÿ ÿÿz"TestMaskedFields.test_getmaskarrayc C s¶ t tt d¡tj d¡}t |¡}t|dtfdtfgd}d|jd< tjdgddg t d}|
|
t¡}t || ¡ t |j| |
|
td f¡}t || t |j| d
|
d ¡ d S ) NrÈ r r® rÉ ru r r é r¨ rÖ )rË rÌ r} r rÍ rÎ r& rÏ r rñ ra r rh rj )r ÚiteratorrÐ r Ú controlmaskrÑ r| r| r Ú test_viewQ s
|
|
|
|
zTestMaskedFields.test_viewc sª dt fdt fg}ttttj d¡t d¡|d tjttddddddddddg
|
ddddddddddg
|
dtfdtfgd _ fdd}|d |d |d d S )
|
Nr r® rÈ rÉ r r c s t t | t t | j j| t | j j| t t | df t t | df j j| df t | df j j| df d S )N.)r r" r^ r r² rÆ r )r rn r| r Ú _test_indexh s z2TestMaskedFields.test_getitem.<locals>._test_indexr )
|
rÏ r& rË rÌ r} rÍ rÎ r rñ r )r r r/ r| rn r Ú test_getitema s $ ÿþ z TestMaskedFields.test_getitemc C s` t dtfdtfg¡}t jjddg|d}d|d d< t|d t ddg¡ d |d d< t|d t dd g¡ t dtfdtfg¡}t jd
|
d g|d}t jj d |d}d |d d< t|j
|
| t jj d |d}d |d d< t|j
|
| t jd d g|d}t jj d |d}| ¡ d |d d< t|j
|
| t jj d |d}| ¡ d |d d< t|j
|
| d S )Nr r® )r
r )r r¨ rÉ r r r
r r` )TTr¸ r¨ r ) r} r~ rÏ r» rà r r r& rñ r, r rV )r r rà r r r r| r| r Ú test_setitemu s. z TestMaskedFields.test_setitemc C s: t jjddd}t j d¡}||d< t|jdddg d S )Nr Tr r§ r F)r} rà rJ r r r )r Zmask_0dr r| r| r Útest_setitem_scalar s z$TestMaskedFields.test_setitem_scalarc C s, | j d D ] }tt|t| j d q
|
d S )Nr r )rÐ r r¼ )r Zrecr| r| r Útest_element_len s z!TestMaskedFields.test_element_lenN)rê rë rì r¦ r r& r' r( r* r. r0 r1 r2 r3 r| r| r| r r þ s
|
r c @ s e Zd Zdd Zdd ZdS )ÚTestMaskedObjectArrayc C sl t j d d g¡}ttfD ]L}t d¡ |¡}t d¡ |¡}||d< ||d< t|d |k t|d |k tt|d t tt|d t t|d d |k t|d d |k t jj
|
|d< t|d |k tt|d t tt|d t t |d j d t|d d |k t |d j | t |d j d t |d d j | t |d d j d qd S ) Nr¨ r§ r r ©r .)r .r| T)r} rà r& rÏ r r rº r
|
rc r rI r r rÐ )r r r Za0Za1r| r| r r0 ¢ s, z"TestMaskedObjectArray.test_getitemc C sx t j d d g¡}t t jjgt¡d |d< t|jd t jjk t|d t jjk t jj|d< t|d t jjk d S )Nr5 r )r} rà r& rI r r
|
rÐ )r r r| r| r Útest_nested_ma¿ s z$TestMaskedObjectArray.test_nested_maN)rê rë rì r0 r6 r| r| r| r r4 s r4 c @ sD e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dS )ÚTestMaskedViewc C sr t tt d¡tj d¡}t |¡}t|dtfdtfgd}d|jd< tjdgddg t d}|||f| _
|
d S ) NrÈ r r® rÉ ru r r r+ ) rË rÌ r} r rÍ rÎ r& rÏ r rñ rÐ )r r, rÐ r r- r| r| r r¦ Ò s
|
|
zTestMaskedView.setup_methodc C sB | j \}}}| ¡ }tt|t t|j|j t|j|j d S rµ )rÐ ra r
|
rc r r r² rÆ ©r rÐ r r- rÑ r| r| r Útest_view_to_nothingÚ s
|
z#TestMaskedView.test_view_to_nothingc C sN | j \}}}| tj¡}tt|t t||j t || |j
|
¡ ¡ d S rµ ) rÐ ra r} r r
|
rc r r r² r r~ r r8 r| r| r Útest_view_to_typeá s
|
z TestMaskedView.test_view_to_typec C sB | j \}}}| t¡}tt|t t|| ¡ t|j| d S rµ ) rÐ ra rÏ r
|
rc r r rh r r8 r| r| r Útest_view_to_simple_dtypeè s
|
|
z(TestMaskedView.test_view_to_simple_dtypec C s | j \}}}| dtfdtfg¡}t|jjjd t|d |d t|d |d |d dtfdtfg¡}tt|t t|jjjd t|d |d d t|d |d d |d dtfdtfg¡}tt|t t|jjd t|d |d d t|d |d d d S )Nrô r )rô r r r® r rÖ )
|
rÐ ra rÏ r r r~ rù r
|
rc r r8 r| r| r Útest_view_to_flexible_dtypeð s z*TestMaskedView.test_view_to_flexible_dtypec C s² | j \}}}| tdf¡}tt|t t|| t|j| dd¡ |d tdf¡}tt|t t||d t|jd |d tdf¡}tt|t t||d d S )Nr¨ rÖ r ru ) rÐ ra rÏ r
|
rc r r r rj r8 r| r| r Útest_view_to_subdtype s
|
z$TestMaskedView.test_view_to_subdtypec C sL | j \}}}| tdftj¡}t|| tt|tj tt|t d S )Nr¨ ) rÐ ra rÏ r} rb r r
|
rc r r8 r| r| r Útest_view_to_dtype_and_type s
|
|
z*TestMaskedView.test_view_to_dtype_and_typeN)
|
rê rë rì r¦ r9 r: r; r< r= r> r| r| r| r r7 Ð s r7 c @ s e Zd Zdd Zdd ZdS )ÚTestOptionalArgsc C sà t d¡ d¡}t jdtd d¡}d|d d
d d
df< t jj||d}dd }d
|
d }d D ] }|||| |||| q^d D ]}|||| qt d¡ d¡d dk}t jj||d}dD ] }|||| |||| q¾d S )Nç 8@©r¨ r§ rÁ rÚ rÉ TrÖ r c S s~ t | ¡}tj | ¡}t||dddd d
f ||dd d
f dd t||dddd d
f ||dd d
f dd d S )Nr r3 .rÖ ri )r Ú__getattribute__r} rà r ©r r r Znumpy_fZma_fr| r| r Útestaxis( s
|
2ÿz4TestOptionalArgs.test_ndarrayfuncs.<locals>.testaxisc S s t | ¡}tj | ¡}t||ddj||ddj t||ddj||ddj t||ddddd d
f ||dd d
f ddd t||ddddd d
f ||dd d
f ddd d S ) NT©ÚkeepdimsFr ©r4 rF .rÖ ri )r rB r} rà r rl rC r| r| r Ú testkeepdims1 s
|
ÿ ÿÿÿz8TestOptionalArgs.test_ndarrayfuncs.<locals>.testkeepdims)rs r r r r rå r¨ r )r r¹ )r} r rj ry rñ rà r& )r r r r rD rH r r| r| r Útest_ndarrayfuncs s z"TestOptionalArgs.test_ndarrayfuncsc C sB t d¡ d¡}t jdtd d¡}d|d d
dd d
f< t jj||d}tt|d tt|d d
|
d t d tt|d d
|
dt d tt|dddt d tt|d ddd t d tt|d dddt d tt|dd
|
d t d t
|
t t|dd
|
t
|
t j t|dd
|
t jj|t d}tt|d tt|d d
|
dt d tt|d d
|
dt d tt|dddt d tt t|dd¡d tt|d dddt d tt|d dddt d tt|dd
|
dt d t
|
t t|dd
|
t
|
t j t|dd
|
ttt jjd t
|
t j tt j d ¡d d
|
d S )Nr@ rA rÚ rÉ Tr r rÄ r r3 r¨ )r¨ rÁ ri rÁ ræ rE r
rG )r¨ r rÁ )r r rÁ r rJ r§ r )r} r rj ry rñ rà r& r r- ra r rè rß r_ rª rI )r r r r r| r| r Ú
|
test_countN s4 zTestOptionalArgs.test_countN)rê rë rì rI rJ 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 dd Z dd Zejjdddd Zejjddd d Zd d Zd!d" Zd#S )$ÚTestMaskedConstantc C s t |tjjdtjjk t dddg¡}|tjj|}t |tjjk t t|tjjj t|j |j ttj
|
|¡tj |j t d d S )Nr r¨ r§ rÉ ) r
|
r} rà rI r& rc rì ÚMaskedConstantr rl r; ra rñ )r r Zvectorrñ r| r| r Ú _do_add_testv s z TestMaskedConstant._do_add_testc C s | tj¡ d S rµ )rM r} r rä r| r| r Ú
|
test_ufunc s z TestMaskedConstant.test_ufuncc C s | dd ¡ d S )Nc S s | | S rµ r| ©r r® r| r| r r· r¸ z2TestMaskedConstant.test_operator.<locals>.<lambda>)rM rä r| r| r Ú test_operator s z TestMaskedConstant.test_operatorc C s: t j t jj¡}tt|t jjj t|t jjk d S rµ )r} rà r& rI r
|
rc rì rL r r| r| r Ú test_ctor s z TestMaskedConstant.test_ctorc C s8 t ttjjd tjj tjjj¡}t t|d d S )NrI )
|
r r! r} rà rI r Ú__new__rì rL r )r Zmasked2r| r| r Ú test_repr s z TestMaskedConstant.test_reprc
|
C sp ddl m} tdtjd D ]N}| .}tjtjj||d | d¡ t
|
|¡}W 5 Q R X t |tjjk q d S )Nr )ÚBytesIOr¨ r rU ) ÚiorT rW r{ rX Údumpr} rà rI ÚseekÚloadr
|
)r rT r\ r rò r| r| r Ú test_pickle s
|
z TestMaskedConstant.test_picklec C s* t tjj ¡ tjjktj ¡ tjk d S rµ )r r} rà rI rÝ ZTrue_rä r| r| r r9 ¥ s þz TestMaskedConstant.test_copyc C s& dd l }t| tjj¡tjjk d S ro )rÝ r
|
r} rà rI ©r rÝ r| r| r Ú
|
test__copy¬ s ÿz TestMaskedConstant.test__copyc C s& dd l }t| tjj¡tjjk d S ro )rÝ r
|
r> r} rà rI rZ r| r| r rA ± s ÿz TestMaskedConstant.test_deepcopyc C s t jj}tt jjjtj|dd tttj|j dd tttj|j
|
dd t jj t jj ¡}tttj|dd tttj|j dd tttj|j
|
dd d S )Nr| r F) r} rà rI r rì r rH Úsetitemrè rÐ r ra r )r Úorigra r| r| r Útest_immutable¶ s z!TestMaskedConstant.test_immutablec C s6 t dt¡}tttj|dt jj tttt jj d S ©Nr| ) r} ry r» r r rH r\ rà rI )r Za_ir| r| r Útest_coercion_intÁ s z$TestMaskedConstant.test_coercion_intc C s8 t dt¡}tttj|dt jj t t
|
|d ¡ d S r_ ) r} ry rÏ r rã rH r\ rà rI r
|
rq )r Za_fr| r| r Útest_coercion_floatÆ s z&TestMaskedConstant.test_coercion_floatz See gh-9750rð c C s* t dd¡}t jj|d< t|d d d S )Nr| ZU10r ©r} ry rà rI r )r Za_ur| r| r Útest_coercion_unicodeË s z(TestMaskedConstant.test_coercion_unicodec C s* t dd¡}t jj|d< t|d d d S )Nr| ZS10s --rb )r Za_br| r| r Útest_coercion_bytesÑ s z&TestMaskedConstant.test_coercion_bytesc C sN G dd dt tjj}| }t|| k t|tjjk tt|d d S )Nc @ s e Zd ZdS )z-TestMaskedConstant.test_subclass.<locals>.SubNr r| r| r| r ÚSubÙ s re rI )r" r} rà rI r
|
r r! )r re r r| r| r Ú test_subclass× s
|
z TestMaskedConstant.test_subclassc C s. t tttjjdd t tttjjdtj d S )Nrl r» r~ )r r¿ Úsetattrr} rà rI r rä r| r| r Útest_attributes_readonlyà s z+TestMaskedConstant.test_attributes_readonlyN)rê rë rì rM rN rP rQ rS rY r9 r[ rA r^ r` ra rá rD Zxfailrc rd rf rh r| r| r| r rK u s$
|
|
|
rK c @ s e Zd Zdd ZdS )ÚTestMaskedWhereAliasesc C s¼ t t dg¡t d¡}t|jdg t tjtj}t|jd tj tjtj ¡}t|jd tjj ddddgd dd
|
}t|jtjj k tjj ddddgd dd
|
}t|jdgd d S ) Ng àÀi ÿÿTFr r¨ r§ rÁ rÊ r )
|
rT r} r& rð r r r- rà r
|
r_ )r rò r| r| r Útest_masked_valuesé s z)TestMaskedWhereAliases.test_masked_valuesN)rê rë rì rj r| r| r| r ri å s ri c C s< t jjddddgddddgd} tt | ¡dgdgg d S )Nr r r¨ r§ r )r} rà r& r Zargwherern r| r| r Útest_masked_arrayú s rk c C sÊ t j ddddg¡} t jj| dk| dd}t| jddddg t jjddddgddddgd } t jj| dk| dd}t| jddddg t j t jddddg¡} t jj| dd}t| jdddddg d S )
|
Nr r¨ r§ rÁ Fr2 Tr r )r} rà r& rU r r r- rþ )r rõ r| r| r Útest_masked_array_no_copyþ s rl c C s t jjdddgdd} t jjdddgdd}t j | |¡}ddddddg}ddddddg}t|j| t|j| t j d¡} t j d ¡}t j | |¡}dgd }dgd dgd }t|jd
|
| t|j| t jj| |d d }t|jd
|
| t|j| d S ) Nr r¨ r§ ©r0 rÁ FTr* )r§ r r÷ r3 ) r} rà rK Úappendr rÐ r r, ra )r r® rñ Z expected_dataZ expected_maskr| r| r Útest_append_masked_array s"
|
ro c C s² t jjdddgdd} t j dddgdd d
|
ggd¡}ttt jj| |d d t jj| t jd d
f |d d }t j dd ¡}t jj |ddg< |
|
d¡}t |j |j t |j |j d S )Nr r¨ r§ rm rÁ rÊ r r) rø r r r3 rÈ r )r} rà rK rT r rè rn Znewaxisr rI rj r rÐ r )r r® rñ r r| r| r Ú#test_append_masked_array_along_axis$ s
|
rp c C s t tddk d S )Ny ð? ð?y@µx ¯D )r
|
r. r| r| r| r Ú test_default_fill_value_complex2 s rq c C s8 t dddgdddgd} tj| d| d}t|| k d S )Nr
r r r r r ra )r& r} r r
|
r¶ r| r| r Útest_ufunc_with_output7 s rr c
|
C s t dddgdddgd} t dddgdddgd}t dddgdddgd}t d d
|
d gdddgd}| ¡ }t | ||¡}| ¡ }tj| ||d }| ¡ }tj| ||fd } t|j|j t|j|j t| j|j t| j|j t|j|j t|j|j d S )z0 Test that masked arrays are immune to gh-10459 r r¨ r§ r r rÈ rI rT r rl r ra N)r& rÝ r} r r r rÐ )
|
r r® rè r Zout_posZres_posZout_kwZres_kwZout_tupZres_tupr| r| r Útest_ufunc_with_out_varied? s rs c C s² t dtdfddtfgfg¡} tdddgdfdddgdfgdddgd fdddgd
|
fgg| d }t jj|d d d < | | ¡}|jjt | ¡jkst |j
|
jjt | ¡jks®t t || t ||j|jd dk t t |j|jd dt j |j|jdd}t |jj t |j
|
jj t j|| dd}|jjt | ¡jks@t |j
|
jjt | ¡jks\t t || t |t j|| d ddk t j||jddd}t |jj t |j
|
jj d S )Nr r§ r r r r¨ r¦ r§ )r )r rÉ r Fr2 )Úsubokr¢ r T)r~ rt )r~ rÝ rt )r~ r rt )r} r~ r» rÏ r& rà rI rº rù rú r r r
|
r" r r Ú f_contiguous)Údescrr Zx_aZx_fZx_a2Zx_f2r| r| r Útest_astype_mask_orderingX s2 þþ
|
|
|
rw r4 r5 r7 z ignore::numpy.ComplexWarningc C s t jjtd| dd}| |¡}t|jdk t|j| k t|jj| k t|jdk t|j|k t|jj|k t|| d S )Nr§ r r ) r} rà r& ra rº r
|
r r~ r )r4 r7 ÚsrcÚdstr| r| r Útest_astype_basicx s
|
rz c C sn t g ¡} t d| ¡}t j |¡}t|j|j t|j|j t jj||d}t|j|j t|j|j d S )NrÁ r )r} r~ r2 rà r& r rl )r r r½ r| r| r Útest_fieldless_void s
|
r{ c C s2 t jj} t jjd| jd}d|_t| jjd d S )Nr r r» r| )r} rà rI r& r rl r rO r| r| r Ú0test_mask_shape_assignment_does_not_break_masked s r| r zHno docstrings present to inspect when PYTHONOPTIMIZE/Py_OptimizeFlag > 1rð c C s( dd } d}t tjj | jd¡| d S )Nc S s dS )zyThis docstring
|
|
Has multiple lines
|
|
And notes
|
|
Notes
|
-----
|
original note
|
Nr| rä r| r| r rÏ £ s z test_doc_note.<locals>.methodzNThis docstring
|
|
Has multiple lines
|
|
And notes
|
|
Notes
|
-----
|
note
|
|
original noteZnote)r r} rà rì r1 Ú__doc__)rÏ Z expected_docr| r| r Ú test_doc_note s r~ )¸r} Ú
|
__author__Úsysrà rH r rK rá Ú functoolsr r r} Z numpy.ma.coreZnumpy.core.fromnumericrì rÀ Znumpy.core.umathr¾ Z numpy.testingr r r r Z numpy.testing._private.utilsr r Z numpy.compatr Znumpy.ma.testutilsr
|
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 rQ rR rS rT rU rV rW rX rY rZ r[ r\ r] r^ r_ r` ra rb rc rd re rf rg rh ri rj rk rl rm rn ro rp rq rr rs rt ru rv rw rx ry rz r{ r rÄ rp rà ZMaskedArrayFutureWarningrF rG r rÅ rM rn r² rä r& r¯ r× rØ r r4 r7 r? rK ri rk rl ro rp rq rr rs rw rD rE râ rz r{ r| r r Úoptimizer~ r| r| r| r Ú<module> sÀ (ÿ þ < o 1 T s 55 . #0NWp
|
ÿ
|
