y}r d dl mZ d dlZd dlZd dlmc mZ d dlmc m Z d dl
m Z m Z m
Z
d dlmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m Z m Z m Z m Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZY d dlZm[Z[ ej\ Z\ddZ] G d d Z^ G d
d Z_ G d d
Z`d ZadS ) )reduceN)assert_
assert_raises assert_equal)KMaskType MaskedArrayabsoluteaddallallcloseallequalalltruearangearccosarcsinarctanarctan2arrayaveragechoose concatenate conjugatecoscoshcountdivideequalexpfilledgetmaskgreater
greater_equalinner
isMaskedArrayless
less_equalloglog10 make_maskmasked masked_array masked_equalmasked_greatermasked_greater_equal
masked_inside masked_lessmasked_less_equalmasked_not_equalmasked_outsidemasked_print_option
masked_values masked_wheremaximumminimummultiplynomasknonzero not_equalonesouterproductputravelrepeatresizeshapesinsinhsometruesortsqrtsubtractsumtaketantanh transposewherezeros)pickle c Z t | | }|st d| d| d| |S )NzNot eq:
z
----)r print)vwmsgresults n/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/ma/tests/test_old_ma.pyeqr\ sC
a^^F -
+++q++++,,, M c e Zd Zd Zd Zej dddg d 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#d"S )#TestMac 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| }
|j } | d ||||||||| |
| f | _ d S )N ?g @ @ @ $ $@ @) rd rg rb g rh re rf ra rh rg ) r r r r r rj r r r r r ) r r rj r r rj rj r r r r rj mask)g rh ?g?r rj r r @x D)npr pirP rD set_fill_valued selfxya10m1m2xmymzzmxfss r[ setup_methodzTestMa.setup_method& s Hb"b#r#vr2tS"b"M N N HJJJ K K
1
1
1
1
1
1
12
12
H&&& ' '
1<<<
(
(
(
Xb%
#
#
G
% QRRQB:r] c | j \ }}}}}}}}} }
} t t | t t | t t | | t |j | t |j |j t |j t d | t t | t | t d | z
t t ||
t t t |d |
t t || d S )Nc | |z S N rv rw s r[ z)TestMa.test_testBasic1d..= s
Q r] c | |z S r r r s r[ r z)TestMa.test_testBasic1d..> s
QU r] ro ) rs r r$ r rD dtypesizer r lenr\ r rt s r[ test_testBasic1dzTestMa.test_testBasic1d5 s. 48F1AsBBAr2qM!$$$%%%
b!!"""U2YY"""RXq!!!RXqw'''RWf%6%6 : :;;;U2YYB&1B1BB*G*G GHHH2r
6"e$$b))***1b r] r ) c | j \ }}}}}}}} }
} }||_ ||_ ||_ ||_ || _ t t | t t | t t | | t |j | t |j t
d | t t | t | t
d | z
t t || t t t |d | t t || d S )Nc | |z S r r r s r[ r z)TestMa.test_testBasic2d..Q s
!a% r] c | |z S r r r s r[ r z)TestMa.test_testBasic2d..R s
a!e r] ro ) rs rD r r$ r r r r r r\ r ) ru r rv rw rx ry rz r{ r| r} r~ r s r[ test_testBasic2dzTestMa.test_testBasic2dC s? 59F1AsBBAr2qM!$$$%%%
b!!"""U2YY"""RXq!!!RWf%7%7 ; ;<<<U2YYB&1C1CR*H*H HIII2r
6"e$$b))***1b r] c | j \ }}}}}}}}} }
} t ddgddgg } t | ddgddgg }
t t | | z | |
z t t | | z | |
z t t | | z
| |
z
dD ]4} | | }| | }| | }| | }|
| }
t t | | t t ||z ||z t t ||z
||z
t t ||z ||z t
j dd 5 t t ||z ||z d d d n # 1 swxY w Y t t ||z ||z t t ||z
||z
t t ||z ||z t
j dd 5 t t ||z ||z d d d n # 1 swxY w Y t t ||z ||z t t ||z
||z
t t ||z ||z t t ||z ||z t t |dz |dz t t t | dz t | dz t t ||z ||z t t t
j || t || t t t
j
|| t || t t t
j || t || t
j dd 5 t t t
j || t || d d d n # 1 swxY w Y 6d S ) Nrj r r r )) r )r r ignorer invalid @)
rs r r+ r r\ reshaperp errstateabsr
rJ r9 r )ru rv rw rx ry rz r{ r| r} r~ r r a2da2dms r[ test_testArithmeticzTestMa.test_testArithmeticW s 48F1AsBBAr2qaVaV$%%C1a&1a&!12239cDj))***39cDj))***39cDj))***( = =A ! A ! AA BA BA B BrB3KK Bq1ub2g&& ' ' ' Bq1ub2g&& ' ' ' Bq1ub2g&& ' ' 'Hh???
,
,1q5"r'**+++
,
,
,
,
,
,
,
,
,
,
,
,
,
,
, BsQwb)) * * * BsQwb)) * * * BsQwb)) * * *Hh???
/
/37C"H--...
/
/
/
/
/
/
/
/
/
/
/
/
/
/
/ Bq3wS)) * * * Bq3wS)) * * * Bq3wS)) * * * Bq3wS)) * * * BqAvrQw'' ( ( ( Bs1vv}c"ggn55 6 6 6 BqAvrRx(( ) ) ) Brva||SR[[11 2 2 2 Br{1a(((2r*:*:;; < < < Br{1a(((2r*:*:;; < < <Hh???
=
=29Q??F2rNN;;<<<
=
=
=
=
=
=
=
=
=
=
=
=
=
=
=7 = =s6 4$G$$G( +G( .$J J" %J" ?SS S c t j dg }t dg }t t ||z t t t ||z t d S )Nrj )rp r r
isinstancer )ru namas r[ test_testMixedArithmeticz TestMa.test_testMixedArithmetic} s\
Xqc]]
A3ZZ
27K00111
27K0011111r] c F
| j \ }}}}}}}}} }
} t t t j | t | t t t j | t | t t t j | t
| t t t j | t | t t t j | t | t t t j | t | t j
dd 5 t t t j t | t | t t t j
t | t | t t t j t | t | d d d n # 1 swxY w Y t t t j | t | t t t j | t! | t t t j | t# | t t t j | t% | t t t j || t' || t t t j | t) | t t t j || t+ || t t t j || t- || t t t j || t/ || t t t j || t1 || t t t j || t3 || t t t j || t5 || t t t j | t7 | t t t j ||f t9 ||f t t t j ||f t9 ||f t t t j ||f t9 ||f t t t j |||f t9 |||f d S )Nr r ) rs r r\ rp r r rE rF rM rN r rI r r' r( r r r r r r r r< r% r! r& r" r r rt s r[ test_testUfuncs1zTestMa.test_testUfuncs1 s 48F1AsBBAr2q26!99c"gg&&'''271::tBxx(()))26!99c"gg&&'''271::tBxx(()))26!99c"gg&&'''271::tBxx(()))
[(
;
;
; 5 5 Brws1vvR11 2 2 2 Brvc!ff~~s2ww// 0 0 0 BrxA''r33 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 26!99c"gg&&'''29Q<<,,---29Q<<,,---29Q<<,,---2:a##WR__556662;q>>8B<<0011128Aq>>5R==112222<1%%yR'8'899:::271a==$r2,,//0002:a##WR__556662=A&&
2r(:(:;;<<<2#Aq))=R+@+@AABBB2<??IbMM223332>1a&));Bx+@+@AABBB2>1a&));1v+>+>??@@@2>1a&));Aw+?+?@@AAA2>1a),,k1b!*.E.EFFGGGGGs C I::I>I>c t g dg d }t t | j j t
j u t dt | t dt d t t dt ddg | d }t t | j j t
j u t t t |d t
j t t | j j t
j u t t dt | t t t |d t u t t ddgt |d d S ) Nrh ra rb rg rj r r r rk r rj r )r r r )r r r r typerp intpr r\ r r ndarrayr r: )ru otts r[ test_xtestCountzTestMa.test_xtestCount ss $$$<<<888c
%0111Qc
###Qa!!!1eAQC((())***kk&!!c
%0111
5a=="*55666c
%01111eCjj!!"""c1
&&&0111Aq65a==))*****r] c b | j \ }}}}}}}}} }
} t j | } t | }
t t t | t
j |
t t t | t j |
d S r )
rs rp rA r r\ maxr7 r minr8 )ru rv rw rx ry rz r{ r| r} r~ r r xrxmrs r[ test_testMinMaxzTestMa.test_testMinMax s 48F1AsBBAr2q
Xa[[Bii 3r77GN3 / /001113r77GN3 / /0011111r] c | j \ }}}}}}}}} }
} t t t j | t j | t t t j | t j | t t dt t d d t t dt t d d t t t j |d t |d t t t j t |d d t |d t t t j |d t |d t t t j
|d t |d t t t j
|d t |d t t t j
t |d d t |d t | dk rt t t j
||fd t ||fd t t t j |d t j |d t t t j |d t |d t t t j
|d t |d d S d S )Nr r axisrj )rs r r\ rp r
r
accumulaterK r r prodr? r r rt s r[ test_testAddSumProdzTestMa.test_testAddSumProd s 48F1AsBBAr2q26==##SZ]]3344426$$Q'' ):):;;<<<1c%((+++,,---1c%((+++,,---26!!$$$c!!nnn5566626&Q--a000#bq///BBCCC26!Q<<Q++,,,2711%%%wqq'9'9'9::;;;271a=='!Q--00111276"a==q111"2A...0 0 1 1 1 q66A:: Br~q!fa00 *B8Q 7 79 9
:
:
: Brv}}Q**CJq!,<,<== > > > Brva||SAYY// 0 0 0 Brwq!}}gamm44 5 5 5 5 5 :r] c t j g d }t |g d }t |g d }t | }t | t | t t t j | t
|d t t |d t |d u t |d |d k t |d t u t t |d |d t t |dd |dd t t |d d |d d t t |dd |dd d
|d< d
|d< t t || d |dd < d |dd < t t || t |d< t t || t |dd < t 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 t g dt |j t 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)rj r r r r rk )r rj r rj r )
fill_valuerj r c r rn rj r r r )r rj rj r ra rg )r r r rj r )rj hellor r )r )rp r strreprr r\ rH r r* r
r r+ r r5 r rl r objectr rD )ru x1x2x3x4s1s2s r[ test_testCIzTestMa.test_testCI s&
Xlll
#
#
2LLL
)
)
)
2LLL
)
)
)
2YY B R272;;RA 6 6 677888RU tBqE{{*+++1A 1 2a5"Q% !!!2ac7BqsG$$%%%2aaa5"QQQ% !!!2abb62abb6""###112r
1Q31Q32r
12r
1Q32r
1111eLLL&9&9::;;; \\\<<<88111eLLL&9&9::;;; \\\<<<88111eLLL&9&9::;;;U<<< 0 011222
Yq\\C
2s
#
#2r
9927CCDDD3
&&'''
%%%v
.
.
X(((&
1
1
U
UT"XXs###T"XXs###R 1Q3
%&&&&&r] c g d}t | }t | }t ||u t |d }t ||u t j d }t || }t |j |u t t
||j t |j |u t |d }t |j j |j j k t ||d }t |j |u t |d t u d |d< t |d t u t |j |u t t
|j
d t ||d
} t | j |u t | d t u d | d< t | d t u t | j |u t t
| j
d t |d z | }
t t |
j |d z j u t d } t | d< t | d
} t t t | | g | t t t! | g d t# | dd }
t t |
g d t# | dd }t t |
| d S )N)r r rj r r T)copyr rk r rl r r r rj ra r ) )r r rj r r r rj r )r r r r r )r r rj rj r r r r )r) r rp r r _datar
_mask__array_interface__r* rl r r rC r\ r r rB )ru nmrz m3r y1y1ay2y2ay3r y4y5y6s r[ test_testCopySizezTestMa.test_testCopySize s OO aLL
q\\R
qt
$
$
$
Yq\\
2A
"###RX&&'''A
BQ -,- . . . 2BQ
'
'
'B 1 11V#$$$B !$$%%%BQQ''' "###A& !!!AAf$%%% "###1%%&&&
28!
$
$
$r
R#X$44555
AYY1
B
;Bx(("--...72;; 8 8 899:::
B 1
-
-
-2///00111
B
"
"
"2r
r] c " t d }g d}t | }| }t || }t |d t
u t |d t
u ddg|ddg< t |j |u t |d t
u t |d t
u t t |g d t ||d
}| g d g d
t |j |u t |d t
u t |d t
u t t |g d d S )Nr )r r r rj rj rk r r
( rj )r r r r Tr )r rj r )r d )r r r r r ) r r) r r r r* r r\ r@ )ru rs r r rz rv s r[ test_testPutzTestMa.test_testPut5 ss 1II OO aLL
VVXX !! ! ! H1a& 1 ! !F"###1((())*** !"4 ( ( ( iii (((r!"""! ! 1***++,,,,,r] c t d }t |g d }t ddgddg }t |d t u t |d t u ||dd
< t |d t u t |d t u t t |g d t d }t |g d }|dd
}t ddgddg }t |d t u t |d t u ||d d < t |d t u t |d t u t t |ddg t |d t u t |d t u t t |g d d S ) Nr )r r r r r rk r r rj r r r r )r rj r r r )r r r r* r\ )ru rs rv r} rw s r[
test_testPut2zTestMa.test_testPut2K s 1II !/// * * * 2r(!Q ( ( (!F"###!F"###!A#! !F"###1'''(())) 1II !/// * * *
acF 2r(!Q ( ( (!F"###!F"###!!!! !F"###1r2h ! !F"###1'''(()))))r] c | j \ }}}}}}}}} }
} g d} t j | d }
t ||
| t t t
||
d | k d S )N) rj r r r r r rj r r rj r rj r r )rs rp r; r@ r r rL )ru rv rw rx ry rz r{ r| r} r~ r r r is r[ test_testMaPutzTestMa.test_testMaPute s| 48F1AsBBAr2q 0 0 0 JqMM! B2DQQ'''2-../////r] c
t d }| dd }d|j d< t |d dk |d|z z }t t |j | t t |j d|z t 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 }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 | }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 |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 g d }t g d }t |d< t ||| }t t |g d t |d
< t ||| }t t |g d t |d
t u t |d t u t |d t u t t t t! |d | t# |d t t t t% |d | t' |d t t t t) |d | t+ |d t t t t- |d | t/ |d t t t t1 |d | t3 |d t t t t5 |d | t7 |d t t t t1 |d | t3 |d t t t9 t; t= d dd g d t t t? t; t= d dd g d t t t9 t t; t= d g d dd j g d t t t? t t; t= d g d dd j g d t t t7 t t; t= d g d d j g d t t t3 t g d g d d j g d t t t g d g d g d! tC d"tD j# # }tI |j tJ }t || }t t || tM || |f }t t |g d t |d
t u t |d t u t |d t u t d$ }t |d< t d$ dz }t |d< t g d%g d& }|' d }t ||| }t ||| } t t || t tQ | tR u t t | g d' t |t d }t t |g d( t |dt }t t |g d) d S )*N r r r )rj r y $@r e rh .r r r r )ra rb rg rc rd )rj rj rj r r r )ra rb rh ri rj )r r r r )r rj r r r )rj r r r r rk )rj rj rj rj r )r rj r r r )rj rj r r rj )rj r rj r r )r r rj r rj )rj r rj r rj )rj rj r r r )rj r r r r )r r r r r )r r r )r r )rj rj rj r r r )rj r r r r r )r rj r r 2 )r r r rj rj rj )r rj rj r r r )*r r flatr r\ realimagr r* r r rP rD r r6 r r! r- r" r. r% r0 r& r1 r< r2 r r, r/ listranger3 rl r= rp float32rQ r r r r r: )
ru rv r} catestbtestctestrw cmr~ s
r[ test_testOddFeatureszTestMa.test_testOddFeaturesl s 2JJ
IIaOOq $2
aK161
1626""###A ! $*C!GaK88999s 2JJ!AaD S[[()))
FeAvv..//14555eAvv..//17:;;; !Q 17"###! ! ! !F"###!F"###1a !VQ 17"###! !F"###!F"###! ! A 17"###! !F"###!F"###! ! 1a &&& ' ' /// " "! !QOO1+++,,---! !QOO1+++,,---! !F"###! <1
q11>!Q3G3GHHIII<
a 3 3Q77'1--/ / 0 0 0<Q
A.. Aq0A0ABBCCC<
1a 0 0!446G16M6MNNOOO< !Q335Ea5K5KLLMMM<a Q// a1C1CDDEEE< !Q335Ea5K5KLLMMM=eAhh A668M8M8MNNOOO>$uQxx..!Q779L9L9LMMNNN=tE!HH~~,;OO"= "= "=>?D DDH"??$ $ % % % >%U1XX-<__#> #> #>?@!E EEI"??$ $ % % % <d588nn+:??!< !< !<=>@ @@D"??$ $ % % % #E////>%@ %@ %@ABD DDH"??$ $ % % % < AA$$$& & ' ' '\444ek8,, UE**5% !!! 1r1g 1+++,,---! !F"###! 1II! 1IIN! $$$+=+=+= > > >
XXa[[ !QNN
2q!__1b v%&&&2,,,--... !VQ 1+++,,--- !Q 1,,,--.....r] c t t t g dg d g d t 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 || || t t j | dk t t j | d k d S )
N)rj r r )r r r )rj r r )r r r r r r r r )
r r\ r8 r7 r r* rP r% r! r )ru rv rw s r[ test_testMinMax2zTestMa.test_testMinMax2 s2 7999iii00)))<<===7999iii00)))<<=== 1II 1IIM!!71a==%Q
Aq"9"9::;;;71a==%1
q!"<"<==>>>q!!Q&'''q!!Q&'''''r] c t d }t j d }t |dd< | ddd }| ddd }t t t j |d t
|d t t t j |dd t |dd t t t j t |d t |d t || 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 )
N r r r r r )r r rj rj r )abcrj defr r )r r r r )
r rp r* r r r\ rO rL r# r r> r r )ru rv rw ts r[ test_testTakeTransposeInnerOuterz'TestMa.test_testTakeTransposeInnerOuter s 2JJ IbMM!A#
IIaA
IIaA 2<9--yI/F/FGGHHH271i++T!Y-B-BCCDDD28F1aLL&A,,77A;; ! ! !28F1aLL&A,,77A;; ! ! ! )))6 2 2! III !
! ! r] c t d }t d }t d }t |d< |dz
}t t ||dz |dz
}t t ||dz t d }t d }t |d< |dz }t t ||dz
|dz }t t ||dz
t d dz }t d dz }t |d< |dz }t t ||dz |dz }t t ||dz t d dz }t d }t |d< |dz }t t || |dz }t t || t d dz }t d dz }t |d< |dz }t t ||dz |t d z }t t |t d t d t j }t d }t |d< |dz
}t t ||dz d S )Nr r rj ra rb )r )r r* r r\ r= astyperp r )ru rw rv r{ s r[ test_testInplacezTestMa.test_testInplace sy 2JJ 2JJ
BZZ1 Q1a!e
a1a!e 2JJ
BZZ1 Q1a!e
a2q1u
2JJ
BZZ#
1 S1a!e
c 2q1u
2JJN
BZZ1 a1a
q1a 2JJ
BZZ#
1 S1a#g
fRjj2tE{{##$$$ 2JJ bj ) )
BZZ1 R1a"f
r] c 8 t d }t |ddd< | dd }t dt j dz D ]I}t j || }t j | }t t || Jd S )Nr r r r r rj )protocol)
r r* r r rR HIGHEST_PROTOCOLdumpsloadsr r\ )ru rv protor rw s r[ test_testPicklezTestMa.test_testPickle$ s 2JJ!Bq&
IIaOO1f 5 9:: E Q///A QA Bq!HH r] c t d }t |d< t t t dk t |d t u t t |d d d d S )Nr rj z--r )r r* r r r r )ru xxs r[ test_testMaskedzTestMa.test_testMasked. si
AYY1F t#$$$1 VBqE1%%q)))))r] c
t g dg d }t t dt |d t t dt |g d t |g dd
\ }}t t d| t |d k t |d d < t t |d t u t g dg d }| d d }t |d d d
f< t t t |d ddg t t |d
d t u t t ddgt |d t |dd \ }}t t |ddg d S )Nr r rk rb r r )ra ra rb ra weightsTr returnedrc r rj rh r r ra )r r r\ r r* r )ru r rZ wtss r[ test_testAverage1zTestMa.test_testAverage16 s $$$<<<8883!,,,--...3-=-=-=>>>??@@@ c+;+;+;dKKK 3 s
AAA!$$$.///$$$<<<888kk!Q AAAqD 73Q'''#s44555!$$$Q'61222B8WSq 1 1 122333 cD999 3R!!"""""r] c
T
g d}g dg dg}t d }t t t |d d t t t |d| d t t d dt d z g }t t t |d t
j t j d d z d
z t t t |d t j d d z dz t t t |d t |d t |d dz g t t t |d | d
t t t |d| g d t t t |d t |d t |d dz g t d }g d}g dg dg}t d }g d} t t t t || d d t t t t || d d t t t || d t u t t t || d d t t t t || d d t || }
t t t |
d d
t t t |
d g d t t t |
d ddg t t t |
d| g d t d } t d dz } t | | g| | ggd d \ }
}t t |
t | t |
j |j t t d ddd gd \ }}t t | t | t t d d \ }}t t | t | t t d t d d \ }}t t | t | k t d d gdd ggt }t |ddgd dgg }t |d }t t! |d d g t |d }t t! |d d g t |d }t t! |d t |d }t t! |d!d"g d S )#N)r rj rj rj rj r )rj r r r r rj r r r r )r r rb rg g (@rj r g
@)rh ra rb rg rc rf )r r rj rj r r )r rj rj rj rj rj rh )rh ra X@r rc g @rd )rh ra r r rc rf r Tr )r r r )r r r )r r r r rm ra g@g ?rc )r r r r r rp r
r rQ r= r+ r* r r rD floatr\ )ru w1w2rv rw ry rz r m4m5r} abr1r2r r a2daa2dmas r[ test_testAverage2zTestMa.test_testAverage2I sF
"4"4"4
5 1II+++S11222B777==>>> 6!99cF1IIo. / /D))!# ry||!-CDDEEE+++")!!"4"4"4gaa6H6H6H6L!MO O P P PD"555x@@AAAB777!:!:!:< < = = =+++")!!"4"4"4gaa6H6H6H6L!MO O P P P
1XX
"4"4"4
5
!WW
a!4!41===sCCDDD a!4!41===sCCDDD Q++!444>???W\!R 0 0q9993???U7<2#6#6Q???@@!DDD B D))844555+++!=!=!=? ? @ @ @+++c3Z88999B777!>!>!>@ @ A A A
1II 1IIM1a&1a&)DAAABU2YYb ***RXrx(((i q1a&4PPPBU2YYb ***i 4888BU2YYb ***i $y//DQQQBb U2YY&'''aVaV$e,,C1a&1a&!122s###4#s$$%%%1%%%53*%%&&&4(((5'""###1%%%53*%%&&&&&r] c d t dt t d t dt t d t dt t dggg t dt t dgg t t
t t ddg t t t t ddg t t t t ddgddg d S )Nrj ra r rk )r intr r r TypeError
ValueErrorboolru s r[ test_testToPythonzTestMa.test_testToPython s QE!HH
&&&S%a//***QEQC5'NN++,,,S%se --...iq!f
666j$q!f
666j$q!fAq6(B(B(BCCCCCr] c t dd }t j d 5 t dt d z j d d d n # 1 swxY w Y t d|z j t | j t | j t t || j t t
|| j t | j |j j u t dd }t | |j k t t | t t d S )Nr rj rk r r )
r rp r r rl r7 r8 r r r r r r4 )ru r{ rv s r[ test_testScalarArithmeticz TestMa.test_testScalarArithmetic sw
11
[
)
)
) ) ) Qq\' ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) )R
"
"
B$%%%B$%%% !RX^3444 !!
ag%&&&SWWc"5 6 677777s %AA A c t g d }t t | |j t t | |j t t | |j t t | |j t t | ddddd |j ddddd t t | g d |j g d t t |
|j
t t | |j t ddgddgg }t t | |j t t |
|j
t t | ddg |j ddg t t | |j d S )Nrj r r r rj r r r rj r rj )r r r\ anyr r argmaxargminr compressconjr diagonalrK rL rO )ru r% r s r[ test_testArrayMethodsz TestMa.test_testArrayMethods s~ ))) 15577AGKKMM**+++15577AGKKMM**+++188::qw~~ / /00111188::qw~~ / /00111188Aq!Q** 7>>!Q1a88: : ; ; ;1::iii((!'*:*:999*E*EFFGGG16688QW\\^^,,---1;;==!'"3"3"5"566777 Aq6Aq6" # #1::<<!1!1!3!34455515577AGKKMM**+++1661a&>>17<<A#7#7889991;;==!'"3"3"5"56677777r] c R t g d }t |j d d S )Nr7 rj )r r ndimru r% s r[ test_testArrayAttributesz TestMa.test_testArrayAttributes s+ ))) QVQ r] c h t d t t j D d S )Nc h g | ]/}|t t v| d -|0S )_)dirr
startswith).0r s r[
z'TestMa.test_testAPI.. sH , , ,1[!1!111 c** 2 111r] )r rG rp r r1 s r[ test_testAPIzTestMa.test_testAPI sF , ,BJ , , , , - - - - -r] c t g d }t g dg d }t |d j d t |d j d t |d j d d S )Nr7 r8 rk r r rj )r r rD )ru r% r& s r[ test_testSingleElementSubscriptz&TestMa.test_testSingleElementSubscript su ))) )))))) , , ,QqTZ$$$QqTZ$$$QqTZ$$$$$r] c ~ t g dg d }|dk }d||< t |d t u d S )Nr )rj r rj r rk r r r )r r r* )ru r% r s r[ test_assignment_by_conditionz#TestMa.test_assignment_by_condition sJ ,,,\\\ 2 2 2
F!! r] c t ddgddg }t ddgddg }|dk }|| ||< ddg}t |j | d S )Nr rj Frk T)r+ r rl )ru r% r& rl
expected_masks r[ test_assignment_by_condition_2z%TestMa.test_assignment_by_condition_2 sm !Quen 5 5 5 !QtTl 3 3 31uD'$
QV]+++++r] N)$__name__
__module__ __qualname__r r pytestmark parametrizer r r r r r r r r r r r r r r r r r r r+ r2 r5 r? rC rK rM rO rR r r] r[ r_ r_ $ s
;
;
; [ S66"233 43&$= $= $=L2 2 2 H H H>
+
+
+ 2 2 26 6 6*0' 0' 0'd1 1 1 f- - -,* * *40 0 0c/ c/ c/J ( ( ( (0 0 0 d * * *# # #&8' 8' 8'tD D D
8
8
8 8 8 8" - - -
% % % , , , , ,r] r_ c , e Zd Zd Zd Zd Zd Zd ZdS )
TestUfuncsc t dddt dz gdz ddgdgdz z t dddt dz gdz ddgdgdz z f| _ d S )Nra r r r rj r rk )r rq rs r1 s r[ r zTestUfuncs.setup_method sr aR!V,q01va7GHHHaR!V,q01va7GHHHKr] c g d}dD ]?} t t | }n%# t $ r t t | }Y nw xY wt t j | }| j d |j }t j 5 ||v rt j d |dv rt j d || } || }d d d n # 1 swxY w Y t t | d | d | t t |j |j Ad S )N)
rI arctanhr r arccoshr] r' r( r true_divide floor_divide remainderfmod)*rI r' r( r r rE r rM r r r rF r rN arcsinhr^ r] r fabsnegativefloorceil logical_notr
rJ r9 r r_ r` ra rb hypotr r r< r& r" r% r! logical_and
logical_or logical_xorr )r )r] r' r( r4 r )getattrumathAttributeError fromnumericrp r rs ninr seterrr r\ r eqmaskrl )ru f_invalid_ignorefufmfargsurmrs r[ test_testUfuncRegressionz#TestUfuncs.test_testUfuncRegression s @ @ @> . .A
- UA&&!
-
-
- [!,,
-""B6'26'?D
(((Ih////333IX....RYRY
Bryy||RYYq\\155 6 6 6 F27BG,, - - - -= . .s! ! AA=CC C c & | j d }t t |d t t |d t t |d d d d t t
|d d d S )Nr r r )rs r r rG r rK r? rB s r[ test_reducezTestUfuncs.test_reduce s F1IGAA&&&&'''###$$$S2A2Q'''+++WQQ'''+++++r] c t dd dd }t |dk | }t | | t | d t
| d | d k t
| d g dk t
| d d j t
| d d j d S )Nrj
r r r r )r r r r ) r r r6 r r r r r rl )ru r% amasks r[ test_minmaxzTestUfuncs.test_minmax s 1bMM ! !!Q ' ' QUA&&UYY[[!%%''***UYY[[!$$$1 q)..001111 -2244555 ! Q$%%% ! Q$%%%%%r] c dD ]B}t g dg d }t t t | dg Cd S )Nz?bhilqpBHILQPfdgFDGO)rj r r r )r r rj rj rk r )r r r\ r; )ru r rv s r[ test_nonzerozTestUfuncs.test_nonzero sX ' ) )Alll 666A BwqzzA3'' ( ( ( ( ) )r] N)rS rT rU r r{ r} r r r r] r[ rZ rZ sc K K K#. #. #.J, , ,& & &) ) ) ) )r] rZ c > e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d S )
TestArrayMethodsc t j g d }| dd }| dddd }t j g d }t || }t || |j }t || |j }|||||||f| _ d S )N)$g @gGz. @g-!@g !@g&1 ?gʡE@g\( @g Q @g{G#@gPn@gh|?!@g!rh@gjt@gJ +@gzGa@g!rh@gʡE@g-@g)\(@g(\B#@g + @guV @g~j@g"~@gK7 @g
ףp=J#@gGz@gMb@g}?5^#@g/$@g!rhm @g-#@gJ +@gMb@gʡ@g-?r r r )$r rj r rj r r rj r rj rj r rj r r r rj r rj r r r rj rj rj rj r r rj r r r r rj r rj r )datarl )rp r r rD rs )ru rv XXXr mxmXmXXs r[ r z TestArrayMethods.setup_method s H @ @ @
A
A
IIaOO
YYq!Q
"
" H ( ( (
)
)
"
"
"
!' 2 2
3
3
3!))BH"5"5666QAr2s+r] c | j \ }}}}}}}| }t | | t
t | | t |j | z d z
d S )Nr r ) rs r> r trace
compressedrK r r\ rl ) ru rv r r r r r r mXdiags r[
test_tracezTestArrayMethods.test_trace& s &*f#Ar1b"c RXXZZ !9!9!;!;!?!?!A!ABBB288:: 7799s6; +E01(3 (3 (3 34 4 5 5 5 5 5r] c | j \ }}}}}}}| dd }t t |j |j t t |j | dd t t |j |j dd d S )Nr r )rs clipr r\ rl r ) ru rv r r r r r r clippeds r[ test_clipzTestArrayMethods.test_clip. s &*f#Ar1b"c''!Q--7<))***7=!&&A,,//0007="(--1"5"56677777r] c | j \ }}}}}}}|j \ }}t | | t j |t
j } t j |t
j }
t | D ]5} |d d | f |
| < 6t | D ]1} || | | < 2t t | d |
t t | d | d S Nr rj ) rs rD r ptpr rp rQ float_r r r\ ) ru rv r r r r r r r rowscolsks r[ test_ptpzTestArrayMethods.test_ptp5 s? &*f#Ar1b"cARVVXXr}} 2 2 4 4555x29%%x29%%q 2 2AAh))++//11DGGq / /Ae&&((,,..DGG266!99d##$$$266!99d##$$$$$r] c | j \ }}}}}}}| dd }t t |d |d d df | dd } t | j d d S )Nr rj r r )r r r r )rs swapaxesr r\ r rD )
ru rv r r r r r r mXswapped
mXXswappeds
r[
test_swapaxesz TestArrayMethods.test_swapaxesB s &*f#Ar1b"cKK1%% 9R="QQQU),,---\\!Q''
Z%|44444r] c | j \ }}}}}}}| d }t t |j | d d | d }t t |j | d d d S r )rs cumprodr r\ r r ru rv r r r r r r mXcps r[ test_cumprodz TestArrayMethods.test_cumprodI s &*f#Ar1b"czz!}}4:ryy|| 3 3A 6 677888zz!}}4:ryy|| 3 3A 6 67788888r] c | j \ }}}}}}}| d }t t |j | d d | d }t t |j | d d d S r )rs cumsumr r\ r r r s r[ test_cumsumz TestArrayMethods.test_cumsumP s &*f#Ar1b"cyy||4:ryy|| 2 21 5 566777yy||4:ryy|| 2 21 5 56677777r] c . | j \ }}}}}}}t t | d | t t | d | t t | d j | d j t t | j | j | d | d } }t d D ]}
t t | |
||
t t ||
|d d |
f t t t j ||
|d d |
f d S )Nr r r rj r )
rs r r\ varr stdrD r rp rI ) ru rv r r r r r r mXvar0mXvar1r s r[ test_varstdz TestArrayMethods.test_varstdW s &*f#Ar1b"c266t6$$bmmoo&9&9&;&;<<===266t6$$bmmoo&9&9&;&;<<===3777??("&&a&..*>??@@@26688>15577=11222 FFFNNBFFFNNq = =A Bvay"Q%"2"2"4"4"8"8":":;; < < < Bvay"QQQT("5"5"7"7";";"="=>> ? ? ? Brwvay)) !!!!Q$x 2 2 4 4 8 8 : :< <
=
=
=
= = =r] N) rS rT rU r r r r r r r r r r] r[ r r s , , ,,5 5 58 8 8 % % %5 5 59 9 98 8 8 = = = = =r] r c z | t u r |t u S |t u r | t u S | |k S r )r: r )ry rz s r[ rs rs e s< V||V| V||V| "H>> r] )rS )b functoolsr rV numpyrp numpy.core.umathcorern numpy.core.fromnumericrp
numpy.testingr r r numpy.mar r r r
r r r
r r r r r r r r r r r r r r r r r r r r! r" r# r$ r% r& r' r( r) r* r+ r, r- r. r/ 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 numpy.compatrR rq r\ r_ rZ r rs r r] r[ r si
, , , , , , , , , U f
, f
, f
, f
, f
, f
, f
, f
,R>) >) >) >) >) >) >) >)BT= T= T= T= T= T= T= T=n r]