yZ>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYmZZZm[Z[m\Z\m]Z]m^Z^m_Z_m`Z`maZambZbmcZcmdZdmeZemfZfmgZgmhZhmiZimjZjmkZkmlZlmmZmmnZnmoZompZpmqZqmrZrmsZsmtZtmuZumvZvmwZwmxZxmyZymzZzm{Z{m|Z|m}Z}m~Z~mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZ dd lmZ e j Z e Ze e j j j d d
dD Zd eD Z G d d Z G d d Z G d d Z G d d Z G d d Z G d d Z G d d Z G d d Z G d d! Z G d" d# Z G d$ d% Z G d& d' Z G d( d) Z G d* d+ Z G d, d- Z G d. d/ Zd0 Zd1 Zd2 Zd3 Zd4 Zd5 Zd6 Zd7 Zej d8ee9 ej d:ee9 ej d; d< Zd= Zd> Zej ej j d?k d@A dB ZdC ZdD ZdE ZdF ZdS )GznTests suite for MaskedArray & subclassing.
:author: Pierre Gerard-Marchant
:contact: pierregm_at_uga_dot_edu
zPierre GF Gerard-Marchant N)reduce)
assert_raises assert_warnssuppress_warningsIS_WASM)requires_memory)ndarray)asbytes)assert_assert_array_equal assert_equalassert_almost_equalassert_equal_records
fail_if_equalassert_not_equalassert_mask_equal)iMAError MaskErrorMaskType MaskedArrayabsabsoluteaddallallcloseallequalalltrueangleanomarangearccosarccosharctan2arcsinarctanargsortarrayasarraychoose concatenate conjugatecoscoshcountdefault_fill_valuediagdividedoc_noteempty
empty_likeequalexp flatten_maskfilled fix_invalidflatten_structured_arrayfromflexgetmask getmaskarraygreater
greater_equalidentityinner
isMaskedArrayless
less_equalloglog10 make_maskmake_mask_descrmask_ormasked masked_array masked_equalmasked_greatermasked_greater_equal
masked_inside masked_lessmasked_less_equalmasked_not_equalmasked_outsidemasked_print_option
masked_values masked_wheremaxmaximummaximum_fill_valueminminimumminimum_fill_valuemodmultiplymvoidnomask not_equalones ones_likeouterpowerproductputputmaskravelrepeatreshaperesizeshapesinsinhsometruesortsqrtsubtractsumtaketantanh transposewherezeros
zeros_like)picklezGsetting an item on a masked array which has a shared mask will not copyc 6 g | ]}t j | S )npdtype.0dt_s l/builddir/build/BUILD/cloudlinux-venv-1.0.12/venv/lib64/python3.11/site-packages/numpy/ma/tests/test_core.py
r @ s
8
8
8S28C==
8
8
8 z?bhilqBHILQefdgFDc g | ] }|j
S r~ )charr s r r r A s
'
'
'38
'
'
'r c e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zed Z d Z d Z d Z d Z d Z!d Z"d Z#d! Z$d" Z%d# Z&d$ Z'd% Z(d& Z)d' Z*d( Z+d) Z,d* Z-ed+ Z.d, Z/d- Z0d. Z1d/ Z2d0 Z3d1 Z4d2 Z5d3 Z6d4 Z7d5 Z8d6 Z9d7 Z:d8 Z;d9 Zz.TestMaskedArray.test_basic1d..x s
Q r c | |z S r r~ r s r r z.TestMaskedArray.test_basic1d..y s
QU r r )r r rB r8 anyr r astypeintrm r
r r r r r. lenr r r r r r r r r r r r ss r test_basic1dz TestMaskedArray.test_basic1dl s 15.AsBBAr2M!$$$%%%
b!!"""b ##''))***bgnnS))27>>#+>+>???
GRXb\\1%%%RXq!!!RXqw'''RXqw'''RWf%6%6 : :;;;U2YYB&1B1BB*G*G GHHH2r"""6"e,,b1111b!!!!!r c
| j \
}}}}}}}}} }
dD ]/} | |_ | |_ | |_ | |_ | |
_ t t | t t | t t | | t |j | t |j t
d | t t | t | t
d | z
t ||
t t |d |
t || 1d S )N) r ) r c | |z S r r~ r s r r z.TestMaskedArray.test_basic2d.. s
QU r c | |z S r r~ r s r r z.TestMaskedArray.test_basic2d.. s
Q r r )
r rm r rB r
r r r. r r8 r s r test_basic2dz TestMaskedArray.test_basic2d~ s= 15.AsBBAr2! AAGAGBHBHBH
a((( ) ) ) M"%% & & & rA & & & 1 % % % &):):A">"> ? ? ? rCGGf5F5F.K.K$K L L L R E**B / / / B r c
| j \
}}}}}}}}} }
t t j ||f t ||f t t j ||f t ||f t t j ||f t ||f t t j |||f t |||f d S r )r r
r r* r s r test_concatenate_basicz&TestMaskedArray.test_concatenate_basic s 15.AsBBAr2R^QF++["b-B-BCCCR^QF++[!Q-@-@AAAR^QF++["a-A-ABBBR^Q1I.. QAJ0G0GHHHHHr c
" | j \
}}}}}}}}} }
d} | x|_ x|_ x|_ |_ t |j t j || t |j t j || t
||fd } t t j ||fd | t t j |j |j fd | j t d }t t d ddg }t
||f }t |g d t |j g d t
||f }t |g d t |j g d
d S ) Nr r r r FTr )r r r r )FFFTr r r r )FTFF) r rm r
r r rk r* _maskrz r' rb r )
r r r r r r r r r r r r xmyms
r test_concatenate_alongaxisz*TestMaskedArray.test_concatenate_alongaxis s~ 15.AsBBAr2 2333!'3BHrxRWbjQ / /000RWbjQ / /000B8Q''R^QFA..555R^RWbg$6::DJGGG !HH $q''
. . . A 1lll+++16#>#>#>??? A 1lll+++16#>#>#>?????r c
> t t t t j d t j d dt fdt fg }t |d d |dd g }t || d S )N
r r r ) rK listzipr randomrandr floatr r* r )r datatests r test_concatenate_flexiblez)TestMaskedArray.test_concatenate_flexible s DRY^^B%7%7%'Yr]]"4 "4 5 5$'<#s"<> > > D!Hd122h/00 T4(((((r c t g dg dd }t |j d t |j g dg t |j g dg d S )Nr r r r r r r )r ndminr r )r' r
rm r r r s r test_creation_ndminz#TestMaskedArray.test_creation_ndmin sf ))))))1 5 5 5QWf%%%QWyyyk***QWyyyk*****r c t g d }t |d< t |dt }t |j |j j t |j |j j d S )Nr r )r r )r' rJ r r
rm r r r xxs r $test_creation_ndmin_from_maskedarrayz4TestMaskedArray.test_creation_ndmin_from_maskedarray sb ))) "
1AU
+
+
+QWagm,,,RXrx~.....r c ` t dt }t |g d< t | }t |j |j t | }t |j |j t |g ddz }t
|j |j t g dd }t |j g d t g dd
}t |j g d t g d|j d
}t t j |j |j t g d|j d }t t j |j |j t g dd }t |j g d d S )
N r )r r r r r r r r r TTTTFFFFr copy) r r rJ r r
r r r' r r r may_share_memory)r r dma_1dma_2dma_3r r s r test_creation_maskcreationz*TestMaskedArray.test_creation_maskcreation s b&&&!ZZZD!!UZ+++E""UZ,,,E q(8999ej%*--- )))$ ' ' 'QW 0 0 0111 )))% ( ( (QW 3 3 3444 )))!' 6 6 6#AFAF33444 )))!' 5 5 5B'777888 )))$ ' ' 'QW 3 3 344444r c t j t j j t j t 5 t j dt j j g d d d d S # 1 swxY w Y d S )Nr )r r' marJ pytestwarns UserWarningr s r *test_masked_singleton_array_creation_warnsz:TestMaskedArray.test_masked_singleton_array_creation_warns s
\+
&
& ) )
Hb"%,' ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s &A00A47A4c t t j d g d }t ||d d d f }t |g dg dg t |j g dg dg t
|_ t ||d d d f }t |g dg dg t |j t
u 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 r r s r 'test_creation_with_list_of_maskedarraysz7TestMaskedArray.test_creation_with_list_of_maskedarrays s ")A,,___ 5 5 5a44R4\""TOOO___=>>>TZ///???!CDDDa44R4\""TOOO___=>>> V#$$$$$r c ~ t j ddgddg }t j d }t j ||gd }t |j ddgddgg G d d
}t j | dgddg }t j t d 5 t j |gt d d d n # 1 swxY w Y t j ||g }t |j ddgddgg d S )
Nr r TFr r U21r c e Zd Zd ZdS )UTestMaskedArray.test_creation_with_list_of_maskedarrays_no_bool_cast..NotBoolc t d )N not a bool!)
ValueErrorr s r __bool__z^TestMaskedArray.test_creation_with_list_of_maskedarrays_no_bool_cast..NotBool.__bool__ s ///r N)__name__
__module__ __qualname__r r~ r r NotBoolr s#
0
0
0
0
0r r r match) r r rK r r( r r r raisesr bool)r
masked_str
normal_intresr
masked_objs r 4test_creation_with_list_of_maskedarrays_no_bool_castzDTestMaskedArray.test_creation_with_list_of_maskedarrays_no_bool_cast s U''c
$ 'GG
Yq\\
emmZ 4EmBB38tUmeU^%DEEE 0 0 0 0 0 0 0 0 U''C(8e}'MM
]:]
;
;
; 1 1 J
|4 0 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 emmZ 45538tUmeU^%DEEEEEs
C66C:=C:c f t j dgddgddgddgd }t | d S )
N)Ar f0f1S4i8r )namesformatsoffsetsr )r r' r s r 'test_creation_from_ndarray_with_paddingz7TestMaskedArray.test_creation_from_ndarray_with_padding sK HhZd 37+34Q%(9 (9
:
:
: ar c t j t d 5 t g dg d d d d d S # 1 swxY w Y d S )Nzunexpected keyword argumentr r r r r )maks)r r TypeErrorr r s r test_unknown_keyword_parameterz.TestMaskedArray.test_unknown_keyword_parameter s
]9,I
J
J
J 3 3 2 2 2 2 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3s ?AAc
4 | j \
}}}}}}}}} }
d|_ d|_ t | } t | j |j t | j |j t | j |j t | j |j d S )NT)r
fill_value _hardmaskr( r
r r ) r r r r r r r r r r r xmms r test_asarrayz TestMaskedArray.test_asarray s 15.AsBBAr2
bkkSY)))SY)))S^R]333S]BL11111r c t j d j }t |j j t
| }t |j j d S )Nr r eyeTr flags c_contiguousr( r mnew_ms r test_asarray_default_orderz*TestMaskedArray.test_asarray_default_order sK F1IIKAG(()))
()))))r c t j d j }t |j j t
|d }t |j j d S )Nr Corderr< rA s r test_asarray_enforce_orderz*TestMaskedArray.test_asarray_enforce_order% sR F1IIKAG(()))%%% ()))))r c ( t j d 5 t t j ddgg d }t | }t |j |j ddg t |j g d d d d d S # 1 swxY w Y d S )Nignoreinvalidr r r r r r )r r r ) r errstaterK nanr9 r
r r7 r )r r
data_fixeds r test_fix_invalidz TestMaskedArray.test_fix_invalid- s
[
*
*
* 9 9 R 0yyyAAAD$T**J )DOR+D E E E )<<< 8 8 8 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9s A$BB B c t d }t |d< t t t dk t |d t u t t |d d d d S )Nr r --r )r rJ r r r
r8 r s r test_maskedelementz"TestMaskedArray.test_maskedelement5 sg 1II!F t#$$$! VAaD!__a(((((r c t dt }d}||d< t |d | t |d |u dd l}|j }||d< t |d |u d S )Nr r r r r r r r )r3 objectr
r datetimenow)r r r rY dts r test_set_element_as_objectz*TestMaskedArray.test_set_element_as_object= s !6 " " " !QqT1 !
"
"
$
$!!
r c t j g d }t |g d }t |g d }t | }t | t | t t j | t |d t
t |d t |d u t
|d |d k t
|d t u t |d |d t |d d
|d d
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< t
t t | t g d t g dg d |d d < t
t t | t g d t g dg d |d d < t
t t | t g d t
t |t g d t j d
dz }t |d }t || t
t t g dt |j t d|j t g dt" }t j g dt" }|d }|d }t t | t t t | t t || t
|dd j dk d S )N)r r r r r r r r r r Fendwithr r r r c r r r r r r )r r r r r r r r r r r )r hellor r r )r r' r reprr
rq r typerJ r r< rK r rU r r r7 rX rm )r x1x2x3x4s1s2s r
test_indexingz TestMaskedArray.test_indexingJ s
Xlll
#
#
2LLL
)
)
)
2LLL
)
)
)
2YY B RRWR[[$r5"9"9"9:::RU tBqE{{*+++1A 1 RUBqE"""R!Wb1g&&&RUBqqqE"""RVRV$$$11R 1Q31Q3R 1R 1Q3R 1111eLLL&9&9::;;; \\\<<<88111eLLL&9&9::;;; \\\<<<88111eLLL&9&9::;;;U<<< 0 011222
Yq\\C
2s
#
#R 9927CCDDDS"-(((
%%%v
.
.
X(((&
1
1
U
UT"XXs###T"XXs###R 1Q3
%&&&&&r c t j d d }t ddd ddgf}t j d }t j |j d< |||< t j j |d< t j d t j
}t j
g d g d
}||d< ||g d< d S )N< )r r r r r r r .r r )gCD pr r TFFr r r r ) r r r rk slice
masked_allinfr rJ float32r' )r r indexvalues r test_setitem_no_warningz'TestMaskedArray.test_setitem_no_warning| s
ELL $ $W - -q!Q !Q( (( 6 C% # ELL2:L . . MMM0D0D0D EE# ))) r c
g d}t | }t | }t ||u t |d }t ||u t j d }t || }t |j j |j t t ||j t |j
j |j t | }t |j j |j j k t |j
j |j
j k t || }t |j j |j k t |j
j |j k t |d t u d|d< t |d t u t |j
j |j k t t |j d t ||d
} t | j j |j k t | j
j |j k t | d t u d| d< t | d t u t | j
j |j k t t | j d t |d z | }
t t |
j |d z j u t d } t | d< t | d
} t t! | | g | t t# | g d t% | dd }
t |
g d t% | dd }t |
| | dd }t |
| | dd }t |
| | }t |j | j t |j
| j
t) g dg d }t) | }t |j j j |j j j t |j
j j |j
j j t) |d }t- |j j j |j j j t- |j
j j |j
j j d S )Nr r r r r Tr r r r ra r r r r r )r+ )r r r r r r r r )r r r r axis)r r r r r r r r r r1 )rG r r r r' r
r __array_interface__r r r rJ r r8 r rl r* r< rj r rK ctypesr )r nrB r m3ri y1y1ay2y2ay3rl y4y5y6y7y8y9r r s r test_copyzTestMaskedArray.test_copy s
OO aLL
q\\R
qt
$
$
$
Yq\\
2A
RX123IJJJRW%%&&&RX113HIIIBii -45 6 6 6 -45 6 6 6 2B
,0FFGGG,0FFGGG1 11V#$$$,0FFGGG!$$%%%BQQ''' -1GGHHH -1FFGGGA& !!!AAf$%%% -1FFGGG1%%&&&
28!
$
$
$r
R#X$44555
AYY1
B
["b**B///WR[[":":":;;;
B 1
-
-
-R111222
B
"
"
"R
YY|!Y
,
,R
YYq!__R
WWYYRXrx(((RXrx((( 3 3 3 OOQW^(!'.*=>>>QW^(!'.*=>>> & & & ,agn.ABBB ,agn.ABBBBBr c t j dd }| }t |j d d S )N+ Tr )r r r' r r
r )r r xcs r test_copy_0dz TestMaskedArray.test_copy_0d s> EKKK & &
VVXXRWd#####r c t t t j g d t t t j d d S )Nr )r rB r r r r s r test_copy_on_python_builtinsz,TestMaskedArray.test_copy_on_python_builtins sT
bejj 1 122333
bejj 1 12233333r c t j g d }t j g d }|j }|j t | g d d S )Nr r r r )r r r' r r
)r r r
a_copy_methods r test_copy_immutablez#TestMaskedArray.test_copy_immutable s^ EKK " " EKK " "
]]__iii00000r c ddl m} t g dg d } || }t |j |j t t
|j t
|j d|d< t |j g d t |j g d || }t |j |j d |j d< t |j g d t |j g d d S )
Nr )deepcopyru FTFr r r r r r1 F)r r r' r
r r idr )r r r copieds r
test_deepcopyz TestMaskedArray.test_deepcopy s !!!!!! )))"6"6"6 7 7 7!V[!&)))AGb &6&6777q V[))),,,QVYYY'''!V[!&))) AV[))),,,QVYYY'''''r c t g dg d }t t | d t t t d t t t d d t t
5 t t d }d d d n # 1 swxY w Y t |d d S )Nru r r [0 -- 2]rT z >5)r' r
formatrJ r
FutureWarning)r r with_format_strings r test_formatzTestMaskedArray.test_format s )))"6"6"6 7 7 7VAYY
+++VF^^T***VFB''... -
(
( 7 7!'!6!6 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7'.....s B..B25B2c t g dg d }t t | d t t | t j d t j d }t j j |dd< t t | t j d t j d
}t t | t j d t g d g d
gt j
}t j j |d< t t | t j d t t |d d t j d t t | t t j d d S )Nru r r r z masked_array(data=[0, --, 2],
mask=[False, True, False],
fill_value=999999) r 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 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 rg textwrapdedentr r r rJ int8r r r r s r
test_str_reprz TestMaskedArray.test_str_repr
s )))"6"6"6 7 7 7SVVZ((( GG O )
*
*
ELL %,!B$ GG O )
*
*
ELL GG O )
*
*
999iii( 8 8 8 # GG O
2A2KK O '
(
(
# O $
%
%
r c t j }t j d t g dg d }t t | d t t
| d t j d }t j j |d d
< t t
| d t j d i | d S # t j d i | w xY w)
Nz1.13)legacyru r r r zcmasked_array(data = [0 -- 2],
mask = [False True False],
fill_value = 999999)
r r r zmasked_array(data = [0 -- -- ..., 1997 1998 1999],
mask = [False True True ..., False False False],
fill_value = 999999)
r~ )
r get_printoptionsset_printoptionsr' r
r rg r r rJ )r oldoptsr s r test_str_repr_legacyz$TestMaskedArray.test_str_repr_legacyP s %''
6**** +iii&:&:&:;;;A Q , , , a #B
C
C
C T""AelAadG Q0
* *' * * * * *B * *' * * * *s B!C C1c d}t | }t j | }t j |d }t || | t || d d S )Nu caféTr rT )rh r r r' r
)r uutype
arr_nomask
arr_maskeds r test_0d_unicodez TestMaskedArray.test_0d_unicodee st QU[[^^
U[[[..
UU:&&***UU:&&-----r c f t t t t fD ]}t d | }d|_ g dddf}t dt j dz D ]}|D ]}||_
t j t j || }t |j |j t |j |j |t t fv rt |j d n t |j |d t! |j
| Đd S ) Nr )
r r r r r r r r r r TFr r protocol)r r r rX r r r7 ranger| HIGHEST_PROTOCOLr loadsdumpsr
r r r )r r r masksprotor a_pickleds r
test_picklingz TestMaskedArray.test_picklingo s8 5#v. = =Er
!!%((A AL333 E q&"9A"=>>
=
= ! = =D !AF & V\!e-L-L-L M MI !'::: !':::
--$Y%93????$Y%955::FFF&y~t<<<< =
= = =r c t j ddgdt fdt fg t j }t
|ddg }t d t j d
z D ]}t j
t j || }t |j
|j
t || t t |j t j d S ) N)r r )r r r r r TFFTr r r r )r r' r r viewrecarrayrK r r| r r r r
r r
isinstancer )r r r r r s r test_pickling_subbaseclassz*TestMaskedArray.test_pickling_subbaseclass s Hh) %L3*5
7
7
77;tBK7H7H
- !? @ @ @1f 5 9:: > >E V\!e%D%D%DEEI !' 2 2 2 A & & & Jy << = = = = > >r c T t j j }t dt j dz D ]x}t j t j || }t |j |j t |j
|j
t |j |j yd S )Nr r r ) r r rJ r r| r r r r
_baseclassr r )r mcr
mc_pickleds r test_pickling_maskedconstantz,TestMaskedArray.test_pickling_maskedconstant s
U\1f 5 9:: 5 5E fl2&F&F&FGGJ .
> > > )28 4 4 4 )28 4 4 4 4 5 5r c 6 t ddgddgdt fdt fg }t dt j d z D ]T}t j t j ||
}t |j |j t || Ud 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 s r test_pickling_wstructuredz)TestMaskedArray.test_pickling_wstructured s 7G$FF+; sc5\2
4
4
41f 5 9:: ' 'E V\!e%D%D%DEEI !' 2 2 2 A & & & & ' 'r c t d }d|_ |j }t dt j dz D ]:}t j t j || }t || ;d S )Nr )r r r r r ) r rm r> r r| r r r r
)r r r r r s r test_pickling_keepalignmentz+TestMaskedArray.test_pickling_keepalignment sw 2JJ
C1f 5 9:: " "E< Q ? ? ?@@D q ! ! ! ! " "r c t g d }t g dg d }t |d j d t |d j d t |d j d d S )Nr r r r r~ r )r' r
rm r s r test_single_element_subscriptz-TestMaskedArray.test_single_element_subscript su ))) )))))) , , ,QqTZ$$$QqTZ$$$QqTZ$$$$$r c j t dt t d t dt t d t dt t dggg t dt t dgg t t
t t ddg t
5 }| t d t t j t t dgdg t g dg d t t
fd t t d d t t j t d
d d d n # 1 swxY w Y t t
t t t d d t t fd d S )
Nr r z$Warning: converting a masked elementr r r c " t S r )r r s r r z/TestMaskedArray.test_topython.. s
U1XX r r r r r c . t d S Nr )r r s r r z/TestMaskedArray.test_topython.. s c!A$ii r )
r
r r' r r r3 r filterr r r isnanr r supr s @r
test_topythonz TestMaskedArray.test_topython s QE!HH
&&&S%a//***QEQC5'NN++,,,S%se --...iq!f
666
+C JJ{$J K K K BHU5!A3#7#7#7 8 899 : : :iiiiii000A )%5%5%5%5 6 6 6 qu r * * * BHU1Q4[[)) * * * + + + + + + + + + + + + + + + ia(((S2ZZ###g / / / /00000s 'C GGGc t d }| dd }d|j d< t |d dk |d|z z }t |j | t |j d|z t |t | z j d|z |z d |j d
< t d }t |d < t t |d t t k |d k }t t t |t t d
k t t t |t t |j k t || }t |j |j u t |d t u t |d t u t |d t u t |d t u t |d t u t || d S )Nr r r r r y $@r e r .r r+ r rr ra )r rk flatr r
realimagr+ rJ r r. ry rm rV r )r r r cs r test_oddfeatures_1z"TestMaskedArray.test_oddfeatures_1 s 2JJ
IIaOOq $2
aKQVQ QVR!V$$$a)A,,&,cAgk:::s 2JJ!AaD S[[()))
FeAvv..//14555eAvv..//17:;;; A 17"###! !F"###!F"###! ! Qr c t g d }t g d }t |d< t ||| }t |g d t |d< t ||| }t |g d t |d t u t |d t u t |d t u d S )Nr r r r r r r r r r r r r r r r r )r' rJ ry r
r r r r r s r test_oddfeatures_2z"TestMaskedArray.test_oddfeatures_2 s &&& ' ' /// " "! !QOOQ---...! !QOOQ---...! !F"###! r c t dgd }t dg }|j }|| ||< t |dg d S )Nr Tr r )r' r r
)r atestbtestidxs r test_oddfeatures_3z"TestMaskedArray.test_oddfeatures_3 sQ rd&&&rd j3Zc
URD!!!!!r c t j dd }t | d d d d S )Nr Or r r )r r rw r
r8 r s r test_filled_with_object_dtypez-TestMaskedArray.test_filled_with_object_dtype sA E Qc * *QXXc]]1%s+++++r c
t dgdt fddt fg }t |d< t | t
j t d t d t d fg|j t |d< t | d t
j d
g|j d S ) Nr r r ir |S8fr r r r r )r 1r ) r' r r rJ r
r8 r r/ r )r flexis r test_filled_with_flexible_dtypez/TestMaskedArray.test_filled_with_flexible_dtype s yk !3Z U| DF F FaU\\^^X 21 5 5 23 7 7 22 6 6 9 :AFN N N O O O aU\\!__X|nEK@@@ B B B B Br c dt fdt fg}t ddg| }| }t t | dt
d f | d }t t | d d
|_ t t | d d S ) Nr r r r r r r r r r )r r r )r r )r r r_ r8 r
tupler/ r7 )r ndtyper r s r test_filled_with_mvoidz&TestMaskedArray.test_filled_with_mvoid s *sEl+ ' 7 7 7xxzzU4[[1&8&<&<"=>>>xx!!U4[['***# U188::&& 22222r c dt fddt fdt fgfg}t ddgddg| }| d
}t j d d g|
}t || |d d
}t j ddg|d j
}t || t j dt
j dg }t |j j t
j dddgdfg t |j
j t
j dddgdfg d S )Nr& BBABBr r r rs r r r r r r r r )r r r r r r r )r& z(2,2)i1,(2,2)i1rs )r' i1rs )r( r rs rs )r' ?rs )r( r! rs ) r r' r8 r r
r numpyr rz r r )r r r r controlZs r test_filled_with_nested_dtypez-TestMaskedArray.test_filled_with_nested_dtype s *sdC[4+$> ?@ ; ,#[1
A
A
Axx{{(K5VDDDT7###v}}Q (FF+1S6<@@@T7###
HNN1ek+J*K L L M MQV\5;7K*>7@AG1I 0J $K $K L L LQV\5;7J*=7?@F1H 0I $J $J K K K K Kr c t t j ddgd t j ddgd d }t |j d t | d j d d S )
Nru r FrG rN r )r rH F_CONTIGUOUSr )r' r r r? r8 r s r test_filled_with_f_orderz(TestMaskedArray.test_filled_with_f_order- s "(Iy1===xI 6cBBB
'((( !.122222r c t g dt }d|j d< | }t |j d d d|j d< t |j d d d S )Nr r ???infoz!!!)r' r _optinfor r
r r r s r test_optinfo_propagationz(TestMaskedArray.test_optinfo_propagation5 sq +++U + + + "
6
FFHHQZ'/// "
6QZ'/////r c t g d }d|j d< t |j d |dk j d t |j d |dk j d t |j d |dk j d t |j d |dk j d t |j d |dk j d t |j d |dz j d t |j d |dz
j d t |j d |dz j d t |j d |dz j d t |j d |d d j d t |j d |g d j d t |j d t j | j d t |j d t j | j d t |j d t |d j d t |j d t j | j d d S )N)r r r r r{ keyr )r r r Tr )r' r- r
r r6 r r{ r s r test_optinfo_forward_propagationz0TestMaskedArray.test_optinfo_forward_propagation> sP ))) #
5QZ&a(9%(@AAAQZ&a(9%(@AAAQZ&Q(8(?@@@QZ&a(9%(@AAAQZ&a(9%(@AAAQZ&Q(8(?@@@QZ&Q(8(?@@@QZ&Q(8(?@@@QZ&Q(8(?@@@QZ&"1"u(=>>>QZ&'''
(;E(BCCCQZ&q (:5(ABBBQZ&q (:5(ABBBQZ&ad(;(;(;(DU(KLLLQZ&
a(8(8(A%(HIIIIIr c T t j dt fddt fdt fgfg }t ddgddg| }d
}t t
| | t d g d g d gd
fdg dg dgdfd }d}t t
| | d S )Nr r tr )r )r r )r )r @r r r z [(--, (2, --)) (4, (--, 6.0))]r )r r r r F)TFTFFTint, (2,3)float, floatr r r z)(0, [[--, 0.0, --], [0.0, 0.0, --]], 0.0))r r r r r' r
r rK )r
fancydtyper r# t_2d0s r test_fancy_printoptionsz'TestMaskedArray.test_fancy_printoptionsQ s XSzC3*sEl1K+L MNN
m]3& 4%' ' ' 3SYY((( Q (7)9$'%) &+-@-@-@-A-A-A-C%*%, &>
? ? ? >SZZ)))))r c dt fdt fg}t j ddg| }t | }t j ddgddggt }t || t |j |j t ddgdd g|
}t | }t ddgddggd d gd d ggt
}t || t |j |j t |j |j dt fdd
t fdt fgfg}t ddgddg|
}t | }t g dg dgg dg dgt
}t || t |j |j t |j |j dt fdt fg}t j dgdgg| }t | }t j ddggddgggt }t || t |j |j d S )Nr r r rs r r r r r r r r babb)r )r 皙?)r )r 皙@r r )r r r? )r r r@ r1 r )r r r r' r: r
r r )r r r r r# s r test_flatten_structured_arrayz-TestMaskedArray.test_flatten_structured_arraye si *sEl+ Hff%V 4 4 4'**(RHr2h/u===T7###TZ /// 66"&&)9 H H H'**"bB8, !fq!f -U< < <T7###TZ ///TY ---*sdC[4-$@ AB =-0#[1
A
A
A'**
6 'ii 35B B BT7###TZ ///TY ---*sEl+ Hvj6*-V < < <'**(b"XLB8,7uEEET7###TZ /////r c dt fdt fg}t j dg| d }t | }t t |t t
dgdg| d }t t |t t
ddgddg| }t |j d |j d }t t |t d S ) Nr r r r r r r r r ) r r r' r_ r r rK r r )r r r r
s r test_void0dzTestMaskedArray.test_void0d s *sCj) HgYf - - -a 0 !HH
1e$$%%% & ? ? ? B
1e$$%%% &&)0@ O O O !'!*agaj ) )
1e$$%%%%%r c dt fdt fg}t ddgddg| }|d }t t |t t |d |d fd t |d d
|d }t t |t t |d t u t |d t u t |d d t dd gfgd
dgfgdg }t |d d |d d t |d d t dd gd
dgd d S )Nr r rC r r r r r r r r r TF)r& >i2r r8 r& rF )r rK r r r_ r
rJ )r r r r
r& s r test_mvoid_getitemz"TestMaskedArray.test_mvoid_getitem s *sCj) '6*&&1A %
'
'
'
aD
1e$$%%%adAcF^V,,,QsVQ
aD
1e$$%%%! #& !!!QqT1
1xj!%u
/ 0 2 3
5
5
5 QqT#Y#q ***QqT#Y 1a& #Um5!: !: !: ; ; ; ; ;r c dt fdt fg}t ddgddg| }t t |d d d
g t t |d t d g d S ) Nr r rC r r r r r r r r )r rK r
r rJ )r r r s r test_mvoid_iterz TestMaskedArray.test_mvoid_iter s} *sCj) '6*&&1A %
'
'
' T!A$ZZ!Q(((T!A$ZZ&!-----r c J t ddgdt fdt fg }t t |d d t |d d< t
j }t j d t t |d d t t |d d t j | n# t j | w xY wt d
d gd g }t t |d d
d S )Nr rs r r r r z(1, 1)z-X-z(1, -X-)r rG )r r z(1,)) r' r r
r rJ rT _display set_displayrg )r mx ini_displays r test_mvoid_printz TestMaskedArray.test_mvoid_print s
FF#S#Jc
+C
D
D
DSAZZ***3
)2 '... 9 RUZ 0 0 0 bej 1 1 1 +K 8 8 8 8 +K 8 8 8 8 D$<
|
4
4
4SAZZ(((((s
8AC C)c 4 t g dfgg dfgg dfdg }t t |d dk t t |d dk t dd gd
d ggfgd d
gd
d ggfgdg }t t |d dk t t |d dk t dgdgddg }t t |d dk t t |d dk t dd gd
d ggdfgd d
gd
d ggd fgddt fg }t t |d dk t t |d dk t dgdgddg }t t |d dk t t |d dk d S ) Nr r )?B rS rS )r %9%9%9$;#<*B*B*B)D%7$8: : : DG /000T!W
0111 q!fqe_$6#7',dmdE]%C$E#F%8$9; ; ; DG 77888T!W
!88999E7$0>%1<$@B B B DG )***T!W
*+++q!fqe_a$8#9',dmdE]%CU$K#L%83,$GI I I DG ;;<<<T!W
!<<===+ "7!8%1>$BD D D DG /000T!W
011111r c t dgdg }t ddg }t ||gddgt }t |d |u t |d |u t t j |d j |j k t t j |d j d |d j d< t |d d k d S )
Nr Tr r Fr r r r )rK rX r r r r r )r mx1mx2rO s r test_object_with_arrayz&TestMaskedArray.test_object_with_array s B4tf---B8$$
3*E4=
G
G
G1 1S !!!r!uzSX-..///r!uz""###1
1
A" r N)>r r r r r r r r r r r r r r r
r r$ r/ r4 r: rD rI rR rU r\ ro r| suppress_copy_mask_on_assignmentr r r r r r r r r r r r r r r r r r r r r
r r% r) r/ r2 r; rA rD rH rJ rQ rZ r^ r~ r r r r D s
8
8
8 $ $ $ 3 3 3" " "$ (I I I@ @ @*) ) )+ + +/ / /5 5 5, ) ) )
%
%
%F F F( 3 3 32 2 2* * ** * *9 9 9) ) ) 0' 0' 0'd $ &FC FC &%FCP$ $ $ 4 4 4
1 1 1( ( ("
/
/
/ D
D
D
L+ + +*. . .= = =, > > >5 5 5' ' '" " "% % %1 1 1( 8 &" " &%", , , B B B 3 3 3 K K K(3 3 30 0 0J J J&* * *(!0 !0 !0F & & & ; ; ;2. . .) ) )""2 "2 "2H
r r c
e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Z d Z d Z d Z d Z d Z!d Z"d! Z#e$j% & d"d#d$g e$j% & d%d&d'g d( Z'e$j% & d"d#d$g e$j% & d%d&d'g d) Z(e$j% & d*e)e*+ e$j% & d,e)e*+ e$j% & d%d&d-g d. Z+e$j% & d*e)e*+ e$j% & d,e)e*+ e$j% & d%d&d-g d/ Z,e$j% & d*e)e*+ e$j% & d,e)e*+ e$j% & d%d&d-g e$j% & d0e-j. e-j/ e-j0 e-j1 g d1 Z2e$j% & d0e-j. e-j/ e-j0 e-j1 g e$j% & d%d&d2g d3 Z3d4 Z4d5 Z5d6 Z6d7 Z7d&S )8TestMaskedArrayArithmeticc t j ddddt dz dddddddg }t j g d }d}g d
}g d }t || }t || }t j g d
}t |g d } t j |d| }
| d ||||||||| |
f
| _ t j | _ t j dd d S )Nr r r r r r r r r r r r r r r rK r1 rM )
r r' r rK ry r r geterr
err_statusseterrr s r r z&TestMaskedArrayArithmetic.setup_method s Hb"b#r#vr2tS"b"M N N HJJJ K K
1
1
1
1
1
1
!"
%
%
%
!"
%
%
% H&&& ' '
!,,,
/
/
/
Xb%
#
#
% QRRQB7 )++
8444444r c 2 t j di | j d S Nr~ r rf re r s r teardown_methodz)TestMaskedArrayArithmetic.teardown_method
$$DO$$$$$r c
X | j \
}}}}}}}}} }
t ddgddgg } t | ddgddgg } t | | z | | z t | | z | | z t | | z
| | z
dD ]}
| |
}| |
}| |
}| |
}|
|
}
t | | t ||z ||z t ||z
||z
t ||z ||z t ||z ||z t ||z ||z t ||z
||z
t ||z ||z t ||z ||z t ||z ||z t ||z
||z
t ||z ||z t ||z ||z t |dz |dz t t | dz t | dz t ||z ||z t t
j || t || t t
j || t || t t
j || t || t t
j
|| t || d S )Nr r r r ))r r )r r @) r r' rK r
rk r r r rs r^ r1 )r r r r r r r r r r r a2da2dmr s r test_basic_arithmeticz/TestMaskedArrayArithmetic.test_basic_arithmetic s 15.AsBBAr2aVaV$%%C1a&1a&!122S3Yd
+++S3Yd
+++S3Yd
+++( : :A ! A ! AA BA BA B !bS ! ! ! QR ( ( ( QR ( ( ( QR ( ( ( QR ( ( ( q#( + + + q#( + + + q#( + + + q#( + + + S"s( + + + S"s( + + + S"s( + + + S"s( + + + aq ) ) ) Q3 B3 7 7 7 ar * * * 1 s2r{{ 3 3 3 Q**HR,<,< = = = Q**HR,<,< = = = 1a &R.. 9 9 9 93 : :r c t dt }d|_ t dt }||z }t |g dg dg t |j g dg dg ||d d d f z }t |g dg dg t |j g dg dg t dt }||d d d f z }t |g d g d
g t |j g d g d g d S )
Nr r r r r )r r r )r r rm r r )r r r )r r r r r )r r rm r
r r r r r s r test_divide_on_different_shapesz9TestMaskedArrayArithmetic.test_divide_on_different_shapes5 s, 1E " " " 1E " " "
EQ 7888QViii 3444
$qqq& MQ 7888QViii 3444 1E " " "
!!!T'
NQ ,,,7888QViii 344444r c t j dg }t dg }t t ||z t t t ||z t d S Nr )r r' r r r )r nar s r test_mixed_arithmeticz/TestMaskedArrayArithmetic.test_mixed_arithmeticG s\
Xqc]]
A3ZZ
27K00111
27K0011111r c t j t j }t |d|z dg }t t
|dz g d t t
d|z g d d S )Nr r r r r )r finfor tinyr' r
r= )r r{ r s r test_limits_arithmeticz0TestMaskedArrayArithmetic.test_limits_arithmeticN sp x# 4dB' ( (\!a%(()))444\!a%(()))44444r c P t dd }t dt d z j t d|z j t | j t t || j t t || j d S )Nr r r )r' r r rX r[ )r r s r test_masked_singleton_arithmeticz:TestMaskedArrayArithmetic.test_masked_singleton_arithmeticT s 11
U1XX #$$$R
"
B$%%%B$%%%%%r c t g dg d }t |d dk t u t |d dk t u t |d dk d t |d dk d d S )Nr r r r r r r FT)r' r rJ r
r s r test_masked_singleton_equalityz8TestMaskedArrayArithmetic.test_masked_singleton_equality^ s )))))) , , ,1v%&&&1v%&&&aeqj5)))aeqj4(((((r c ` t ddg }|t z }t |j |j t |j ddg |d t z }t |t u |t z }t |j |j t |j ddg d S )Nr r Tr )rK rJ r
rm r r r. s r %test_arithmetic_with_masked_singletonz?TestMaskedArrayArithmetic.test_arithmetic_with_masked_singletonf s !Q
JQWag&&&QWtTl+++
aD6MV
JQWag&&&QWtTl+++++r c t dg }|t z }t |j |j t |j dg d S )Nr T)rK rJ r
rm r r. s r 5test_arithmetic_with_masked_singleton_on_1d_singletonzOTestMaskedArrayArithmetic.test_arithmetic_with_masked_singleton_on_1d_singletonr sH !
JQWag&&&QVdX&&&&&r c t dd }t | j j |j j t d dz }t |j d t |j ddg d S )Nr r r r rG r )r' r
r8 r r rm r )r r r s r test_scalar_arithmeticz0TestMaskedArrayArithmetic.test_scalar_arithmeticy st !! QXXZZ&+QX];;;
6]]R
RXt$$$RWq!f%%%%%r c
| j \
}}}}}}}}} }
t t j | t | t t j | t | t t j | t | t t j | t
| t t j | t | t t j | t | t t j t | t | t t j t | t | t t j t | t | t t j
| t | t t j | t | t t j | t | t t j | t! | t t j || t# || t t j | t% | t t j |d|z z t' |d|z z t t j |d|z z d t' |d|z z d t t j || t) || t t j || t+ || t t j || t- || t t j || t/ || t t j || t1 || t t j || t3 || t t j | t5 | d S )N ?T)deg)r r
r r, r- rn ro rv rw rr r rE rF r6 r$ r! r% r# r r r5 ra rC r> rD r? r+ r s r test_basic_ufuncsz+TestMaskedArrayArithmetic.test_basic_ufuncs sg 15.AsBBAr2RVAYYB(((RWQZZb***RVAYYB(((RWQZZb***RVAYYB(((RWQZZb***RWSVV__d2hh///RVCFF^^SWW---RXc!ff%%uRyy111RVAYYB(((RYq\\6"::...RYq\\6"::...RYq\\6"::...RZ1%%wr2777R[^^Xb\\222RXa"Q$h''rBrEz):):;;;RXa"Q$hD1115be3N3N3NOOORXa^^U2r]]333R\!Q''2r):):;;;RWQ]]DRLL111RZ1%%wr2777R]1a((*R*<*<===R%a++]2r-B-BCCCR\!__imm44444r c t dt d t dt ddg t g dg d }t | }t |j j t j u t d| | d }t | }t |j j t j u t d| t |d }t t |t t ddg| t t | t u t g d }t |d }t t |t t |j j t j u t t j |j d d S )
Nr r r )r r r r r r rs r r )r
r. r' r r rh r intprk r r r< r` r AxisError)r ottr" s r test_count_funcz)TestMaskedArrayArithmetic.test_count_func s Qa!!!Qaqc***+++$$$<<<888Cjj "')***Q kk&!!Cjj "')***Q Cmm
3(()))aVS!!! &'''$$$%%Cmm
3(())) "')***blCIA666666r c ~ t dt g d t dt d d S )Nr r r rC )r
r. r s r test_count_on_python_builtinsz7TestMaskedArrayArithmetic.test_count_on_python_builtins s8 Qggg'''Qe %%%%%r c
H | j \
}}}}}}}}} }
t j | } t | } t t | t j | t t | t j | t t g dg d g d t t g dg d g d t d }t d dz
}t |d< t |d< t t || t t || || t t || t t || || t t j | dk t t j | d k t d dd }t |d
< t t j |d d d S ) Nr )r r ra r r r )r r ra r r r r r r r )r r ri r
rW rX r rZ r[ r rJ ry rC r> r rk )
r r r r r r r r r r r xrxmrs
r test_minmax_funcz*TestMaskedArrayArithmetic.test_minmax_func s 15.AsBBAr2
Xa[[BiiSWWgnS 1 1222SWWgnS 1 1222WYYY 22III>>>WYYY 22III>>> 1II 1IIM!!WQ]]E$q!**a$;$;<<<WQ]]E'!Q--A$>$>???q!!Q&'''q!!Q&''' 1II a # #& W^AD111155555r c t j d }t || }t t |t
t
|t j || t j || }t t |t
t
|t j || t || }t t |t
t
|t j || t j || }t t |t
t
|t j || d S )Nrs ) r rb r[ r r r r
rd rX )r r aminimumamaximums r test_minimummaximum_funcz2TestMaskedArrayArithmetic.test_minimummaximum_func s- GFOO1a==
8[11222Xrz!Q / /000=A&&
8[11222Xrz / /1 5 56661a==
8[11222Xrz!Q / /000=A&&
8[11222Xrz / /1 5 566666r c t g dg d }t j | }t |d d S )Nr r r r )r' r rX r r
r s r test_minmax_reducez,TestMaskedArrayArithmetic.test_minmax_reduce sJ )))"7"7"7 8 8 8 J a Qr c t j d }t t j ddd | }d|_ dD ]}t t | }t t j j
| }t j dt } ||d| }n# t $ r Y nw xY wt j dt } ||d| }t ||u | d
||d| }t ||u d S ) Nr r r r r rZ rW r r r outr )r r r roundr' uniformrm getattrr" r corer3 r r r r fill)r r r funcnamenpfuncmafuncnoutresults r test_minmax_funcs_with_outputz7TestMaskedArrayArithmetic.test_minmax_funcs_with_output sP y~~b!!''))
29$$QB//d
;
;
;& $ $H R**F UX]H55F8D,,,D
555
8D...DVBQD111F FdN # # # IIdOOOVBQD111F FdN # # # #! $ $s =C
CCc
n | j \
}}}}}}}}}}|j f|_ t | d t |d t u t |d d t u t |d d t u t | d t |d t u t |d d t u t |d d t u t | d t |d t u t |d d t u t |d d t u t g dd }t | t u t | t u t | t u d S ) Nr r r r 4@r Tr )
r r rm r
rW r rJ rZ ptpr' )r _r r s r test_minmax_methodsz-TestMaskedArrayArithmetic.test_minmax_methods s *.&'Aq!QAq!QG:RVVXXr"""1 v%&&&1 ! &'''1 "
'(((RVVXXt$$$1 v%&&&1 ! &'''1 "
'(((RVVXXs###1 v%&&&1 ! &'''1 "
'((( )))$ ' ' '6!"""6!"""6!"""""r c H t j ddddt dz dddddddg }d}d}g d }t ||
}| d t j t j t j t j t j t j
t j g}|D ]x}t t |||
|| t t ||| || yt | | t |
| |d d
D ]x}t t |||
|| t t ||| || y|dd D ]}t g dg d| }t | |d t |
|d t t j dz t j dz t j dz
gg d| } t | |t j dz
t |
|t j dz t j dt j t j j
z z
}
t |
dgddg
|
k sJ t |
dgddg
|
k sJ d S )Nr r r r r r r r r r r r r )@x D ?y@x D @x D r1 r r y @y @r r r )r r' r rK r halfsingledouble
longdoublecfloatcdouble clongdoubler
rW rZ rx rz float64) r r r an10r r float_dtypes float_dtyper r cmaxs r test_minmax_dtypesz,TestMaskedArrayArithmetic.test_minmax_dtypes s Hb"b#r#vr2tS"b"M N N
1
1
1
!"
%
%
%
% BI
ry"*bnN ' , ,K ab DDDHHJJ$S))
+
+
+ ab DDDHHJJ$T**
,
,
,
, RVVXXt$$$RVVXXs### (+ , ,K ab DDDHHJJ$S))
+
+
+ ab DDDHHJJ$T**
,
,
,
, (,
F
FK 9 9 9 +- - -B ;;w#7#7 8 8 8 ;;w#7#7 8 8 8 rvby"&)bfWRZ @yyy$/1 1 1B ;;wrz#:#: ; ; ; ;;rvby#9#9 : : :6B"*!5!5!9 99D $
!Q888<<>>4%GGGG q A777;;==EEEEE
F
Fr c
| j \
}}}}}}}}} }
t t j | t j | t t j | t j | t dt
t d d t dt
t d d t t j |d t
|d t t j t |d d t
|d t t j |d t
|d t t j |d t |d t t j |d t |d t t j t |d d t |d d} | x|_ x|_ x|_ |_ t | dk rt t j
||fd t ||fd t t j |d t j |d t t j |d t
|d t t j |d t |d d S d S )Nr r r r r )r r
r r r
accumulatert r' r8 prodrf rm r r* r s r test_addsumprodz)TestMaskedArrayArithmetic.test_addsumprodC s 15.AsBBAr2RV]]1%%sz!}}555RV&&q))3>!+<+<===QE!HH1---...QE!HH1---...RVAA&&&AA777RVF2qMM222COOODDDRVAq\\3q!99---RWQQ''');););<<<RWQ]]GAqMM222RWVB]]333WRa5H5H5HIII 2333!'3BHrx q66A:: A22KR!4L4L M M M q!,,cjA.>.> ? ? ? 1 s1ayy 1 1 1 A 1
6 6 6 6 6 :r c @ t dgdgdggdgdgdgg }t ddgddgd d
gg }||z }t ddgddgddggd d gd d gd d gg }t || t |j |j t |j |j ||z }t ddgddgd d
ggd d gd d gd d gg }t || t |j |j t |j |j t dgdgdgg }t ddgddgd d
ggd d gd d gd d gg }||z }t d
dgddgddggd d gd d gd d gg }t || t |j |j t |j |j ||z }t d
dgddgddggd d gd d gd d gg }t || t |j |j t |j |j d S )Nr r r FTr r r r5 @r r r r r+ r rr r' r
r r r r r r r# s r test_binops_d2Dz)TestMaskedArrayArithmetic.test_binops_d2DX s B4"t$UGdVdV+D E E E B8b"XBx0 1 11u"bB8b"X6 !fq!fq!f 57 7 7T7###TY ---TY ---1u"bB8b"X6 !fq!fq!f 57 7 7T7###TY ---TY --- B4"t$ % % B8b"XBx0AAA/
1
1
11u!Q!R2q'2 !fq!fq!f 57 7 7T7###TY ---TY ---1u!Q!R2q'2 !fq!fq!f 57 7 7T7###TY ---TY -----r c @ t dgdgdggdgdgdgg }t ddgddgd d
gg }||z }t d d gddgddggd
d
gddgddgg }t || t |j |j t |j |j ||z }t ddgddgd d
ggd
d
gddgddgg }t || t |j |j t |j |j t dgdgdgg }t ddgddgd d
ggd
d
gd
d
gd
dgg }||z }t d d gd dgd dggd
d
gd
d
gd
dgg }t || t |j |j t |j |j ||z }t ddgddgddggd
d
gd
d
gd
dgg }t || t |j |j t |j |j d S )Nr r r FTr r r r5 r r gUUUUUU?r r g?r rm rr r r s r test_domained_binops_d2Dz2TestMaskedArrayArithmetic.test_domained_binops_d2D| s B4"t$UGdVdV+D E E E B8b"XBx0 1 11u'7+b"XBx@ !fq!fq!f 57 7 7T7###TY ---TY ---1u'7+b"XBx@ !fq!fq!f 57 7 7T7###TY ---TY --- B4"t$ % % B8b"XBx0AAA/
1
1
11u&&)FF+;fa[I !fq!fq!f 57 7 7T7###TY ---TY ---1u&&)FF+;fa[I !fq!fq!f 57 7 7T7###TY ---TY -----r c t g dg dd }|dz }t |j g d |dz
}t |j g d |dz }t |j g d |dz }t |j g d d S )Nr r r r Fr shrinkr r r rK r
r r s r test_noshrinkingz*TestMaskedArrayArithmetic.test_noshrinking s
,A,A,A %
'
'
'
EQVYYY''' QQVYYY'''
FQVYYY''' RQVYYY'''''r c t j dg }t t j |d j j d d S )Nr r r~ )r r r' r
true_divider rm r rB s r test_ufunc_nomaskz+TestMaskedArrayArithmetic.test_ufunc_nomask s? EKK R^Aq)).4b99999r c | t j g ddd }t |j g d d S )N)r rm @ ?Fr r )r r rU r
r r s r test_noshink_on_creationz2TestMaskedArrayArithmetic.test_noshink_on_creation s> E E B BQVYYY'''''r c
& | j \
}}}}}}}}} }
t t || t || t || } t | t j || t | j t |j |j t || } t | t j || t | j t t |j |j |dk d S r )r r
r] r r rI ) r r r r r r r r r r r r s r test_modz"TestMaskedArrayArithmetic.test_mod s 15.AsBBAr2SAYYB ,,,2r{{T26"b>>***TY 9 9:::2r{{T26"b>>***TY(A(AB!G M MNNNNNr c t d }t j d }t |dd< | ddd }| ddd }t t j |d t |d t t j |dd t
|dd t t j t |d t |d t || t t j t |d t |d t || t g d
t }t |d< t
|g d }t |d d k t |d dk t |d dk d S )
Nr r r r r r r r r r r )abcr defr r )r r r r )
r r rJ rk r
rx ru rA r8 rd r' rX r )r r r r4 s r test_TakeTransposeInnerOuterz6TestMaskedArrayArithmetic.test_TakeTransposeInnerOuter s 2JJ IbMM!A#
IIaA
IIaA R\!Y//1i1H1HIIIRWQ 1--tAy!/D/DEEERXfQllF1aLL991a[[ " " "RXfQllF1aLL991a[[ " " " )))6 2 2! III !
! ! r c t ddgddg }t |j ddg t |j ddg t |j j |j j j t |j dd g t |j dd g t |j j |j j j d S )
Ny ? $@y 4@ @r r r r r r r )r' r
r r8 r r r )r r s r test_imag_realz(TestMaskedArrayArithmetic.test_imag_real s
GW%QF
3
3
3RWr1g&&&RW^^%%qz222RW]BHM$7888RWq"g&&&RW^^%%r{333RW]BHM$788888r c t t j ddd dd }t
x|d d df< x|d< |d< d}|D ]}t
t | }t
|| }t j dt }| d ||d|
}t ||u t | |d|
t dt } |d|
}t ||u t |d t
u d S ) Nr r r r r r )rt r varstdrW rZ r meanr r6 r
r' r r r rk rJ r r3 r r r r
r )r r funclistr r xmmethoutputr s r test_methods_with_outputz2TestMaskedArrayArithmetic.test_methods_with_output sS
29$$QB//
0
0
8
8A
>
>(..111a4.2a52f:N ) )H R**F R**FXau---F KK VBQF333F Ff$ % % % QF!;!;!; < < <1C(((FV///F Ff$ % % % F1I' ( ( ( ( ) )r c dt fdt fg}t ddgddg| }||k }t |j ddg t |j d d g t |j dk ||d
k }t |j dd g t |j d d g t |j dk t ddgd dg| }||k }t |j d dg t |j dd g t |j dk |d
|k }t |j d d g t |j dd g t |j dk t ddgdd g| }||k }t |j ddg t |j d d g t |j dk dt fdd t fd
t fgfg}t ddgddggddgddgg| }|d |k }t |j dd gd d gg t |j d d gd dgg t |j dk d S )Nr& r r rs r r r TFr r r r r r r r r r r r r r r r r r' r
r r r r7 r r r r r s r test_eq_on_structuredz/TestMaskedArrayArithmetic.test_eq_on_structured s *sCj) 66"&&)9 H H HQTYt ---TY///4'(((QqT TYu
...TY///4'((( 66"&&)9 H H HQTY
...TYu
...4'(((! TY///TYu
...4'((( 66"&&)9 H H HQTYt ---TY///4'((( *sdC[4+$> ?@ K- -/$k2$k24;A
C
C
C $1 TY$ ?@@@TY% % ?@@@4'(((((r c dt fdt fg}t ddgddg| }||k }t |j ddg t |j ddg t |j d k ||d
k }t |j dd g t |j ddg t |j d k t ddgd dg| }||k }t |j d dg t |j d dg t |j d k |d
|k }t |j d d g t |j d dg t |j d k t ddgdd g| }||k }t |j ddg t |j ddg t |j d k dt fdd t fd
t fgfg}t ddgddggddgddgg| }|d |k }t |j dd gd d gg t |j ddgdd gg t |j d k d S )Nr& r r rs r r r FTr r r r r r r r r r r r r s r test_ne_on_structuredz/TestMaskedArrayArithmetic.test_ne_on_structured& s *sCj) 66"&&)9 H H HQTY///TY///4'(((QqT TY
...TY///4'((( 66"&&)9 H H HQTYu
...TYu
...4'(((! TYt ---TYu
...4'((( 66"&&)9 H H HQTY///TY///4'((( *sdC[4+$> ?@ K- -/$k2$k24;A
C
C
C $1 TY% t =>>>TY% % ?@@@4'(((((r c 8
t j d }t dd| t dd| t dd| t dd| fD ]; t }| d }t j d| t d| t dd| t d
d| fD ]
t }| d }||k }t
k | t
k | t |
k | t |k | t ||k |
fd |j D }t t |t | ||k
}t
k | t
k | t |
k | t |k | t ||k |
fd |j D }t t |t
| =d S )
NrU rC r r r r r 2i4r r r c 4 g | ]}| | k S r~ r~ r namer r s r r zITestMaskedArrayArithmetic.test_eq_ne_structured_extra..h & FFFTBtH4 0FFFr c 4 g | ]}| | k S r~ r~ r s r r zITestMaskedArrayArithmetic.test_eq_ne_structured_extra..p r r ) r r r_ r r r' r r
r, r r ) r r[ ma1r1ma2r2 eq_expectedel_by_el ne_expectedr r s @@r test_eq_ne_structured_extraz5TestMaskedArrayArithmetic.test_eq_ne_structured_extraQ s Xg
fB777fB777fB777fB7779 M MB ''+&&C% Bxb111 V2... V&;;; V&;;;=
M
M ggk**XXe__ !Rxnn . . R2X{333 R2X{333 SBY 444 R3Y 444 SCZ555FFFFFRXFFF U84 8 8 8 < < > > LLL !Rxnn . . R2X{333 R2X{333 SBY 444 R3Y 444 SCZ555FFFFFRXFFF U84 8 8 8 < < > > LLLL-
M
M Mr r[ SUr Nr& c ~ t ddg|ddg| }||k }t |j ddg t |j ddg t |j dk ||d k }t |j ddg t |j ddg t |j dk t ddg|ddg| }||k }t |j ddg t |j ddg t |j dk |d |k }t |j ddg t |j ddg t |j dk ||d k }t |j ddg t |j ddg t |j dk d S )Nr r r r r r r7 TFr' r
r r r r7 r r[ r r r r s r test_eq_for_stringsz-TestMaskedArrayArithmetic.test_eq_for_stringss s
3*BaV E E EQTYt ---TY
...4'(((QqT TYu
...TY
...4'((( 3*BaV E E EQTY///TYt ---4'(((! TY///TYu
...4'(((QqT TY///TYu
...4'(((((r c ~ t ddg|ddg| }||k }t |j ddg t |j ddg t |j dk ||d k }t |j ddg t |j ddg t |j dk t ddg|ddg| }||k }t |j ddg t |j ddg t |j dk |d |k }t |j ddg t |j ddg t |j dk ||d k }t |j ddg t |j ddg t |j dk d S )Nr r r r r FTr r s r test_ne_for_stringsz-TestMaskedArrayArithmetic.test_ne_for_strings s
3*BaV E E EQTY///TY
...4'(((QqT TY
...TY
...4'((( 3*BaV E E EQTYt ---TYt ---4'(((! TYt ---TYu
...4'(((QqT TYt ---TYu
...4'(((((r dt1idsdt2r c ~ t ddg|ddg| }||k }t |j ddg t |j ddg t |j dk ||d k }t |j ddg t |j ddg t |j dk t ddg|ddg| }||k }t |j ddg t |j ddg t |j dk |d |k }t |j ddg t |j ddg t |j dk ||d k }t |j ddg t |j ddg t |j dk d S )Nr r r TFr r r r r r r r s r test_eq_for_numericz-TestMaskedArrayArithmetic.test_eq_for_numeric s
1a&1a&T B B BQTYt ---TY
...4'(((QqT TYu
...TY
...4'((( 1a&1a&T B B BQTY///TYt ---4'(((! TY///TYu
...4'(((QqT TY///TYu
...4'(((((r c ~ t ddg|ddg| }||k }t |j ddg t |j ddg t |j dk ||d k }t |j ddg t |j ddg t |j dk t ddg|ddg| }||k }t |j ddg t |j ddg t |j dk |d |k }t |j ddg t |j ddg t |j dk ||d k }t |j ddg t |j ddg t |j dk d S Nr r r FTr r s r test_ne_for_numericz-TestMaskedArrayArithmetic.test_ne_for_numeric s
1a&1a&T B B BQTY///TY
...4'(((QqT TY
...TY
...4'((( 1a&1a&T B B BQTYt ---TYt ---4'(((! TYt ---TYu
...4'(((QqT TYt ---TYu
...4'(((((r opc t ddg|ddg| } ||| }t |j ||j |j t |j ddg t |j dk |||d }t |j ||j |j d t |j ddg t |j dk t ddg|ddg| } ||| }t |j ||j |j t |j ddg t |j dk ||d | }t |j ||j d |j t |j ddg t |j dk |||d }t |j ||j |j d t |j ddg t |j dk d S r )r' r
r r r r r7 )r r r r r r r r s r test_comparisons_for_numericz6TestMaskedArrayArithmetic.test_comparisons_for_numeric s*
1a&1a&T B B Br!QxxTY17AG 4 4555TY
...4'(((r!QqT{{TY17AGAJ 7 7888TY
...4'((( 1a&1a&T B B Br!QxxTY17AG 4 4555TYt ---4'(((r!A${{TY171:qw 7 7888TYu
...4'(((r!QqT{{TY17AGAJ 7 7888TYu
...4'(((((r zN/Ac t g dg d| }t g dg d| }t ||| j ||j |j d S )N)r r cder1 )r r7 )r# r r )rK r
r )r r r r r s r test_comparisons_stringsz2TestMaskedArrayArithmetic.test_comparisons_strings su ,,,999NNN,,,999NNNRRS\\'CIsy)A)ABBBBBr c t 5 }| t d t d dgddg }t |d k t ddgddg t |j d k ddg t |d k t ddgddg t d dgd }t |d k ddg t |d k ddg t d dgd }t |d k t ddgd t |d k t ddgd t }t |d k t d d d d S # 1 swxY w Y d S )NzComparison to `None`r r r TFr )r r r r' r
r rJ r s r test_eq_with_Nonez+TestMaskedArrayArithmetic.test_eq_with_None# s
,C JJ}&< = = =tQiq!f---A dE4-q!f$E$E$E F F F 4 $ 7 7 7 dE5$-q!f$E$E$E F F FtQie,,,A dT5M 2 2 2 dUDM 2 2 2tQid+++A dE5$-d$C$C$C D D D dE4-d$C$C$C D D DA dF + + +! , , , , , , , , , , , , , , , , , ,s EE22E69E6c t d }t |dk d t |dk d t |dk d t |dk d t dd }t |dk t t |dk t t |dk t t |dk t d S )Nr Tr Fr )r' r
rJ r s r test_eq_with_scalarz-TestMaskedArrayArithmetic.test_eq_with_scalar< s !HHQ!VT"""Q!VU###Q!VU###Q!VT""" !$ Q!VV$$$Q!VV$$$Q!VV$$$Q!VV$$$$$r c t ddgddg }t ddgddgg t j ddgddgg fD ]>}||k }t |j ddgddgg t |j ddgddgg ?d S )Nr r r r FT)r' r r
r r )r r r r s r test_eq_different_dimensionsz6TestMaskedArrayArithmetic.test_eq_different_dimensionsH s
Aq6A
'
'
'1a&1a&)**8aVaV ,--/ 5 5B"HD eU^&*E]%4
5
5
5 eT]&+T]%4
5
5
5
5 5 5r c t g dg d }t t j t j dt j d dgg d }t | }t || t |j |j t |j g d t j | }t || t |j |j t |j g d d S )Nr r r r r r r r r r r r r r r )rK r rP rE r
r )r r r# r s r test_numpyarithmeticz.TestMaskedArrayArithmetic.test_numpyarithmeticS s ))) @ @ @ 26!99b A$3OO5 5 5 1vvT7###TY ---QV___---vayyT7###TY ---QV___-----r )8r r r r rj rp rt rx 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 parametrizer r num_dtsnum_idsr r operatorleltgegtr! r$ r& r( r* r. r~ r r ra ra s 5 5 5"% % %!: !: !:F5 5 5$2 2 25 5 5 & & &) ) )
,
,
,' ' '& & &5 5 587 7 72& & &
6 6 627 7 7$ $ $ $.# # #,(F (F (FT7 7 7*". ". ".H". ". ".H( ( (": : : ( ( (
O O O (9 9 9) ) ),)) )) ))V)) )) ))V M M MD [ TC:.. [ VdC[11 ) ) 21 /. )< [ TC:.. [ VdC[11 ) ) 21 /. )< [ UG 99 [ UG 99 [ VdAY// ) ) 0/ :9 :9 )< [ UG 99 [ UG 99 [ VdAY// ) ) 0/ :9 :9 )< [ UG 99 [ UG 99 [ VdAY// [ T
[(+x{HK @B B ) )B B 0/ :9 :9
)< [ T
[(+x{HK @B B [ VdE]33C C 43B BC, , ,2
%
%
% 5 5 5. . . . .r ra c > e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d S )
TestMaskedArrayAttributesc t g dg d }t | }t |j |j t |g dd }t |j g d t |g dd }t |j g d t |g d }t |j g d d S ) Nr r r r1 F)r keep_maskTr r r r rO s r
test_keepmaskz'TestMaskedArrayAttributes.test_keepmaskf s 3 3 3
!__RWaf%%%
!)))u
=
=
=RWiii(((
!)))t
<
<
<RWiii(((
!)))
,
,
,RWiii(((((r c D t d }g d}t | }t ||d }t ||dd }ddg|d d
g< ddg|d d
g< t |j g d t |j g d t |j g d
t
|j t
|j g d|d d
< g d|d d
< t |j g d t |j g d t |j t t |d< t |d< t |j g d t |j g d d |d d < d |d d < t |j g d t |j g d t |j g d t |j t |
t d |d d < t |j g d t |j t | t ||dk < t |j g d t |j g d d|t |d k d < t |j g d t |j g d t d dgdd
ggd dgddggd }d|d< t |j d dgdd
gg t |j d dgddgg d|d< t |j d dgddgg t |j d dgddgg d|t |dk d < t |j d dgddgg t |j d dgddgg d S ) Nr r r r r r Tr hard_maskFr rA r r ( r r )r r r r r )r r r r rC rd r r )r r r r r )r r r rE rC 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 rG r' r
r r r r8 r` rJ soften_mask harden_maskr r8 )r r r rB xhxss r
test_hardmaskz'TestMaskedArrayAttributes.test_hardmasks s 1II OO aLL
11
-
-
-
11D
9
9
9"XAq6
"XAq6
RX / / /000RX 0 0 0111RWooo... BL !!! ,,1Q3 ,,1Q3RX 0 0 0111RX 2 2 2333RWf%%%11RWooo...RWooo...111111RX///RX///RWooo...RWf%%%
q 111RX///RWf%%%
26
RX///RX///$%6"q&% !RX///RX///
QFQF#Aq6Aq6*:d
K
K
K1RXAA /000RXAA /0006
RXAA /000RXAA /000$%6"q&% !RXAA /000RXAA /00000r c t d }g d}t | }t ||d }d|dd< d|dd< t |j g d d S )
Nr r? Tr@ r r r r )r r r r r )r rG r' r
r r r r rB rH s r test_hardmask_againz-TestMaskedArrayAttributes.test_hardmask_again sn 1II OO aLL
11
-
-
-1Q31Q3RX 0 0 011111r c t g dg d }| }t || d|d< t || t |t g dg d | }d|d< t || t |t g dg d d S )Nr r r r r r r r )r' rG r
rF r s r test_hardmask_oncemore_yayz4TestMaskedArrayAttributes.test_hardmask_oncemore_yay s
)))))) , , ,
MMOOQ!QQiiiiii888999
MMOO!QQiiiiii88899999r c t d }t |d< d|d< t |j t t d }d|_ t |d< d|d< t |j t
d d S )Nr r F)r rJ r
r r`
_smallmaskrz r s r test_smallmaskz(TestMaskedArrayAttributes.test_smallmask ss 2JJ!!QWf%%% 2JJ !!QWeBii(((((r c t g dg d }| }t || t |j t t
j dgdt fdt fg }| }| t |j |j d S )Nr r r r r r )
r' shrink_maskr
r r` r r r r r r s r test_shrink_maskz*TestMaskedArrayAttributes.test_shrink_mask s )))))) , , ,
MMOOQQVV$$$
EKK
c3Z#u $> ? ?
FFHH
QVQV$$$$$r c T t g dg dgdt fdt fdg }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 ]7}t ||||f |dz
}||j d k rd }|dz
}8d S )N))r r? one)r r@ two)r ffffff
@thr))r 皙@fou)r @fiv)r gffffff@sixr r r r r r r r )r r r r r r r r r r r rC r )r' r r rJ r r
rm )r r xflatr jr s r test_flatz#TestMaskedArrayAttributes.test_flat s
FFFFFFH sc5\<@
B
B
B #t #t #t & U1Xqw'''U1Xqw'''U1Xqw'''U2A2Y!%%%U1Xqw'''U1Xqw'''U1Xqw'''U122Y!%%%U2Y& ***
B Qq!tW % % %
FAAGBK Q r c t j dd t j }t j d |_ t | t
|j t j d fd}t t | dt j j
}t
|j t j d t j dd t j }t j d |_ t
|j t j d t
|j t j j d S )Nr zf4,i4r f4c x t j } t j d | _ d S )Nf8)r r r' r )rB r s r assignz;TestMaskedArrayAttributes.test_assign_dtype..assign s' AAhtnnAGGGr )r rh )
r rz r r' r rg r
r r r r r r` )r rB ri r r s @r test_assign_dtypez+TestMaskedArrayAttributes.test_assign_dtype s+ HQg & & & EKKNN(4.. QQWbhtnn--- % % % % % j&)))
FFBE$5F 6 6QWbhtnn---
HQd # # # EKKNN(7##QWbhw / /000QWbel+++++r N) r r r r= rJ rM rP rS rV rd rj r~ r r r9 r9 d s ) ) )51 51 51n2 2 2 : : :
)
)
) % % % <, , , , ,r r9 c z e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zd Zd Zd Zd ZdS )TestFillingValuesc t j j j } |dt }t |d |d t }t |t
d |dd }t |d |d d }t |t
d t t |dt t t |dt d S )Nr |S3 0s camelot!r stuff) r r r _check_fill_valuer r
r/ r r3 )r rq fvals r test_check_on_scalarz&TestFillingValues.test_check_on_scalar s EJ 8 C((T1 s++T-a00111 E**T4 u--T-k::;;;i!2E3???i!2GSAAAAAr c
r t j j j }dt fdt
fdg} |g d| }t
t |t t |
g d |d | }t
t |t t |
t d t d t t d g t j
d| } ||| }t
t |t t |
g d t j
dd
t fd t
fd g } ||| }t
t |t t |
g d t j d
t }d|d< ||t }t
t |t t |
g d dt fg} |d| }t
t |t t |
d d S )Nr r )r rn )r )gr+ )r ru ???r r r r r& r )rF rn rL rm r 6e)rx )r r r rq r r r r r r
itemr/ r
r' rX )r rq r rr fill_vals r test_check_on_fieldsz&TestFillingValues.test_check_on_fields& s EJ 8*sElL9 !;!;!;VDD
4))***TYY[["="="=>>> v..
4))***TYY[[#5a#8#8#5b#9#9#*+=c+B+B#C#C#E F F F 8 6fEEE 622
4))***TYY[["="="=>>>
8 6$':U|\"JL L L 622
4))***TYY[["="="=>>> :D7771 622
4))***TYY[["="="=>>> * V44
4))***TYY[[-00000r c $ t g d }|j ddi t |t }t |j g d t |j t d t |t }t |j g d t |j t d | t }t |j g d t |j t d t |j d d | dg }t |d j |j t |d j |j d S )
N) 3 4 5commentzupdated!r )r r r r r )r rn r )
r' r- updater r
r r7 r/ r r r s r test_fillvalue_conversionz+TestFillingValues.test_fillvalue_conversionU sZ
$$$ % %
9Z0111 !3 QWiii(((Q\#5a#8#8999 !5 ! ! !QWiii(((Q\#5b#9#9:::
HHSMMQWiii(((Q\#5a#8#8999QZ *J777
HHl^ $ $QsV\17+++QsV& 55555r c t d }t t j d }t t j d j }t || t || d S )Nr )r/ r r' r r
)r r( f2f3s r test_default_fill_valuez)TestFillingValues.test_default_fill_valuem sb
#
#
-
-
2
3
3R R r c \ t dgdt fddt fg }t | }t |j }t j t d t d t d f|j }t
|| t
|| d S ) Nr r r r
r r r r )r' r r r/ r r r
)r fieldsr( r expecteds r "test_default_fill_value_structuredz4TestFillingValues.test_default_fill_value_structuredu s { !3Z U| DF F F
'
'
-
-8 / 2 2 / 4 4 / 3 3 5>>'(( .r 2 24#'- 2 2 228&.. &&tV44g>>>(4u
777>>vFF&&q&117;;;233( .q 1 1 .r 2 2 .r 2 2 5 "56 6 6 7=fVnn ! v..T7###(9,?@@@GGOO&&q&117;;;
! !QtW',a00000r c dD ]}t j d| }t t j d|z dz }t j || t j d| }t t j d|z dz }t j || d S )N)
asfspsnsusmsr rB hDWr YNaTz > >!!$$TZ)))T!Wbga);AgJ)G)G H HIII!!$$TZ)))T!Wbga);AgJ)G)G H HIIIIIr c dt fdt fg}t t t g dg d d| }|d }| d t |j t j d t t |j d d |j d< t t |j d
t t t g dg d | }|d }| d t |j t j d t t |j dt d f d S )
Nr r r r r r7 r r )r r )r r r r ) r r' r r r r
r r r r7 r/ )r r r aar4 tts r test_fillvalue_individual_fieldsz2TestFillingValues.test_fillvalue_individual_fields sQ *sCj) $s999iii0011 )
9
9
9
sV
" R^RXb\\222U1<((*555 SU1<(()444 $s999iii0011 @ @ @
sV
" R^RXb\\222U1<((2/A!/D/D*EFFFFFr c dt f}dt f}t ddgddgt j ft j |g }t |j ||g }|d |d< |d |d j |j d }t t j |d t |d
t d d S ) Nr r r r )F)r r7 r r r~ r r r )r r' r rP r r3 rm r r7 r r r r
r/ )r r adtyper r r
s r (test_fillvalue_implicit_structured_arrayz:TestFillingValues.test_fillvalue_implicit_structured_array
s u u 5%.('; fYbhx.@.@
B
B
B !'&&!1 2 2 23# # af /000
M" 1 QrU .r 2 233333r c Z t dd }t |j d t ddt }t |j d t ddt }t |j d t ddt }t |j d d S )Nr r r r r )r3 r
r7 rb r rz complexr@ r s r test_fillvalue_as_argumentsz-TestFillingValues.test_fillvalue_as_arguments s ! % % %Q\4((( t5 1 1 1Q\4((( !' 2 2 2Q\2&&& Q2W 5 5 5Q\2&&&&&r c t d }t |j d t dt }t |j d t dt }t |j d d S )Nr rm rw )r3 r
rm rb r rz r r s r test_shape_argumentz%TestFillingValues.test_shape_argument( st
QWe$$$ uE * * *QWe$$$ W - - -QWe$$$$$r c " t g ddt j }| }t |j dk | t }t |j dk | t }t |j dk | t j }| t j }| t d }t |j dk | t d }t |j dk | t j }t |j d k d S )
Nr r r )rh r r )rh r7 r rS ) r' r int64r r r7 r r int32r. s r test_fillvalue_in_viewz(TestFillingValues.test_fillvalue_in_view4 sH
))) : : :
FFHH !"""
FF; !"""
FF F $ $ !"""
FF2:
FF
F # #
FF;1F - - !"""
FF F 2 2 !"""
FFF " " &'''''r c t dd }t |d j t d t |d j t d d S )Nr z (2)3S,(2)3Urw r' s spamr( eggs)r3 r
r7 r/ r s r test_fillvalue_bytes_or_strz-TestFillingValues.test_fillvalue_bytes_or_str^ s\
] 3 3 3QtW');G)D)DEEEQtW');F)C)CDDDDDr N)r r r rs r{ r r r r r r r r r r r r r r r r r~ r r rl rl s5 B B B -1 -1 -1^6 6 60
#
#
#M M M
4 4 40 , , , 1 1 1>
3
3
3 : : :0 J J JG G G& 4 4 4 ' ' '
%
%
%(( (( ((TE E E E Er rl c > e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d S )
TestUfuncsc t dddt dz gdz ddgdgdz z t dddt dz gdz ddgdgdz z f| _ t j | _ t j dd d S )
Nr r r r r r r rK rc )r' r r r rd re rf r s r r zTestUfuncs.setup_methodi s aR!V,q01va7GHHHaR!V,q01va7GHHHK )++
8444444r c 2 t j di | j d S rh ri r s r rj zTestUfuncs.teardown_methodp rk r c dD ]} t t | }n%# t $ r t t | }Y nw xY wt t j j | }| j d |j } || } || }t |
d |
d | t |j |j | d S )N)*rr rE rF r6 r+ rn r, rv r$ r! r% ro r- rw arcsinhr" arctanhr fabsnegativefloorceil logical_notr rs r^ r1 r floor_divide remainderfmodhypotr# r5 ra rD r? rC r> logical_and
logical_or logical_xorr )err_msg)
r umathAttributeError fromnumericr" r r r ninr
r8 r r )r r
ufmfargsurmrs r test_testUfuncRegressionz#TestUfuncs.test_testUfuncRegressions s ; ;A"
- UA&&!
-
-
- [!,,
- **B6'26'?DTBTB 1 ryy||Q 7 7 7 bgrw : : : : :5 ; ;s >>c t | j d }t t |d t t |d t t |d d d d t t
|d d t t j | t d S )Nr r r )
r r r rp r
rt rf r r r r s r test_reducezTestUfuncs.test_reduce s F1IGAA&&&&'''###$$$S2A2Q'''+++WQQ'''+++SZ]]B'''''r c t dd dd }t |dk | }t | | t | d t | d | d t | d g d t
| d d j t
| d d j d S )Nr
r r r r )r r rr r+ )r rk rV r
rW rZ r r )r r amasks r test_minmaxzTestUfuncs.test_minmax s 1bMM ! !!Q ' ' QUA&&UYY[[!%%''***UYY[[!$$$UYYq\\15588,,,UYYq\\<<<000 ! Q$%%% ! Q$%%%%%r c P t g dg d }t j | }t dddt j d dgg d }t || t |j |j t t
|j t d S ) Nr, r r r r r r r r r r r )rK r rr r
r r r r )r r r r# s r test_ndarray_maskz TestUfuncs.test_ndarray_mask s ))) @ @ @wqzz Aq"'!**b 9$3OO5 5 5T7###TY ---Jty+66677777r c t ddgddg t t t j d t t t j d G fdd } | }t
|z d k t
|z d
k G d d } | }t
| t u t
t t j | d
k t
| t u t
|z dk t
|z d
k t
|z dk d S )Nr r r r r r c 0 e Zd Z j dz Zd Zd ZdS ).MyClassr c dS )NMy mulr~ r others r __mul__zDTestUfuncs.test_treatment_of_NotImplemented..MyClass.__mul__ xr c dS )NMy rmulr~ r s r __rmul__zETestUfuncs.test_treatment_of_NotImplemented..MyClass.__rmul__ yr N)r r r __array_priority__r r r s r MyClassr sA !"!5!9
!
!
!
!
!r r r r c ( e Zd ZdZd Zd Zd ZeZdS )=TestUfuncs.test_treatment_of_NotImplemented..MyClass2d c dS )NMe2mulr~ r s r r zETestUfuncs.test_treatment_of_NotImplemented..MyClass2.__mul__ r r c dS )NMe2rmulr~ r s r r zFTestUfuncs.test_treatment_of_NotImplemented..MyClass2.__rmul__ r r c dS )NMe2rdivr~ r s r __rdiv__zFTestUfuncs.test_treatment_of_NotImplemented..MyClass2.__rdiv__ r r N)r r r r r r r __rtruediv__r~ r r MyClass2r sH !$
!
!
!
!
!
! $LLLr r r r r
)
rK r r3 r3 multruedivr r NotImplementedr r^ rd __truediv__)r r mer me_toor s @r test_treatment_of_NotImplementedz+TestUfuncs.test_treatment_of_NotImplemented s
"bA / / /i q%888i!11e<<< ! ! ! ! ! ! ! ! ! ! WYYQ("###B)#$$$ $ $ $ $ $ $ $ $ &!!^3444HN1f--:;;<<<
f%% 7888
h&'''F
i'(((F
i'(((((r c " t j dt j gddg }t j 5 t j d t | t |d |dk t | t | t | t | t | t | t |d t! |d d d d d S # 1 swxY w Y d S )Nr r r r errorr )r r r' rP warningscatch_warningsfilterwarningsr6 r rr rE rv r$ r! r" r1 r r s r test_no_masked_nan_warningsz&TestUfuncs.test_no_masked_nan_warnings s1
EKKbf
QqEK 2 2
$
&
& #G , , ,
FFF 1III
EE
GGG FFF FFF 1III 1III AJJJ
1aLLL
Q ) s B4DD DN) r r r r rj r r r r r r r~ r r r r f s 5 5 5% % % ; ; ;<( ( ( & & &8 8 8)) )) ))V r r c e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Z d Z d Z d Z d Z d S ) TestMaskedArrayInPlaceArithmeticc n t d }t d }t d }t |d< |||f| _ | t | t | t f| _ t j d t j d z | _ d | j D | _ | t j | t j | t j f| _
d S )Nr r
AllIntegerAllFloatc @ g | ]}t j | j S r~ r r rh )r r s r r zATestMaskedArrayInPlaceArithmetic.setup_method..
s# EEE28A;;+EEEr ) r rJ intdatar r floatdatar typecodes
othertypesuint8 uint8datar r r r s r r z-TestMaskedArrayInPlaceArithmetic.setup_method s 2JJ 2JJ
BZZ11bz ((5//188E??BIIe | j \ }}}|j }t dt }t |d< ||z }||z }t |||z t |||z t |j t
t
||j |dk d S )Nr r r r r7 r2 s r !test_inplace_division_array_floatzBTestMaskedArrayInPlaceArithmetic.test_inplace_division_array_float^
s ^
Ar G 2U # # #" Q
aQA RQ RWgga&8&816 C CDDDDDr c ddddt dz dddddddg }g d }g d
}g d }t || }t || }||z }t |j g d
t |j ddddt dz dddddddg | }||z }t |j g d
t |j ddddt dz dddddddg 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 rK r
r r r )r r r r r r r r s r test_inplace_division_miscz;TestMaskedArrayInPlaceArithmetic.test_inplace_division_miscj
s>
Rb2gr2tS"b" E A A A
1
1
1
1
1
1
!"
%
%
%
!"
%
%
% GQW B B BCCCQW"b#sRxRRRL N N N WWYY
bRX C C CDDDQW"b#sRxRRRL N N N N Nr c h t g dg d }|dz }t |j g d t |j g d |dz
}t |j g d t |j g d t g dg d }|t g dg d z }t |j g d t |j g d t g dg d }|t g dg d z
}t |j g d t |j g d d S ) Nr rN r r )r r r r r r r r r r s r test_datafriendly_addz6TestMaskedArrayInPlaceArithmetic.test_datafriendly_add~
sV )))))) , , ,
URWiii(((RWiii((( QQVYYY'''QVYYY''' )))))) , , ,
yyyyyy111
1RWiii(((RWiii((( )))))) , , , U999999
-
-
--QVYYY'''QVYYY'''''r c t g dg d }|dz
}t |j g d t |j g d t g dg d }|dz }t |j g d t |j g d t g dg d }|t g dg d z
}t |j g d t |j g d t g dg d }|t g dg d z }t |j g d t |j g d d S ) Nr rN r r )r r r r r r r r s r test_datafriendly_subz6TestMaskedArrayInPlaceArithmetic.test_datafriendly_sub
sr
)))))) , , ,
URWiii(((RWiii((( )))))) , , , QQVYYY'''QVYYY''' )))))) , , ,
yyyyyy111
1RWiii(((RWiii((( )))))) , , , U999999
-
-
--QVYYY'''QVYYY'''''r c t g dg d }|dz }t |j g d t |j g d t g dg d }|dz }t |j g d t |j g d t g dg d }|t g dg d z }t |j g d t |j g d t g dg d }|t g dg d z }t |j g d t |j g d d S )
Nr rN r r r r r rD r )r rC r r r r s r test_datafriendly_mulz6TestMaskedArrayInPlaceArithmetic.test_datafriendly_mul
sr
)))))) , , ,
URWiii(((RWiii((( )))))) , , , QQVYYY'''QVYYY''' )))))) , , ,
|||)))444
4RWjjj)))RWiii((( )))))) , , , U<<@r )r g?r r r r s r test_datafriendly_divz6TestMaskedArrayInPlaceArithmetic.test_datafriendly_div
s~
)))))) , , ,
VRW 1 1 1222RWiii((( ,,,YYY / / / RQV 0 0 0111QVYYY''' ,,,YYY / / /
YYY777
7RW 0 0 0111RWiii((( ,,,YYY / / / U???
3
3
33QV . . .///QVYYY'''''r c t g dg d }|dz }t |j g d t |j g d |dz }t |j g d t |j g d d S )Nr rN r rm )r ;f@r )r rQ r r r s r test_datafriendly_powz6TestMaskedArrayInPlaceArithmetic.test_datafriendly_pow
s
,,,YYY / / /
#XRW 1 1 1222RWiii((( c QV / / /000QVYYY'''''r c t ddgddgg }t ddgddg }||z
}t |ddgddgg |j t urt |j ddgddgg t ddgddgg }t ddgddg }||z
}t |ddgddgg t |j ddgddgg d S )Nr r r r r r r' r
r r` r s r test_datafriendly_add_arraysz=TestMaskedArrayInPlaceArithmetic.test_datafriendly_add_arrays
Aq6Aq6" # # 1a&1v & & & QQ!Q!Q())) 6 1a&1a&!1 2 2 2 Aq6Aq6" # # 1a&1v & & & QQ!Q!Q()))QVq!fq!f -.....r c t ddgddgg }t ddgddg }||z }t |ddgddgg |j t urt |j ddgddgg t ddgddgg }t ddgddg }||z }t |ddgddgg t |j ddgddgg d S )Nr r r r r rT r s r test_datafriendly_sub_arraysz=TestMaskedArrayInPlaceArithmetic.test_datafriendly_sub_arrays
rV r c t ddgddgg }t ddgddg }||z }t |ddgddgg |j t urt |j ddgddgg t ddgddgg }t ddgddg }||z }t |ddgddgg t |j ddgddgg d S )Nr r r r rT r s r test_datafriendly_mul_arraysz=TestMaskedArrayInPlaceArithmetic.test_datafriendly_mul_arrays rV r c | j D ]t j 5 t j d fd| j D \ }}}t
|d< | d z
}t
|| d z | d z
}t
|| d z d d d n # 1 swxY w Y d S )Nr c 3 B K | ]}| V d S r r r r r4 s r zUTestMaskedArrayInPlaceArithmetic.test_inplace_addition_scalar_type.. - B Baahhqkk B B B B B Br r r )r( r r r r* rJ r
r r r r r4 s @r !test_inplace_addition_scalar_typezBTestMaskedArrayInPlaceArithmetic.test_inplace_addition_scalar_type s + +A(**
+
+'000 B B B B4> B B B
Ar 1QQqTT QAAaDD)))aadd
RQQqTT***
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+ + +s B
B44B8 ;B8 c | j D ]Ҋt j 5 t j d fd| j D \ }}}|j }t
d }t |d< ||z
}||z
}t |||z t |||z t |j t ||j d d d n # 1 swxY w Y d S )Nr c 3 B K | ]}| V d S r r] r^ s r r_ zTTestMaskedArrayInPlaceArithmetic.test_inplace_addition_array_type.. r` r r r r
r( r r r r* r r rJ r
rI r r r r rB r r4 s @r test_inplace_addition_array_typezATestMaskedArrayInPlaceArithmetic.test_inplace_addition_array_type / : :A(**
:
:'000 B B B B4> B B B
ArG2Q''' "Qa QA&&& RQ''' RWga&8&8999
:
:
:
:
:
:
:
:
:
:
:
:
:
:
: : : B&CC C c p | j D ]t j 5 t j d fd| j D \ }}}| d z }t || d z
| d z }t || d z
d d d n # 1 swxY w Y d S )Nr c 3 B K | ]}| V d S r r] r^ s r r_ zXTestMaskedArrayInPlaceArithmetic.test_inplace_subtraction_scalar_type../ r` r r r( r r r r* r
ra s @r $test_inplace_subtraction_scalar_typezETestMaskedArrayInPlaceArithmetic.test_inplace_subtraction_scalar_type* + +A(**
+
+'000 B B B B4> B B B
ArQQqTT QAAaDD)))aadd
RQQqTT***
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+ + + B B**B. 1B. c | j D ]Ҋt j 5 t j d fd| j D \ }}}|j }t
d }t |d< ||z }||z }t |||z
t |||z
t |j t ||j d d d n # 1 swxY w Y d S )Nr c 3 B K | ]}| V d S r r] r^ s r r_ zWTestMaskedArrayInPlaceArithmetic.test_inplace_subtraction_array_type..: r` r r r r re rf s @r #test_inplace_subtraction_array_typezDTestMaskedArrayInPlaceArithmetic.test_inplace_subtraction_array_type5 rh ri c p | j D ]t j 5 t j d fd| j D \ }}}| d z }t || d z | d z }t || d z d d d n # 1 swxY w Y d S )Nr c 3 B K | ]}| V d S r r] r^ s r r_ z[TestMaskedArrayInPlaceArithmetic.test_inplace_multiplication_scalar_type..I r` r r rl ra s @r 'test_inplace_multiplication_scalar_typezHTestMaskedArrayInPlaceArithmetic.test_inplace_multiplication_scalar_typeD rn ro c | j D ]Ҋt j 5 t j d fd| j D \ }}}|j }t
d }t |d< ||z }||z }t |||z t |||z t |j t ||j d d d n # 1 swxY w Y d S )Nr c 3 B K | ]}| V d S r r] r^ s r r_ zZTestMaskedArrayInPlaceArithmetic.test_inplace_multiplication_array_type..T r` r r r r re rf s @r &test_inplace_multiplication_array_typezGTestMaskedArrayInPlaceArithmetic.test_inplace_multiplication_array_typeO rh ri c L d t j d D }| j D ]t j 5 t j d fd| j D \ }}}t d d z }t d d z }t |d< | d z }| d z }t || t || n$# t $ r d d }|v s
J | Y nw xY wd d d n # 1 swxY w Y d S )
Nc @ h | ]}t j | j S r~ r$ r r4 s r z[TestMaskedArrayInPlaceArithmetic.test_inplace_floor_division_scalar_type..a # IIIArx{{'IIIr Complexr c 3 B K | ]}| V d S r r] r^ s r r_ z[TestMaskedArrayInPlaceArithmetic.test_inplace_floor_division_scalar_type..e r` r r r r Supported type throwing TypeError) r r' r( r r r r* r rJ r
r3 )r unsupportedr r r msgr4 s @r 'test_inplace_floor_division_scalar_typezHTestMaskedArrayInPlaceArithmetic.test_inplace_floor_division_scalar_type^ s JI i1HIII 1 1A(**
1
1'000 B B B B4> B B B
Ar2Q'''!!A$$.Ba(((11Q44/ 11!!A$$JA11Q44KB A&&& Q'''' 1 1 1BABBBC +++S+++++1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1 1 1s6 A0D+.u r} r r~ r c 3 B K | ]}| V d S r r] r^ s r r_ zZTestMaskedArrayInPlaceArithmetic.test_inplace_floor_division_array_type..y r` r r r r r r r )
r r' r( r r r r* r r rJ r
rI r3 ) r r r r r rB r r r4 s @r &test_inplace_floor_division_array_typezGTestMaskedArrayInPlaceArithmetic.test_inplace_floor_division_array_typer s JI i1HIII 1 1A(**
1
1'000 B B B B4> B B B
ArG2Q''' " 1!GA1HB AF+++ Q!V,,, 16 2 2Q!!A$$Y@@ ! 1 1 1BABBBC +++S+++++1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1 1 1s7 AD.
A3C>=D.> D D. D D..D2 5D2 c | j D ]Ɋt 5 }| t fd| j D \ }}}t d d z }t d d z }t |d< | d z }t || nA# t t f$ r-}t j t | d Y d }~nd }~ww xY w | d z }t || nA# t t f$ r-}t j t | d Y d }~nd }~ww xY wt t j r(t t! |j dd d n't t! |j d d d d d d n # 1 swxY w Y d S )
Nc 3 B K | ]}| V d S r r] r^ s r r_ zUTestMaskedArrayInPlaceArithmetic.test_inplace_division_scalar_type.. r` r r r r r
stacklevelFailed on type=.r )r( r recordr r* r rJ r
DeprecationWarningr3 r warnr
issubclassr integerr rE )r r r r r er4 s @r !test_inplace_division_scalar_typezBTestMaskedArrayInPlaceArithmetic.test_inplace_division_scalar_type s_ J JA"$$
J
;''' B B B B4> B B B
Ar2Q'''!!A$$.Ba(((11Q44/ 181IA A&&&&*I6 8 8 8 M#a&&Q77777777788!!A$$JB Q''''*I6 8 8 8 M#a&&Q7777777778 a,, J SW q2HA2H2H2HIIII SW q2HA2H2H2HIII?
J
J
J
J
J
J
J
J
J
J
J
J
J
J
J J Jsg A6G B0/G0C.#C)$G)C..G2 DGE"#E
G
EA,GG G c | j D ]ފt 5 }| t fd| j D \ }}}|j }t
d }t |d< ||z }t |||z nA# t t f$ r-}t j t | d Y d }~nd }~ww xY w ||z }t |||z t |j t t ||j | d k nA# t t f$ r-}t j t | d Y d }~nd }~ww xY wt t j r(t t% |j dd d
n't t% |j dd d
d d d n # 1 swxY w Y d S ) Nc 3 B K | ]}| V d S r r] r^ s r r_ zTTestMaskedArrayInPlaceArithmetic.test_inplace_division_array_type.. r` r r r r r r r r r r )r( r r r r* r r rJ r
r r3 r r r rI r r r r rE ) r r r r r rB r r r4 s @r test_inplace_division_array_typezATestMaskedArrayInPlaceArithmetic.test_inplace_division_array_type ss # J # JA"$$ "
J
;''' B B B B4> B B B
ArG2Q''' "8FA AE*****I6 8 8 8 M#a&&Q77777777788!GB QU+++ 16 2 2Q!!A$$Y@@ +I6 8 8 8 M#a&&Q7777777778 a,, J SW q2HA2H2H2HIIII SW q2HA2H2H2HIIIE"
J "
J "
J "
J "
J "
J "
J "
J "
J "
J "
J "
J "
J "
J "
J# J # Jsh AG /BG C#C<G CG
AD&%G &E$7#E G E$$A,G G #G c | j D ]}t j 5 t j d t g dg d| }| |d z }t g dg d| }t |j |j t |j |j | |d z }t |j |j t |j |j d d d n # 1 swxY w Y d S )Nr r rN r r rG )r( r r r r' r
r r )r r4 r r xx_rs r test_inplace_pow_typez6TestMaskedArrayInPlaceArithmetic.test_inplace_pow_type s= 0 0A(**
0
0'000))))))1===!!A$$Y ^^^)))1EEE RWdi000 RWdi000aadd
QVTY/// QVTY///
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0 0 0s CC..C2 5C2 N)!r r r r r. r3 r5 r8 r; r= r? rA rC rE rH rJ rM rO rR rU rX rZ rb rg rm rr ru rx r r r r r r~ r r r r s
2
2
2
2
2
2
2
2
2 & & &
E
E
EN N N(( ( (,( ( (.( ( (.( ( (.
(
(
( / / / / / / / / /
+
+
+
:
:
: + + +
:
:
: + + +
:
:
: 1 1 1(1 1 10"J "J "JH%J %J %JN0 0 0 0 0r r c e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zed Zd Zd Zd Zd Zej dd ej dd d Zd Zd Z d Z d Z d Z d Z d Z!d Z"d! Z#d" Z$d# Z%d$ Z&d% Z'd& Z(d' Z)d( Z*d) Z+d* Z,d+ Z-d,S )-TestMaskedArrayMethodsc
t j g d }| dd }| dddd }t j g d }t || }t || |j }t || |j }t j g d }t || } t || |j }
t || |j } |||||||| |
| f
| _ d S N$g @gGz. @g-!@g !@g&1 ?ʡE@\( @ Q @{G#@Pn@h|?!@!rh@jt@J +@zGa@!rh@ʡE@-@)\(@(\B#@ + @uV @~j@g"~@gK7 @
ףp=J#@Gz@Mb@}?5^#@羟/$@!rhm @-#@J +@Mb@ʡ@g-?r r r $r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r $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' rk rm r r r XXXrB rO mXmXXr m2xm2Xm2XXs r r z#TestMaskedArrayMethods.setup_method Y H @ @ @
A
A
IIaOO
YYq!Q
"
" H ' ' '
(
(
"
"
"
!' 2 2
3
3
3!))BH"5"5666
X ( ( ( ) ) $$$AG!4!4555"2::bh#7#7888QAr2sCd;r c
t g d }t | |j t | |j t | |j t | |j t | ddddd |j ddddd t | g d |j g d 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 r r argmaxargminr) compressconjr+ diagonalrt ru rx r r rB s r test_generic_methodsz+TestMaskedArrayMethods.test_generic_methods s ))) QUUWWagkkmm,,,QUUWWagkkmm,,,QXXZZ !1!1222QXXZZ !1!1222QXXaAq!,,agnnQ1a.K.KLLLQZZ **AG,<,r r T)rL Fr ) r r r r r rx rK rJ iinfoint_rZ r s r
test_allclosez$TestMaskedArrayMethods.test_allclose sQ INN2
r""T) )A v!HQNN"###v!A OO"AD111222HQ6666777 T !AD111222
"(27++/0 @ @ @A r c t j g dgd }t ||d sJ t ||t j dd sJ d S )Nrc zm8[ns]r r )atolr r )r r' r r r s r test_allclose_timedeltaz.TestMaskedArrayMethods.test_allclose_timedelta2 sf
Hlll^8 4 4 411%%%%%%12>!T#:#:;;;;;;;;r c ( t j g dg dg dg }t j g dg dg dgt j }t || }|d k }|d k }t | t | t | d
g d t | d g d t | d
g d t | d g d
t | t | t | d
g d t | d g d t | d
g d t | d g d d S )N)gp=
ף?gp=
ף?g?)gQ ?g Q ?g)\(?)gףp=
?gףp=
?gffffff?rt r )TTFr r r r r6 r r )r r' bool_rK r r r r
)r r rB rO mxbigmxsmalls r test_allanyz"TestMaskedArrayMethods.test_allany9 s H(((((((((*
+
+
H***+++)))+24(
<
<
< !!
$
$
$c8EIIKK UYYq\\#7#7#7888UYYq\\#7#7#7888UYYq\\#7#7#7888UYYq\\#5#5#5666GKKMM!"""
W[[^^%8%8%8999W[[^^%:%:%:;;;W[[^^%8%8%8999W[[^^%8%8%899999r c f t dt }t g dd }t | t
u | | t | t |j d t |t
u t dt }t | t
u | | t | t |j d t |t
u d S )Nr~ r r Tr r )r3 r r' r r rJ r r )r storer s r test_allany_odditiesz+TestMaskedArrayMethods.test_allany_odditiesS s b%%%YYYT***
f$%%% U T"""V#$$$b%%%
f$%%% UE T"""V#$$$$$r c
| j \
}}}}}}}}} }
t | d t | d t | d t | d t | d t | d t | d t | d t | d g d t | d g d t | d g d t | d g d t | d
g d t | d
g d t | d
g d
t | d
g d d S )N# r r )r r r r r r )r r r r r r )r r r r r r )r r r r r r r )r r r r r r )r r r r r r )r r r r r r )r r r r r r )r r
r r ) r r r r rB rO r r r r r s r test_argmax_argminz)TestMaskedArrayMethods.test_argmax_argmine s 59V2Ar1b"c3TRYY[["%%%RYY[["%%%SZZ\\1%%%SZZ\\1%%%RYY[["%%%RYY[["%%%SZZ\\2&&&SZZ\\2&&&RYYq\\#5#5#5666SZZ]]$6$6$6777RYYq\\#5#5#5666SZZ]]$6$6$6777RYYq\\#7#7#7888SZZ]]$6$6$6777RYYq\\#5#5#5666SZZ]]$6$6$677777r c t j g d }t j g d }t || }| dd }t |j |j t |j | dd t |j |j dd d S )Nr r r r r+ )r r' clipr
r r )r r rB rO clippeds r test_clipz TestMaskedArrayMethods.test_clip| s H @ @ @
A
A
H : : :
;
; 11
''!Q--W\27+++W]AFF1aLL111W]BHMM!Q$7$788888r c t j d }t j |ddgdz }| dd| t |j ddgdz d S )Nr r r r r r )r r r r r r
r r s r
test_clip_outz$TestMaskedArrayMethods.test_clip_out si IbMM E aq!fqj 1 1 q!QVaVaZ(((((r c & t g dd }|dk |dk z }t | | ddg t |dd g< | | }t |j ddg t |j d
d g t |j d t ||| |d k }| | }t |j g d
t |j g d t |j d t ||| t g dg dgg dg dg }| | dk }t |j g d t |j g d t j g d }| |dk d }t |j ddgddgg t |j d
d gd d
gg d S ) Nr i' r r @r r r r r r r r rN rD rC r rq r r )rE rC r rq r )r r r r r rE rC rq )
rK r
r rJ r r r7 ri r r' )r r conditionr r s r
test_compressz$TestMaskedArrayMethods.test_compress s ---$ ? ? ?WS) QZZ **RH5551a&
JJy ! !QWr2h'''QWq!f%%%Q\4(((Q) %%%V
JJy ! !QWlll+++QWiii(((Q\4(((Q) %%% ,,, 5 (yy))) 4
6
6
6
JJqwwyyB ' 'QW . . .///QWlll+++ HYYY
JJqAvAJ & &QWBx"b 2333QW1v1v ./////r c t g dg d }| }t || t |d< | }t |g d d S )Nrc r r r r r r r )r'
compressedr
rJ r s r test_compressedz&TestMaskedArrayMethods.test_compressed si ,,,\\\ 2 2 2
LLNNQ!
LLNNQ """""r c , dt fdt fdg}t g d| }t t |j t | t | }t |j |j t |j |j t t | | }t |j |j t |j |j t g dg d }t | }t t j |j
|j
| t }t t j |j
|j
d S Nr r ra ))r r? z1.1)r r@ z2.2)r rZ z3.3r r r r )r r rK r
r r7 ry r4 rm r3 r r r r r r datatyper r s r
test_emptyz!TestMaskedArrayMethods.test_empty sW #Je l; LLL '
)
)
)S **,,--s8}}=== qMMQWag&&&Q\1<000 #a&& ) ) )QWag&&&Q\1<000
)=)=)= > > > qMMB'777888
FF< #AFAF3344444r c dt fdt fdg}t g d| }t t |j t | t t | | }t |j |j t |j |j t | }t |j |j t |j |j t g dg d }t | }t t j |j
|j
| }t t j |j
|j
d S r )r r rK r
r r7 ry rz rm r{ r r r r r r s r
test_zerosz!TestMaskedArrayMethods.test_zeros sS #Je l; LLL '
)
)
)S **,,--s8}}=== #a&& ) ) )QWag&&&Q\1<000 qMMQWag&&&Q\1<000
)=)=)= > > > qMMB'777888
FFHH#AFAF3344444r c dt fdt fdg}t g d| }t t |j t | t t | | }t |j |j t |j |j t | }t |j |j t |j |j t g dg d }t | }t t j |j
|j
| }t t j |j
|j
d S r )r r rK r
r r7 ry rb rm rc r r r r r r s r test_onesz TestMaskedArrayMethods.test_ones sS #Je l; LLL '
)
)
)S **,,--s8}}=== Qx ( ( (QWag&&&Q\1<000 aLLQWag&&&Q\1<000
)=)=)= > > > aLLB'777888
FFHH#AFAF3344444r c ~ t d }g d}t | }t || }t |d t u t |d t u ddg|ddg< t |d t u t |d t u t |g d t
t d g d
d z }g d }| |g d
t |t g d t |j g d | |t
g d g d t |g d t |j g d t
t d g d
d z }t ||g d
t |t g d t |j g d t ||t
g d g d t |g d t |j g d d S )Nr r? r r r r rC r )r r r r rC r r )r r r r )r r r r )
r r r r r r r rr r+ ra )
r r r r r r r r r r )r r r r )
r r r r r r r rr r+ ra )
r r r r r r r r r r ) r rG r' r rJ r
rK rg r( r r )r r r rB r r s r test_putz TestMaskedArrayMethods.test_put sO
1II OO aLL !! ! ! H1a& ! !F"###Q***+++ ///A*= > > > LL a Q @ @ @AABBBQV ; ; ;<<< a lllLLL99:::1 > > >???QV ; ; ;<<< ///A*= > > > Aq,,, Q @ @ @AABBBQV ; ; ;<<< Aq,|||\\\::;;;1 > > >???QV ; ; ;<<<< r' r r? fncr r8 r ri r )r rH r arrr r raveleds r test_ravel_orderz'TestMaskedArrayMethods.test_ravel_orderh
s
ggZ000AqqqD wwd*===?QT
#D ! ! !w} AGM$55555 q %%'''''''%..q!!Q&++----- 399U++QWWU^^-ABBBBBr c t d }t |d< | dd }t |j d t |j j d t |j d t |j j d d S )Nr r r rs r )r rJ rk r
rm r r. s r test_reshapez#TestMaskedArrayMethods.test_reshapez
sy 1II!
IIaOOQWg&&&QW]G,,,QWd###QW]D)))))r c t g dg dt j }t | }t |j g d t |j g d t |d }t |j g d t |j g d | t |j g d t |j g d t g dg dt j }| d t |j g d t |j g d g d}t | }t t t t t g d
t t j }t |d }t |j g d t g d
g d t j }t |d }t |j g d
t |j g d t ddgt j }t |d }t |t ddgt j d S )Nr r r r r r rc r r r r Fr_ )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 stablekind)
r' r r) rq r
r r r r sortedr r` r )r r sortedxs r test_sortz TestMaskedArrayMethods.test_sort
sX ,,,\\\ B B Bq''W]LLL111W]LLL111q%(((W]LLL111W]LLL111 QWlll+++QWlll+++ ,,,\\\ B B B u QWlll+++QWlll+++ LLq''Jv{333444 ###& @ @ @q%(((W]$5$5$5666 ###/// I I Iq%(((W]$5$5$5666W]OOO444 1b' ) ) )qx(((WeRG27 ; ; ;<<<<
> !a%1*
VVXX
XXb\\
R
VVXX
XXb\\
R
VVXX
XXb\\
R r c v t g dg ddt fdt fg }t g dg ddt fdt fg }t g dg d dt fdt fg }t | }t || t |j |j t |d
}t || t |j |j t j d t d
fg }| | }t | }t |d
}d S )N)r r rs r r r r rC )r r r r r r r r& r r8 )r rC r4 rs r r r )r r r r r r r )r r rC r4 rs r r )r r r r r r r Fr_ r r )r' r rq r
r r r r )r r mask_last
mask_firstr r[ s r test_sort_flexiblez)TestMaskedArrayMethods.test_sort_flexible
sX IIIIII:Sz*
,
,
, IIIIII:Sz*, , , IIIIII:Sz*, , ,
AwwT9%%%TY ///Au%%%T:&&&TY
000 XS!}o
&
&
FF2JJAwwAu%%%r c t g dg d }t t j | t | d S )N)r r r r r )r r r r r r )r' r
r r& r s r test_argsortz#TestMaskedArrayMethods.test_argsort s@ /// 8 8 8RZ]]GAJJ/////r c t g dg }t | g d t g dgg dg }t | g d t | j g d t j dgg }| }t |d d|d< t |d d t dggd }| }t
|t j j u t |j d d|d< t |d d d S ) Nr r r r r .r T)
rK r
squeezer r r' r r rJ r )r r r arr_sqm_arrm_arr_sqs r test_squeezez#TestMaskedArrayMethods.test_squeeze
sH YYYK((T\\^^YYY///YYYKyyyk:::T\\^^YYY///T\\^^)999555 huoo VQ s SXq!!! qcU...==?? ,---X]D)))
U3Z#####r c t j g d }t j g d }t || dd }| dddd }| dd }t |d |d d d f | dd }t |j d
d S ) Nr r r r r r r r r )r r r r )r r' rk swapaxesr
rm )r r rB r r mXswapped
mXXswappeds r
test_swapaxesz$TestMaskedArrayMethods.test_swapaxes! s H @ @ @
A
A
H ( ( (
)
) 11
%
%a
+
+jjAq!$$KK1%% Yr]Bqqq"uI...\\!Q''
Z%|44444r c f t g dg d }t | g d t g dg d t | g d |g d t | ddgddgg t dd gdd ggddgddgg t |d t j j u t | d t j j u t g d
g d gg dg d g
}t | ddgd t ddgddggddgddgg
t t |ddgd t ddgddggddgddgg
d S )N)r r rE rC r^ )r r r )r r rC rN r r r r rD r r r r r rE rC rq )rK r
ru r r r rJ r' r s r test_takez TestMaskedArrayMethods.test_take8 s )))<<< 8 8QVVIII&& \\\999(M(MNNNQVVIII&&))) 555QVVaVaV ,--!B8b"X"6!Q!Q8HII K K K ! $%%%q RU\)*** <<< .iii5M N N NQVVQFV++RHr2h/1v1v6FGGG I I IT!aV!,,,RHr2h/1v1v6FGGG I I I I Ir c t j d }t j d d d d f t j d d d d f z }t ||t | k }t ||d }t g dg dg d g d
g d g }t || t || }t g dg dg d g d g d
g }t x|d< x|d< |d< t || t |j |j t dd }t || }t x|d< |d< t || t |j |j d S )N)rC r % ra r r r r r mode)rC r rH )r rH ra )rH ra r )ra r r )r r r )ra r rC )r rC rC r )r r r r r r r )r r' r r ru r
rJ r )r r indicesmindicesr ctrls r test_take_masked_indicesz/TestMaskedArrayMethods.test_take_masked_indicesJ s H( ) ))A,,tAAAv&1 aaag)>>3q66(9;;;Axf---lll!kk!kk!kk"ll $ % %
T4 Ax lll!kk!kk!kk"ll $ % %
065T
5T$Z$t*T4 TY *** %O < < <Ax "((T
T$ZT4 TY *****r c
t t j d }t |ddg< | }t |d d u t |d d u d|_ | }g dg dg dg}t |d g d t |d g d t |d g d t || t t t g d
g d g d d
t fdt fdg }t |d< t | g d t ddgddgd
t fdt fg }| }t |dd gddgg |d }| }t |dd g d S )Nr r r r )r Nr r )r r r rr )r+ ra N r r r )r? r@ rZ )rX rY r[ r r ra r r ))r r? s one)r r@ two)NNNrC r r r r r ) r' r r rJ tolistr rm r
r r r r )r r xlistrN r r s r test_tolistz"TestMaskedArrayMethods.test_tolisth s
")B-- 1b'
aD !!!b T!"""
/?/?/?@U1X///U1X|||,,,U1X / / /000UD!!! $s999*??0002 2 3 3 sc5\<@
B
B
B "QXXZZ* * * + + +
7F#66*: sc3Z0
2
2
2xxzzTQI1v.///
aDxxzzTAt9%%%%%r c p t ddgdt fdt fg }|D ]A}| D ]*}t t |t
j +Bd|j d< |D ]A}| D ]*}t t |t
j +Bd S )Nr rr r r r r )r' r rS r r r genericr )r r entryry s r test_tolist_specialcasez.TestMaskedArrayMethods.test_tolist_specialcase s 66"C:Sz*B C C C : :E
:
:JtRZ 8 889999
: q : :E
:
:JtRZ 8 889999
: : :r c
t d }| }t |d |j t |d |j t
|g d< | }t |d |j t |d |j dt fddt fg}t d t t j d d t j d D |
}t
|g d< | }t |d |j t |d |j t j
d }t d t t j d t j d t j d D |
}t
|g d< | }t |d |j t |d |j d S )
Nr r r )r r r r r r r
c g | ] \ }}}|||f S r~ r~ )r r r r
s r r z6TestMaskedArrayMethods.test_toflex.. s6 J J JIQ1q!Qi J J Jr
ABCDEFGHIJKLMr r7 c g | ] \ }}}|||f S r~ r~ )r r r
ffs r r z6TestMaskedArrayMethods.test_toflex.. s6 L L LZaBq!Rj L L Lr )r toflexr
r r rJ r r r' r r r r r r )r r r r s r test_toflexz"TestMaskedArrayMethods.test_toflex s bzz VG_dj111VG_dj111 $]]] VG_dj111VG_dj111*lS%L9 J JRYr]]5D57Y^^B5G5G2I 2I J J J "# # # %]]] VG_dj111VG_dj111233 L L3ry}}79y~~b7I7I79y~~b7I7I4K 4K L L L "# # # %]]] VG_dj999 VG_dj99999r c ` t g d }t | }t || t |j |j t g dg d }t | }t || t |j |j t g dg ddt
fdt fg }t | }t || t |j |j d S ) Nr rN r )r r )r r )r r r r& r r )r' r; r_ r
r r r r r s r
test_fromflexz$TestMaskedArrayMethods.test_fromflex s )))
##T1 TY''' )))))) , , ,
##T1 TY''' ---4L4L4L sc5\2
4
4
4
##T1 TY'''''r c P t g dgg d }t dgdgdgdgdggg d }t |j | t | | t t j |j d | d d S )
NrW r~ r r r r r r r )rK r
r> rx r cumsum)r marrayr# s r test_arraymethodz'TestMaskedArrayMethods.test_arraymethod s /oooFFF aS1#sQC 8$3OO5 5 5VXw'''V%%''111['!44gnnQ6G6GHHHHHr c t j dd }t |j j |j t |j j |j d S )N* Tr )r r r' r
r> r r r s r test_arraymethod_0dz*TestMaskedArrayMethods.test_arraymethod_0d sK EKKK & &QSXqv&&&QSXqv&&&&&r c 4 t j g dg dg }t j j |d< |j }d|d< t j j |d< t |j |j j t |j |j j d S )Nr r r r r )r r r' rJ r> r
r r )r r xts r test_transpose_viewz*TestMaskedArrayMethods.test_transpose_view s~ EKKIII. / / #
S3%,3QVRTY'''QVRTY'''''r c b t j d }d|d< t j j |d< d|d< | }d|d< t |j | j t |j | j d S )Nr r r r r rs r )r r rz rJ r r
r r )r r xds r test_diagonal_viewz)TestMaskedArrayMethods.test_diagonal_view s EKK # ##
ZZ\\#RWajjll /000RWajjll /00000r N).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 r r r% r( r- r2 r7 r9 r? rD rF rO rU rY r` rb rf ri rl ro r~ r r r r s < < <@9 9 9$ 0< < <: : :4% % %$8 8 8.9 9 9") ) ) 0 0 0># # #5 5 5,5 5 5,5 5 5, & = = &% =< 0 0 0 0 0 0<7 7 7, [ Wf-- [ \400C C 10 .-C * * *#= #= #=J) ) ) : : : & & & P & & &@0 0 0
$ $ $.5 5 5.I I I$ + + +<"& "& "&H : : : : : :@( ( ($I I I' ' ' ( ( (1 1 1 1 1r r c e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zd Zed Zd Zd Zd Z ed d Zd Zd Zd Zd Z dS ) TestMaskedArrayMathMethodsc
t j g d }| dd }| dddd }t j g d }t || }t || |j }t || |j }t j g d }t || } t || |j }
t || |j } |||||||| |
| f
| _ d S r r r s r r z'TestMaskedArrayMathMethods.setup_method r r c
| 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 rd r
r r8 cumprod) r r r r rB rO r r r r r mXcps r test_cumsumprodz*TestMaskedArrayMathMethods.test_cumsumprod s 59V2Ar1b"c3Tyy||TZ1 !4!4Q!7!7888yy||TZ1 !4!4Q!7!7888zz!}}TZ1 !5!5a!8!8999zz!}}TZ1 !5!5a!8!899999r c J t t j ddd dd }t
x|d d df< x|d< |d< dD ]}t
t | }t
|| }t j dt }| d
||d| }t ||u t | |d| t dt } |d| }t ||u d S ) Nr r r r r r )rd ru r r r6 r r )r r r r r r r s r test_cumsumprod_with_outputz6TestMaskedArrayMathMethods.test_cumsumprod_with_output! s9
29$$QB//
0
0
8
8A
>
>(..111a4.2a52f:- & &H R**F R**F XfE222F KK VBQF333F Ff$ % % % QF!;!;!; < < <6---FV///F Ff$ % % % % & &r c
| j \
}}}}}}}}} }
|j \ } }t | | t j | t } t j |t }
t | D ]5}|d d |f |
|< 6t | D ]1}|| | |< 2t | d |
t | d | d S rt ) r rm r
r r r rz r r )r r r r rB rO r r r r r r rowscolsks r test_ptpz#TestMaskedArrayMathMethods.test_ptp6 s1 59V2Ar1b"c3TARVVXXr}} 2 2 4 4555x5!!x5!!q 2 2AAh))++//11DGGq / /Ae&&((,,..DGGRVVAYY%%%RVVAYY%%%%%r c t ddgddgt }|dz }t |d d t |j d d S )Nr r r r r r bx)rK rX r
r r r. s r test_add_objectz*TestMaskedArrayMathMethods.test_add_objectD sU #s1a& ? ? ?
GQqT4 q r c t g dg dt }t | d t g dg dgt }t | d g d d S )
Nr r r r r r r r r rr ra )rK rX r
rt r s r test_sum_objectz*TestMaskedArrayMathMethods.test_sum_objectJ s{ & A A AQUUWWa )))YYY/v > > >QUUU]]III.....r c t g dg dt }t | d t g dg dgt }t | d g d d S )
Nr r r r r r r r r r r )rK rX r
r r s r test_prod_objectz+TestMaskedArrayMathMethods.test_prod_objectQ s{ & A A AQVVXXu%%% )))YYY/v > > >QVVV^^[[[11111r c t g dt }t | d t | g d d S )Nr r r )r r r )rK rX r
r r r s r test_meananom_objectz/TestMaskedArrayMathMethods.test_meananom_objectX sR & 1 1 1QVVXXq!!!QVVXXzzz*****r c b t g d }t | j |j d|_ t | j |j t t j | d S )Nr T) rK r
r rm r r r r is_maskedr s r test_anom_shapez*TestMaskedArrayMathMethods.test_anom_shape^ s| # #QVVXX^QW---QVVXX^QW---))*****r c t t j dd dd }t | g dg dg t | d g d g d
g t | d g d g d g g d g d
g|_ d}t | | dd|gd|dgg t | d | dd|gd|dgg t | d | dd|gd|dgg d S )Nr rr r r )g r )r r rm r r )r r r )r r r )r r r rN r1 r g g g ?g @r r r r r r r )rK r r rk r r r r8 )r r mvals r test_anomz$TestMaskedArrayMathMethods.test_anome s 1a 00A66 7 7AFFHH / / / A C C CAFFFNN / / / A C C CAFFFNN *]]MMM : < < <))YYY'AFFHHOOD 1 1 #UD 1D$3E F H H HAFFFNN 1 1$ 7 7 "C .dC0@ A C C CAFFFNN 1 1$ 7 7 "C .tS0A B D D D D Dr c
0 | j \
}}}}}}}}} }
| } t | | t
| | t | j | z d z
t t j | | t j d
ddd } t j | d }
t | dd |
dd d S ) Nr r r r Fr )axis1axis2)
r r r
tracer rt r r r r rk r rK )r r r r rB rO r r r r r mXdiagr r= s r
test_tracez%TestMaskedArrayMathMethods.test_tracev sF 59V2Ar1b"c3T RXXZZ !9!9!;!;!?!?!A!ABBBBHHJJ GGIIFK!**,,,F12)4 )4 )4 4 5 5 5 RXb\\288::... i &&q1--""3..SYYQaY00%++AQ+2O2OPPPPPr c
| j \
}}}}}}}}} }
| d } | | } t | d | | t | j t u | d }
| | } t | d |
|
t | j d t | }| || t | | | dd }| d | d }}| | } t | d | | t | }| || t | | d S )Nr r r r r ) r r8 dotr r r r` r4 rA )r r r r rB rO r r r r r fxrfXr mYYfXXfYYs r test_dotz#TestMaskedArrayMathMethods.test_dot s 59V2Ar1b"c3T
YYq\\ FF2JJAHHQKK444& !!!
YYq\\ FF2JJAHHQKK444s
]]
rrAr"""ll2r""::a==#**Q--S GGCLLAHHQKK 666
]] Ar"""""r c Z t ddgddggddgddgg }t ddgddggddgddgg }t ddgddgg }| || t | d ddgdd gg t |j ddgddgg d S )
Nr r r r r r r r )rK r r r8 r rs s r test_dot_shape_mismatchz2TestMaskedArrayMathMethods.test_dot_shape_mismatch s 1Q%1 qeQqE] ; ; ; 1Q%1 qeQqE] ; ; ; 1Q%1 ' ' aQAHHQKK1a&2r();<<<AFaVaV$455555r c t g dd }t g dd }t t | t j t t | t j | | k t j d u sJ t t j d d d }t dd }t t | d
t t t | d
t t | d | |u t | d | |u d S ) Nrc rh r Tr r r rf r r r )
r' r
rh r r r r r r rk r3 r r )r foobars r test_varmean_nomaskz.TestMaskedArrayMathMethods.test_varmean_nomask sp IIIT***IIIT***T#((**%%rz222T#''))__bj111
chhjj(RXd^^;;;; BIbMM))%00===AT"""T#(((**++[999T#''q'//**K888aS))S0111QC((C/00000r c
| j \
}}}}}}}}} }
t | d | t | d | t | d d | d t | d d | d t | d j | d j t | j | j | d | d } } t | d d | d t | d d | d t d D ]}
t | |
||
t | |
|d d |
f t t j | |
|d d |
f d S ) Nr r r ddofr r r r r
r r r r r r
rm r r rr r r r r rB rO r r r r r mXvar0mXvar1r} s r test_varstdz&TestMaskedArrayMathMethods.test_varstd s 59V2Ar1b"c3TBFFF - -r}}/B/B/D/DEEEBFFF - -r}}/B/B/D/DEEEBFF1F 5 5 MMOO / /Q / 7 7 9 9 9BFF1F 5 5 MMOO / /Q / 7 7 9 9 9SWW!W__*BFFFNN,@AAARVVXX^QUUWW]333 FFFNNBFFFNNBFF1F 5 5 MMOO / /Q / 7 7 9 9 9BFF1F 5 5 MMOO / /Q / 7 7 9 9 9q = =A q 2a5+;+;+=+=+A+A+C+C D D D q 2aaad8+>+>+@+@+D+D+F+F G G G q 2 2 "111a4 3 3 5 5 9 9 ; ;
=
=
=
= = =r c N t j dt }t dt }t t d d }dD ]}t || }t | t u t |d t u t |d t u || t |t u t |j d || t t j | t t d d }d |d< dD ]}t || }t |d
t u t |dd
t u t |dd
t u ||d
t |t u t |j d ||d
t t j | d S )
Nr r r Tr )r r r r ra r r )r r )
r r' r r r r rJ r
r r )r r moutr
methodnamemethods r test_varstd_specialcasesz3TestMaskedArrayMathMethods.test_varstd_specialcases s% x%(((Ru%%% &**4 ( ( (( $ $J Q
++F FFHH& ' ' ' FF1II' ( ( ( FF2JJ&( ) ) ) Ft D& ' ' ' D ) ) ) Ft BHTNN # # # # &**4 ( ( ("( $ $J Q
++F FFNNNf, - - - FF11%%%/ 0 0 0 FF2A&&&&0 1 1 1 Ft! $ $ $ $ D& ' ' ' D ) ) ) Ft! $ $ $ $ BHTNN # # # # $ $r c 2 t g dg dgg dg dg }| dd }t | d g d t |j g d | dd }t | d g d t |j g d | dd }t | d g d t |j g d d S )
Nr rN r r r r r r r )r' r r
r8 r r s r test_varstd_ddofz+TestMaskedArrayMathMethods.test_varstd_ddof s 999iii( 999/E F F Fuu!!u$$T[[^^YYY///TY ***uu!!u$$T[[^^YYY///TY ***uu!!u$$T[[^^YYY///TY *****r c t d d }t |d< t j | }t |g d t | }t |g d t |j g d t | }t g dg dg dgg dg dg dg }t || d S )
Nra r r )r r r+ r1 r )r r r )r r r+ r )r rk rJ r r0 r
r r' )r r r r# s r test_diagz$TestMaskedArrayMathMethods.test_diag s 1II f % %$gajjS)))$$$1ggS)))$$$SXyyy)))3iiIIIyyy9 'iiIII >@ @ @S'"""""r c t g dg dg }t | d g d t | d ddg t | d ddg t | d g d t | d dd
g t | d dd
g t | d g d t | d dd g t | d dd g t | d g d t | d d dg t | d d dg d S )
Nr r r r r r r r r x r r )r' r
rt r rZ rW r s r test_axis_methods_nomaskz3TestMaskedArrayMathMethods.test_axis_methods_nomask s 999iii( ) )QUU1XXyyy)))QUU2YYB(((QUU1XX2w'''QVVAYY ,,,QVVBZZ!S***QVVAYYC)))QUU1XXyyy)))QUU2YYA'''QUU1XX1v&&&QUU1XXyyy)))QUU2YYA'''QUU1XX1v&&&&&r i Zb)
free_bytesc t t j ddt j t j d }t | d d S )N)' r i r r g @)rK r r uint16rz r
r r s r test_mean_overflowz-TestMaskedArrayMathMethods.test_mean_overflow sX
biHHH h~ 6 6
8
8
8QVVXXw'''''r c t j g d }t j |dd d }t j |d d d }t j ||d }t j |d }t j |d }t t j || d S )Nr r r r r r r r r r r{ r )prependr r r r' r rL diffr r )r r r a_prepdiff1r diff2s r test_diff_with_prependz1TestMaskedArrayMathMethods.test_diff_with_prepend% s H--- . . E quA . .##AbqbE#33
1f1
55 E q * *
11
%%ue,,-----r c t j g d }t j |d d d }t j |dd d }t j ||d }t j |d }t j |d }t t j || d S )Nr r r r r )appendr r r )r r r a_appr r r s r test_diff_with_appendz0TestMaskedArrayMathMethods.test_diff_with_append2 s H--- . . E q!uA . .""1QRR5"22
1U
33 E q * *
11
%%ue,,-----r c t j t d 5 t j t j d d d d d S # 1 swxY w Y d S )Nz4diff requires input that is at least one dimensionalr r )r r r r r r r' r s r test_diff_with_dim_0z/TestMaskedArrayMathMethods.test_diff_with_dim_0? s
] H $ $
EJJrx{{ # # # $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $s 2AA "A c t j g dd }t j |dd }t t j || d S )Nr r r r )r r )r r rL r r r )r r r s r test_diff_with_n_0z-TestMaskedArrayMathMethods.test_diff_with_n_0F s` E 7 7 7q A Auzz!qqz))q$''(((((r N) r r r r rw ry r~ r r r r r r r r r r r r_ r r r r r r r r r r r~ r r rq rq s < < <@ : : :& & &* & & & / / /2 2 2+ + + + + +D D D"
Q
Q
Q # # #06 6 61 1 1 = = =, & $ $ &% $D
+
+
+ # # # ' ' '( _ 4555( ( 65( . . . . . .$ $ $) ) ) ) )r rq c e Zd Zd Zd ZdS )!TestMaskedArrayMathMethodsComplexc
t j g d }| dd }| dddd }t j g d }t || }t || |j }t || |j }t j g d }t || } t || |j }
t || |j } |||||||| |
| f
| _ d S )N)$y @y Gz. @y -!@y !@y &1 ?r r r r r r r r r r r r r r r r r r y "~@y K7 @r r r r r r r r r r y -?r r r r r r r r s r r z.TestMaskedArrayMathMethodsComplex.setup_methodO sY H A A A
B
B
IIaOO
YYq!Q
"
" H ' ' '
(
(
"
"
"
!' 2 2
3
3
3!))BH"5"5666
X ( ( ( ) ) $$$AG!4!4555"2::bh#7#7888QAr2sCd;r c
| j \
}}}}}}}}} }
t | d | t | d | t | d j | d j t | j | j | d | d } } t | d d | d t | d d | d t d D ]}
t | |
||
t | |
|d d |
f t t j | |
|d d |
f d S ) Nr r r r r r r r r r s r r z-TestMaskedArrayMathMethodsComplex.test_varstdo sJ 59V2Ar1b"c3TBFFF - -r}}/B/B/D/DEEEBFFF - -r}}/B/B/D/DEEESWW!W__*BFFFNN,@AAARVVXX^QUUWW]333 FFFNNBFFFNNBFF1F 5 5 MMOO / /Q / 7 7 9 9 9BFF1F 5 5 MMOO / /Q / 7 7 9 9 9q = =A q 2a5+;+;+=+=+A+A+C+C D D D q 2aaad8+>+>+@+@+D+D+F+F G G G q 2 2 "111a4 3 3 5 5 9 9 ; ;
=
=
=
= = =r N)r r r r r r~ r r r r M s3 < < <@= = = = =r r c | e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zd Zd Zej ed d Zd Zd Zd Zd Zd Z d Z d Z d Z d Z ej ! d d d g d Z"d! Z#d" Z$d# Z%d$ Z&d% Z'd& Z(d' Z)d( Z*d) Z+d* Z,d+ Z-d,S )-TestMaskedArrayFunctionsc t j ddddt dz dddddddg }t j g d }g d
}g d }t || }t || }| d
||f| _ d S )Nr r r r r r r r r r r r r )r r' r rK r r, )r r r r r r r s r r z%TestMaskedArrayFunctions.setup_method s Hb"b#r#vr2tS"b"M N N HJJJ K K
1
1
1
1
1
1
!"
%
%
%
!"
%
%
%
% H r c ~ ddg}t d| }t |ddg t |d d d S )Nr r F)rV r
r. s r test_masked_where_boolz/TestMaskedArrayFunctions.test_masked_where_bool sH
F " "QA QqT1 r c g d}t |d }t || t |j g d t |d }t || t |j g d d S )Nr r rN r )rL r
r rR r< s r test_masked_equal_wlistz0TestMaskedArrayFunctions.test_masked_equal_wlist s{ II
!Q
RRXyyy)))
a
#
#RRXyyy)))))r c g d}t |d }t |j g d t |j d d S )Nr r rN )rL r
r r7 r< s r test_masked_equal_fill_valuez5TestMaskedArrayFunctions.test_masked_equal_fill_value sH II
!Q
RXyyy)))R]A&&&&&r c t g d }t |d< t t t |d | t |d t t t
|d | t |d t t t |d | t |d t t t |d | t |d t t t |d | t |d t t t |d | t |d t t t |d | t |d t t g dg d g d d S )Nr r r rW )rb rb r r r )r' rJ r
rV r> rM r? rN rC rP rD rQ ra rR r5 rL r s r test_masked_where_conditionz4TestMaskedArrayFunctions.test_masked_where_condition s &&& ' '!\'!Q--33^Aq5I5IJJJ\-1"5"5q99)!Q// 1 1 1\$q!**a00+a2C2CDDD\*Q"2"2A66&q!,, . . .\)Aq//1557G17M7MNNN\%1++q11<13E3EFFF\)Aq//1557G17M7MNNN\///???CC&&& ( ( ( ( (r c t dt }t |j t }t || }t
|| d S )N)r r r r )rb r rz rm r rV r
)r r r ctests r test_masked_where_odditiesz3TestMaskedArrayFunctions.test_masked_where_oddities sJ \///ek8,, UE**UE"""""r c t d }t t 5 t d| d d d n # 1 swxY w Y t |d }t |j g d d S )Nr r )
r r r r r r r r r r )r r
IndexErrorrL r
r r s r "test_masked_where_shape_constraintz;TestMaskedArrayFunctions.test_masked_where_shape_constraint s 2JJ
:
&
& A Aq!!TY > > >?????s AAAc t j dddg }t j d 5 t j |d dk | }d d d n # 1 swxY w Y t |j j j |j j t |d t j t j d t j
d d S ) Nr )r& z>1a88*** , , ,^DqNNAq99;N;N;NOOO]5eAhh .=oo$? $? $?@A1F FFJ$__ & & & ^E$uQxx../>%@ %@ %@ABAG GGK$__ & & & \%U1XX-<__#> #> #>?@B BBF$__ & & & %eOOO1@ 'B 'B 'BCDF FFJ$__ & & & & &r c t g dg d }t | g d t | d g d t | d g d t | }| | t |g d t g d
}t g d }t |d < t ||| }t |g d
t |d< t ||| }t |g d
t
|d t u t
|d t u t
|d t u d S )N)g82?g @g,E| @gE@gŏ1@r r )r r r r r5 r )g333333?gffffff@r gffffff@g@r )g(\?gS㥛@g- @gʡE@gK@r r r r r r )r' r
r r4 rJ ry r )r r r r r r s r
test_roundz#TestMaskedArrayFunctions.test_round s ???&
(
(
(QWWYY 4 4 4555QWWQZZ!:!:!:;;;QWWQZZ!D!D!DEEE qMM AQ,,,--- &&& ' ' /// " "! !QOOQ---...! !QOOQ---...! !F"###! r c 0 t t j ddd dd }t
x|d d df< x|d< |d< t j dt }| d t j |d
| }t ||u t || d
| t
dt }| d
| }t ||u d S ) Nr r r r r r r r r6 r )decimalsr ) r' r r r rk rJ r3 r r r r r
)r r r r s r test_round_with_outputz/TestMaskedArrayFunctions.test_round_with_output s 29$$QB//
0
0
8
8A
>
>(..111a4.2a52f: &... E"qf555& !!!VRXXqfX = =>>>vU+++1&11& !!!!!r c t ddg }t | d t ddg }t | t u t ddg }t j dt }| d | | t |d t ddg }t d dg }| | t |d
d t ddg }t d dg }| | t |d
t u d S ) Nr? Fr r Tr r6 r g r~ ) r' r
r r rJ r r3 r r )r r r s r test_round_with_scalarz/TestMaskedArrayFunctions.test_round_with_scalar sf
#UG $ $ $QWWYY""" #TF # # # V#$$$ #UG $ $ $!5))) E FVQ #UG $ $ $vTF+++ FVBZ### #TF # # #vUG,,, Fr
f$%%%%%r c t d }t t |t t |t j d d S Nr )r@ r r r r
r r s r
test_identityz&TestMaskedArrayFunctions.test_identity' sC QKK
1k**+++Q A'''''r c d}t t |d d t t |t t u t g d }t g dg d }t || }t |g d t |j g d t |_ t || }t |j g d ||z }t |j |j t || t |j |j ||z }t |j |j t || t |j |j d S )
N皙r \(\?)r r r? r? r )r r r r r r r )r r gKTj?r r r )
r re r rJ r' r
r r` r r )r r r r r s r
test_powerz#TestMaskedArrayFunctions.test_power, sY E!RLL$///a F*+++ ,,, - - ))) @ @ @ !QKKA A A ABBBQWooo... !QKKQWooo...
FQWag&&&Aq!!!AGQW--- aQWag&&&Aq!!!AGQW-----r c 8 t j g dg dg }t |g dg dg }t j g d }t j ||g }t |g dg dg }t g dg d gg d
g d g }||z }t || t |j |j ||z }t || t |j |j ||z }t || t |j |j t g d g d gg dg dg }||z }t || t |j |j ||z }t || t |j |j d S )
Nr )r r r5 r rN r rL r1 )r r )r iq r r r r )r r )r r' r
r )r a2a2mb1b2b2mrN r s r test_power_with_broadcastingz5TestMaskedArrayFunctions.test_power_with_broadcastingA s
X|||\\\2
3
3Biii 3444
Xiii
Xr2h
Biii 3444...0H0H0HI$99iii02 2 2 czT4 TY ***byT4 TY)))SyT4 TY)))...0H0H0HI$99iii02 2 2SyT4 TY ***byT4 TY *****r z fp errors don't work in wasmreasonc j t j ddddt dz dddddddg }t j g d }g d
}g d }t || }t || }| d
t |dk |d }t
|g d t
|j |j t |dk d| }t
|g d t
|j g d t |dk |t }t
|g d |j }d||dk
d < t
|j | t j d 5 t j
t d 5 | t } d d d n # 1 swxY w Y d d d n # 1 swxY w Y t | dk | t }t
|g d t
|j | j 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 r r r r r ) r r r r r r r r r r r r Tr rL z
invalid valuer ) r
r
r
r
r
r r
r
r r
r
r )r r' r rK r ry r
r rJ r r8 rO r r RuntimeWarningr r r )
r r r r r r r r tmpixms
r
test_wherez#TestMaskedArrayFunctions.test_wherea s
Hb"b#r#vr2tS"b"M N N HJJJ K K
1
1
1
1
1
1
!"
%
%
%
!"
%
%
%
% "q&"b ! !Q 6 6 6 7 7 7QWbh''' "q&"b ! !Q 6 6 6 7 7 7QW B B BCCC "q&"f % %Q 6 6 6 7 7 7hmmoo&*R1W T " "#QWc"""
[
(
(
( % % nODDD
%
%iinn
%
%
%
%
%
%
%
%
%
%
%
%
%
%
% % % % % % % % % % % % % % % %
#'3 ' 'QFFFGGGQWci(((((s6 G"$G ? G" G G"G G""G&)G&c 0 t j d }t d }| }t t j d|| | t t j d|| | d S )NT)r r' rK r r
r ry )r r r r s r test_where_objectz*TestMaskedArrayFunctions.test_where_object ss HTNN
FFHHRU[[q!,,a000RU[[q!,,a00000r c X t d }t |d< |dk }t ||t }t |j |j u t |d t u t |d t u t |d t u t |d t u t |d t u t || t |t | }t |j |j u t |d t u t |d t u t |d t u t |d t u t |d t u d S )Nr r r+ r rr ra )r rJ ry r r r
r s r test_where_with_masked_choicez6TestMaskedArrayFunctions.test_where_with_masked_choice se 2JJ!
F !Q 17"###! ! ! !F"###!F"###Q !VQ 17"###! !F"###!F"###! ! r c t g d }t g d }t |d< t ||| }t |g d t |d< t ||| }t |g d t |d t u t |d t u t |d t u t dd }t |d< t dd d z }t |d< t g dg d
}| d }t ||| }t ||| }t || t t | t u t |g d t |t d }t |g d
t |dt }t |g d d S )Nr r r r r r r r r r r )r r r rC r )rb rb rb r r )rb r r rb rb ) r' rJ ry r
r r r8 r< r` )r r r r r cmr s r test_where_with_masked_conditionz9TestMaskedArrayFunctions.test_where_with_masked_condition s &&& ' ' /// " "! !QOOQ---...! !QOOQ---...! !F"###! 1aLL" 1aLL2 ! /// 8 8 8
XXa[[ !QNN
2q!__Q v%&&&R***+++ !VQ Q***+++ !Q Q***+++++r c 2 t j dt j }t j dt j dz }t |dk || j }t j t j t j }t || d S )Nr r r@ r )r r r ry ry r result_typer
)r r r r r# s r test_where_typez(TestMaskedArrayFunctions.test_where_type sv Iarx ( ( ( Iarz * * *S 0QWa##).2:66T7#####r c ( t j d dd }t j d }t j g d|| }t g d|| }t || t |j |j d S )Nra r r )r r rk rz ry r
r )r r r r r s r test_where_broadcastz-TestMaskedArrayFunctions.test_where_broadcast s IaLL A & & HQKKx 1a((
999a
#
#T2 TZ*****r c ^ t j dt fdt fg }t j g d| }t j d| }t j g d|| }t j g d|| }t || t |j |j d S )Nr r rC r r r )r r r )r r r r' ry r
)r r[ r r r r s r test_where_structuredz.TestMaskedArrayFunctions.test_where_structured s
XSzC:.
/
/ H---R 8 8 8 HXR ( ( (x 1a((
XiiiA
&
&T2 TZ*****r c Z t j dt fdt fg }t j g d| }t g d|t
}t
g d| }t |j |j t || t |j |j d S )Nr r r r r r ) r r r r' ry rJ rV r
r )r r[ r r r s r test_where_structured_maskedz5TestMaskedArrayFunctions.test_where_structured_masked s
XSzC:.
/
/ H---R 8 8 8
999a
(
( 1--RXx~...R"""RWhm,,,,,r c t j dt }t j |d< t j |d< t j t d 5 t j | d d d d S # 1 swxY w Y d S )Nr r r r z!not supported for the input typesr )
r r rX PINFNaNr r r3 r masked_invalidr s r test_masked_invalid_errorz2TestMaskedArrayFunctions.test_masked_invalid_error s Iav & & &w!v!
]9!DF F F $ $ E # # # $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $s BB Bc G d d }t j | }t |j t j | t |j g d d S )Nc e Zd ZdZd ZdS )CTestMaskedArrayFunctions.test_masked_invalid_pandas..Seriesnonsensec X t j dt j t j g S r )r r' rP rx r s r __array__zMTestMaskedArrayFunctions.test_masked_invalid_pandas..Series.__array__ s xBFBF 3444r N)r r r r r. r~ r r Seriesr+ s( E
5
5
5
5
5r r/ )FTT)r r r' r r r' r )r r/ r s r test_masked_invalid_pandasz3TestMaskedArrayFunctions.test_masked_invalid_pandas s 5 5 5 5 5 5 5 5 e""6688,,39bhvvxx&8&899939&9&9&9:::::r r TFc 4 t j g d }|j t u sJ t j || }|j t usJ |j t u sJ t j |j |j |k sJ d S )Nrc r ) r r r' r r` r' r r r )r r r r" s r test_masked_invalid_full_maskz6TestMaskedArrayFunctions.test_masked_invalid_full_mask s
EKK % %w& e""14"00xv%%%%v "17CI66$>>>>>>r c g dg dg dg dg}t g d| }t |t g d t g d|d }t |t g d t g d|d
}t |t g d t g dg d
}t ||d
}t |t g d t |j g d t |g dg dg dg dg
}g d}t ||d
}t |t g d t |j g d d S )Nr r r r r rQ r r r r rE r r ! r r r r r r r r )r r r r r rI wrap)r r r r r r r r r )rb r r rb r r r r r r r )r) r
r' r )r choiceschosenindices_s r test_choosez$TestMaskedArrayFunctions.test_choose s <>> 2 2333 LLL999'777VU??? 3 3444V[,,,///|||\\\'3||\\\'C D D D <<'777VU??? 3 3444V[,,,/////r c g dg dg dg dg}t dt }t g d|| }t |t g d t ||u t dt }t g dg d
}t ||d |
}t |t g d t |j g d
t |g dg dg dg dg }g d}t dt t }t ||d |
}t |t g d d S )Nr4 r5 r6 r9 r r r; r r< r> r r= )rJ r )rb r r rb r r? r r )rS r r rS ) r3 r r) r
r' r r r r )r r@ r rA rB s r test_choose_with_outz-TestMaskedArrayFunctions.test_choose_with_out s < )r r )r= r )r> r )r& r r r ) r rH r
r r r r r r )r ntyper r# base_type
base_mtypesub_types r test_make_mask_descrz-TestMaskedArrayFunctions.test_make_mask_descrN s u U|, u%%TS$K#t5666---...
u%%TD!9%%%---... u%%T28D>>***---... u tUmdE]%C DE u%%(K# l/K)LMNNT7###---... qz"# u%%T28cD!9%5$677888---... e$% u%%T28j$%7$899:::---... HsCm_-- $Y//
8c3Z#z): ;<< x((T28c4[3#tQ 8H2I$JKKLLL C #z122222r c & ddg}t | }t |j t t |ddg t j ddgt }t | }t |j t t |ddg dt fdt fg}t j ddg| }t | }t |j t t |ddg dt fdt fg}t j ddg| }t ||j }t |j | t || dt fdt fg}dt fdt fg}t j ddg| }t ||j }t |j | t |t j ddg| t j dd d
}t t |t j
t ||j }t || t ||u t j dd d
}t ||j }t || d d
g}t t g}d}t j ||| D ];\ } }
} t t | |
| } t | t u || |
| fz > > > ? ?r c l dt fdt fg}t j g d| }t |t }t || t t | }t || t |d }t || t j g d| }t || }t j g d| }t || dt fd t fg}t j g d| } t || }n# t $ r Y nw xY wdt fdd
t fd t fgfg}t j d d g| }t j d
dg| } t j dd g| }
t t || |
d S )Nr r )r r r r r F)r r r r )r r r r r& r r= r> r r r r r )r r r' rI r` r
r ) r r r r r r# othertyper r bmaskcntrls r test_mask_orz%TestMaskedArrayFunctions.test_mask_or s tsDk*x888FFFtV$$T4 vt$$T4 tU##T4 999GGGtU##(;;;5IIIT7###4[3+. 999KKK 4''DD D tsdD\D$<$@ AB+{35AAA+{35AAA+{35AAAWUE**E22222s D! !
D.-D.c ( t j g dt }t t | | t j ddgdt fdt fg }t | }t j g dt }t || dt fddt fd t fgfg}d
d g}t j || }t | }t j g d t }t || d S )
NrN r r r r r r r= r> rZ r )r r r r r r )r r' r r
r7 )r r r r# rP r s r test_flatten_maskz*TestMaskedArrayFunctions.test_flatten_mask s x ...\$''...x(#tsDk0JKKKD!!(<<.ANr r r r~ r r r& rh Dr r& r r c e Zd ZdS )3TestMaskedArrayFunctions.test_compressed..MNri r~ r r r rl rj r r c e Zd Zd ZdS )rl c dS )Nrh r~ r s r r z>TestMaskedArrayFunctions.test_compressed..M.compressed s rr N)r r r r r~ r r r z3TestMaskedArrayFunctions.test_compressed..M s#
r rh )
r r r' r r rh r r
r r )r r r r& r s r r z(TestMaskedArrayFunctions.test_compressed s
EKKA u ""T
bj()))
EKK
# #u ""T
a u !aS **TY"""u QC5' ,,TY""" u B4",00TY""" u B4",00T2 r c 6 t t j d d }t j ddg }t j || }t
|t g dd t j ||d }t
|t g dd t j ddgg d }t
|t g d
d ddg}t g d d }t j ||d }t
|t g d d t j ||d
}t
|t g dd d S )Nr r r )r r r r rr r r F)propagate_mask)r r r r rr r r )r r r r )r r r r )r r r r r T)r r r r r )rL r r r' r convolver
)r r r r s r
test_convolvez&TestMaskedArrayFunctions.test_convolve sm 1 q ) ) HaV u~~a##T<(<(<(>>u~~a4~88T<(<(<($ $ $$, , , $ $ $" " " HC C C C Cr r c P e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d S )
TestMaskedFieldsc g d}g d}g d}dt fdt fdg}dt fdt fdt fg}g d}t t t ||| || }t
||||
| _ d S ) NrW )r? r@ rZ r\ r^ )rX rY threefourfiver r ra r r r )baser ddtyperP )r r r r' r r r* r )r ilistflistslistr{ rP r rz s r r z TestMaskedFields.setup_method- s )))777*sElL9+T{S$K8 T#eUE2233$fMMMdfVLLL r c n | j d }| j d }t |_ t |j t j |j | t |_ t |j t j |j | d|_ t |j t j |j | d|_ t |j t j |j | g d|_ t |j t j
d dD | d S )Nrz rP r FTr? c g | ]}|||fS r~ r~ )r r s r r z;TestMaskedFields.test_set_records_masks..G s &J&J&JQ1ay&J&J&Jr ) r r` r r r r rz rm rJ rb r' )r rz rP s r test_set_records_masksz'TestMaskedFields.test_set_records_masks7 s y 8$ TZ$*F)K)K)KLLL TZ6)J)J)JKKK TZ$*F)K)K)KLLL TZ6)J)J)JKKK#OO TZ X&J&J/&J&J&J,2 4 4 4 5 5 5 5 5r c | j d }|d |d |d }}}t t df|d< t |j t t |j g d t |j t t |j t dd d
d g t |j d t |j g d
d S )Nrz r r r r r )r r r r r r@ rZ r\ r^ r ) pirR s three four fiver r r
r r r r r rz base_abase_bbase_cs r test_set_record_elementz(TestMaskedFields.test_set_record_elementJ s y $(ItCy$s)r4.QV\3'''V\???333V\5)))V\BS#s#;<<<V\5)))V\@@@ B B B B Br c | j d }|d |d |d }}}t t df|d d< t |j t t |j g d t |j t t |j t t t dd g t |j d
t |j g d d S ) Nrz r r r r r )r r r r r r\ r^ r )r r r r r r r s r test_set_record_slicez&TestMaskedFields.test_set_record_sliceZ s y $(ItCy$s)D>RaRV\3'''V\???333V\5)))V\BBS#9:::V\5)))V\<<< > > > > >r c | j d }t |d< dD ]E}t || j g d t || j |j | FdS )zCheck record accessrz r )r r r r- N)r rJ r
r r )r rz r s r test_mask_elementz"TestMaskedFields.test_mask_elementi si y Q 7 7A a 7 7 7 a
1
6 6 6 6 7 7r c 8 dt fdt fg}t d| }t t | t j g dddg t |d d < t t | t j g dddg d S ) Nr r r r )r r r )r |b1)r r )r r r )r r r3 r
r= r r' rJ )r r r s r test_getmaskarrayz"TestMaskedFields.test_getmaskarrayr s *sEl+Qf%%%\$''X 6 6 6%1<$@B B B C C C QQQ\$''X 6 6 6%1<$@B B B C C C C Cr c t t t j d t j d }t j | }t
|dt fdt fg }d|j d< t j dgddgz z t }|
t }t || t |j | |
t d f }t || t |j |
d
d d S ) Nr r r r r r r r r )r r r r r r r' r r r r r
ri rk )r iteratorr r controlmaskr s r test_viewzTestMaskedFields.test_view~ s BIbMM29>>"+=+=>>??x!! (C<#u "> ? ? ?q hsR1#X~T::: vve}}T4::<<(((TY ,,,vvuaj!!T4 TY 3 3B : :;;;;;r c
dt fdt fg}t t t t j d t j d | t j t t g dg d dt fdt fg _ fd} |d |d |d
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 c 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
rh r_ r r r r )r r s r _test_indexz2TestMaskedFields.test_getitem.._test_index s adU + + + 1QWQZ 8 8 8 1QWQZ 8 8 8 a3i + 6 6 6 1S5 3 @ @ @ 1S5 3 @ @ @ @ @r r r r )
r r' r r r r r r r r )r r r r s @r test_getitemz TestMaskedFields.test_getitem s ,e - $s29>>"--ry}}==>>f M M M$s#A#A#A#A#A#A C C D D"%tsDk :< < < A A A A A A A Br c h t j dt fdt fg }t j ddg| }d|d d< t
|d t j ddg d |d d< t
|d t j dd g t j dt fdt fg }t j d
d g| }t j d | }d
|d d< t
|j
| t j d | }d
|d d< t
|j
| t j d d g| }t j d | }| d
|d d< t
|j
| t j d | }| d
|d d< t
|j
| d S )Nr r )r r )r r r r r r r r )TTrG r r ) r r r r r r r
r' r rw r rG )r r r rP r# r s r test_setitemz TestMaskedFields.test_setitem s C<#s455
U
( 36
B
B3
RWbhSz 2 23331c
RWbhSz 2 2333C;d 455(M<8GGG E T 0 0#q QVW%%% E T 0 0!S QVW%%%(L,7vFFF E T 0 0
#q QVW%%% E T 0 0
!S QVW%%%%%r c t j dd }t j d }||d< t |j g d d S )Nr Tr r r rt )r r rK r r r )r mask_0dr s r test_setitem_scalarz$TestMaskedFields.test_setitem_scalar sU %$$QT$22ell1ooA38%9%9%9:::::r c | j d D ]7}t t | t | j d 8d S )Nrz r{ )r r
r )r recs r test_element_lenz!TestMaskedFields.test_element_len sJ 9V$ = =C S3ty':#;#; < < < < = =r N)r r r r r r r r r r r r r r r~ r r ru ru + s M M M5 5 5&B B B
>
>
> 7 7 7
C
C
C< < < ( & & &<; ; ;= = = = =r ru c e Zd Zd Zd ZdS )TestMaskedObjectArrayc t j d d g }t t fD ]>}t j d | }t j d | }||d< ||d< t |d |u t |d |u t t |d t t t |d t t |d d |u t |d d |u t j j
|d< t |d |u t t |d t t t |d t t |d j d t |d d |u t |d j
| t |d j d t |d d j
| t |d d j d @d S ) Nr r r r r .)r .r~ T)r r r' r rX r= r r r r rJ r
r r )r r r[ a0a1s r r z"TestMaskedObjectArray.test_getitem s ekk4,''&/ 4 4B!!"%%B!!"%%BCFCF CFbL ! ! ! CFbL ! ! ! Js5z;77 8 8 8 Js5z;77 8 8 8 CJrNb( ) ) ) CJrNb( ) ) )U\CF CFbL ! ! ! Js5z;77 8 8 8 Js5z;77 8 8 8 U $ / / / CJrNb( ) ) )
Qb ) ) ) Qd + + + UB ,b 1 1 1 UB ,d 3 3 3 33 4 4r c t j d d g }t j t j j gt d |d< t |j d t j j u t |d t j j u t j j |d< t |d t j j u d S )Nr r )r r r' rJ rX r r )r r s r test_nested_maz$TestMaskedObjectArray.test_nested_ma s ekk4,''Xru|nf55e<E
ru|+,,, A"%,&''' AA"%,&'''''r N)r r r r r r~ r r r r s2 4 4 4:( ( ( ( (r r c 8 e Zd Zd Zd Zd Zd Zd Zd Zd Z dS ) TestMaskedViewc v t t t j d t j d }t j | }t
|dt fdt fg }d|j d< t j dgddgz z t }|||f| _
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 s r r zTestMaskedView.setup_method s BIbMM29>>"+=+=>>??x!! (C<#u "> ? ? ?q hsR1#X~T::: 1k* r c | j \ }}}| }t t |t t |j |j t |j |j d S r )r r r r r r
r r r r r r r s r test_view_to_nothingz#TestMaskedView.test_view_to_nothing sc !% q+vvxx
4--...TZ)))TZ)))))r c > | j \ }}}| t j }t t |t t ||j t || |j
d S r ) r r r r r r r r
r r r r; r s r test_view_to_typez TestMaskedView.test_view_to_type s !% q+vvbj!!Jt[111222T17### T499QW#5#5#=#=#?#?@@@@@r c | j \ }}}| t }t t |t
t
|| t
|j | d S r ) r r r r r r r
ri r r s r test_view_to_simple_dtypez(TestMaskedView.test_view_to_simple_dtype sg !% q+vve}}
4--...T4::<<(((TY ,,,,,r c | j \ }}}| dt fdt fg }t |j j j d t |d |d t |d |d |d dt fdt fg }t t |t t |j j j d t |d |d d t |d |d d |d dt fdt fg }t t |t t |j j d t |d |d d t |d |d d d S )Nr& r )r& r r r r r )
r r r r
r r r, r r r r s r test_view_to_flexible_dtypez*TestMaskedView.test_view_to_flexible_dtype s !% q+vvU|c5\233TY_*J777T#Y#'''T#Y#'''tyy3,e 566
4--...TY_*J777T#Y#q ***T#Y#q ***uzzC<#u 677
4--...TZ%z222T#Y#r
+++T#Y#r
+++++r c | j \ }}}| t df }t t |t
t
|| t
|j | dd |d t df }t t |t
t
||d t
|j d |d t df }t t |t
t
||d d S )Nr r r r ) r r r r r r r
r rk r s r test_view_to_subdtypez$TestMaskedView.test_view_to_subdtype1 s !% q+vvuaj!!
4--...T4 TY 3 3B : :;;;tyy%$$
4--...T47###TY'''uzz5!*%%
4--...T48$$$$$r c | j \ }}}| t dft j }t || t
t |t j t
t |t d S )Nr ) r r r r r r
r r r r s r test_view_to_dtype_and_typez*TestMaskedView.test_view_to_dtype_and_typeB st !% q+vvuaj"+..T4
4--...Jt[11122222r N)
r r r r r r r r r r r~ r r r r s + + +* * *A A A- - -, , ,(% % %"3 3 3 3 3r r c e Zd Zd Zd ZdS )TestOptionalArgsc T t j d d }t j dt d }d|d d d d df< t j || }d }d }d
D ] } |||| |||| d D ]} |||| t j d d d z d
k }t j || }dD ] } |||| |||| d S )N 8@r r r Tr r c T t j | }t j | }t ||d dd df ||dd df d t ||d dd df ||dd df d d S )Nr r .r r )r" __getattribute__r r r
r
r r numpy_fma_fs r testaxisz4TestOptionalArgs.test_ndarrayfuncs..testaxisU s ,Q//G5))!,,D
aa SbS1771S"W:A3N3N3N O O O ae,,,S"W5 3ss7%888
:
:
:
:
:r c t j | }t j | }t ||d j ||d j t ||d j ||d j t ||dd dd df ||dd df dd t ||dd dd df ||dd df dd d S ) NTkeepdimsFr r r .r r )r" r r r r
rm r s r testkeepdimsz8TestOptionalArgs.test_ndarrayfuncs..testkeepdims^ sW ,Q//G5))!,,D
a$///5 T2228
:
:
: a%0006 U3339
;
;
;
aa$777CRC@ 3ss7!dCCC
E
E
E aed;;;CGD 3ss7%$GGG
I
I
I
I
Ir )rt r r r r r r r )r r )r r rk rz r r r' )r r rB r r r r
s r test_ndarrayfuncsz"TestOptionalArgs.test_ndarrayfuncsL sr
IdOO # #G , , HRt $ $ $ , ,W 5 5!!!AAAb& EKKK " " : : : I I I 7 " "A HQ1 LAq ! ! ! ! A HQ1
Yr]]
"
"7
+
+A
-
2 EKKK " " " "A HQ1 LAq ! ! ! ! " "r c n t j d d }t j dt d }d|d d dd d f< t j || }t t | d t t |d
d t d z t t |d
dt d z t t |d dt d z t t |d d d t d z t t |d
d dt d z t t |d
d t d z t t t |d
t t j t |d
t j |t }t t | d t t |d
dt d z t t |d
dt d z t t |d dt d z t t j t |d d t t |d d dt d z t t |d
d dt d z t t |d
dt d z t t t |d
t t j t |d
t t t j j d t t j t t j d d
d S )Nr r r r Tr 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 rk rz r r r' r
r. rb r r r r` r rJ )r r rB r s r
test_countzTestOptionalArgs.test_count{ sd
IdOO # #G , , HRt $ $ $ , ,W 5 5!!!Aaaa% EKKK " "U1XXr"""U11%%%qe}555U15)))1T$ZZ<888U1t,,,bg .>???U11t444aW
oFFFU154888!DMM/JJJU12&&&$u++
666j%7777blE115555
EKKK ' 'U1XXr"""U11%%%qe}555U15)))1T$ZZ<888U1t,,,bg .>???RWU1t 4 4 455q999U11t444aW
oFFFU154888!DMM/JJJU12&&&$u++
666j%7777blE115555 U25<((!,,, blE25;;q>>BBBBBBr N)r r r r r r~ r r r r K s8 -" -" -"^$C $C $C $C $Cr r c e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zej d
d Zej d
d Zd Zd ZdS )TestMaskedConstantc H t |t j j d t j j u t j g d } |t j j | }t |t j j u t t |t j j j t |j |j t t j
| t j |j t d S )Nr r r )
r r r rJ r' r r MaskedConstantr
rm r< rb r )r r vectorr s r _do_add_testz TestMaskedConstant._do_add_test s BEL!$$ 4555 )))$$RU\6** bel*+++Jvruz'@AAABBBV\6<000RU]]6**BGFL,M,M,MNNNNNr c D | t j d S r )r r r r s r
test_ufuncz TestMaskedConstant.test_ufunc s "&!!!!!r c 2 | d d S )Nc | |z S r r~ r r s r r z2TestMaskedConstant.test_operator.. s
q1u r )r r s r
test_operatorz TestMaskedConstant.test_operator s ,,-----r c t j t j j }t t |t j j j t |t j j u d S r )r r r' rJ r r r r r s r test_ctorz TestMaskedConstant.test_ctor sZ EKK % % Jq"%*";<<<=== %&&&&&r c t t t j j d t j j t j j j }t t | d d S )NrJ )
r
rg r r rJ r __new__r r r )r masked2s r test_reprz TestMaskedConstant.test_repr s] T"%,''222 %#++BEJ,EFFg 11111r c t ddl m} t dt j dz D ]} | 5 }t j t j j || | d t j
| }d d d n # 1 swxY w Y t |t j j u d S )Nr )BytesIOr r r ) ior r r| r dumpr r rJ seekloadr )r r r r
r" s r test_picklez TestMaskedConstant.test_pickle s 1f 5 9:: ) )E
%a BEL!e<<<<q k!nn
%
%
%
%
%
%
%
%
%
%
%
%
%
%
%
C25<' ( ( ( ( ) )s AB B B c t t j j t j j u t j t j u d S r )r
r r rJ r True_r s r r z TestMaskedConstant.test_copy sP EL 25< / HMMOOrx ' ) ) ) ) )r c dd l }t |j t j j t j j u d S r )r r r r rJ r r s r
test__copyz TestMaskedConstant.test__copy sC DIbel # #ru| 3 5 5 5 5 5r c dd l }t |j t j j t j j u d S r )r r r r r rJ r s r r z TestMaskedConstant.test_deepcopy sC DM"%, ' '25< 7 9 9 9 9 9r c l t j j }t t j j j t j |dd t t t j |j dd t t t j |j
dd t j j t j j }t t t j |dd t t t j |j dd t t t j |j
dd d S )Nr~ r F)
r r rJ r r r r3 setitemr r r r r )r origr s r test_immutablez!TestMaskedConstant.test_immutable s u|bej*H,.SubNri r~ r r Subr s r r rJ )rh r r rJ r r rg )r r r s r
test_subclassz TestMaskedConstant.test_subclass s~ +++++$ru|$$+++ CEESSUU
%&&&a(+++++r c t t t t j j dd t t t t j j dt j d S )Nrm rL r )r r setattrr r rJ r r s r test_attributes_readonlyz+TestMaskedConstant.test_attributes_readonly
s> ngru|WdKKKngru|WbhOOOOOr N)r r r r r r r r r r r r r r r r r/ xfailr r r r r~ r r r r sO O O O " " ". . .' ' '2 2 2) ) )) ) )5 5 5
9 9 9
J J J4 4 4
# # #
[m,,$ $ -,$
[m,,% % -,%
, , ,P P P P Pr r c e Zd Zd ZdS )TestMaskedWhereAliasesc t t j dg t j d }t |j dg t t j t j }t |j d t j t j t j }t |j d t j g ddd }t |j t j j u t j g ddd }t |j dgdz d S ) Ng i TFrc r r r )
rU r r' r1 r
r rx r r r` )r r" s r test_masked_valuesz)TestMaskedWhereAliases.test_masked_values s BHhZ 0 0"(62B2BCCSXv&&&BFBF++SXt$$$e!!"&26'22SXu%%%e!!,,,$!??BEL()))e!!,,,%!@@SXw{+++++r N)r r r r r~ r r r
r
s# , , , , ,r r
c t j g dg d } t t j | dgdgg d S )Nr4 )r r r r r r r )r r r' r
argwherer s r test_masked_arrayr ' sG
LLL||| 44AQ 1#s,,,,,r c R t j g d } t j | dk | d }t | j g d t j g dg d } t j | dk | d }t | j g d t j t j d d
dd g } t j | d }t | j g d d S )
Nrc r Fr )FFTFr r )TFTFr r r )TFFFF)r r r' rV r r rx r' )r r s r test_masked_array_no_copyr + s
LLL!!A
161511Aqv : : :;;;
LLL||| 44A
161511Aqv 9 9 9:::
RVQ1a())A
QU ++Aqv A A ABBBBBr c t j g dd } t j g dd }t j | | }g d}g d}t |j | t |j | t j d } t j d }t j | | }d gd
z }d gd z d
gd
z z }t |j d | t |j | t j | |d }t |j d | t |j | d S )Nr r r )r r r )r r r r r r )FTFFFTrs )r r r r Tr Fr r ) r r rL r r r r rw rb )r r r
expected_data
expected_masks r test_append_masked_arrayr 9 sZ
777!,,A
777!,,A
U\\!Q
F&&&M<<%:; < 28E??#8 8 8 8 8C A!'.. .///ahhqweh4455rzBBB
((17#(
&
&C CI "### CHN '((( 8AU$ / / /D : rx 4 4 4 4 4 9? BHUOO$9 9 9 9 9D A!5uDAAA ABBB
8AQWCt < < 1r
c d } d}t t j j | j d | d S )Nc dS )zyThis docstring
Has multiple lines
And notes
Notes
-----
original note
Nr~ r s r r z test_doc_note..method s
r zNThis docstring
Has multiple lines
And notes
Notes
-----
note
original notenote)r
r r r r2 __doc__)r expected_docs r
test_doc_noter; sF
L $$V^V<>Q r c t j ddgdt j } t j d }t j | |d }t j |t j |j t j }t j ddt j }t j
| t j
| d S ) Nr Tr8 r Fr r r ) r r rK r r' r rb rm r r r )r? r r r' s r
test_gh_21022rB s
U
b"XD
K
KF
8A;;D
WV$
7
7
7F
U
%'WV\%J%J%J L LF E BT D DEM%M&r c t j ddgddgddgdggddgddgddggt } t j | }|d dd g t | d dk sJ t |d d
k sJ t |j | j d|j d < | j d dk sJ d S ) Nr dogr r catr r r= r r r )
r r r' rX r r extendr r
r r> s r test_deepcopy_2d_objrG s
U[[1e* a& q65/+ !!f !f !f & %
& &F }V$$H TN=qABBB vd| ! ! ! ! x~ ! # # # # ... HN4 < " " " " " "r c t j ddgt } t j | }d|d< t
| d t
|d d S )Nr r .)r r r' rX r r r
r> s r test_deepcopy_0d_objrJ s[
U[[!F[
3
3F}V$$HHSM2 r )r9
__author__sysr r r3 rO r r functoolsr r" r
numpy.ma.corenumpy.core.fromnumericr r numpy.core.umathr
numpy.testingr r r r numpy.testing._private.utilsr r numpy.compatr
numpy.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/ 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 r_ r r MaskedArrayFutureWarningr1 r2 r ra r9 rl r r r rq r r ru r r r r r
r r r r r r r$ r- r/ r0 r r1 r3 r5 rs r? optimizer; r@ rB rG rJ r~ r r rW s
)
, , , , , , , , , 9 8 8 8 8 8
" U $5#4#6#6 ' ' HM*MO O O 9
8$7
8
8
8
'
'w
'
'
't t t t t t t t n f
. f
. f
. f
. f
. f
. f
. f
.Rm, m, m, m, m, m, m, m,`OE OE OE OE OE OE OE OEd
N N N N N N N N `m0 m0 m0 m0 m0 m0 m0 m0`I 1 I 1 I 1 I 1 I 1 I 1 I 1 I 1XX) X) X) X) X) X) X) X)v
2= 2= 2= 2= 2= 2= 2= 2=jf
C f
C f
C f
C f
C f
C f
C f
CR_= _= _= _= _= _= _= _=D-( -( -( -( -( -( -( -(`K3 K3 K3 K3 K3 K3 K3 K3\TC TC TC TC TC TC TC TCnmP mP mP mP mP mP mP mP`, , , , , , , ,*- - - C C C 3 3 30 3 3 3 8 8 8
. . .2 * * *@ W55W55 :;;
<; 65 65
$ $ $ # # # CI&*e g gL Lg gL: # # #" r