y d dl Z d dlZd dlmZ d dlmZ d dlmZ d dl m
Z
m Z d dl m
Z
mZmZmZmZmZmZmZmZmZmZmZmZmZmZ ej j j Z ej j j Z ej j j! Z!ej j j" Z#ej j j$ 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-dS ) N)
MaskedRecords) assert_equal)assert_
assert_raises) drop_fields
rename_fieldsget_fieldstructurerecursive_fill_fieldsfind_duplicates merge_arrays
append_fields stack_arraysjoin_by
repack_fieldsunstructured_to_structuredstructured_to_unstructuredapply_along_fieldsrequire_fieldsassign_fields_by_namec \ e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
ZdS )TestRecFunctionsc t j ddg }t j g d }t j ddgddt fg }t j d d
gd t fd d
t fdt fgfg }||||f| _ d S N
A ?B @r! z|S3r$ dtyper r @ @abbabbnparrayfloatintdataselfxyzws u/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/lib/tests/test_recfunctions.py setup_methodz TestRecFunctions.setup_method s HaX H\\\ " " Hi+ (3,7
9
9
9 Hm]3 #Je}tSk.J(KL
N
N
N1aL c | j \ }}}}t ||fd }t |t j dt
fdt
fg t ||fd }t |t j dt
fdt
fg t ||fd }t |t j dt
fddt fg t ||fd }t |t j dt
fdddt fgfg t ||fd }t |t j dt
fdt
fdt fd t
fg t ||fd }t |t j dt
fddt
fd
dt fd t
fgfgfg d S ) NTflatten Fr& r$ r1 r3 r4 r2 )r: zip_descrr r6 r( r9 r8 )r< r@ r= r> r? tests rA test_zip_descrz TestRecFunctions.test_zip_descr s y Aq! !Q...TXCy2s) 455 7 7 7!Q///TXCy2s) 455 7 7 7 !Q...TXCy,e EFF H H H!Q///TXCy "\C<$@ A C D D E E E
!Q...TXCy #Sz $e}tSk ; < < = = = !Q///TXCy "c3Z&)T5MD#;+G%H%J K L M M N N N N NrC c t j ddgdt fddt fdt fgfg }t |d }t j dd gddt fdt fgfg }t || t |d }t j d
d gdt fg }t || t |dg }t j d d
gdt fddt fgfg }t || t |ddg }t j d
d gdt fg }t || t |ddg }t j ddgg }t || d S )Nr) r, r1 r2 r3 r4 r' )r* )r. r r- )r )r+ )r- )r0 )r6 r7 r9 r8 r r r< r1 rI controls rA test_drop_fieldsz!TestRecFunctions.test_drop_fieldsA s Hm]3 #Je}tSk.J(KL
N
N
N 1c""(K5#&$ s (D"E!FH H HT7### 1c""(D$<c
|<<<T7### 1th''(K5#&*sdC[M.B!CE E ET7### 1tTl++(D$<c
|<<<T7### 1sCj))(B82...T7#####rC c ` t j ddddgffddddgffgd t fd
d t fd t dffgfg
}t |ddd }dt fd
d t fdt dffgfg}| | }t
|j | t
|| d S )Nr r r+ >@r- r/ r0 g N@r1 r2 r3 r4 r' r! BB)r1 r4 )r6 r7 r9 r8 r viewr r( )r< r1 rI newdtyperP s rA test_rename_fieldsz#TestRecFunctions.test_rename_fieldsa s Hq1sCj/*QS#J,@A #J D%=4%2D"E F H
I
I
I Qc 6 677#Je}teQZ6H&I JK&&""TZ***T7#####rC c t j ddt fg }t | }t |d t j dt
fddt fdt
fgfg }t | }t |d t j dt
fdg fg }t | }t |d t j g }t | }t |d
d S ) Nr& r$ r! r$ r1 r2 r3 r4 )r1 )r2 )r3 r4 )r1 )r2 rN rN )r6 r( r8 get_namesr r9 r< ndtyperI s rA test_get_namesz TestRecFunctions.test_get_namesl s <#u 677 T:&&&C:tUmdC[-I'JKLL T5666C:Ry122 T+,,," T2 rC c t j ddt fg }t | }t |d t j dt
fddt fdt
fgfg }t | }t |d t j dt
fdg fg }t | }t |d t j g }t | }t |d
d S ) Nr& r$ rY r1 r2 r3 r4 )r1 r2 r3 r4 r1 r2 rN )r6 r( r8 get_names_flatr r9 r[ s rA test_get_names_flatz$TestRecFunctions.test_get_names_flat~ s <#u 677 f%%T:&&&C:tUmdC[-I'JKLL f%%T1222C:Ry122 f%%T:&&&" f%%T2 rC c V t j ddt fg }t | }t |g g d t j dt
fddt fdgfg }t | }t |g g dgdgd t j dt
fddt
fdd t
fd
t
fgfgfg }t | }g g dgdgddgddgd }t || t j g }t | }t |i d S ) Nr& r$ rY r! BArT z|S1)r! r$ rc rT rT BBABBB)r! r$ rc rT re rf )r6 r( r8 r r r9 )r< r\ rI rP s rA test_get_fieldstructurez(TestRecFunctions.test_get_fieldstructure sX <#u 677!&))T"--... C:tUm]-K'LMNN!&))T"SGC5IIJJJ C: 4+#'5#, )E"F"H IJ K K "&))C5 +sDk; ;T7### " !&))T2 rC c t j g dg ddt fddt fdgfg }t |dd
}d d g}t t
|d
| t |d ||d
t |dd }g d}t t
|d
| t |d ||d
t |dd }g d}t t
|d
| t |d ||d
t |dd }g d}t t
|d
| t |d ||d
t |dd }g d}t t
|d
| t |d ||d
d S )N)r r% r$ )r rj ri )r )r" r$ ri )r )r% C)r r r rl rl rl )r rm )r r r r! r$ rc rd maskr( FT
ignoremask return_indexr r )keyrs )r r r r/ )r r r r- rT )r r r rv r- )mar7 r9 r8 r r sortedrO s rA test_find_duplicatesz%TestRecFunctions.test_find_duplicates s
H F F FB B B #Je}m.L(MN
P
P
P qUFFFa&VDH%%w///T!WaRk*** qc===!//VDH%%w///T!WaRk*** qc=== ,,VDH%%w///T!WaRk*** qd>>> ,,VDH%%w///T!WaRk*** qd>>>!//VDH%%w///T!WaRk*****rC c dt fg}t j g dg d | }t |dd }g d}t t
|d | t |d ||d t |d
d }g d }t t
|d | t |d ||d d S ) Nr1 )r r r r r rv rv )r r r r r r r rp Trq )r r rv r- rt r F)r r r rv r- )r9 rw r7 rU r r rx )r< r\ r1 rI rP s rA test_find_duplicates_ignoremaskz0TestRecFunctions.test_find_duplicates_ignoremask s * H***///
1
1
115f
qTEEE ,,VDH%%w///T!WaRk*** qUFFF$$$VDH%%w///T!WaRk*****rC c t j dd }t j d| }t t | t j d t t | j d t t t | d | t j t j |f }t t | j t j u d S )Nzu1,f4,i8Talignr r'
) r6 r( zerosr r itemsizerecordr type)r< dtr1 s rA test_repack_fieldsz#TestRecFunctions.test_repack_fields s
Xj
-
-
- HQb ! ! !]2&&(<(<===]1%%.333]=#4#4DAAA2FFF Xry"o
&
&
b!!&")344444rC c t j dg d }t | }t |t j dd t j g dg d }t j t |dd g d
}t |t j g d t j t |dg d
}t |t j g d
t j d d }t ||j }t j ddddgfddddgfddddgfdd d d gfgd d d!d"gfd#g }t || t j g dg d }t t t j | t j g d$ t t t j |dd g t j g d t j g dg d% }t | }t ||j } t t j || t t j | | t |d dg }
t |
dd&gd'dgd(d'gd)dgg t t j |
| t j d&d*d+gddgd,d'ggfd-d.dgd(d)gd/d0ggfgg d1 }t | }t ||j } t t j || t t j | | |d d d
} t | } t | g d2g d3g |g d4 }
t |
}
t |
g d5g d6g t j d&d*d+gddgd,d'ggd7fd-d.dgd(d)gd/d0ggd8fgg d9 }t |g d: }t t j || t |g d3g d2g t j dt fd;t fg }t j d<|fd |fd=|fg }t j d| }t |t }t |t j d>t d? }d@ }dPdA} | | |t j t j d+ }t || dBt j |f | | |t j d* d* }t || dCt j |f |t j }t || dDt j |f |t j | |t j d* d* }t || dEt j |f | }t t t t j d+| t t" t t j d+| t j t t" t t j dFt j t j dGdHgd dIg }t |dJK }| t j }t |dLK }t |dJK } t t j || t t) | t j u t t) | t j u t || t | | t j |dMz dN|j |j O }||d d < t |dLK }t |dJK } t t j || t t) | t j u t t) | t j u t || t | | d S )QNr- )r1 i4)r2 zf4,u2)cf4r r' )r- r/ f8))r r r/ )r- r/ )r )r r )r= r )r> r )r? r r= r? rt )axis)r+ g @ "@ &@)r" @ @ $@r r )r" r r+ r r/ )r0 r @r r )r r g *@g ,@ )g 0@ g 2@g 3@r r2 )f0r )f1u2)r r r )gUUUUUU@gUUUUUU@gUUUUUU!@r )r )r> r )r? r r r r r r rv r| r r )x0r x1)r r x2)r r r )r r r r r r r )r r rv r- r/ r| r )r r r )r- r/ r| r r rv r )r r r r r r r @ )r r r )ignoredu1)r r r r> r1 r )r r| c . t j | |f S N)r6 r( )r shapes rA subarrayzBTestRecFunctions.test_structured_to_unstructured..subarray; s 8RK(( (rC c X t j d t | D S )Nc B g | ] \ }}d | |f S )zx{})format).0ir s rA
zXTestRecFunctions.test_structured_to_unstructured..structured..? s+ O O Ouq"ell1oor 2 O O OrC )r6 r( enumerate)dtss rA
structuredzDTestRecFunctions.test_structured_to_unstructured..structured> s' 8 O O # O O OPP PrC c t j d| }t || }t || }|j |j |j fS )NrN r' )r6 r r r r r( )r r( arrretbackarrs rA inspectzATestRecFunctions.test_structured_to_unstructured..inspectA sG (2r""C,S>>>C0b99G9ci 6 6rC )r| rM rL )r/ )rv r r r rv r- )r2 r T)copyFmemmapzw+)moder( r r )r6 r r r r7 meanarangereshaper r( r r
shares_memoryr9 int32r
ValueErrorNotImplementedErrorrU recarrayr r r )r< tmp_pathr1 outr2 r wantdddddd
dd_attrib_revd_revdd_rev d_attrib_revpointtriangler resr r r r d_plain dd_expecteds rA test_structured_to_unstructuredz0TestRecFunctions.test_structured_to_unstructured sK HQ K K K L L L(++S"(5555666 HEEEBBB
D
D
Dg0C: ??bIIIS"(#;#;#;<<===g0C5::DDDS"(#;#;#;<<=== IbMM ! !% ( ((AG44x1i24 i24 i#s4 i#s46 ( < "> ? - / 0 0 0 S$ HEEEBBB
D
D
D'33X > > >?? A A A'C: ??X 6 6 677 9 9 9
HEEEBBB
D
D
D
'
*
*(QW55 Q''((( a(())) 31c3Z=AA
]aVaVb!Wr2h$GHHH 22333
Hq1q'a!HAh#781b'RHr2h#78: 4 4 4
5
5
5 (
*
*(QW55 Q''((( a(())) $$B$+E22V 8 8 8 3 3 3 5 6 6 6 +++, 2<@@
]%:%:%:%?%?%?%A B B B
Hq1q'QFQF#3R81b'RHr2h#7<> G G G
H
H
H (*<*<*<(=
>
> Q''(((R///4446 7 7 7 3*sCj1228c5\C<#u FGGhr8$$(C888S"(7#666777 ) ) ) Q Q Q 7 7 7 7 ZBHbh!?!?CC
D
DWWR[[42"6777
Z"(A!6!6::
;
;WWR[[42"6777
Z
!
!WWR[[42"6777
Z((88BHa+@+@!"D"D
E
EWWR[[42"6777
Z\\j" r' )r6 r r8 r( r r )r< r1 test_dtype_args
test_dtypefield1field2s rA test_unstructured_to_structuredz0TestRecFunctions.test_unstructured_to_structuredy sl HW <#u 6Xo..
+A_EEE+AZ@@@VV$$$$$rC c t j dg d }ddg}t t || t j d| t j ddg| }t ||d t |t j d
d g|j t || t |t j d d
g|j t j ddddgfg }ddgfg}t t || t j d| t j ddg| }t ||d t |t j ddg|j t || t |t j ddg|j t j d t j d }}t || t |d d d S )Nr )r r2 r r r r' r2 r r r r F)zero_unassigned)r r r )r rv r- )r r r )r rv r- r1 r )r )rv )r )r rv ))r r ))r rv rv r rN )r6 onesr r r7 r r( )r< r1 newdtr2 s rA test_field_assignment_by_namez.TestRecFunctions.test_field_assignment_by_name s GA D D D E E E k*^Au--rwq%/@/@AAA HeU^5 1 1 1 aE::::Q''!2!'BBBCCC a###Q''!2!'BBBCCC
GAs[+$> ? @ A A A }%&^Au--rwq%/@/@AAA Hgw'u 5 5 5 aE::::Q(H!5QWEEEFFF a###Q(H!5QWEEEFFF x{{BHQKK1 a###QrUA rC N)__name__
__module__ __qualname__rB rJ rQ rW r] ra rg ry r} r r r r rN rC rA r r s ! ! ! N N NB $ $ $@ $ $ $ $ $ 6 + + +B
+
+
+
5
5
5P' P' P'd% % % rC r c e Zd Zd Zd ZdS )TestRecursiveFillFieldsc t j ddgdt fdt fg }t j d|j }t
|| }t j g ddt fdt fg }t || d S )Nr r r 4@r! r$ r' r r r )r )r6 r7 r9 r8 r r( r
r r< r1 r2 rI rP s rA test_simple_flexiblez,TestRecursiveFillFields.test_simple_flexible s Hh)3*sEl1K L L L HT ) ) )$Q**(888#&*sEl!;= = =T7#####rC c ( t j ddgddgdt fdt fg }t j d|j }t
|| }t j g d
g d dt fdt fg }t || d S ) Nr r r r rn r! r$ ro r r' r )r rn rm )rw r7 r9 r8 r r( r
r r s rA test_masked_flexiblez,TestRecursiveFillFields.test_masked_flexible s Hh)0@ #Je 5
7
7
7 HT ) ) )$Q**(888 8 8 8#&*sEl!;= = = T7#####rC N)r r r r r rN rC rA r r s2 $ $ $ $ $ $ $ $rC r c J e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d S ) TestMergeArraysc " t j ddg }t j g d }t j ddgddt fg }t j d d
gd t fd d
t fdt fdg fgfg }||||f| _ d S )Nr r r r r# r& r$ r' r )r r+ rN r- )r/ r0 rN r1 r2 r3 r4 bcr5 r; s rA rB z TestMergeArrays.setup_method s HaX H\\\ " " H
"<#u *F
H
H
H H
1 2:tUmdC[4*%M NO
Q
Q
Q 1aL rC c ` | j \ }}}}t | }t j ddgdt fg }t || t |f }t || t |d }t || t |d }t || d S )NrL r r r' FrE T)r: r r6 r7 r9 r r< _r= r? rI rP s rA test_solozTestMergeArrays.test_solo s y Aq!A(D$<s }===T7###QD!!T7###Au---T1 At,,,T1 rC c | j d }t |d }t || t |d }t j ddgdt
fdt fd t
fg
}t || d S ) Nr FrE T)r r r+ )r- r/ r0 r1 r3 r4 r' )r: r r r6 r7 r9 r8 r< r@ rI rP s rA test_solo_w_flattenz#TestMergeArrays.test_solo_w_flatten s IaLAu---T1 At,,,(K5#&*tUmdC[!IK K KT7#####rC c | j \ }}}}t ||fd }t j g ddt fdt fg }t || t ||fd }t
j g dg ddt fdt fg }t || t |j |j d S )
NFusemask)r r )r r )rt r r r r' Trm rm rn ro )r: r r6 r7 r9 r rw rp r< r r= r> rI rP s rA
test_standardz TestMergeArrays.test_standard s y Aq!QFE222(777#'+c{!;= = =T7###QFD111(777 8 8 8#'+c{!;= = = T7###TY -----rC c T | j \ }}}}t ||fd }t j ddgdt fddt
fg }t
|| t ||fd }t j d
d gdt fd ddt
fgfg }t
|| d S )
NTrE )r r! r" )r r$ r% r r& r$ r' F)r r )r r# r r: r r6 r7 r9 r8 r r s rA test_flattenz TestMergeArrays.test_flatten s y Aq!QFD111(L,7#'+|c5\!JL L LT7###QFE222(NN;#'+#',e )E"F"HI I I T7#####rC c | j \ }}}}t ||fd }t j ddgdt fdt fdt
fdt fg }t
|| t ||fd
}dt fd dt fd dt
fdt fd
g fgfgfg}t j ddg| }t
|| d S )NTrE )r r r r+ )r r- r/ r0 r r1 r3 r4 r' Fr r2 r )r" r )r r r )r< r@ r= r rI rP controldtypes rA test_flatten_wflexiblez&TestMergeArrays.test_flatten_wflexible s y Aq!QFD111(NN;#'+#&*tUmdC["JK K K T7###QFE222 s !%c
),e}tSkDRT:.V(W(Y !Z[ ( 35KL!-/ / /T7#####rC c z | j \ }}}}t j g dg d }t ||fd }t j g dg ddt fd t fg
}t || t ||fdd }t || t
t |t d S ) Nr r rv r r r r{ Tr
)r r r )rt rv )r rm rn r r ro )r
asrecarray) r: rw r7 r r9 r r
isinstancer )r< r r= mxrI rP s rA test_wmasked_arraysz#TestMergeArrays.test_wmasked_arrays s y Aq!
Xiiiiii
0
0
0QGT222(444 8 8 8#'+c{!;= = = T7###QGTdCCCT7###
4 //00000rC c $ t t j ddg dt fg t j g d f }t j g dg ddt fdt fg }t || d S ) Nr r r1 )r r rS )r r )rt rS r
r ro )r r6 r7 rU r9 rw r8 r )r< rI rP s rA test_w_singlefieldz"TestMergeArrays.test_w_singlefield s RXq!f - - 2 2S#J< @ @ Xooo 6 6 8 : :(::: 8 8 8#&*tUm!<> > > T7#####rC c | j d }t |t j g d dt
fg f t j g dddt fdt
fg d S )Nrt r rk )r! r" r r$ r% r )z-1rt r r& r$ r' )r: r r6 r7 rU r9 r8 )r< r? s rA test_w_shorter_flexz#TestMergeArrays.test_w_shorter_flex! s IbM a,,,//44sCj\BBCDDD
???$sElS#J? A A A A A ArC c | j \ }}}}t |d |d |d fd }t j dgdt fdt fddd t
fgfg
}t
|| d S ) Nr Fr
)r r r! r r r f2r& r$ r' r )r< r r= r> r? rI rP s rA test_singlerecordz!TestMergeArrays.test_singlerecord0 s y Aq!QqT1Q41 .>>>( -.#'+#'+#',e )E"F"HI I I T7#####rC N)
r r r rB r r r r r r r r# r' rN rC rA r r s ! ! !
$ $ $. . . $ $ $ $ $ $" 1 1 1$ $ $
A
A
A $ $ $ $ $rC r c , e Zd Zd Zd Zd Zd Zd ZdS )TestAppendFieldsc t j ddg }t j g d }t j ddgddt fg }t j d d
gd t fd d
t fdt fgfg }||||f| _ d S r r5 r; s rA rB z TestAppendFields.setup_method= HaX H\\\ " " H
"<#u *F
H
H
H Hm]3 #Je}tSk.J(KL
N
N
N1aL rC c | j \ }}}}t |dg d }t j g dg ddt fdt fg }t || d S )Nr! r r: r r
r ro r: r
rw r7 r9 r r< r r= rI rP s rA test_append_singlez#TestAppendFields.test_append_singleF sy y Aq! Q,,,777(777 8 8 8#'+Sz!:= = = T7#####rC c | j \ }}}}t |dg dddgg }t j g dg ddt fd t fd
t fg }t || d S ) NrY r d r- ))r r r2 )r r r3 )rt r rt )r r r r4 )r r r r r! r$ ro r. r/ s rA test_append_doublez#TestAppendFields.test_append_doubleO s y Aq! Q
,,,c
1KLLL(EEE A A A#'+SzC:!FI I I T7#####rC c | j d }t |dg d }t j g dg dddt fdt
fg }t
|| d S )
Nrt rk r r- )r! r" )rt r )r4 r4 r r r r& r$ ro )r: r
rw r7 r8 r9 r )r< r? rI rP s rA test_append_on_flexz$TestAppendFields.test_append_on_flexX sz IbM Q,,,777(HHH A A A".e sCj!IL L L T7#####rC c | j d }t |dg d }t j g dg ddt fdd t
fd
t fgfdt fg }t
|| d S ) Nr rk r r- ))r r* r )r- r. r )rt )rt r7 r )r rm r r; )r r r r1 r2 r3 r4 ro )r: r
rw r7 r9 r8 r r s rA test_append_on_nestedz&TestAppendFields.test_append_on_nesteda s IaL Q,,,777( 1 1 1!N !N !N#&*#&$ s (D"E#&*". 0 0 0 T7#####rC N)r r r rB r0 r5 r9 r< rN rC rA r) r) : s_ ! ! !$ $ $$ $ $$ $ $ $ $ $ $ $rC r) c D e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
S ) TestStackArraysc t j ddg }t j g d }t j ddgddt fg }t j d d
gd t fd d
t fdt fgfg }||||f| _ d S r r5 r; s rA rB z TestStackArrays.setup_methodr r+ rC c | j \ }}}}t |f }t || t ||u t | }t || t ||u d S r )r: r r r )r< r r= rI s rA r zTestStackArrays.test_solo{ su y Aq!QD!!T1 AT1 rC c t | j \ }}}}t ||fd }t j g d }t || t ||fd }t j g d }t || t ||fd }t j g d }t || d S )NFr
)r r r r )r r r r r )r r r r r )r: r r6 r7 r r s rA test_unnamed_fieldsz#TestStackArrays.test_unnamed_fields s y Aq!QFE222(<<<((T7###QFE222(---..T7###QFE222(---..T7#####rC c V | j \ }}}}t ||f }t j g dg ddt fddt
fg }t
|| t
|j |j t ||f }t j g dg dddt
fd t fg }t
|| t
|j |j t |||f }t j g d
g d ddt
fd t fg }t
|| d S ) N))r rt rt )r rt rt )rt r! r )rt r$ r )r r r rD r r r r& r$ ro )r! r rt r$ r rt rt rt r rt rt r )r r r rI r8 r8 r& )rE rF rE rF rG rH )rI rI rI rI r8 r8 )r: r rw r7 r9 r8 r rp r s rA test_unnamed_and_named_fieldsz-TestStackArrays.test_unnamed_and_named_fields s y Aq!QF##( 8 8 8!7 !7 !7#'+|c5\!J L L L
T7###TY ---QF##( 8 8 8!7 !7 !7".e tSk!J L L L
T7###TY ---Q1I&&( 8 8 8!7 !7 !7 #/e tSk!J
L L L T7#####rC c " | j \ }}}}t j g dddt fdt fg }t ||f }t j g dddt fdt fgg d }t
|| t
|j |j t |||f }ddt fdt fd t fg}t j g d
|g d }t
|| t
|j |j d S ) Nr1 r Y@r2 r i@r rS r@r& r$ rk r' )rE rF rM rO rQ rI rI r4 r4 r4 r( rp f3))r! r rt rt )r$ r rt rt )r1 r rN rt )r2 r rP rt )r rS rR rt )rt rt rt r )rt rt rt r )r r r r rV r r r r rW rW r r r r rX ) r: r6 r7 r8 r rw r rp r9 )r< r r= r? zzrI rP r\ s rA test_matching_named_fieldsz*TestStackArrays.test_matching_named_fields s_ y Aq!
XLLL )C<#u FH H HQG$$( U U U #/e sEl!K!B !B !B C C C T7###TY ---QAJ'' e sElT3KH( > > > "(!= != != > > > T7###TY -----rC c | j \ }}}}t j g dddt fdt fg }dddd d
}t ||f| }t j g d ddt fdt fgg d
}t
|| t
|j |j t
|j |j d S )NrL r& r$ rk r' z???g 8 g i)r! r$ rk D)defaults))r! r r\ )r$ r r\ rM rO rQ rS rT )r: r6 r7 r8 r rw r rp )r< r r? rY r^ rI rP s rA
test_defaultsz TestStackArrays.test_defaults s y Aq!
XLLL )C<#u FH H H UgFFQGh777( U U U #/e sEl!K!B !B !B C C C T7###TY ---TY -----rC c dt fdt fdt fg}t j dgdg| }dt fdt fdt fg}t j dg| }t j ddgdd g| }t ||fd
}t
|| t
|j |j t t 5 t ||fd d d d d S # 1 swxY w Y d S )
Nr! r$ rk r r r r ro r- r/ r| r' r4 T) autoconvertF)
r9 boolr8 rw r7 r r rp r TypeError)r< adtyper1 bdtyper2 rP rI s rA test_autoconversionz#TestStackArrays.test_autoconversion sL *sDkC<8 Hi[ {& A A A*sElS%L9 Hi[ / / /(Iy1I8N!') ) )QF555T7###TY ---
9
%
% 4 4 !QU 3 3 3 3 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4s C22C69C6c z dt fdt fdt fg}t j dgdg| }dt fdt fdt fg}t j dg| }t ||f }t j ddgdd g| }t
|| t
|j |j d S )
N)r1 r! )r2 r$ )r rk r ra ro rb r' r4 )r9 rd r8 rw r7 r r rp )r< rf r1 rg r2 rI rP s rA test_checktitlesz TestStackArrays.test_checktitles s s#j$%7*e9LM Hi[ {& A A A s#j$%7*e9LM Hi[ / / /QF##(Iy1I8N!') ) )T7###TY -----rC c
t j ddgddt dfg }t j ddgd fd
d gd fd
dgdfgddt dfdt fg }t ||f }t j ddgdfddgdfddgd fdd gd fddgdfgddgdfddgdfddgdfddgdfddgdfg|j }t
|j |j t
|| t
|j |j d S ) Nr% )r$ r r& r$ rL r' r1 r rN r2 r rP r rS rR rk Ar" r Br% a b cFT)r: rp r( )r6 r7 r8 r rw r( r rp )r< r? rY r expecteds rA
test_subdtypez TestStackArrays.test_subdtype s| H h
eT 23
5
5
5 X
3% seT 2S3%4F
eT 2S%LAC C C Ar7##8ua ua vu%vu%vu% ' $'$'''' (
SY///S(###SXx}-----rC N) r r r rB r rB rJ rZ r_ rh rj rr rN rC rA r> r> p s ! ! ! $ $ $ $ $ $@. . .8. . ." 4 4 4
.
.
.. . . . .rC r> c V e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
S )
TestJoinByc $ t j t t t j d t j dd t j dd dt
fdt
fdt
fg | _ t j t t t j d
d t j d d
t j dd dt
fdt
fdt
fg | _ d S )Nr 2 < r2 n r1 r2 r r' r/ r A K r r6 r7 listzipr r9 r1 r2 r< s rA rB zTestJoinBy.setup_method s $s29R==")B2C2C#%9S##6#6 8 8 9 9"%sc3Z#s DF F F $s29Q#3#3RYr25F5F#%9S##6#6 8 8 9 9"%sc3Z#s DF F FrC c | j | j }}t d||d }t j g ddt
fdt
fdt
fdt
fdt
fg }t
|| d S )
Nr1 inner)jointype))r/ 7 ry i r2 )r| 8 B j e )r 9 C k f )r : D l g )r ; E m h b1b2r r r' r1 r2 r r6 r7 r9 r r s rA test_inner_joinzTestJoinBy.test_inner_join$ s vtv1sAq7333( 3 3 3 $'*tSkD#;#&*sCj":; ; ;
T7#####rC c | j | j }}t d|| t j g ddt
fdt
fdt
fdt
fg d S )Nr_ ))r/ r r r2 )r| r r r )r r r r )r r r r )r r r r r1 r2 r r r' )r1 r2 r r6 r7 r9 r< r1 r2 s rA test_joinzTestJoinBy.test_join0 sv vtv1
Aq!!!
% % % sc3Z sc3Z1 2 2 2 2 2 2rC c , t j dgdt fg }t j dt j g d fgdt fdg }t d|| }t || t j d S )NrL ru r' r r )valueuint16rv )r6 r7 r9 r r rU rw MaskedArray)r< foobarr s rA test_join_subdtypez TestJoinBy.test_join_subdtypeD s hv $cl^- - -hBHWWW - -./ $cl,B CE E EeS#&&S#((2>2233333rC c | j | j }}t d||d }t j g dg ddt
fdt
fdt
fdt
fg }t
|| d S )
Nr_ outer)r rv r2 rt r 3 r rt r 4 r rt rv 5 r rt r- 6 r rt r/ r r rt )r/ ry rt r2 r| r r rt )r| r rt r r r r rt )r r rt r r r r rt )r r rt r r r r rt )r r rt r )r F rt r )r G rt r )r H rt r )r I rt r )r J rt r )rW rW rW rW rW rW r r r r rW r rW r rW r rW r r r r r r r1 r2 r r ro r1 r2 r rw r7 r9 r r s rA test_outer_joinzTestJoinBy.test_outer_joinM s vtv1z1a11( B B B != != != $'*sCj#&*sCj":); ; ;, T7#####rC c | j | j }}t d||d }t j g dg ddt
fdt
fdt
fdt
fg }t
|| d S )
Nr_ leftouter)
r r r r r r r r r r )
rW rW rW rW rW rW rW rW rW rW r1 r2 r r ro r r s rA test_leftouter_joinz TestJoinBy.test_leftouter_joini s vtv1z1a55( @ @ @
!= != !=
$'*sCj3*sCj!Q
S
S
S T7#####rC c t j dg d }t j dg d }t ddg||dd }t |j j g d
d S ) Nrv r r r r' r r r r r2 r F)r r )r2 r a1a2)r6 r r r r r( names)r< r1 r2 js rA test_different_field_orderz%TestJoinBy.test_different_field_orderz su HQ E E E F F F GA D D D E E E S#J1w F F FQW]$:$:$:;;;;;rC c t j dg d }t j dg d }t t t
g d|| d S )Nrv r r' r )r1 r2 r2 )r6 r r r r r r s rA test_duplicate_keysz TestJoinBy.test_duplicate_keys sX HQ E E E F F F GA D D D E E Ej'???AqAAAAArC c 6 t j ddg }t j ddg }t j g d }t j ddg| }t j dd g| }t d
|| }t |j | d S ) N)ru S5r > #B #B #B C C
Ho}5W E E E Ho}5W E E EeQ""SY/////rC c 6 t j ddg }t j ddg }t j g d }t j ddg| }t j dd g| }t d
|| }t |j | d S ) Nr r )r >(N,<=>> #C #C #C D D
Ho}5W E E E Ho}5W E E EeQ""SY/////rC c t j dt dfdg }t j g dt j fg ddfg| }t j dt dfdg }t j g ddfg d dfg| }t j dt dfddg }t j g dt j dfg| }t d|| }t
|j | t
|| d S )
Nposrv )fr )r r r r r r' )gr )rv r r )r6 r( r9 r7 pir r )r< r r1 r r2 r rq r s rA test_subarray_keyz TestJoinBy.test_subarray_key s (UCO\:;; Hyyy"%(999c*:;7 K K K(UCO\:;; Hyyy!nyyy#&67w G G GE3?L,"OPP8iii 2 3>JJJeQ""SY///S(#####rC c X t j dd }d|_ t t |j d t j ddg| }t j dd g| }t d
|| }t j g d }t |j | d S ) Nzi1,f4Tr )kvrv r )rv r r r r ))r i1)v1r )v2r ) r6 r( r r lendescrr7 r r )r< r r1 r2 r r s rA test_padded_dtypez TestJoinBy.test_padded_dtype s
XgT
*
*
* BH
q!!! Hff%r * * Hff%r * *c1a #
#
#
SY/////rC N)r r r rB r r r r r r r r r r r rN rC rA rt rt s F F F
$
$
$2 2 2(4 4 4$ $ $8$ $ $"< < <B B B
0
0
0 0 0 0 $ $ $ 0 0 0 0 0rC rt c < e Zd Zed Zd Zd Zd Zd ZdS ) TestJoinBy2c " t j t t t j d t j dd t j dd dt
fdt
fdt
fg | _ t j t t t j d t j d
d t j dd dt
fdt
fd t
fg | _ d S )
Nr rv rw r2 rx r1 r2 r r' ry rz r r{ )clss rA rB zTestJoinBy2.setup_method s c")B--2r1B1B"$)C"5"5 7 7 8 8!$c
S#Jc
CE E E c")B--2r1B1B"$)C"5"5 7 7 8 8!$c
S#Jc
CE E ErC c | j | j }}t d||ddd }t j g ddt
fdt
fdt
fd t
fd
t
fg }t
|| d S ) Nr1 rG 2r r1postfix r2postfixr
)r rv ry r2 r2 )r r r r r )r r r r r )rv 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 r2 r r r r' r r s rA test_no_r1postfixz TestJoinBy2.test_no_r1postfix s vtv1 AsWF F F( J J J
$'*sCj4+#&*sCj": ; ; ; T7#####rC c Z t t t d| j | j dd d S )Nr1 rG )r r )r r r r1 r2 r~ s rA test_no_postfixzTestJoinBy2.test_no_postfix s6 j'3 "b 2 2 2 2 2 2rC c | j | j }}t d||ddd }t j g ddt
fdt
fdt
fd t
fd
t
fg }t
|| d S ) Nr1 1rG r r r r r2 r r r' r r s rA test_no_r2postfixz TestJoinBy2.test_no_r2postfix s vtv1 AWF F F( J J J
$'*tSkC:#&*sCj": ; ; ; T7#####rC c
t j t t t j ddgd t j t j d d t j dd t j dd dt fd t fd
t fd t fg }t j t t t j ddgd t j t j d d t j d
d t j dd dt fd t fd
t fd t fg }t j g ddt fd t fdt fdt fdt fdt fg }t d dg||ddd }t |j
|j
t || d S )Nr r r/ r rv rw r r r1 r2 r r' ry rz r )
)r r rv ry r r )r r r r r r )r r r r r r )r r r r r rv )r r r r r r- )r r r r r r/ )r rv r r r| )r rv r r r r )r r- r r r )r r- r r r r r c1c2r r r r ) r6 r7 r| r} tilerepeatr r9 r r r( )r< r1 r2 rP rI s rA test_two_keys_two_varsz"TestJoinBy2.test_two_keys_two_vars s HT#bgr2h 2 2BIbillA4N4N iB / /2r1B1BD D E E #Jc
S#Jc
K
M
M
M
HT#bgr2h 2 2BIbillA4N4N iB / /1b1A1AC C D D #Jc
S#Jc
K
M
M
M ( L L L
$'*sCj4+#'+c{T3K"I J J J
#J1sWN N NTZ ///T7#####rC N) r r r classmethodrB r r r r rN rC rA r r sl E E [E
$
$
$ 2 2 2
$
$
$ $ $ $ $ $rC r c e Zd ZdZd Zd ZdS )TestAppendFieldsObjz;
Test append_fields with arrays containing objects
c R ddl m} t |ddd | _ d S )Nr )datei r )obj)datetimer dictr: )r< r s rA rB z TestAppendFieldsObj.setup_method s8 !!!!!!TT$1 - -... rC c d | j d }t j |df|dfgdt fdt fg }t j ddgt
}t
|d |d
}t j |ddf|ddfgdt fdt fd t
fg }t || d S )
z7Test append_fields when the base array contains objectsr r" r% r! r$ r' r r rk F)r: r N)r: r6 r7 objectr8 r9 r
r )r< r r= r> rI rP s rA test_append_to_objectsz*TestAppendFieldsObj.test_append_to_objects
s i HsBi#r+ !6]S%L 9
;
;
; Hb"XS ) ) ) Q!U;;;(S#rNS#rN;#&-#u Sz!JL L LT7#####rC N)r r r __doc__rB r rN rC rA r r s<
/ / / $ $ $ $ $rC r ).pytestnumpyr6 numpy.marw numpy.ma.mrecordsr numpy.ma.testutilsr
numpy.testingr r numpy.lib.recfunctionsr r r r
r r r
r r r r r r r r lib recfunctions_get_fieldspec
get_fieldspecrZ r`
_zip_descrrH
_zip_dtype zip_dtyper r r r) r> rt r r rN rC rA r! s
+ + + + + + + + + + + + 0 0 0 0 0 0 0 0? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?
#2
F ) $3 F * F * G G G G G G G G T $ $ $ $ $ $ $ $.B$ B$ B$ B$ B$ B$ B$ B$J3$ 3$ 3$ 3$ 3$ 3$ 3$ 3$lh. h. h. h. h. h. h. h.Va0 a0 a0 a0 a0 a0 a0 a0H?$ ?$ ?$ ?$ ?$ ?$ ?$ ?$B$ $ $ $ $ $ $ $ $ $rC