U
|
P ±d8¦ ã @ s. 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 dd dZ&G dd d Z'G d
|
d d Z(G d d d Z)G dd dZ*G dd dZ+G dd dZ,G dd dZ-dS )é N)Ú MaskedRecords)Ú assert_equal)Úassert_Ú assert_raises)Ú drop_fieldsÚ rename_fieldsÚget_fieldstructureÚrecursive_fill_fieldsÚfind_duplicatesÚ merge_arraysÚ append_fieldsÚ stack_arraysÚjoin_byÚ repack_fieldsÚunstructured_to_structuredÚstructured_to_unstructuredÚapply_along_fieldsÚrequire_fieldsÚassign_fields_by_namec @ st e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd ZdS ) ÚTestRecFunctionsc C st t ddg¡}t dddg¡}t jddgdd tfgd
|
}t jd d gd tfddtfdtfgfgd
|
}||||f| _d S ©Né é é
|
é é ©ÚAç ð?©ÚBç @©r z|S3r ©Údtype©r ©r ç @©é ©é ç @ÚaÚbÚbaÚbb©ÚnpÚarrayÚfloatÚintÚdata©ÚselfÚxÚyÚzÚw© r= úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/lib/tests/test_recfunctions.pyÚ setup_method s
|
|
ÿ
|
ÿz TestRecFunctions.setup_methodc C sH | j \}}}}t||fdd}t|t dtfdtfg¡ t||fdd}t|t dtfdtfg¡ t||fdd}t|t dtfddtfg¡ t||fdd}t|t dtfdddtfgfg¡ t||fdd}t|t dtfdtfdtfd tfg¡ t||fdd}t|t dtfddtfd
|
dtfd tfgfgfg¡ d S ) NT©ÚflattenÚ Fr" r r- r/ r0 r. )r6 Ú zip_descrr r2 r$ r5 r4 )r8 r< r9 r: r; Útestr= r= r> Útest_zip_descr sF ÿÿÿ
|
ÿÿ
|
þÿ
|
ÿÿÿz TestRecFunctions.test_zip_descrc C s& t jddgdtfddtfdtfgfgd}t|d}t jdd gddtfdtfgfgd}t|| t|d}t jd
|
d gdtfgd}t|| t|dg}t jd d gdtfddtfgfgd}t|| t|ddg}t jd
|
d gdtfgd}t|| t|ddg}t jddgg d}t|| d S )Nr% r( r- r. r/ r0 r# )r& )r* ©r ©r) )r )r' )r) )r, r= )r2 r3 r5 r4 r r ©r8 r- rD Úcontrolr= r= r> Útest_drop_fieldsA s,
|
ÿ
|
|
ÿ
|
|
|
|
ÿ
|
|
z!TestRecFunctions.test_drop_fieldsc C s t jddddgffddddgffgd tfd
|
d tfd tdffgfgd }t|ddd}dtfd
|
d tfdtdffgfg}| |¡}t|j| t|| d S )Nr r r' ç >@r) r+ r, g N@r- r. r/ r0 r# r ÚBB)r- r0 )r2 r3 r5 r4 r Úviewr r$ )r8 r- rD ZnewdtyperI r= r= r> Útest_rename_fieldsa s "ÿÿ
|
z#TestRecFunctions.test_rename_fieldsc C s t ddtfg¡}t|}t|d t dtfddtfdtfgfg¡}t|}t|d t dtfdg fg¡}t|}t|d t g ¡}t|}t|d
|
d S ) Nr" r ©r r r- r. r/ r0 )r- )r. )r/ r0 )r- )r. r= r= )r2 r$ r4 Ú get_namesr r5 ©r8 ÚndtyperD r= r= r> Útest_get_namesl s
|
"
|
|
|
z TestRecFunctions.test_get_namesc C s t ddtfg¡}t|}t|d t dtfddtfdtfgfg¡}t|}t|d t dtfdg fg¡}t|}t|d t g ¡}t|}t|d
|
d S ) Nr" r rO r- r. r/ r0 )r- r. r/ r0 ©r- r. r= )r2 r$ r4 Úget_names_flatr r5 rQ r= r= r> Útest_get_names_flat~ s
|
"
|
|
|
z$TestRecFunctions.test_get_names_flatc C sæ t ddtfg¡}t|}t|g g d t dtfddtfdgfg¡}t|}t|g g dgdgd t dtfddtfdd tfd
|
tfgfgfg¡}t|}g g dgdgddgddgd }t|| t g ¡}t|}t|i d S ) Nr" r rO r ÚBA©rL z|S1)r r rW rL rL ÚBBAÚBBB)r r rW rL rY rZ )r2 r$ r4 r r r5 )r8 rR rD rI r= r= r> Útest_get_fieldstructure s(
|
ÿÿ ÿ
|
|
z(TestRecFunctions.test_get_fieldstructurec C s t jddddddgddddddgdtfd d
|
tfd gfgd }t|d dd}ddg}tt|d | t|d ||d t|ddd}dddddg}tt|d | t|d ||d t|d dd}ddddg}tt|d | t|d ||d t|d
|
dd}ddddg}tt|d | t|d ||d t|ddd}dddddg}tt|d | t|d ||d d S )N)r ©r! r )r r\ )r )r r )r )r! ÚC)r ©r r )r r^ )r ©r r r r rW rX ©Úmaskr$ FT©Z
|
ignoremaskÚ return_indexr r éÿÿÿÿ)Úkeyrc r é r+ r) rL )Úmar3 r5 r4 r
|
r ÚsortedrH r= r= r> Útest_find_duplicates« s@
|
ÿ ÿü z%TestRecFunctions.test_find_duplicatesc C sÆ dt fg}tjdddddddgdddddddgd |¡}t|ddd}dddd g}tt|d
|
| t|d ||d
|
t|d dd}ddddd d g}tt|d
|
| t|d ||d
|
d S ) Nr- r r rf r ©ra Trb r) rd Fé )r5 rg r3 rM r
|
r rh )r8 rR r- rD rI r= r= r> Ú test_find_duplicates_ignoremaskÌ s
|
ÿÿ z0TestRecFunctions.test_find_duplicates_ignoremaskc C s~ t jddd}t jd|d}tt|t d¡ tt|jd ttt|dd| t t j|f¡}tt|jt jk d S )Nzu1,f4,i8T©Zalignr r# é ) r2 r$ Úzerosr r ÚitemsizeÚrecordr Útype)r8 Údtr- r= r= r> Útest_repack_fieldsÛ s z#TestRecFunctions.test_repack_fieldsc C s¶ t jddddgd}t|}t|t jddd t jdd d
|
d gd d dgd}t jt|ddg dd}t|t ddddg¡ t jt|dg dd}t|t ddddg¡ t d¡ d¡}t||j }t jd d ddgfd d d dgfd!d"d#d$gfd%d&d'd(gfgdd)d*d+gfd,gd}t|| t jdd d
|
d gd d dgd}tt
|
t j|t d-d.d/dg¡ tt
|
t j|ddg t ddddg¡ t jdd d
|
d gd d0d1gd}t|}t||j }t |j |k t |j |k t jd2d3d4gdd gd5d6ggfd7d8d!gd9d:gd;d<ggfgd=d>d?gd}t|}t||j }t |j |k t |j |k t dt fd@t fg¡} t dA| fd)| fdB| fg¡}
|
t d!|
|
¡} t| t d} t| t jdCt d dDdE } dFdG }dOdHdI}|| |t jt jd4}t||dJt j|f || | t jd3d3}t||dKt j|f |t j}t||dLt j|f |t j| | t jd3d3}t||dMt j|f | }tttt d4|¡ tttt d4|¡t jd tttt jdNt jd d S )PNr) ©r- Úi4)r. zf4,u2)ÚcÚf4r r# )r) r+ Úf8)r r r+ )r) r+ é )rz é é )r r| é )r9 rv )r: rx )r; ry r9 r; rd )Zaxisr' g @g "@ç &@r g @ç @ç $@r r )r r r+ )r, rz ç @r )r~ r} g *@g ,@é )g 0@é g 2@g 3@r. )Úf0rx )Úf1Úu2)rw rx ©r gUUUUUU@gUUUUUU@gUUUUUU!@)r: rv )r; rv r r rf rk rz r{ é r| r} rn é )Zx0rv )Úx1)rv r )Zx2)rv ©r r r: r- rw )r rk c S s t | |f¡S ©N)r2 r$ )rs Úshaper= r= r> Úsubarray s zBTestRecFunctions.test_structured_to_unstructured.<locals>.subarrayc W s t dd t| D ¡S )Nc S s g | ]\}}d |¡|fqS )zx{})Úformat)Ú.0Úirs r= r= r> Ú
|
<listcomp>$ s zXTestRecFunctions.test_structured_to_unstructured.<locals>.structured.<locals>.<listcomp>)r2 r$ Ú enumerate)Zdtsr= r= r> Ú
|
structured# s zDTestRecFunctions.test_structured_to_unstructured.<locals>.structuredc S s2 t d| ¡}t||d}t|| }|j|j|jfS )Nr= r# )r2 ro r r r r$ )rs r$ ÚarrÚretZbackarrr= r= r> Úinspect& s
|
zATestRecFunctions.test_structured_to_unstructured.<locals>.inspect)rk rG rF )r+ )rf r )N)r2 ro r r r3 ZmeanÚarangeZreshaper r$ r r Úbaser5 Úint32r Ú
|
ValueErrorÚNotImplementedError)r8 r- Úoutr. rw ZwantÚdÚddZdddZpointZtriangler Úresr r r rs r= r= r> Ú test_structured_to_unstructuredç s ÿ ý
|
þü
|
ÿ ÿÿÿ ÿÿþ
|
|
|
ÿÿz0TestRecFunctions.test_structured_to_unstructuredc C sJ t d¡}dtfdtfg}t |¡}t||d}t||d}t|| d S )N)r r r9 r: r# )r2 ro r4 r$ r r )r8 r- Ztest_dtype_argsZ
|
test_dtypeZfield1Zfield2r= r= r> Ú test_unstructured_to_structuredA s
|
|
z0TestRecFunctions.test_unstructured_to_structuredc C s^ t jddddgd}ddg}tt||t d|¡ t jddg|d}t||d d
|
t|t jd d g|jd t|| t|t jd dg|jd t jddddgfgd}ddgfg}tt||t d|¡ t jddg|d}t||d d
|
t|t jddg|jd t|| t|t jddg|jd t d¡t d¡ }}t|| t|d d d S )Nr ru )r. ry ©rw Úu1r# ©r. rx ©r r )rf r) F)Zzero_unassigned)r r r )r rf r) )r r r )r rf r) r- )r )©rf )r¦ )©r rf ))r r ))r rf rf r r= )r2 Úonesr r r3 r r$ )r8 r- Znewdtr. r= r= r> Ú test_field_assignment_by_nameJ s&
|
|
|
z.TestRecFunctions.test_field_assignment_by_nameN)Ú__name__Ú
|
__module__Ú __qualname__r? rE rJ rN rS rV r[ ri rl rt r¡ r¢ rª r= r= r= r> r s ! ! Z r c @ s e Zd Zdd Zdd ZdS )ÚTestRecursiveFillFieldsc C sf t jddgdtfdtfgd}t jd|jd}t||}t jdddgdtfdtfgd}t|| d S )N©r r ©r ç 4@r r r# r§ ©r ç )r2 r3 r5 r4 ro r$ r r ©r8 r- r. rD rI r= r= r> Útest_simple_flexibleh s
|
ÿz,TestRecursiveFillFields.test_simple_flexiblec C st t jddgddgdtfdtfgd}t jd|jd }t||}t jddd
|
gddd gdtfdtfgd}t|| d S ) Nr¯ r° ©r r r_ r r r` r§ r# r² r^ )rg r3 r5 r4 ro r$ r r r´ r= r= r> Útest_masked_flexibleq s ÿ
|
þz,TestRecursiveFillFields.test_masked_flexibleN)r« r¬ r rµ r· r= r= r= r> r® f s r® c @ s\ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dS )ÚTestMergeArraysc C sz t ddg¡}t dddg¡}t jddgdd tfgd
|
}t jd d gd tfddtfdtfdg fgfgd
|
}||||f| _d S )Nr r r r r r r r" r r# ©r )r r' r= ©r) )r+ r, r= r- r. r/ r0 Úbcr1 r7 r= r= r> r? s
|
ÿ þz TestMergeArrays.setup_methodc C s| | j \}}}}t|}tjddgdtfgd}t|| t|f}t|| t|dd}t|| t|dd}t|| d S )NrF r r r# Fr@ T)r6 r r2 r3 r5 r ©r8 Ú_r9 r; rD rI r= r= r> Ú test_solo s
|
|
|
|
zTestMergeArrays.test_soloc C s^ | j d }t|dd}t|| t|dd}tjddgdtfdtfd tfgd
|
}t|| d S ) Nr Fr@ T)r r r' )r) r+ r, r- r/ r0 r# )r6 r r r2 r3 r5 r4 ©r8 r< rD rI r= r= r> Útest_solo_w_flatten s
|
|
|
ÿz#TestMergeArrays.test_solo_w_flattenc C s | j \}}}}t||fdd}tjdddgdtfdtfgd}t|| t||fd d}tjdddgd
|
d
|
d gdtfdtfgd }t|| t|j|j d S ) NF©Úusemask©r r ©r r ©rd r r r
r# Tr^ r_ r` )r6 r r2 r3 r5 r rg ra ©r8 r½ r9 r: rD rI r= r= r> Ú test_standard¤ s ÿ
|
þ
|
z TestMergeArrays.test_standardc C s | j \}}}}t||fdd}tjddgdtfddtfgd}t|| t||fd d}tjd
|
d gdtfd ddtfgfgd}t|| d S ) NTr@ )r r r )r r r! r r" r r# F)r r )r r r
©r6 r r2 r3 r5 r4 r r¼ r= r= r> Ú test_flatten´ s
|
ÿ
|
|
ÿÿz TestMergeArrays.test_flattenc C s° | j \}}}}t||fdd}tjddgdtfdtfdtfdtfgd }t|| t||fd
|
d}dtfd dtfd dtfdtfd g fgfgfg}tjddg|d }t|| d S )NTr@ )r r r r' )r r) r+ r, r r- r/ r0 r# Fr
r. r» )r r¹ )r rº rÈ )r8 r< r9 r½ rD rI Z controldtyper= r= r> Útest_flatten_wflexible s(
|
ÿÿ
|
ÿÿ
|
ÿz&TestMergeArrays.test_flatten_wflexiblec C s | j \}}}}tjdddgdddgd}t||fdd}tjdd d
|
gd d d gdtfdtfgd}t|| t||fddd}t|| tt|t d S )Nr r rf r rj TrÁ ©r r r )rd rf r¶ r^ r_ r r
r` )r Z
|
asrecarray) r6 rg r3 r r5 r r Ú
|
isinstancer )r8 r½ r9 ZmxrD rI r= r= r> Útest_wmasked_arraysÓ s þ
|
|
z#TestMergeArrays.test_wmasked_arraysc C sd t t ddg¡ dtfg¡t dddg¡f}tjddd gd
|
d
|
d gdtfd tfgd }t|| d S )Nr r r- r r± rK r¯ r° )rd rK r^ r_ r
r` )r r2 r3 rM r5 rg r4 r )r8 rD rI r= r= r> Útest_w_singlefieldà s ÿ þz"TestMergeArrays.test_w_singlefieldc C sT | j d }t|t dddg¡ dtfg¡f tjdddgd d
|
tfdtfgd d S ) Nrd r r r r] ©r r r ©r r! r )z-1rd r r" r r# )r6 r r2 r3 rM r5 r4 )r8 r; r= r= r> Útest_w_shorter_flexé s
|
|
$ ÿz#TestMergeArrays.test_w_shorter_flexc C sd | j \}}}}t|d |d |d fdd}tjdgdtfdtfddd tfgfgd
|
}t|| d S ) Nr FrÁ )r r ©r r r r
Úf2r" r r# rÈ )r8 r½ r9 r: r; rD rI r= r= r> Útest_singlerecordø s þÿz!TestMergeArrays.test_singlerecordN) r« r¬ r r? r¾ rÀ rÇ rÉ rÊ rÍ rÎ rÑ rÔ r= r= r= r> r¸ } s
|
r¸ c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) ÚTestAppendFieldsc C st t ddg¡}t dddg¡}t jddgdd tfgd
|
}t jd d gd tfddtfdtfgfgd
|
}||||f| _d S r r1 r7 r= r= r> r? s
|
ÿ
|
ÿz TestAppendFields.setup_methodc C sX | j \}}}}t|ddddgd}tjdddgd d d
|
gd tfdtfgd }t|| d S ) Nr r r r ©r6 rà rÄ rÅ r^ r_ r r` ©r6 r rg r3 r5 r ©r8 r½ r9 rD rI r= r= r> Útest_append_single s þz#TestAppendFields.test_append_singlec C sf | j \}}}}t|ddddgddggd}tjdd d
|
gd d d gd tfdtfdtfgd}t|| d S )NrO r r r éd éÈ rÖ )r r rÚ )r r rÛ )rd r rd ©r r r )r r r r r r r` r× rØ r= r= r> Útest_append_double s þz#TestAppendFields.test_append_doublec C sV | j d }t|ddddgd}tjddd gd
|
d
|
d gd d tfdtfgd}t|| d S )Nrd r] r r r rÖ rÏ rÐ )rd ç ð¿r rÜ ©r r r r" r r` )r6 r rg r3 r4 r5 r )r8 r; rD rI r= r= r> Útest_append_on_flex s
|
þz$TestAppendFields.test_append_on_flexc C sf | j d }t|ddddgd}tjddd gd
|
d
|
d gd tfd dtfdtfgfdtfgd}t|| d S )Nr r] r r r rÖ )r r& r )r) r* r )rd )rd rÞ r )r r^ r )r rË r r- r. r/ r0 r` )r6 r rg r3 r5 r4 r r¿ r= r= r> Útest_append_on_nested) s
|
þ ÿþûz&TestAppendFields.test_append_on_nestedN)r« r¬ r r? rÙ rÝ rà rá r= r= r= r> rÕ s
|
rÕ c @ sT e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dS )ÚTestStackArraysc C st t ddg¡}t dddg¡}t jddgdd tfgd
|
}t jd d gd tfddtfdtfgfgd
|
}||||f| _d S r r1 r7 r= r= r> r? : s
|
ÿ
|
ÿz TestStackArrays.setup_methodc C sP | j \}}}}t|f}t|| t||k t|}t|| t||k d S r )r6 r r r )r8 r½ r9 rD r= r= r> r¾ C s
|
|
|
zTestStackArrays.test_soloc C s | j \}}}}t||fdd}t ddddg¡}t|| t||fdd}t dddddg¡}t|| t||fdd}t dddddg¡}t|| d S )NFrÁ r r r r r )r6 r r2 r3 r rÆ r= r= r> Útest_unnamed_fieldsN s
|
|
z#TestStackArrays.test_unnamed_fieldsc C s | j \}}}}t||f}tjddddgddddgdtfdd tfgd
|
}t|| t|j|j t||f}tjd d d dgddddgdd tfdtfgd
|
}t|| t|j|j t|||f}tjd d d d d dgddddddgdd tfdtfgd
|
}t|| d S )N)r rd rd )r rd rd )rd r r )rd r r )r r r )r r r r r" r r` ©r r rd ©r r rd )rd rd r )rd rd r ©r r r rß rÓ )r6 r rg r3 r5 r4 r ra r¼ r= r= r> Ú test_unnamed_and_named_fields^ sV ÿ ÿü
|
ÿ ÿü
|
þ þúz-TestStackArrays.test_unnamed_and_named_fieldsc
|
C sô | j \}}}}tjdddgddtfdtfgd}t||f}tjdd dddgddtfdtfgd
|
d
|
d d d gd }t|| t|j|j t|||f}ddtfdtfd tfg}tjdddddddg|dddddddgd }t|| t|j|j d S )N©r- r ç Y@©r. r± ç i@©rw rK ç Àr@r" r r] r# rä rå ræ rÜ ©r$ ra Zf3)r r rd rd )r r rd rd )r- r ré rd )r. r± rë rd )rw rK rí rd )rd rd rd r )rd rd rd r )r r r r ©r r r r )r r r r ) r6 r2 r3 r4 r rg r ra r5 )r8 r½ r9 r; ÚzzrD rI rR r= r= r> Útest_matching_named_fields~ sN ÿ þ ÿü
|
ý þû
|
z*TestStackArrays.test_matching_named_fieldsc C sª | j \}}}}tjdddgddtfdtfgd}dd d
|
d d }t||f|d }tjdddddgddtfdtfgdddddgd}t|| t|j |j t|j|j d S )Nrè rê rì r" r r] r# z???g 8Àç ÃÀg ðiøÀ)r r r] ÚD)Údefaults)r r rò )r r rò ræ rÜ rî )r6 r2 r3 r4 r rg r ra )r8 r½ r; rð rô rD rI r= r= r> Ú test_defaults s* ÿ þ ÿü
|
z TestStackArrays.test_defaultsc C s¸ dt fdtfdtfg}tjdgdg|d}dt fdtfdtfg}tjdg|d}tjddgdd g|d}t||fd
|
d }t|| t|j|j tt t||fd d W 5 Q R X d S ) Nr r r] ©r r rf ©r r r r` ©r) r+ rk r# rÜ T)Z autoconvertF)
|
r5 Úboolr4 rg r3 r r ra r Ú TypeError)r8 Úadtyper- Úbdtyper. rI rD r= r= r> Útest_autoconversion« s ÿ
|
|
z#TestStackArrays.test_autoconversionc C s dt fdtfdtfg}tjdgdg|d}dt fdtfdtfg}tjdg|d}t||f}tjddgdd g|d}t|| t|j|j d S )
|
N)r- r )r. r )rw r] rö r÷ r` rø r# rÜ )r5 rù r4 rg r3 r r ra )r8 rû r- rü r. rD rI r= r= r> Útest_checktitles¹ s ÿ
|
z TestStackArrays.test_checktitlesc C s t jddgddtdfgd}t jddgd fd
|
d gd fd dgdfgddtdfdtfgd}t||f}tjddgdfddgdfddgd fdd gd fddgdfgddgdfddgdfddgdfddgdfddgdfg|jd}t|j|j t|| t|j|j d S ) NrÒ )r r r" r rF r# r- r ré r. r± rë rw rK rí r] ó Ar r ó Br! ó aó bó cFT)r6 ra r$ )r2 r3 r4 r rg r$ r ra )r8 r; rð r Úexpectedr= r= r> Ú test_subdtypeÅ s@ ÿ þ
|
|
|
ÿþ
|
|
|
|
|
û
|
|
|
|
|
ûò
|
z TestStackArrays.test_subdtypeN) r« r¬ r r? r¾ rã rç rñ rõ rý rþ r r= r= r= r> râ 8 s râ c @ sl e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd ZdS )Ú
|
TestJoinByc C s t jttt d¡t dd¡t dd¡dtfdtfdtfgd | _t jttt d
|
d ¡t d d ¡t dd¡dtfdtfdtfgd | _d S )Nr é2 é< rÚ én r- r. rw r# r+ r éA éK r ©r2 r3 ÚlistÚzipr r5 r- r. ©r8 r= r= r> r? ä s
|
ÿþ
|
ÿþzTestJoinBy.setup_methodc C sb | j | j }}td||dd}tjdddddgdtfd tfd
|
tfd tfd tfgd }t|| d S )Nr- Úinner)Újointype)r+ é7 r
|
éi rÚ )rk é8 éB éj ée )rz é9 éC ék éf )r{ é: éD él ég )r é; éE ém éh Úb1Úb2rw r r# ©r- r. r r2 r3 r5 r r´ r= r= r> Útest_inner_joinì s þ ÿýzTestJoinBy.test_inner_joinc C sN | j | j }}td|| tjdddddgdtfdtfd tfd
|
tfgd d S ) NrT )r+ r r rÚ )rk r r r )rz r r r )r{ r r r )r r r" r# r- r. rw r r# )r- r. r r2 r3 r5 ©r8 r- r. r= r= r> Ú test_joinø s þ ÿýzTestJoinBy.test_joinc C s` t jdgdtfgd}t jdt dddg¡fgdtfdgd}td||}t|| tj¡ d S )NrF re r# r r rf )ÚvalueZuint16rf )r2 r3 r5 r r rM rg Z MaskedArray)r8 ZfooÚbarr r= r= r> Útest_join_subdtype s ÿ
|
ÿ z TestJoinBy.test_join_subdtypec C s¢ | j | j }}td||d}tjddddddd d
|
d d d dddddddddgddddddddddddddddddddgdtfdtfdtfd tfgd }t|| d S ) NrT Úouter©r r rÚ rd ©r é3 r rd ©r é4 r rd ©rf é5 r rd ©r) é6 r# rd ©r+ r r rd )r+ r
|
rd rÚ ©rk r r rd )rk r rd r ©rz r r rd )rz r rd r ©r{ r r rd )r{ r rd r ©r r r" rd )r r! rd r# )r éF rd r )r| éG rd r )r} éH rd r )rn éI rd r )r éJ rd r" rï )r r r r r- r. rw r r` ©r- r. r rg r3 r5 r r´ r= r= r> Útest_outer_join s` ÷
|
÷
|
ÿìzTestJoinBy.test_outer_joinc C sz | j | j }}td||d}tjddddddd d
|
d d g
|
d d d d d d d d d d g
|
dtfdtfdtfdtfgd}t|| d S )NrT Z leftouterr. r/ r1 r3 r5 r7 r8 r9 r: r; rï r- r. rw r r` rA r´ r= r= r> Útest_leftouter_join1 s2 ü üö z TestJoinBy.test_leftouter_joinc C sX t jddddgd}t jddddgd}tddg||dd d
|
}t|jjddd d g d S ) Nrf ru r¥ r£ r# rw r. r F)r r Za1Za2)r2 ro r© r r r$ Únames)r8 r- r. Újr= r= r> Útest_different_field_orderB s z%TestJoinBy.test_different_field_orderc C sB t jddddgd}t jddddgd}tttdddg|| d S )Nrf ru r¥ r£ r# r- r. )r2 ro r© r r r r( r= r= r> Útest_duplicate_keysJ s z TestJoinBy.test_duplicate_keysc C sl t ddg¡}t ddg¡}t dddg¡}t jddg|d}t jd d
|
g|d}td ||}t|j| d S ) N)re ZS5©r* ú<f4©re ZS10©Zvalue1rI )Úvalue2rI ©ÚSarahr ©ÚJohnr, r# ©rN r ©rP r re ©r2 r$ r3 r r ©r8 Úa_dtypeÚb_dtypeÚexpected_dtyper- r. r r= r= r> Ú#test_same_name_different_dtypes_keyO s ÿ z.TestJoinBy.test_same_name_different_dtypes_keyc C sl t ddg¡}t ddg¡}t dddg¡}t jddg|d }t jd
|
d g|d }td ||}t|j| d S ) NrJ rH )r* ú<f8)re z|S10rK )rL rY rM rO r# rQ rR re rS rT r= r= r> Ú test_same_name_different_dtypes[ s ÿ z*TestJoinBy.test_same_name_different_dtypesc C sÐ t dtdfdg¡}t jdddgt jfdddgdfg|d}t dtdfdg¡}t jdddgdfdddgdfg|d}t dtdfddg¡}t jdddgt jdfg|d}td||}t|j| t|| d S ) NÚposrf )ÚfrI r r r³ r# )ÚgrI )r2 r$ r5 r3 Úpir r )r8 rU r- rV r. rW r r r= r= r> Útest_subarray_keyh s (& z TestJoinBy.test_subarray_keyc C sp t jddd}d|_tt|jd t ddg|¡}t dd g|¡}td
|
||}t d d d g¡}t|j| d S )Nzi1,f4Trm )ÚkÚvrf r¨ )rf r rË r r` )r` Úi1)Zv1rx )Zv2rx ) r2 r$ rD r ÚlenÚdescrr3 r r )r8 rs r- r. r rW r= r= r> Útest_padded_dtypev s ÿz TestJoinBy.test_padded_dtypeN)r« r¬ r r? r' r) r, rB rC rF rG rX rZ r_ re r= r= r= r> r ã s r c @ s8 e Zd Zedd Zdd Zdd Zdd Zd d
|
Zd S ) Ú TestJoinBy2c C s t jttt d¡t dd¡t dd¡dtfdtfdtfgd | _t jttt d¡t d
|
d ¡t dd¡dtfdtfd tfgd | _d S ) Nr r r rÚ r r- r. rw r# r
|
r r r )Úclsr= r= r> r? s
|
ÿþ
|
ÿþzTestJoinBy2.setup_methodc C sp | j | j }}td||dddd}tjdddd d
|
d d d ddg
|
dtfdtfdtfdtfdtfgd}t|| d S )Nr- rB Ú2r ©Ú r1postfixÚ r2postfixr ©r r r
|
rÚ rÚ ©r r0 r r r ©r r2 r r r ©rf r4 r r r ©r) r6 r! r# r# ©r+ r r< r r ©rk r r= r r ©rz r r> r r ©r{ r r? r r ©r r r@ r" r" r. r% rw r r# r& r´ r= r= r> Útest_no_r1postfix s2 ÿ ü ÿûz TestJoinBy2.test_no_r1postfixc C s t ttd| j| jddd d S )Nr- rB )rj rk )r r r r- r. r r= r= r> Útest_no_postfix s ÿzTestJoinBy2.test_no_postfixc C sp | j | j }}td||dddd}tjdddd d
|
d d d ddg
|
dtfdtfdtfdtfdtfgd}t|| d S )Nr- Ú1rB r ri rl rm rn ro rp rq rr rs rt ru r$ r. rw r r# r& r´ r= r= r> Útest_no_r2postfix¤ s2 ÿ ü ÿûz TestJoinBy2.test_no_r2postfixc C s6 t jttt ddgd¡t t d¡d¡t dd¡t dd¡dtfd tfd
|
tfd tfgd }t jttt ddgd¡t t d¡d¡t d d¡t dd¡dtfd tfd
|
tfd tfgd }t jddddddddddg
|
dtfd tfdtfdtfd tfd tfgd }td dg||d d d d!}t |j
|
|j
|
t || d S )"Nr r| r+ r r r r r` r- r. rw r# r
|
r r )r r r r
|
r r )r| r r0 r r| r )r r r2 r r} r )r| r r4 r rn rf )r r r6 r! r r) )r| r r r< r r+ )r rf r r= é rk )r| rf r r> r rz )r r) r r? é r{ )r| r) r r@ é r r$ r% Zc1Úc2rx rh r ri ) r2 r3 r r ZtileÚrepeatr r5 r r r$ )r8 r- r. rI rD r= r= r> Útest_two_keys_two_vars³ sL &
|
|
ÿþ&
|
|
ÿþ ü ÿû ÿz"TestJoinBy2.test_two_keys_two_varsN) r« r¬ r Ú classmethodr? rv rw ry r r= r= r= r> rf s
|
rf c @ s e Zd ZdZdd Zdd ZdS )ÚTestAppendFieldsObjz;
|
Test append_fields with arrays containing objects
|
c C s$ ddl m} t|dddd| _d S )Nr )ÚdateiÐ r )Úobj)Údatetimer Údictr6 )r8 r r= r= r> r? Î s z TestAppendFieldsObj.setup_methodc C s | j d }tj|df|dfgdtfdtfgd}tjddgtd}t|d |d
|
d }tj|ddf|ddfgdtfdtfd tfgd}t|| d S ) z7Test append_fields when the base array contains objectsr r r! r r r# r r r] F)r6 r N)r6 r2 r3 Úobjectr4 r5 r r )r8 r r9 r: rD rI r= r= r> Útest_append_to_objectsÒ s
|
ÿÿz*TestAppendFieldsObj.test_append_to_objectsN)r« r¬ r Ú__doc__r? r r= r= r= r> r È s r ).ZpytestÚnumpyr2 Znumpy.marg Znumpy.ma.mrecordsr Znumpy.ma.testutilsr Z numpy.testingr r Znumpy.lib.recfunctionsr r r r r
|
r r r r r r r r r r ÚlibZ recfunctionsZ_get_fieldspecZ get_fieldspecrP rU Z
|
_zip_descrrC Z
|
_zip_dtypeZ zip_dtyper r® r¸ rÕ râ r rf r r= r= r= r> Ú<module> s0 D
|
|
|
|
|
T 6 , %A
|
