U
|
ý°d ã @ s¨ d dl Zd dlZd dlZd dlmZmZmZmZm Z m
|
Z
|
m Z m Z d dl mZ d?ddZG dd dZd@d d Zdd Zdd Zdd ZdAddZdd Zej ddddgfddd gfg¡d d Zd d Zej d!d"d#gd"d#d#gfd$d%gd$d%d%gfd&d'gd&d'd'gfe
|
d(e
|
d)ge
|
d(e
|
d)e
|
d)gfg¡d*d+ Z ej d,d-d.d/d0d1d2d3d4d5d6g
|
¡d7d8 Z ej d9d:d d%d$gfd;d d%d$gfd d d%d$gfd<d d$d%gfg¡d=d> Z dS )Bé N)Ú CategoricalÚ DataFrameÚIndexÚ
|
MultiIndexÚSeriesÚ TimestampÚconcatÚmergeé é2 c C s` t t| }t t ||| ¡¡}t||k rPt t ||d |t|
¡}tj |¡ |S ©N)ÚlistÚrangeÚnpZasarrayÚtileÚlenÚrandomÚshuffle)ÚngroupsÚnZ unique_groupsZarr© r ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/reshape/merge/test_join.pyÚ get_test_data s " r c
|
@ s~ e Zd Zejdd Zejdd Zejdd Zdd Zd d
|
Z d d Z
|
d d Z dd Z dd Z dd Zdd Zdd Zdd Zej dd d d e d d g¡g¡d!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Z d/d0 Z d1d2 Z d3d4 Z d5d6 Z d7d8 Z!d9d: Z"d;d< Z#d=d> Z$d?d@ Z%dAdB Z&dCdD Z'dEdF Z(dGdH Z)dIdJ Z*dKdL Z+dMdN Z,dOdP Z-dQdR Z.dSdT Z/dUdV Z0d S )WÚTestJoinc C s: t t t tj d¡tj d¡d}||d dk }|S )Nr )Úkey1Úkey2Zdata1Zdata2r é ©r r r r Úrandn)ÚselfÚdfr r r r s
|
|
üÿ
|
z TestJoin.dfc C s& t tddtdddtj d¡dS )Né
|
)r é )r r )r r Úvaluer )r r r r Údf2. s
|
|
ýÿz TestJoin.df2c C s> t ¡ \}}t||d}t|d |d d|d d}||fS )N©ÚindexÚAÚD)ÚMergedAÚMergedDÚC)ÚtmZgetMixedTypeDictr )r r& ÚdataÚtargetÚsourcer r r Ú target_source8 s ÿzTestJoin.target_sourcec C sF t ||dd}t|||dgdd t ||}t|||ddgdd d S )Nr ©ÚonÚleft©Úhowr ©r Ú _check_join©r r r$ Z joined_key2Z joined_bothr r r Útest_left_outer_joinD s
|
z TestJoin.test_left_outer_joinc C sL t ||ddd}t|||dgdd t ||dd}t|||ddgdd d S )Nr Úright©r2 r5 r4 r r6 r8 r r r Útest_right_outer_joinK s z TestJoin.test_right_outer_joinc C sL t ||ddd}t|||dgdd t ||dd}t|||ddgdd d S )Nr Úouterr; r4 r r6 r8 r r r Útest_full_outer_joinR s z TestJoin.test_full_outer_joinc C sL t ||ddd}t|||dgdd t ||dd}t|||ddgdd d S )Nr Úinnerr; r4 r r6 r8 r r r Útest_inner_joinY s zTestJoin.test_inner_joinc C s, t ||ddd}d|ks td|ks(td S )Nr ©z.fooz.bar)r2 Úsuffixesúkey1.foozkey1.bar©r ÚAssertionError©r r r$ Újoinedr r r Útest_handle_overlap` s z TestJoin.test_handle_overlapc C s. t ||dddd}d|ks td|ks*td S )Nr r rA )Úleft_onÚright_onrB rC zkey2.barrD rF r r r Ú!test_handle_overlap_arbitrary_keyf s û z*TestJoin.test_handle_overlap_arbitrary_keyc C s |\}}|j |dd}tj|d |d dd tj|d |d dd td d
|
d
|
d d d gi}td dddgid
|
d d gd}|j |d d}td
|
d
|
d d d gdddddgd}t ||¡ tdgdgdggd
|
d d gdgd} tdgdggddgdgd}
|
tdgdggddgdgd} | j |
|
dd}|j | dd}t |d d ¡sBtt |d d ¡sZttj t
|
dd |j |d d W 5 Q R X | ¡ } d| d< d } tj t | d |j | dd W 5 Q R X d S ) Nr+ r1 r) r' F)Z check_namesr* r( ÚkeyÚaÚbÚcr# r r é r% ©rL r# é Úone©r& ÚcolumnsÚfooÚbarÚtwoÚthreez^'E'$©ÚmatchÚEzfYou are trying to merge on float64 and object columns. If you wish to proceed you should use pd.concat) Újoinr, Úassert_series_equalr Úassert_frame_equalr ÚisnanrE ÚpytestÚraisesÚKeyErrorÚcopyÚ
|
ValueError)r r0 r. r/ Úmergedr r$ rG ÚexpectedZdf_aZdf_bZdf_cZ source_copyÚmsgr r r Ú test_join_onq s4 ÿ " ÿzTestJoin.test_join_onc C s t tjjddgddtj d¡d}t tjjddgddtj d¡dt dd¡d}d }tjt |d
|
t
|
||d d d W 5 Q R X d S )NÚmÚfrR ©Úsize©rM rN r! rP r% úFlen\(left_on\) must equal the number of levels in the index of "right"rZ rM T©rI Ú right_index© r r r Úchoicer r, ZmakeCustomIndexra rb re r ©r r r$ rh r r r Ú-test_join_on_fails_with_different_right_index s ÿ
|
þz6TestJoin.test_join_on_fails_with_different_right_indexc C s t tjjddgddtj d¡dt dd¡d}t tjjddgddtj d¡d}d }tjt |d
|
t
|
||d d d W 5 Q R X d S )Nrj rk rR rl rn rP r% r! zFlen\(right_on\) must equal the number of levels in the index of "left"rZ rN T)rJ Ú
|
left_indexrr rt r r r Ú,test_join_on_fails_with_different_left_index¤ s
|
þ ÿz5TestJoin.test_join_on_fails_with_different_left_indexc C s t tjjddgddtj d¡d}t tjjddgddtj d¡dt dd¡d}d }tjt |d
|
t
|
||d d d gd W 5 Q R X d S )Nrj rk rR rl rn r! rP r% z)len\(right_on\) must equal len\(left_on\)rZ rM rN )rJ rI rr rt r r r Ú/test_join_on_fails_with_different_column_counts° s ÿ
|
þz8TestJoin.test_join_on_fails_with_different_column_countsÚ
|
wrong_typerP ÚstrNr r c C sx t dddgi}dt| d}tjt|d t||ddd W 5 Q R X tjt|d t||ddd W 5 Q R X d S )NrM r z.Can only merge Series or DataFrame objects, a z was passedrZ )rI rJ )r Útypera rb Ú TypeErrorr )r ry r rh r r r Ú)test_join_on_fails_with_wrong_object_type¼ s ÿz2TestJoin.test_join_on_fails_with_wrong_object_typec C sD |\}}|j |dd}|d= | d¡}|j ||d}t ||¡ d S )Nr+ r1 )r] Úpopr, r_ )r r0 r. r/ rg Újoin_colÚresultr r r Útest_join_on_pass_vectorÍ s
|
z!TestJoin.test_join_on_pass_vectorc C s |\}}|j | g ¡dd}|D ]$}||ks0t|| ¡ ¡ s tq |j | g ¡ddd}t |j|j¡ t|dks|td S )Nr+ r1 r? r; r ) r] ÚreindexrE ÚisnaÚallr, Úassert_index_equalrU r )r r0 r. r/ rf ÚcolZmerged2r r r Útest_join_with_len0Ö s z TestJoin.test_join_with_len0c C s t dddddddgi}t dddgiddgd }|j|dd
|
d }|j|dd }||d ¡ }t |d |d ¡ tj|d |d d d t |j|j¡ d S )NrL rM ÚdrN rO r# r r r% r? r; r1 F)Z check_dtype)r r] Únotnar, r^ r
r& ©r r r$ rG rg r r r Útest_join_on_innerâ s zTestJoin.test_join_on_innerc C s` t ddddddgi}t ddddgidddgd }|j|dgd
|
}|j|dd
|
}t ||¡ d S ) NrL rM rN rO r# r r rP r% r1 ©r r] r, r_ r r r r Útest_join_on_singlekey_listî s
|
z$TestJoin.test_join_on_singlekey_listc C s> |\}}|j |d dd}|j |dg dd}t ||¡ d S )Nr) r+ r1 ©r] r, r_ )r r0 r. r/ r rg r r r Útest_join_on_seriesø s z TestJoin.test_join_on_seriesc C s\ t dddgi}tdgdgdd}|j|dd}t ddgddgd|jd}t ||¡ d S ) NrM r rP rN ©r& Únamer1 rn r% )r r r] r& r, r_ )r r Zdsr rg r r r Útest_join_on_series_bugletþ s
|
z#TestJoin.test_join_on_series_bugletc C s t t d¡d}d|d< d|d< t t dd¡d}d |d
|
< d |d < |j||d }t|||d }t ||¡ |j||d }t|||d }t ||¡ d S )Nr! r% TÚboolrV Ústringé é r Úintç ð?Úfloatr4 )r r Úaranger] Ú _join_by_handr, r_ )r Ú join_typeÚdf1r$ rG rg r r r Útest_join_index_mixed s z TestJoin.test_join_index_mixedc C sÒ t dddddt d¡ddd d
|
gd }|d jtjks:t|d
|
jtjksNtt dddddt d dd¡ddd d
|
gd }|j|d dd}ddddddddg}|d d
|_|dd
|_t ||}t
|
||¡ d S )Nr rP rV T©r' ÚBr+ r( r! r' r r+ r( rT r Z_oneZ_two©ÚlsuffixÚrsuffixZA_oneZB_oneZC_oneZD_oneZA_twoZB_twoZC_twoZD_twor" ) r r r ÚdtypeÚint64rE Zbool_r] rU r r, r_ )r r r$ rG Zexpected_columnsrg r r r Ú test_join_index_mixed_overlap s4
|
ý
|
ýø
|
|
z&TestJoin.test_join_index_mixed_overlapc C s( t }|jt dgdgdgddd d S )NrR r r' rT r= r4 )r r] )r Úxr r r Útest_join_empty_bug8 s z TestJoin.test_join_empty_bugc C sb t tj dd¡ddgd}ttj d¡}||d< t tj dd¡dgd}| |¡ | |¡ d S ) Né rP rM rN ©rU rO r Úq)r r r r r r] )r rM rO r r r r Útest_join_unconsolidated= s
|
z!TestJoin.test_join_unconsolidatedc C s t jddddddgddddddggddgd}t jdddd d d gddddddggddgd}ttj d
|
¡|d gd }ttj d
|
¡|d gd }|jdd}|jdd}|j|dd}t|j
|
t|j ¡}| |¡ | |¡¡}|j |j _ t ||¡ |j j |j kst|jdd}|jdd}|j|ddjdd}t|j
|
t|j ¡}| |¡ | |¡¡}|j |j _ t ||¡ |j j |j kstd S )NrM rN r rP rR ÚfirstÚsecond©ÚnamesrO é zvar X)r- r& rU zvar Yr )Úlevelr= r4 )r Z from_arraysr r r r Ú
|
sort_indexr] r ÚvaluesÚunionr r° r& r, r_ rE )r Zindex1Úindex2r r$ rG Zex_indexrg r r r Útest_join_multiindexH s4 þ þ
|
|
z TestJoin.test_join_multiindexc
|
|
C s ddddddddddg
|
}ddddddddddg
|
}t j t|¡}t|||d }|}tt j d
|
d ¡|d d dgd}|j|ddgdd}t|| ¡ ddgddgddd}t||ddgdddd} t ||
|
|¡¡ t||ddgdddd} |j ddgdd}|j |_ |j j stt ||¡ d S ) NrW rV ÚbazZquxZsnaprX rS rY )r r r- r! rR Zj_oneZj_twoZj_threerT r r r? r; r r® F)rI rJ r5 ÚsortT)rJ rv r5 r¹ r ©Zaxis)r r r r r r] r Ú reset_indexr, r_ Z reindex_likeZdropr& Zis_monotonic_increasingrE )
|
r Z%lexsorted_two_level_string_multiindexr r r- r& Zto_joinrG rg Z expected2r r r Útest_join_inner_multiindexk sd ö ÿú úú z#TestJoin.test_join_inner_multiindexc C s t ddgdddgd}| dg¡ dtjtjgi¡}t ddgdddgd}|jdd d
|
tjt j
|
j d d t ||d d d W 5 Q R X d S )N)r rP rR )r" r r± rM rN rO rª )é r! r± r T)Zinplacez-Not allowed to merge between different levelsrZ ©rv rq ) r ÚgroupbyZaggr ÚmeanÚsumÚ set_indexra rb ÚpdÚerrorsÚ
|
MergeErrorr )r r Znew_dfZother_dfr r r Ú#test_join_hierarchical_mixed_raises§ s ÿz,TestJoin.test_join_hierarchical_mixed_raisesc
|
C sp t tj dd¡ddgtjd}t tj dd¡dgtjd}| |¡}|jd dksVt|jd dksht|jd d kszttj d
|
d d ¡
|
d ¡}tj d ¡
|
d¡}tj d ¡
|
d ¡}t |||d}t |||d}t tj d ¡
|
d ¡dgd}|j |ddd}|jd d kst|jd dks(t|jd d ks<t|jd d ksPt|j |ddd} t || ¡ d S )Nr! rP rM rN ©rU r¤ r rO Úfloat64Úfloat32r r éd r¥ ©rM rN rO Zmdrª Trp )r r r r rÈ rÉ r] ZdtypesrE ÚrandintÚastyper r, r_ )
|
r rM rN rG rO r ZxpdfÚsÚrsZxpr r r Útest_join_float64_float32´ s&
|
z"TestJoin.test_join_float64_float32c
|
C s° t ddgddgddgd}t ddgddgddgd}t ddgddgd d
|
gd }| d d g¡}| d d g¡}| d d g¡}|j||gdd}t||d d gdd}t||d d gdd} | ¡ }| |j } | j d¡| d < | j d¡| d < t
|
|| ¡ t dddgdddgdddgd}t dddgdddgdddgd}t dddgdddgd d
|
dgd }| d d g¡}| d d g¡}| d d g¡}|j||gdd}t||d d gdd}t||d d gdd} | ¡ }t
|
|| j d d
|jf ¡ t ddddddddgddddddddgt j d¡t j d¡d }
|
tt t d¡d¡t t d¡d¡d d } |
|
j| dd} |
|
j| dd} |
|
j| d d}|
|
j| d d}t
|
| | ¡ t
|
| |¡ t
|
| |¡ d S )!Nr r! é rË rP rÊ éÈ ©rM rN r iè iÐ )rM rN ÚerM rN r= r4 r; r¥ r© i, i¸ r? rV rW rS rX rY r
|
r ZTESTr r3 r: )r r r] r r» rU rM rÍ rN r, r_ Úlocr r r r Úrepeatr )r r r$ Údf3Zidf1Zidf2Zidf3r Zdf_partially_mergedrg r rÎ r? r= r3 r: r r r Ú test_join_many_non_unique_indexË sX
|
"""
|
|
üÿ ÿ z(TestJoin.test_join_many_non_unique_indexc C s¸ t ddddgddddgd}t d d
|
d d gidddgd }|j|ddd}t ddddgddddgd
|
d d d gdddddgd }t ||¡ |j|ddd}tj|jttddd d S )NrV rW r¸ r rP rR r" rQ Úvalue2rM rN rO r% rL T)r2 r¹ )rL r# rÙ r F)Úexact)r r] r, r_ r
r& r r )r r3 r: rG rg r r r Útest_join_sort s
|
|
|
ý
|
ú zTestJoin.test_join_sortc C s t dddddgiddddgd}t ddd d
|
d giddddgd}| |¡}t dddddgdtjd d
|
tjgd dddddgd}t ||¡ t dddddgiddddgd}t ddd d
|
d giddddgd}| |¡}t ddddgdd d tjgd ddddgd}t ||¡ d S ) NrM r rP rR r" r% rN r r± r½ r
|
rn )r r] r Únanr, r_ )r r r$ r rg r× Zdf4r r r Ú test_join_mixed_non_unique_index s
|
þ
|
|
ÿz)TestJoin.test_join_mixed_non_unique_indexc C s t jdddd}tttt||dgd}t||g}|j|ddd }tt tj
|
dtj d
|
d ¡ d d ¡d ¡ddg| ¡ jd}t ||¡ d S )Nz
|
2016-01-01é ÚM)ÚperiodsÚfreqZpnumrT r? Z_df2)r5 r£ ©r¤ rP éÿÿÿÿr Zpnum_df2©rU r& )rà Z period_ranger r r r r r] r r r r¥ rÖ Úreshaper³ r& r, r_ )r r& r r$ r rg r r r Ú!test_join_non_unique_period_index' s $ýz*TestJoin.test_join_non_unique_period_indexc C s t tj dd¡ddddddgd }| d
|
d d
|
¡ | d d d¡ | d ¡}tjtdd | ¡ W 5 Q R X |j dd}|
|
¡ }|j |dd d S )NrÑ r± rM rN rO r rÔ rk rª r Úidr ÚdtrV zCould not convertrZ T)Z numeric_onlyÚ_right©r£ ) r r r r Úinsertr¿ ra rb r| rÀ Úcountr] )r r ZgroupedÚmnZcnr r r Ú test_mixed_type_join_with_suffix4 s "
|
z)TestJoin.test_mixed_type_join_with_suffixc C s\ t tj dd¡tdd}|ddg |ddg |d d
|
g g}|d |d d
¡}t ||¡ |ddg d d
|ddg dd
|d d
|
g d d
g}dd }|d j|d d
dd}||||j |d |d d
¡}||||d j |d j|d d
dd}||||jdd
d}t j
|
t |d |d j|d d
dd W 5 Q R X d S )Nr! r± Zabcdefrª rM rN rO r rÔ rk r r éþÿÿÿrP é c s8 fdd| D }|d |dd
¡}t ||¡ d S )Nc s g | ]}| ¡qS r )r )Ú.0r§ ©Ú exp_indexr r Ú
|
<listcomp>M s zFTestJoin.test_join_many.<locals>._check_diff_index.<locals>.<listcomp>r r r )Údf_listr ró Z reindexedrg r rò r Ú_check_diff_indexL s z2TestJoin.test_join_many.<locals>._check_diff_indexr= r4 r? r
|
z?Joining multiple DataFrames only supported for joining on indexrZ r1 ) r r r r r r] r, r_ r& ra rb re )r r rõ rG rö rh r r r Útest_join_manyC s " :zTestJoin.test_join_manyc C s t tj dd¡ddddgd}dd gd |d
|
< |jd d
ddgf }|jd d
ddgf }|jd d
d
|
gf }| ||g¡}t ||¡ d S ) Nr
|
r" r' r r+ r( rª rV rW rL )r r r r rÕ r] r, r_ )r r r r$ r× r r r r Útest_join_many_mixed_ s z TestJoin.test_join_many_mixedc C s6 t ttj dd¡ddddgdttjjdddd dd ¡dd
|
gdgd d }t ||gd d }|j|d d}|j|_t
|
||¡ ttj dd ¡ddgd}ttj dd ¡ddgd}ttj dd ¡ddgd}ttj dd ¡ddgd}|j |dddj |dddd}t j tjjdd |j |ddd W 5 Q R X d S )Nr! r" r' r rª r rÑ rl rP r+ r rº Z_2rê r§ ÚyTr¾ r= )rv rq r5 z0Passing 'suffixes' which cause duplicate columnsrZ )r r r r r rÌ rå r] rU r, r_ r ra rb rà rÄ rÅ ) r r rg r Úwr§ rù ÚzZdtar r r Útest_join_dupsi s8 ÿþù
|
ÿþzTestJoin.test_join_dupsc C s t jtdtdddggdddgd}tdtdi|d }t jtdtdgddgd}td
|
d d tdd D i|d }|j|ddg|d}| ¡ j| ¡ ddg|d dddg¡}t
|
||¡ d}t j t |d |j|d|d W 5 Q R X t j t |d |j|ddg|d W 5 Q R X d S )NÚabcZxyr rP Únumr¯ Zv1é r% Zv2c S s g | ] }d | qS )rÊ r )rñ Úir r r rô s z5TestJoin.test_join_multi_to_multi.<locals>.<listcomp>r½ r; ro rZ )r Z from_productr r r r] r» r r r, r_ ra rb re ) r r Z leftindexr3 Z
|
rightindexr: r rg rh r r r Útest_join_multi_to_multi s2 ÿ ÿ ÿþÿ z!TestJoin.test_join_multi_to_multic C s t tjddddtdd}t tjddddtdd }|j| d
|
¡d
|
d }| ¡ }ttj gd td t
|
d|d< t ||¡ d S )Nz
|
2018-01-01r zAmerica/Chicago)Ústartrà ÚtzZabcde)ÚdateÚvalsz
|
2018-01-03Ztuvwx)r Úvals_2r r1 rP Ztuvrâ r ) r rà Z
|
date_ranger r] r rd r r rÜ Úobjectr, r_ )r r r$ r rg r r r Ú#test_join_on_tz_aware_datetimeindex¥ s, ÿüÿ ÿüÿ z,TestJoin.test_join_on_tz_aware_datetimeindexc C sà t dddgdddgdddgdddggd d
|
d gd }t |d ¡|d < t ddd gdddggd d
|
dgddgd}t |d ¡|d < |j| d d
|
g¡d d
|
gd}t tddd dgtddddggddgd d
|
dd gd}t ||¡ d S )Nz
|
2012-08-02ÚLr! ÚJr z
|
2013-04-06rÑ é r§ rù rM rª r rP rû r" rä r1 z2012-08-02 00:00:00z2013-04-06 00:00:00rT )r rÃ Ú to_datetimer] r r r, r_ )r ZdfaZdfbr rg r r r Útest_join_datetime_string½ s0 üù ý þ
|
úz"TestJoin.test_join_datetime_stringc C s ddg}t j||d}t jt|t|d}td|i|d}tdt|i|d}| |¡}tddgddgdt jddgddgdd}t ||¡ d S )NrM rN ©Ú
|
categoriesÚc1r% Úc2)r r )rà ZCategoricalIndexÚreversedr r] r, r_ )r ÚixZid1Zid2r r$ r rg r r r Ú test_join_with_categorical_indexÚ s
|
þz)TestJoin.test_join_with_categorical_index)1Ú__name__Ú
|
__module__Ú __qualname__ra Zfixturer r$ r0 r9 r< r> r@ rH rK ri ru rw rx ÚmarkÚ parametrizer Úarrayr} 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 sX
|
|
|
'
|
|
#< 6
|
! r r3 Ú_xÚ_yc C sF |D ]}|| ¡ ¡ stq| |¡}| |¡} | t|dkrF|n|d ¡D ]ð\}
|
} t| | j|} t| |j|} z| |
|
¡}W nJ tk
|
rÌ } z,|dkr®td|
|
d|t | | j| W 5 d }~X Y n X t
|
| | z| |
|
¡}W nN tk
|
r4 } z.|dkrtd|
|
d|t | |j| W 5 d }~X Y qPX t
|
| | qPd S )Nr r )r3 r? zkey z! should not have been in the join)r: r? ) r r rE r¿ r Ú_restrict_to_columnsrU Z get_grouprc Ú_assert_all_naÚ_assert_same_contents)r3 r: r r r5 r¢ r£ rO Z left_groupedZ right_groupedZ group_keyÚgroupZl_joinedZr_joinedZlgroupÚerrZrgroupr r r r7 ê s>
|
|
ÿ
|
ÿþ
|
|
|
ÿþ r7 c sR fdd| j D }| jd d
|f } | jfddd} | jd d
f } | S )Nc s( g | ] }| ks | d ¡ kr|qS )Ú ©Úreplace)rñ rO ©rU Úsuffixr r rô s z(_restrict_to_columns.<locals>.<listcomp>c s | d¡S )Nr" r# )r§ )r& r r Ú<lambda> ó z&_restrict_to_columns.<locals>.<lambda>rª )rU rÕ Úrename)r rU r& Úfoundr r% r r s ÿr c sd d}| |¡ ¡ j}| |¡ ¡ j}dd |D t t|ksFtt fdd|D s`td S )Niy)íÿc S s h | ] }t |qS r ©Útuple©rñ Úrowr r r Ú <setcomp>( s z(_assert_same_contents.<locals>.<setcomp>c 3 s | ]}t | kV qd S r r+ r- ©Úrowsr r Ú <genexpr>* s z(_assert_same_contents.<locals>.<genexpr>)ZfillnaZdrop_duplicatesr´ r rE r )Ú
|
join_chunkr/ Z NA_SENTINELZjvaluesZsvaluesr r0 r r " s r c C s, |D ]"}||krq| | ¡ ¡ stqd S r )r r rE )r3 Zsource_columnsr rO r r r r - s r c C sZ | j j|j |d}| |¡}| |¡}| j |j¡}| ¡ D ]\}}|||< q<|j|dS )Nr4 rª )r& r] r rU ÚappendÚitems) rM rN r5 Z
|
join_indexZa_reZb_reZresult_columnsr rÎ r r r r 4 s
|
|
|
r c C sx t dditjdgddd} t dditjd gd
|
dd}| j|d d }t dgdgd tjdgddd}t ||¡ d S )NrÔ r )r rP r" rÓ r¯ ©r- r& rk r± )rP rR )rN rO r? r4 )rÔ rk )rP r r" rR )rN rM r rO r% ©r r Ú from_tuplesr] r, r_ )r3 r: r rg r r r Ú.test_join_inner_multiindex_deterministic_orderA s þ ÿ þr9 )Ú input_colÚ output_colsrN rM Za_xZa_yc C sj t dddgi}t | ddgi}|j|dddd}t |d ddddg|d ddddgi}t ||¡ d S )
|
NrM r rR r" Úcrossr r )r5 r¢ r£ r r )r: r; r3 r: r rg r r r Útest_join_crossR s
|
(r= c C s| t dditjdgddd}t dditjd gd
|
dd}|j|| d }t dgdgd tjd gddgdd}t ||¡ d S )NrO rR )r rP rn r¯ r6 r r" )rP )rN r4 )rO r )rP r rN rM r% r7 )r r3 r: r rg r r r Ú test_join_multiindex_one_level^ s ÿ þr> zcategories, valuesÚYÚXrP r g @g ø?z
|
2020-12-31z
|
2019-12-31c C s¨ t ddgt| | dddgd ddg¡}t dddgt|| dd d
|
d gd ddg¡}|j|d d d}t ddgt| | dddgd d
|
gd ddg¡}t ||¡ d S )Nr' r r rP )r r® r# r r® r rR r" r Ú_leftré r¡ )r r® Z
|
value_leftZ value_right©r r r r] r, r_ )r r´ r3 r: r rg r r r Ú1test_join_multiindex_not_alphabetical_categoricall s6
|
ýÿú
|
ýÿú
|
üÿùrC zleft_empty, how, exp)Fr3 r3 )Fr: Úempty)Fr? rD )Fr= r3 )Fr< rD )Tr3 rD )Tr: r: )Tr? rD )Tr= r: )Tr< rD c C sþ t ddgddgddd d¡}t dgd gd
|
dd d¡}| rL| d ¡}n
|
| d ¡}|j||d }|d krt ddgddgtjtjgd}| d¡}nV|dkrÄt tjgdgd gd}| d¡}n*|dkrît ddgdd}|dkrî| d¡}t ||¡ d S )NrP r rR r" )r' r r¥ râ r' r )r' r+ r r4 r3 )r' r r+ r: )r r' r+ rD r r+ rÇ r< ) r r Úheadr] r rÜ Z rename_axisr, r_ )Z
|
left_emptyr5 Úexpr3 r: r rg r r r Útest_join_empty s"
|
|
rG z how, valuesr? r= r: c C sª t tdddgtdddgdddgd ddg¡}t tdddgtdddgdddgd ddg¡}t t|t|||d ddg¡}|j|| d }t ||¡ d S )
|
Nr r rP rË rM rN rÓ )rM rN rO r r4 rB )r5 r´ r r$ rg r r r r Ú3test_join_multiindex_categorical_output_index_dtype½ s6 ýÿú ýÿúüÿù rH )r
|
r )r3 r r )r3 ) Únumpyr ra Zpandasrà r r r r r r r r Zpandas._testingZ_testingr, r r r7 r r r r r9 r r r= r> rC rG rH r r r r Ú<module> st (
|
|
R
|
'
|
ÿ
|
þüþ
|
öþ
|
üþ
|
