U
|
ý°dh ã @ sæ d dl Zd dlZd dlmZmZmZmZmZm Z d dl
|
m Z ej dedddgedgeddgg¡dd Zd d
|
Zd d Zd d Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zej d d d g¡d d Zd!d" ZdS )#é N)Ú DataFrameÚIndexÚIntervalÚ
|
MultiIndexÚSeriesÚ StringDtypeÚotherÚthreeÚoneÚtwoc sæ |j | |ddd\}}}|j | jd |d |jd | jd ¡sFt|jd ¡sZttj fdd| D td }| j| }t |j|¡ |d
|
krâ| j ||ddd\}} }
|
| |¡s¼tt ||
|
¡ t || ¡ t |j|¡ d S ) NÚsecondT)ÚhowÚlevelÚreturn_indexersé ©r r c s g | ]}|d kqS )r © )Ú.0Úx©Z exp_levelr ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/multi/test_join.pyÚ
|
<listcomp> s z#test_join_level.<locals>.<listcomp>©Zdtype)ÚouterÚinner)
|
ÚjoinÚlevelsÚequalsÚAssertionErrorÚnpÚarrayÚboolÚvaluesÚtmÚassert_numpy_array_equal) Úidxr Ú join_typeZ
|
join_indexÚlidxÚridxÚmaskZ
|
exp_valuesZ join_index2Zridx2Zlidx2r r r Útest_join_level s. ÿ
|
ÿ r* c C sV t dddg}|j| dd}t|ts*ttjtdd | j| dd W 5 Q R X d S ) Nr r
|
r r )r zJoin.*MultiIndex.*ambiguous)Úmatchr )r r Ú
|
isinstancer r ÚpytestZraisesÚ TypeError)r% ÚindexÚresultr r r Útest_join_level_corner_case+ s
|
r1 c C s | j | |d}t || ¡ d S ©Nr )r r# Úassert_index_equal©r% r& Zjoinedr r r Útest_join_self5 s r5 c C s´ t jt d¡t d¡gddgd} tdddgdd}| j|d d
|
d \}}}t jt d¡ddggddgd}tjdddd d dddgtjd}tjddddddddgtjd}t ||¡ t
|
||¡ t
|
||¡ |j| d d
|
d \}}}t ||¡ t
|
||¡ t
|
||¡ | j|dd
|
d \}}}tjddddddddddddddddgtjd}t || ¡ |d ksht t
|
||¡ |j| dd
|
d \}}}t || ¡ |d ks¤t t
|
||¡ d S )Né ÚaÚb©Únamesr é é )Únamer T)r r é é é
|
é é r r ÚleftéÿÿÿÿÚright) r Ú from_productr Zaranger r r Úintpr# r3 r$ r )Úmidxr% Zjidxr' r( Zexp_idxZexp_lidxÚexp_ridxr r r Útest_join_multi: s2 " " ÿ rJ c C s( | j r$| j| |d}| |k ¡ s$td S r2 )Z is_uniquer Úallr r4 r r r Útest_join_self_unique\ s rL c C s t jddgddggddgd} t jddgddggddgd}| j|dd \}}}tjd
|
d
|
d
|
d
|
gtjd }t | |¡ |d kstt ||¡ d S ) Nr r; é r6 r7 r8 r9 T©r rD r )
|
r rF r r r rG r# r3 r r$ )Úmidx1Úmidx2Zjoin_idxr' r( rI r r r Útest_join_multi_wrong_orderb s rQ c C sb t jddgddgddggddd gd
|
} t jddgddggddgd
|
}| j|d d }t || ¡ d S ) Nr r; rM r6 r< r> r7 r8 Úcr9 FrN )r rF r r# r3 )rO rP r0 r r r Ú test_join_multi_return_indexersr s & rS c
|
C s, t jdtddfdtddfdtddfdtddfdtddfdtddfgdd gd
|
} t jdtddfdtddfdtddfdtddfdtddfdtddfgdd gd
|
}t jdtddfdtddfdtddfdtddfdtddfdtddfgdd gd
|
}| j|d d }t ||¡ d S ) Nr g ç ð?ç @g @r; g @ÚnumÚintervalr9 r r )r Ú from_tuplesr r r# r3 )Zidx_1Zidx_2Úexpectedr0 r r r Ú$test_join_overlapping_interval_level| s@ ú÷ ú÷ ú÷ rZ c C s¸ t jtdddgddtdddgddgddgd} t jtdgddtdgddgdd gd}| j|d
|
d }t jtddgddtddgddtddgddgddd gd}t ||¡ d S ) Nr rM ZInt64r r; r7 r8 r9 rR r r )r Ú from_arraysr r r# r3 ©rH rP r0 rY r r r Útest_join_midx_ea¦ s" "þ ÿýúr] c C sÆ t jtdddgt dtdddgt dgddgd} t jtdgt dtdgt dgddgd}| j|dd}t jtddgt dtddgt dtddgt dgdddgd}t ||¡ d S )Nr7 rR r r8 r9 r r )r r[ r r r r# r3 r\ r r r Útest_join_midx_string» s& þû þýúr^ c C s¨ t dddgitjdgddggddgd d
|
} t d d d gitjdgtjdggddgd d
|
}| |¡}t ddgtjd gdtjdgddggddgd d
|
}t ||¡ d S )NÚcol1gñ?g333333ó?ÚArT rU Zid1Zid2r9 )Údatar/ Úcol2gÍÌÌÌÌÌ @g@)r_ rb ) r r rF r ÚNaNr Únanr# Zassert_frame_equal)Zdf1Zdf2r0 rY r r r Útest_join_multi_with_nanÔ s
|
þ
|
þ
|
þre Úvalr< c C s t tddg| dddgg¡}t td||g| ddddgg¡}|j|dd}t t||ddg| dddddgg¡ ¡ }t ||¡ d S )Nr r; r rM r6 r r )r r[ r r Z sort_valuesr# r3 )Úany_numeric_ea_dtyperf rH rP r0 rY r r r Útest_join_dtypesæ s ÿ ÿrh c C s t tddg| dtjtjgg¡}t tdddg| dtjtjtjgg¡}|j|dd}t tddddg| dtjtjtjtjgg¡}t ||¡ d S )Nr r; r r r r )r r[ r r rd r r# r3 )rg rH rP r0 rY r r r Útest_join_dtypes_all_nanô s ÿ ÿþÿri ) Únumpyr r- Zpandasr r r r r r Zpandas._testingZ_testingr# ÚmarkZ parametrizer* r1 r5 rJ rL rQ rS rZ r] r^ re rh ri r r r r Ú<module> s* ÿ
|
|
"
|
*
|
|
