fNC3 d Z ddlZddlmZ ddlmZ ddlmZmZ ddlm Z ddl
m Z m
Z ddlmZmZm
Z
mZm Z mZ ddlmZmZmZ dd lmZ G d
d Z G d d
Z G d d Z d Z dS )z^Tests suite for mrecords.
:author: Pierre Gerard-Marchant
:contact: pierregm_at_uga_dot_edu
N)recarray)maskednomask)temppath) fromrecords
fromarrays)
MaskedRecords mrecarrayr fromtextfiler addfield)assert_ assert_equalassert_equal_records)picklec e Zd Zg dZg dZg dZdefdefdgZg dZ e
j e e
eee e e Zd Zd
Zd Zd Zd
Zd Zd Zd Zd Zd Zd Zd Zd Zd Z d Z dS ) TestMRecords 皙?皙@ffffff
@皙@ @ one two three four fiveabc|S8r r r r r maskdtypec | j }| t }t |j |j t |j |j t t |j t t |j |j t dD ] }t || || t | t | d S )Nr% r& r( ) baseviewr
r
recordmaskr _maskr
isinstance_datar )selfr0 mbasefields p/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/ma/tests/test_mrecords.py test_byviewzTestMRecords.test_byview$ s y )$$U%t777 U[$*555
5;11222 U[$*//(*C*CDDD$ 4 4E eeEl 3 3 3 3 UZZ 2 2E::::: c . | j }| t }dD ]B}t t || || t || || C|d }t
t |t t |j |j t | d t |j
t t |j
d t |d |d d |d }t
t |t t |j |j t | d t |j
d t |j
d t |d |d d t
|d t u |d d
}t
t |t t |j |j t |j
dd g t |j t# j dd g|j j t ||d d
t dD ].}t t || |d d
| /d S )
Nr/ r )r r r )FFFr% )NNNT)TTTr r r- )r0 copyr1 r
r getattrr
r4 r- tolistr2 r r3 itemr r r, nparray)r6 r0 r7 r8 mbase_first
mbase_lastmbase_sls r9 test_getzTestMRecords.test_get0 s y~~ )$$$ 4 4E ..e = = = eeEl 3 3 3 3Ah
; 22333[& 444[''))+;<<<[+V444[&++--/DEEE[%uSz!}5552Y
:y11222Z%u{333Z&&((*<===Z*D111Z%**,,.@AAAZ_eCjn555C F*,,,!9
8Y//000X^U[111X(1a&111 X] X'<'9';,1K,= ? ? ? @ @ @ XtBQBx}}Y'?'?@@@$ D DE 51148E? C C C C D Dr; c | j }| t }| }d|_ d|j j d d < t |d j g d t |d j g d d|_ t |d j dgdz t t j
|d dgdz t |j dgdz t |j t j g d t
t |_ t |j j dgdz t |j j dgdz t t j
|d dgdz t t j |d d gdz t |j t j g d
t
| t }d|j dd < t |j g d t |j j g d t |j dd < t |j |d t |j j g d ddt* fg}t j g d|
}| t, }t j g dg d }||d< t |j | t |j j g d d S )N)i?B g@x DN/Ar r% )r r r r r r* r r F)r r r r r r rK rK rL r> r( N/A)r r r rL rN rN rL r r r r r r )r r r r r r& r r r r r )alphaz|S1num))r% r )r& r )r( r )
r r r r, )r0 r? r1 r
fill_valuer% r5 r r3 ma getmaskarrayr2 rA rC rD boolr r( r, getdatar& intr rR )r6 r0 r7 ndtypedatardatavals r9 test_set_fieldsz TestMRecords.test_set_fieldsX s$ y~~ )$$
0
aaaU3Z%777U3Z%777U3Z%s1u---R_U3Z001#a%888U%wqy111U[''))X * * *
%) * * * + + + UW\A3q5)))UW'!Q///R_U3Z001#a%888RZc
++fXaZ888U[''))X * * *
%) * * * + + + )$$))++ UWooo...UW]OOO444 UWd3i(((UW]OOO444"UCL1x666fEEE -((h|||)))444e UY$$$UY^YYY/////r; c | j }| t }t |d d< t |j g d t |j j g d t t j
d t j d gdt fdt fg }t |d d< t |j g d t |j j g d d S )
Nr% r rP r r& r> )r r r r r )r r r r r )r0 r? r1 r
r r r% r3 r rC arangerandomrandr] floatr6 r0 r7 s r9 test_set_fields_maskz!TestMRecords.test_set_fields_mask s y~~ )$$ c
2UWooo...UW]OOO444BIaLL")..*;*;<#&*sEl!;= = = c
2UWooo...UW]OOO44444r; c | j }| t }t |_ t
t j |d dgdz t
|d j |d j t
|d j |d j t
|j
t j dgdz t t |_ t
t j |d dgdz t
|j
t j d gdz t d S )
Nr& r r r% r( r r r r> r rK )r0 r? r1 r
r r, r rY rZ r3 rA rC rD r[ r ri s r9
test_set_maskzTestMRecords.test_set_mask s6 y~~ )$$
R_U3Z001#a%888U3Z%uSz'7888U3Z%uSz'7888U[''))Xyk!m4888 : : :
R_U3Z001#a%888U[''))Xyk!m4888 : : : : :r; c | j }| t }g d|_ t |j j g d t |j j g d t |j j g d g d|_ t |j j g d t |j j g d t |j j g d d S )N)r r r r r r r r r r ) r0 r? r1 r
r, r r% r& r( ri s r9 test_set_mask_fromarrayz$TestMRecords.test_set_mask_fromarray s y~~ )$$$__
UW\???333UW\???333UW\???333$__
UW\???333UW\???333UW\???33333r; c B | j t }t j g ddt fdt fdt fg }||_ t |j j g d t |j
j g d t |j j g d d|_ ||_ t |j j g d t |j
j g d t |j j g d d S ) N)r r r rr r r r rs rK r% r& r( r> )r r r r r )r r r r r F)
r0 r? r1 r
rC rD r[ r, r r% r& r( fieldmask)r6 r7 nmasks r9 test_set_mask_fromfieldsz%TestMRecords.test_set_mask_fromfields s %%i00 C C C;d c4[9; ; ;
UW\???333UW\???333UW\???333
UW\???333UW\???333UW\???33333r; c | j }| t }t |d< t |j t j g dt t |j g d | t }d|d d< t |j j
g d t |j j g d t |j j
g d t |j j g d t |j j
g d
t |j j g d | t }t |d d< t |j j
g d t |j j g d t |j j
g d
t |j j g d t |j j
g d t |j j g d d S )Nrd )rK rl rK rl rl r> rP r r r r )r r r r r ro ) @ry r r r ) 5rz r" r# r$ r )r r r r r r r )r0 r? r1 r
r r r3 rA rC rD r[ r2 r% r5 r& r( ri s r9 test_set_elementsz TestMRecords.test_set_elements s* y~~ )$$))++b K HLLL
!
!
! " " "
U%777 )$$))++ bqb UW]OOO444UW]OOO444UW]$;$;$;<<<UW]OOO444UW]=== ? ? ?UW]OOO444 )$$))++bqb UW]OOO444UW]OOO444UW]$=$=$=>>>UW]OOO444UW]AAA C C CUW]OOO44444r; c | j }| t }| d|dd < t |j j g d t |j j g d t |j j g d t |j j
g d t |j j
|j j
t |j j
|j j
t d # t $ r Y nt $ r w xY w d|dd < t d # t t f$ r Y d S w xY w)
Nrx rd rO )r r r r r )r r! r" rz r$ r* z)Flexible hard masks should be supported !r z.Should have expected a readable buffer object!)r0 r? r1 r
harden_maskr r% r5 r& r( r3 ExceptionNotImplementedErrorAssertionError TypeErrorri s r9 test_setslices_hardmaskz$TestMRecords.test_setslices_hardmask s} y~~ )$$
I"E"##J 8 8 8 (?(?(? @ @ @ BBB
D
D
D 8 8 8
6 6 6
6 6 6 GHH H
# D
NE"##J LMM M $Y/ DD s% B?D
D.# D.2E E E c | j }| t }| t |j t |_ t |j
|j
| t |j t |_ t |j
t j
|j |j t t j |d j
t u t# |d j
|d j
d S )Nr& r% )r0 r? r1 r
r} r
_hardmaskr r, r r3 soften_maskrY make_mask_noneshaper- make_maskr ri s r9
test_hardmaskzTestMRecords.test_hardmask
s y~~ )$$
U[$*555
EO#$$$
U[ .tz4: F F H H H U3Z -..&8999U3Z%uSz'788888r; c | j }| t }t dt
j dz D ]}t j || }t j | }t |j
|j
t |j |j t |j
|j
t |j
|j
d S )Nr r )protocol)r0 r? r1 r
ranger HIGHEST_PROTOCOLdumpsloadsr r- r r5 r3 )r6 r0 mrecproto_mrec_s r9
test_picklingzTestMRecords.test_pickling s y~~ yy##1f 5 9:: : :E TE222ALOOE dj 1 1 1 dj 9 9 9 dj 1 1 1 dj 9 9 9 9
: :r; c D t j g dg dt }t j g dg dt }t j g dg dd }dt fdt fd g}t |||g|d
}| }t
|d t j d t
t
|d t j dt
t
|d t j dd
d S )Nr r r rN r+ r r r onetwothreer) r% r& r' 韆 i@rJ r- rX )r r r r> )r r r r( )r r rJ )rY rD r] rh r filledr rC )r6 _a_b_cddtyper
mrecfilleds r9 test_filledzTestMRecords.test_filled' s0
Xiiiiiis
;
;
;
XoooIIIU
C
C
C
X---IIIU
K
K
K*sElL92r2,f%;= = =[[]]
Z_bh}C&H&H&HIIIZ_bh/A5:'< '< '< = = =Z_bh/D5:'< '< '< = = = = =r; c T t j g dg dt }t j g dg dt }t j g dg dd }dt fd t fd
g}t |||g|d }t | g d
d S )Nr rN r+ r r rV r) r% r& r' r r ))r r N)r r r! )NNr" )rY rD r] rh r r rA )r6 r r r r r s r9 test_tolistzTestMRecords.test_tolist6 s
Xiiiiiis
;
;
;
XoooIIIU
C
C
C
X---IIIU
K
K
K*sElL92r2,f%;= = = T[[]]. . . / / / / /r; c t dt d }d|d d< t |d d d d S )Nr r0 )formatsnamesrS r )r
rh r )r6 xs r9 test_withnameszTestMRecords.test_withnamesC sA af 5 5 5!V QvYq\2&&&&&r; c t ddt fddt fg }t |d< t | d d t ddg }t |d< t | d t j d |j t d
d }t |d< d |d< | d t | d t j d
d g|j d S )Nr i)sr) fr> r )r 1g ?)f0z > > "9:::QQ A T[[^^ Xy)&<,0J 8 8 8 9 9 9 9 9r; N) __name__
__module__ __qualname__ilistflistslistr] rh r r, rY rD listzipr0 r: rH rb rj rm rp rv r{ r r r r r r r r; r9 r r sj OOE % % %E 8 8 8ECj3,
5F ??D 28DDUE51122V L L LD
;
;
;&D &D &DP50 50 50n
5
5
5 : : : 4 4 4 4 4 4" 5 5 5@N N N:9 9 9"
:
:
:
=
=
= / / /' ' ' 9 9 9 9 9r; r c & e Zd Zd Zd Zd Zd ZdS )TestViewc D t j d t j d }}dt fdt fg}t j t
t || | }t ||g|d }d|j d< ||||f| _
d S ) NrS r% r& r> )g "g Xr )FTr ) rC re rf rg rh rD r r r r, r_ )r6 r% r& r^ arrr s r9 setup_methodzTestView.setup_method_ s )B-- !3!3A,e -htC1IIf5551a&;GGG$ ! 1a% r; c | j \ }}}}| }t t |t t || t |j |j d S )N)r_ r1 r
r4 r r r3 )r6 r r% r& r tests r9 test_view_by_itselfz TestView.test_view_by_itselfh sa Iq!Syy{{
4 //000 T4((( TZ44444r; c r | j \ }}}}t df}| | }t t |t
j t |t j t t || t t |d t
j u d S )Nr r> )r r )
r_ rh r1 r
r4 rY MaskedArrayr rC rD r r r )r6 r r% r& r ntyper s r9 test_view_simple_dtypez TestView.test_view_simple_dtypeo s Iq!S
yy
4 00111T28DQOO5AAABBBT
bi'(((((r; c | j \ }}}}dt fdt fg}| | }t t |t
t
|| | t |d d t u t |j t j | t |j d u d S )NABr ) r_ rh r1 r
r4 r r r r r- rC _fill_value)r6 r r% r& r alttyper s r9 test_view_flexible_typez TestView.test_view_flexible_typew s Iq!S<#u .yy!!
4 //000 T388G#4#4555S ! &'''TZ'!2!2333 D()))))r; N)r r r r r r r r r; r9 r r ] sP & & &5 5 5) ) )* * * * *r; r c 2 e Zd Z ej g dg de Z ej g dg de Z ej g dg dd Z defdefd gZ
e eee ge
d
Z e
ej ej e j fe
Ze ee
fZd
Zd Zd Zd Zd ZdS )TestMRecordsImportr rN r+ r )r r! r" r) r% r& r' )s 99999s 99999.rM r r> c t j g dg dt }t j g dg dt }t j g dg dd }| j \ }}}t d|||f D ]-\ }}t
t || j |j .t j g dg d t } t t | |j
|d d S ) Nr rN r+ r r r) r/ )r r r rV r> r )
rY rD r] rh r_ r r r@ r3 objectr r r- )
r6 r r r r nrecr r l_xs
r9 test_fromarraysz"TestMRecordsImport.test_fromarrays s
Xiiiiiis
;
;
;
XoooIIIU
C
C
C
X---IIIU
K
K
K )tQ/BB<88 : :FQ q))/ 9 9 9 9
XoooIIIV
D
D
D Z$* = = =tAwGGGGGr; c N | j \ }}}g d}t |d }t |d }t || t | }t |j |j |j j D ]3}t t || t |j | 4t | d }t |j dt fdt fdg t dd D ]6\ } }
t t || t |j |
7t | }t |j |j t |j |
t |j |j d S )
N))r abcg
@r )r xyg @r )r g ?r zc1, c2, c3, c4)r zc1,c2,c3c1c2)c3z|S5)r r r r/ )r_ recfromrecordsr r r r- r r@ r5 rA r] rh r r r3 ) r6 r r r palistpampa_mrecr8 r ns r9 test_fromrecordsz#TestMRecordsImport.test_fromrecords s #y tV4 4 4 F*:
;
;
;&(8999 R%%%D!!U[$*---[& L LE ..
E0J0J K K K KDKKMM<<<U[D#;u
}"MNNN,o>> D DFQ **GDJ,B,B C C C CD!!U[$*--- U[$++--888 U[$*55555r; c | j \ }}}t | |g d }t |j |j t |j g d t | |d }t |j |j t |j g d t | ||j }t |j |j t |j |j t | ||j }t |j |j t |j |j d S )Nrr )r- r, )rK rl rK T)rl rl rl )r_ r rA r r5 r r3 )r6 r r r r s r9 test_fromrecords_wmaskz)TestMRecordsImport.test_fromrecords_wmask s #y tVDKKMMjjjIII U[$*555U[''))+L+L+LMMMDKKMMdCCC U[$*555U[''))+L+L+LMMMDKKMMdjIII U[$*555U[''))4:+<+<+>+>???DKKMM!%!2!2!4!46 6 6 U[$*555U[''))4:+<+<+>+>?????r; c d}t 5 }t |d 5 }| | d d d n # 1 swxY w Y t |dd }d d d n # 1 swxY w Y t t |t t |j g d t |j j
g d t |j g d d S )Nz#
'One (S)','Two (I)','Three (F)','Four (M)','Five (-)','Six (C)'
'strings',1,1.0,'mixed column',,1
'with embedded "double quotes"',2,2.0,1.0,,1
'strings',3,3.0E5,3,,1
'strings',4,-1e-10,,,1
w,ABCDEFG) delimitervarnames)r r r r )r r g OAg|۽) r openwriter r
r4 r r FEr3 C)r6 fcontentpathr mrectxts r9 test_fromtextfilez$TestMRecordsImport.test_fromtextfile sg ZZ L4dC
"A!!!
"
"
"
"
"
"
"
"
"
"
"
"
"
"
""43KKKG L L L L L L L L L L L L L L L
7M22333WY ---WY_lll333WY 5 5 566666s3 A-A A-A A- A A--A14A1c | j \ }}}g dg d}}t |t j || }t |j | t |j j | d S )N)d i, rV rW )r_ r rY rD r f3r3 )r6 r r r dms r9
test_addfieldz TestMRecordsImport.test_addfield sk #y tV!//999Abhqq 1 1 122TWa TW]A&&&&&r; N)r r r rY rD r] r rh r r r r r
recfromarraysr5 r r_ r r r r r r r; r9 r r s7 ))))))3 7 7 7B /// ? ? ?B ,,, yy
/
/
/BCj3,
5F :r2rl&"* + + +D ="(BHbh7v F F FD $ D H H H6 6 64@ @ @*7 7 7&' ' ' ' 'r; r c l t j ddgddgdt fdt fg } | d d S ) N)r 2)r 4)r r )r r r% r& r+ r )rY masked_arrayr] r )ys r9 #test_record_array_with_object_fieldr sG
8 f
SzC=) + + +A
aDDDDr; ) __doc__numpyrC numpy.marY r r r
numpy.testingr numpy.core.recordsr r r r numpy.ma.mrecordsr r
r r numpy.ma.testutilsr
r r numpy.compatr r r r r r r; r9 r s # # # # # # # # " " " " " " 9 9 9 9 9 9 9 9D
"* "* "* "* "* "* "* "*L`' `' `' `' `' `' `' `'F r;