Wlv d dl Z d dlmZ d dlZd dlmZ d dlmc mZ d dl m
Z
ej Z G d d Z e
dk rdZd ez Zd
\ ZZ ed d D ]Zd
ez Z e j ee eedz Z ej e Z edez dz e edez eded dded d dS dS ) N)reduce)float_)
build_err_msgc 0 e Zd Zd Z ddZddZ ej d d Z ej d d Z ej d d
Z
ej d d Z ej d d Z ej d d
Z
ej d d Z ej d d Z ej d d Z ej d d ZdS ) ModuleTesterc || _ |j | _ |j | _ |j | _ |j | _ |j | _ |j | _ |j | _ |j | _ |j | _ t | _
|j | _ |j | _ |j
| _
|j | _ |j | _ |j | _ |j | _ |j | _ |j | _ |j | _ |j | _ |j | _ |j | _ |j | _ |j | _ |j | _ |j | _ n!# t8 $ r |j j | _ Y nw xY wg | _ d S N) moduleallequalarangearray concatenatecountequalfilledgetmask getmaskarrayidinner make_maskmasked masked_array
masked_valuesmask_ornomaskonesouterrepeatresizesorttake transposezerosMaskTypeumathAttributeErrorcore testnames)selfr
s m/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/ma/timer_comparison.py__init__zModuleTester.__init__ sA
m \
!-\
\
m ~ " /\
) m " / # 1 ~ m K \
m m K K )\
+ DJJ + + + *DJJJ +s E E,+E, Tc T | | }| | }| | | | | } | | || | }| | || | }|j j dk rm| t }t |t j
r#|j dk rd|t j | < nt j | rd}|j j dk rm| t }t |t j
r#|j dk rd|t j | < nt j | rd} |j
dk p|j
dk p|j
|j
k }
|
s5t ||g|d|j
d|j
dz |d
} |
s
J | ||| } | | j ur|r| | | } t | t r| }
dg}
n<| }
|
}
|
}
|
sSd d |
d z t+ |
z z
}t ||g|d
|dz |d
} |
s J | dS dS # t, $ r*}t ||g||d
} t- | |d}~ww xY w)zZ
Assert that a comparison of two masked arrays is satisfied elementwise.
maskO r z
(shapes z, z
mismatch)xy)headernamesd g Y@z
(mismatch z%)N)r r r r dtypecharastyper
isinstancenpndarraysizeisnanshaper r boolravelalltolistr len
ValueError)r)
comparisonr4 r5 err_msgr6
fill_valuexfyfmcondmsgvalreducedmatches r* assert_array_comparez!ModuleTester.assert_array_compare0 s? [[^^
[[^^ LL a $,,q// : : KK))"1)55z B B KK))"1)55z B B
GLC A!RZ((
QVaZZ!""(1++!
GLC A!RZ((
QVaZZ!""(1++!
)GrM2QW]Iqw!'7ID
!#QF$+&Q17&Q&Qag&Q&Q&Q%R+1*4 6 6 6
S t *Q""C ##
#''!'44#t$$
+ # ))++ {{}}!..**
!E'--"2"2 23w<< ??#QF$+G=BUU&D%E+1*4 6 6 6
S t
!
! ! ) ) ) AjQQQC S//q ( )s 8D7K3 3
L'=%L""L'c D | | j |||d dS )zH
Checks the elementwise equality of two masked arrays.
zArrays are not equal)rI r6 N)rT r )r) r4 r5 rI s r* assert_array_equalz ModuleTester.assert_array_equalg s=
!!$*aG)? " A A A A A ignore)rD c t j ddddt dz dddddddg }g d }| ||
}|d d S )
!
Tests creation
? @ @ @ $ $@ @ r1 r r r r r r1 r r r r r r. r N)r= r
pir )r) r4 rM xms r* test_0zModuleTester.test_0o s`
Hb"b#r#vr2tS"b"M N N 0 0 0
qq
)
)
1rW c t j ddddt dz dddddddg }t j g d }g d
}g d }| || }| || }t j |d
| }| d
||z
d sJ |j }|j t d | k sJ | | t | t d | z
k sJ dD ]_}||_ ||_ ||_ ||_ ||_ | | t | t d | z
k sJ `dS )rZ r[ r\ r] r^ r_ r` ra rb ) r_ rb r] g g rh r` ra r[ rh rb rc ) r r r1 r r r1 r1 r r r r r1 r. g@x Dr c | |z S r r2 r3 s r* z%ModuleTester.test_1.. s
QqS rW c | |z S r r2 r3 s r* rj z%ModuleTester.test_1.. s
ac rW )) ) c | |z S r r2 r3 s r* rj z%ModuleTester.test_1.. s
!A# rW N)
r= r
rd r whereset_fill_valuer anyrA r? r r rF ) r) r4 r5 m1m2re ymrK ss r* test_1zModuleTester.test_1z s
Hb"b#r#vr2tS"b"M N N HJJJ K K
1
1
1
1
1
1
qr
*
*
qr
*
*
Xb&!
$
$
&!!!2~~a $$&&''&
Gw& !444444zz"~~R6//2+F+F!FFFFF! L LAAGAGBHBHBH::b>>SWWvoor/J/J%JJJJJJ
L LrW c Z t j g d }| |g d }| |g d }| | }t | t | t |d t |d u sJ |d |d k sJ d|d< d|d< | || d|dd < d|dd < | j |d< | j |dd < ||d
d
< | j |d< | g d g d |d
d
< | g d g d |d
d
< t j d
dz }| |d }| g dt }t j g dt }|d |d |dd j dk sJ g d}| | }| | }||u sJ | |d }||usJ d
S )z2
Tests conversions and indexing.
)r1 ro rl rm r1 r r r r. )r r1 r r1 r1 ro c rm N)r1 ro rm rl )r r1 r1 r r[ rb )r1 helloro rm )r )r r r1 r r )copy)
r= r
strreprtyperV r r r r objectrA r ) r) x1x2x3x4nrM ru m3s r* test_2zModuleTester.test_2 sJ Xlll
#
#
ZZ Z
.
.
ZZ Z
.
.
ZZ^^ B RBqE{{d2a5kk))))!u1~~~~11 B'''1Q31Q3 1+1Q3111 1!!,,, ==111!!,,, ==111
Yq\\#
C
(
(
ZZ***F
3
3
X(((&
1
1
1
1!A#w}$$$$ OO NN1
^^A
Bwwww
^^AA^
&
&{{{{{{rW c | d }| j |d< | |d }| | ||g | sJ | | | g d sJ | |dd }| |g d | |dd }| || sJ | dd }| || sJ | dd }| || sJ d S )
z&
Tests resize/repeat
rl ro ) )r r r1 r r r r1 r )ro ro ro ro r axis)r r r1 r1 ro ro rm rm N)r r r r r r r rV )r) r y4y5y6y7y8s r* test_3zModuleTester.test_3 sO [[^^ 1
[[T
"
"}}T - -r2h 7 7<<<<<}}T\\" - -/G/G/GHHHHH
[[\[
2
2 $<$<$<===
[[QQ[
'
'}}R$$$$$
YY|!Y
,
,}}R$$$$$
YYq!__}}R$$$$$$$rW c : | d }t j d }| j |dd< | ddd }| ddd }| t j |d | |d sJ | t j |dd | |dd sJ | t j | |d | |d | || sJ | t j | |d | |d | || sJ |
g d
t }| j |d< | |g d }|d d k sJ |d dk sJ |d dk sJ d
S )zB
Test of take, transpose, inner, outer products.
r} rn ro rm rl )ro r r1 r1 r )abcr1 defro rm )r rm rl r N) r r= r reshaper r" r! r r r r
r )r) r4 r5 ts r* test_4zModuleTester.test_4 s
KKOO IbMM!A#
IIaA
IIaA }}R\!Y 7 7 99U9UVVVVV}}RWQ 1 5 5tyyIq7Q7QRRRRR}}RXdkk!Q&7&7Q9J9J K K JJq! , ,. . . . .}}RXdkk!Q&7&7Q9J9J K K JJq! , ,. . . . . JJ... 7 7{! IIa # #tu}}}}tqyyyytqyyyyyyrW c | d }| d }| d }| j |d< |dz
}| ||dz sJ |dz
}| ||dz sJ | d }| d }| j |d< |dz }| ||dz
sJ |dz }| ||dz
sJ | d dz }| d dz }| j |d< |dz }| ||dz sJ |dz }| ||dz sJ | d dz }| d dz }| j |d< |dz }| || sJ |dz }| || sJ | d dz }| d dz }| j |d< |dz }| ||dz sJ || d z }| || d | d t }| d }| j |d< |dz
}| ||dz sJ dS )z*
Tests inplace w/ scalar
ro r1 r[ r] )r N)r r r rV r r; r )r) r4 r5 re s r* test_5zModuleTester.test_5 s
KKOO KKOO
[[__ 1 Q}}Q!$$$$$
a}}R1%%%%% KKOO
[[__ 1 Q}}Q!$$$$$
a}}R1%%%%% KKOOC
[[__S
1 S}}Q!$$$$$
c }}R1%%%%% KKOOA
[[__Q
1 Q}}Q"""""
a}}R##### KKOOC
[[__S
1 S}}Q#&&&&&
dkk"oo DIIe$4$4555 KKOO " "6 * *
[[__ 1 R}}QB'''''''rW c | dt }| d }| dt }| j |d< |j }| dt }| j |d< ||z
}||z
}| |||z sJ | |||z sJ | |j | ||j sJ | dt }| dt }| j |d< |j }| dt }| j |d< ||z }||z }| |||z
sJ | |||z
sJ | |j | ||j sJ | dt }| dt }| j |d< |j }| dt }| j |d< ||z }||z }| |||z sJ | |||z sJ | |j | ||j sJ | dt }| dt }| j |d< |j }| dt }| j |d< ||z }||z }dS )z)
Tests inplace w/ array
r )r9 ro N)r r r r/ r r )r) r4 r5 re rM as r* test_6zModuleTester.test_6" s
KK&K ) ) KKOO
[[6[
*
* 1 G KK&K ) ) " Q
a}}Q!$$$$$}}R1%%%%%}}RWdll1af&=&=>>>>> KK&K ) )
[[6[
*
* 1 G KK&K ) ) " Q
a}}Q!$$$$$}}R1%%%%%}}RWdll1af&=&=>>>>> KK&K ) )
[[6[
*
* 1 G KK&K ) ) " Q
a}}Q!$$$$$}}R1%%%%%}}RWdll1af&=&=>>>>> KK&K ) )
[[6[
*
* 1 G KK&K ) ) " Q
arW c Z | dddt dz gdz ddgdgdz z | dddt dz gdz ddgdgdz z f}dD ]} t | j | }n%# t $ r t t
| }Y nw xY wt | j | }|d |j } || } || }| | d | d | | |j
|j
d S )
z Tests ufuncr[ r r ro r1 rn r. )sqrtloglog10exp conjugateN) r
rd getattrr% r& fromnumericr
ninrV r _mask)r) dfufmfargsurmrs r* test_7zModuleTester.test_7V sR ZZaRT*1,Aq61#a%%>??? T\\#?O?O?O\%P%PQQQll30@0@0@1lMM V,,,czzzzAAA||Ca|((DK7777jj))) j==kk!Q KAAAqD Sq 9 9C:FFF||Ca|((+t{:::: R$,,s,*C*CDDDll3Ql;; b"X...
"4"4"4
5 KKNN QQ 7 7=== QQ C CSIII JJ ADKKNN(:; < < Q 5 5rv}}RYq\\7R7RSU7UVY7YZZZ QQ 7 71 9J29MNNN QQ 7 7$,,qq,:Q:QSWS_S_`ahiS_SjSjmpSp9qrrr Qb A A6JJJ QQ C CE^E^E^___ QQ 7 7$,,qq,:Q:QSWS_S_`ahiS_SjSjmpSp9qrrr
ZZ]]
"4"4"4
5
YYq\\
T->->q"-E-EA N NPSTTT T->->q"-E-EA N NPSTTT T->->q"-E-EA N NPSTTT
4<<8I8I!R8P8PWX<+Y+Y Z Z\]^^^ a $ $ Q 5 5v>>> QQ 7 79U9U9UVVV QQ 7 7#sDDD QQ C CEbEbEbcccccrW c v | d }| j |dd< | ddd }d S )Nr r} rn ro rm rl )r r r )r) r4 s r* test_AzModuleTester.test_A s9 KKOO!A#
IIaA rW N)r, r, T)r, )__name__
__module__ __qualname__r+ rT rV r= errstaterf rx r r r r r r r r r2 rW r* r r
s D IK$(5) 5) 5) 5)nA A A A R[X R[X L L L6 R[X ( ( ( T R[X % % %& R[X . R[X 2( 2( 2(h R[X 1 1 1f R[X 8 8 8D R[X +d +d +dZ R[X rW r __main__zNfrom __main__ import ModuleTester
import numpy
tester = ModuleTester(module)
z import numpy.ma.core as module
)r r r1 r ztester.test_%i()r z#%iz2..................................................z ModuleTester.test_%i.__doc__zcore_current : z.3fz - )timeit functoolsr numpyr= r numpy.core.fromnumericr' r
numpy.testingr rd r r
setup_base setup_curnrepeatnlooprangeifuncTimerr curr printevalr2 rW r* r s
, , , , , , , , , ' ' ' ' ' ' U^ ^ ^ ^ ^ ^ ^ ^ B
z4J 3Z?I We
U1a[[ = =!A%fl4++227E"HEEbgcll
eai& !!!
dd1A566777
;A;;;s1v;;;<<<< = =rW