U
|
ý°d´E ã @ sÀ d dl m Z d dlZd dlZd dlmZ d dlZd dlmZm Z m
|
Z
|
m Z m Z d dl mZ d dlmZ ejdd Zejdd Zejd
|
d Zejd d Zejdd Zejdd Zejdd Zej dddeddgddgdd d gd fdd eddgddgdd d gd fd deddd gddej gdd d d gd fd d ed ddgej ddgdd d d gd fd deej ddgd ddgdd!d d gd fd d eddej gddd gdd d d!gd fd"ded ddej gej ddd gdd d d d!gd fd"d ed ddej gej ddd gdd d d d!gd fg¡d#d$ Z d%d& Z d'd( Z d)d* Z d+d, Z!d-d. Z"d/d0 Z#d1d2 Z$d3d4 Z%d5d6 Z&d7d8 Z'd9d: Z(d;d< Z)d=d> Z*ej d?d dg¡d@dA Z+G dBdC dCZ,dS )Dé )ÚdatetimeN)Ú
|
MergeError)Ú DataFrameÚIndexÚ
|
MultiIndexÚ
|
date_rangeÚ period_range)Úconcatc C s* t t d¡ dd¡tdtdddddS ) Né é é ZabcdeÚ2000ÚA)ÚstartÚfreqÚperiods©ÚdataÚcolumnsÚindex)r ÚnpZarangeZreshapeÚlistr © r r ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_join.pyÚframe_with_period_index s
|
ýr c C s t ddddgidddgdS )NÚar
|
é
|
r é é ©r ©r r r r r Úleft s r! c C s t ddddgidddgdS ) NÚbé, éd éÈ é r r r r r r r r Úright" s r' c C s& t ddddgddddgd td
|
d S ) Nr r" ÚcÚdÚcatÚdogÚweaselÚhorse©r r" r r )r Úranger r r r Ú left_no_dup' s þr0 c C s0 t dddddgdddd d
|
gd td d d¡S )Nr r" r( r) ÚeÚmeowÚbarkúum... weasel noise?ZnayZchirp©r r( r r )r r/ Ú set_indexr r r r Ú right_no_dup/ s þûúr7 c C s$ t | tdgdgddgdgddS )Nr Zcowr. r& r T)Úsort)r r )r0 r r r Ú left_w_dups: s ÿr9 c C s& t | tdgdgddgdg d¡S )Nr1 Zmoor5 r& r r )r r r6 )r7 r r r Ú right_w_dupsA s
|
ÿþr: zhow, sort, expectedÚinnerFr
|
r r% r$ r. r r r Tr# r& Úouterc C s" | j |||dd}t ||¡ d S )Nú1:1)Úhowr8 Úvalidate)ÚjoinÚtmÚassert_frame_equal)r! r' r> r8 ÚexpectedÚresultr r r Ú test_joinH s ,rE c C s0 t ddddddgi} t dddddgdddddgd}t dddddgdddddgd }d
|
}tjt|d | j|gd d W 5 Q R X tjt|d | j||gdd W 5 Q R X tjt|d | j||gd dd W 5 Q R X tjtdd | ||g¡ W 5 Q R X | |g¡}| |¡}t ||¡ d S )NÚkeyr r r& r r é )rF Zv1)ÚkeysZv2z7Suffixes not supported when joining multiple DataFrames©ÚmatchÚy)ÚlsuffixÚx)Úrsuffix©rL rN z Indexes have overlapping values)r ÚpytestÚraisesÚ
|
ValueErrorr@ rA rB )ÚfirstÚsecondÚthirdÚmsgZ
|
arr_joinedZ norm_joinedr r r Útest_suffix_on_list_joinx s ""
|
rW c C s2 d}t jt|d | j|ddd W 5 Q R X d S )Nz"invalid" is not a valid argument. Valid arguments are:
|
- "1:1"
|
- "1:m"
|
- "m:1"
|
- "m:m"
|
- "one_to_one"
|
- "one_to_many"
|
- "many_to_one"
|
- "many_to_many"rI r Úinvalid©Úonr? )rP rQ rR Úmerge)r0 r7 rV r r r Útest_join_invalid_validate s ÿ r\ c C sB | j |ddd d}tjt|d | j |ddd W 5 Q R X d S )Nr Ú one_to_manyrY zBMerge keys are not unique in right dataset; not a one-to-one mergerI Ú
|
one_to_one©r@ rP rQ r )r0 r: rV r r r Ú$test_join_on_single_col_dup_on_right¡ s ýýr` c C sB | j |ddd d}tjt|d | j |ddd W 5 Q R X d S )Nr Ú many_to_onerY zAMerge keys are not unique in left dataset; not a one-to-one mergerI r^ r_ )r9 r7 rV r r r Ú#test_join_on_single_col_dup_on_left´ s ýýrb c C sp | j |ddd d}tjt|d | j |ddd W 5 Q R X d}tjt|d | j |ddd W 5 Q R X d S ) Nr Z many_to_manyrY zCMerge keys are not unique in right dataset; not a many-to-one mergerI ra zBMerge keys are not unique in left dataset; not a one-to-many merger] r_ )r9 r: rV r r r Ú#test_join_on_single_col_dup_on_bothÇ s ýýrc c C sÊ t ddddgddddgddddgd td
|
d ddg¡} t dddgdddgd d dgdtdd ddg¡}t dddgdddgdddgd d dgdtdd ddg¡}| j|ddd}t ||¡ d S )Nr r" r r r* r+ r, r- ©r r" r( r r r2 r3 r4 )r r" r) r& )r r" r( r) r; r= )r> r? )r r/ r6 r@ rA rB )r! r' Zexpected_multirD r r r Ú test_join_on_multi_col_check_dupß s<
|
|
|
ýúù ýúù üùø re c
|
C s | j | jd d
ddgf }| j | jdd
ddgf jd d d
}| |¡}t |j|j¡ tddddg}t |j|¡ |j|dd }t |j|j¡ t |j|¡ |j|d
|
d }t |j|j¡ t |j|¡ |j|d d }t |j|jdd
¡ t |j|¡ |j|d d }t |j| j ¡ ¡ t |j|¡ t j
|
t d d |j|dd W 5 Q R X d}dD ].}t j
|
t |d | j| |d W 5 Q R X qhd S )Nr r ÚBr ÚCÚDéÿÿÿÿr! ©r> r' r; r< z join methodrI Zfooz columns overlap but no suffix)r< r! r; ) Úlocr Zilocr@ rA Úassert_index_equalr r Z sort_valuesrP rQ rR )Ú float_frameÚfÚf2ÚjoinedZexpected_columnsrV r> r r r Útest_join_index s0 (
|
rq c C sÊ | j d d
ddgf }| j d d d
ddgf }| ¡ }| d d d d
|d< | d d d d
|d< | |¡}t ||¡ |j|dd}t ||d d d
¡ |j|dd}t ||j d d
|jf ¡ d S )Nr rf r rg rh r' rj )rk Úcopyr@ rA rB r )rm ÚafZbfrC rD r r r Útest_join_index_more, s
|
rt c C s\ | ¡ }| | jd ¡}| |¡}t || ¡ d |_tjt dd | |¡ W 5 Q R X d S )Nri zmust have a namerI )
|
rr Úpopr r@ rA rB ÚnamerP rQ rR )rm ZdfZserrp r r r Útest_join_index_series> s
|
rw c C s¾ | j d d
dddgf }| j d d
dddgf }|j|ddd}|j d d
ddgf d¡}|j d d
ddgf d¡}| j d d
ddgf }| |¡ |¡}t ||j d d
|jf ¡ d S )Nr rf rg rh Z_df1Z_df2rO )rk r@ Z
|
add_suffixrA rB r )rm Údf1Údf2rp Zdf1_sufZdf2_sufZ
|
no_overlaprC r r r Útest_join_overlapJ s rz c C s^ | j dd d}tj| jgd dd}| j |j¡}| |¡}t||| jd}t
|
||¡ d S )Nc S s | | S )Nr )rF r r r Ú<lambda>Z ó z(test_join_period_index.<locals>.<lambda>©r r r )Zaxisr ) Úrenamer Z concatenateÚvaluesr Úappendr@ r r rA rB )r ÚotherZ joined_valuesZ joined_colsrp rC r r r Útest_join_period_indexY s
|
ÿr c C s° t ddddgidddgd} t d d
|
d d gid ddgd}t ddddgiddd gd}| j||gdd}t ddddgtjd d d gtjddtjgdddddgd}t ||¡ d S )Nr r r r
|
r r r& r r" r$ r% r# r r( i iô iX r! rj rd )r r@ r ÚnanrA rB )rx ry Údf3rp rC r r r Ú(test_join_left_sequence_non_unique_indexh s
|
ý
|
ú r
c C s6 | j ¡ }| j| ddg g}| |¡}t || ¡ d S )Nrg rh )r Zto_framerf r@ rA rB )rm r! r' rD r r r Útest_join_list_series| s
|
|
r Úsort_kwc C sÐ t dddgiddgd}t ddd gid
|
dgd}t d d d gidd
|
gd}t dtdddddtddddtddd}| dkr|jddd
|
gd}t d ¡ |j||gd| d}W 5 Q R X t ||¡ d S )NÚcol1r r r( r r Úcol2r r r" Úcol3é rG ç @r ç ð?rd g @g @g @g @©r r r Fr< )r> r8 )r ÚfloatÚreindexrA Zassert_produces_warningr@ rB )r r r" r( rC rD r r r Ú)test_suppress_future_warning_with_sort_kw
s ýÿ r c @ sD e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dS )ÚTestDataFrameJoinc C s¦ |}|j |jd d
dgf }|j |jdd
ddgf }|j|dd |j¡}| ¡ j ¡ }tj|t |j¡< t ||j|j
|
d}t |j¡ ¡ rt t ||¡ d S ) Nr r r rf rg r< rj ©r r )rk r r@ r rr r r r Úisnanr r ÚallÚAssertionErrorrA rB )ÚselfZ multiindex_dataframe_random_dataÚframer r" rp rC r r r rE s zTestDataFrameJoin.test_joinc C sp t ddgddgddgd}t ddgddgddgd}| ddg¡}| ddg¡}dD ]}|j||d qXd S ) Nr r )r r" rM )r r" rK r r" )r! r' r< rj )r r6 r@ )r rx ry r> r r r Útest_join_segfault¬ s z$TestDataFrameJoin.test_join_segfaultc C sr ddg}t dddt dddg}t|tddgd}td dgd
|
d gg||d}|j|dd }t|jd
|
ksntd S ) NZ20120209Z20120222iÜ r é é Zaar r r& r ©rZ )r r r/ r@ Úlenr r )r Z str_datesZdt_datesr rg Ztstr r r Útest_join_str_datetime¶ s z(TestDataFrameJoin.test_join_str_datetimec C s t dddgdddgdddgddd gddd
|
gddd gd dd gd ddgd ddgg dddgd ddg¡}t ddgddggddgd dg¡}t ddgddgddgd dgd
|
dgd dgdtjgdtjgdtjgg |jddgd}t |j|dd |¡ t |j|d d |ddg ¡ tj ddgdddggddgd }t ddgddgddgd dgd
|
dgd dgg|ddgd}t |j|d d |¡ t |j|dd |ddg ¡ d S ) Nr rM gR
|
º½¤1Þ?rK g¡Ü Ìè?Úzg¼<+J â?r" gµá°4ð£Ö¿g^9ëS×?g? ¨û¿r( r r r& rS rT Zvalue1r} r r
|
Zvalue2r r g @r r! rj r' ©Únames)
|
r r6 r r r rA rB r@ r Z from_product)r rx ry ÚexpZexp_idxr r r Ú test_join_multiindex_leftrightÁ sf ÷ ô ó ÿ÷ ó ÿúö z0TestDataFrameJoin.test_join_multiindex_leftrightc C sà t ddd¡ ¡ }tjd|fgddgd}tddgi|d}tjd|fgddgd}td dgi|d}tjd|fgddgd}td
|
dgi|d}| ||g¡}tjd|fgddgd} tdgdgdgd | d}
|
t ||
|
¡ d S ) NiÐ r r Zindex_0Údater r r r r r ) ÚpdÚ Timestampr¤ r Ú from_tuplesr r@ rA Z assert_equal) r r¤ Z df1_indexrx Z df2_indexry Z df3_indexr rD Zexpected_indexrC r r r Útest_join_multiindex_datesý s ÿz,TestDataFrameJoin.test_join_multiindex_datesc C s t ddgddgddggd}t dd g¡}t |dd
|
gdd ggd}tjtd d tj||dd W 5 Q R X tjtd d |j|dd W 5 Q R X d S )Nr r" r é r r )r r )r Ú )r( Zc1é! é, z-Not allowed to merge between different levelsrI r ) r r r§ rP rQ r r¥ r[ r@ )r rx r ry r r r Ú'test_merge_join_different_levels_raises s ÿ ÿz9TestDataFrameJoin.test_merge_join_different_levels_raisesc C s t t d¡tdddddd}t t d¡tdd d
|
ddtd dd }|j|d d }|j |j¡}t |j|¡ |jj
|
j dkst d S )N)é r& z2012-11-15 00:00:00r® Z100Lz
|
US/Central)r r Útzr )r& r& r& Z250Lr r< rj ) r r Úzerosr r/ r@ r ÚunionrA rl r¯ Úzoner )r Ztest1Ztest2rD rC r r r Útest_frame_join_tzaware( s. ÿþ ÿûz)TestDataFrameJoin.test_frame_join_tzawareN)
|
Ú__name__Ú
|
__module__Ú __qualname__rE r r r£ r¨ r r³ r r r r r s
|
<r )-r Únumpyr rP Z pandas.errorsr Zpandasr¥ r r r r r Zpandas._testingZ_testingrA Zpandas.core.reshape.concatr Zfixturer r! r' r0 r7 r9 r: ÚmarkZ parametrizer rE rW r\ r` rb rc re rq rt rw rz r r
r r r r r r r Ú<module> s
|
|
|
|
|
|
|
|
""$ý$ý$ý$ý
|
þý
|
þýáþ+
|
$)
|
|
