U
|
ý°d¬ ã @ s d dl mZm Z mZ d dlZd dlZd dlZd dlmZm Z d dl
|
m Z d dl Z d dl mZmZmZmZmZmZmZmZmZmZmZ d dlmZ d dl m Z d dl m Z d dl m!Z!m"Z" dÃd
|
d Z#d d Z$dd Z%ej&e$ dd ddd Z'ej&e$ dd ddd Z(ej&e% dd ddd Z)ej&dd Z*G d d d Z+d d Z,G d!d" d"Z-ej&d#d$ Z.ej&d%d& Z/G d'd( d(Z0ej&d)d* Z1ej&d+d, Z2G d-d. d.Z3ej4j5d/d0d1 ej6D ed2d3gd2d3gd4d5ed d6d4d7ed8d9gd4d:g d;d d<d=d> Z7ej4 5d?d@dAgdddBdBd3fdCdd@dAgddBdDd3fddd@dAgdDdBd3fd@dAgdddBdBdfdEdd@dAgddBdDdfddd@dAgdDdBdfg¡dFdG Z8dHdI Z9ej4 5dJd d dKdLidMdNgfd d dKdOid dNgfd d dKdPidQdRgfd d dKdSdTgidQdRgfdUd dKdVidUd gfdWdWdKdXidYdWgfdZdZdKdVidZd[gfdUdUdKdXid\dUgfdUdZdKdXidUdZgfdUdUdKd]idUd\gfd d dKd^id_d gfdUdUi d\d`gfd d i dQdRgfg ¡dadb Z:ej4 5dcd&edddedfgdgdhej;gdidjdkgdlfd@edddedmdfgdgdhdnej;gdidjej;dkgdlfg¡dodp Z<ej4 5dqdrdsdtg¡dudv Z=ej4 5dKd$d&hd d dwg¡dxdy Z>ej4 5dzd{dUdUe?dUd|fg¡d}d~ Z@ej4 5dddg¡ej4 5ddDdBg¡dd ZAd
d ZBdd ZCdd ZDdd ZEej4 5dddg¡dd ZFdd ZGej4 5dddg¡ej4 5ddUdDdddgfdUdDdddgfg¡dd ZHej4 5dddg¡ej4 5dddUiddgfddUiddgfg¡d d¡ ZIej4 5dddg¡ej4 5d¢ddDiddDig¡d£d¤ ZJd¥d¦ ZKd§d¨ ZLd©dª ZMd«d¬ ZNdd® ZOd¯d° ZPd±d² ZQd³d´ ZRej4 5dµdd¶g¡d·d¸ ZSd¹dº ZTd»d¼ ZUd½d¾ ZVej4 5dµd¿dÀg¡dÁd ZWdS )Äé )ÚdateÚdatetimeÚ timedeltaN)Úis_categorical_dtypeÚis_object_dtype)ÚCategoricalDtype) Ú CategoricalÚCategoricalIndexÚ DataFrameÚ DatetimeIndexÚIndexÚ IntervalIndexÚ
|
MultiIndexÚ PeriodIndexÚ
|
RangeIndexÚSeriesÚTimedeltaIndex)Úconcat)Ú
|
MergeErrorÚ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_groupsÚarr© r# ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/reshape/merge/test_merge.pyÚ get_test_data( s " r% c C sT t dgddt dgddt dgt dgt dgt t d¡gt tjdd d
|
ggS ) Né Úint64©ÚdtypeÚInt64g®Gáz®ó?ÚfooTú
|
2018-01-01ú
|
US/Eastern©Útz)r ÚpdÚ Timestampr# r# r# r$ Ú
|
get_series3 s ùr2 c C s8 t tjgddt tjgddt tjgddt tjggS )Nr* r( ÚfloatÚobject)r r Únanr0 ÚNaTr# r# r# r$ Ú get_series_na? s
|
|
ür7 c C s | j jS r ©r) Úname©Úxr# r# r$ Ú<lambda>H ó r< )ÚparamsÚidsc C s | j S )zV
|
A parametrized fixture returning a variety of Series of different
|
dtypes
|
©Úparam©Úrequestr# r# r$ Úseries_of_dtypeH s rD c C s | j jS r r8 r: r# r# r$ r< Q r= c C s | j S )zk
|
A duplicate of the series_of_dtype fixture, so that it can be used
|
twice by a single function
|
r@ rB r# r# r$ Úseries_of_dtype2Q s rE c C s | j jS r r8 r: r# r# r$ r< Z r= c C s | j S )zU
|
A parametrized fixture returning a variety of Series with all NA
|
values
|
r@ rB r# r# r$ Úseries_of_dtype_all_naZ s rF c C sR t ddgddgddgd} t ddddd gddddd gdddddgd
|
}| |fS ) Nr r& é ÚaÚb)Úcol1Ú col_conflictÚcol_lefté é é )rJ rK Ú col_right©r
|
)Údf1Údf2r# r# r$ Údfs_for_indicatorc s ýÿrT c @ s e Zd Zejdd Zejdd Zejdd Zejdd Zd d
|
Z d d Z
|
d d Z dd Z dd Z dd Zdd Zdd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zej d/d0d0d1d0d2d3d4d0d5d4d2d6g¡d7d8 Z d9d: Z d;d< Z d=d> Z!d?d@ Z"dAdB Z#ej dCdDdEdEdFdEdEgfd0dFdEdEdEdEgfg¡ej dGdHdIg¡dJdK Z$dLdM Z%dNdO Z&dPdQ Z'dRdS Z(dTdU Z)ej dVdWdXdYdZd[d\d]g¡d^d_ Z*ej dVdWdXdYdZd[d\d]g¡d`da Z+dbdc Z,ddde Z-dfdg Z.dhdi Z/djdk Z0dldm Z1dndo Z2dpdq Z3drds Z4dtdu Z5dvdw Z6dxdy Z7dzd{ Z8d|d} Z9d~d Z:ej dGdIdg¡ej de;dEddge;dEdddddgfe<d
ddge<d
dde=j>e=j>e=j>gffdd e?j@D eA Bdddg¡eA BdddeCjDeCjDeCjDg¡feEd
ddgdWdeEd
dde=j>e=j>e=j>gdWdfeFdddgeFddde=j>e=j>e=j>gff¡dd ZGdd ZHej dGdHdIg¡dd ZIdd ZJdd ZKdS )Ú TestMergec C s: t t t tj d¡tj d¡d}||d dk }|S )Nr )Úkey1Úkey2Údata1Údata2rW r& ©r
|
r% r r Úrandn)ÚselfÚdfr# r# r$ r] q s
|
|
üÿ
|
z TestMerge.dfc C s& t tddtdddtj d¡dS )Né
|
)r! rN )r r! )rV rW ÚvaluerZ ©r\ r# r# r$ rS s
|
|
ýÿz TestMerge.df2c C s$ t dddddddgtj d¡dS )NrH rI ÚcÚdÚeé ©ÚkeyÚv1©r
|
r r r[ r` r# r# r$ Úleft s ÿzTestMerge.leftc C s t dtj d¡iddddgdS )NÚv2rN rb rI ra rH ©Úindexrh r` r# r# r$ Úright s zTestMerge.rightc C sP t }t dddgiddgdd}t||ddd}t dg idd }t ||¡ d S )
|
NrH r& rG r r' )rl r) T©Ú
|
left_indexÚ right_indexr( ©r
|
r ÚtmÚassert_frame_equal)r\ Údf_emptyZdf_aÚresultÚexpectedr# r# r$ Útest_merge_inner_join_empty s
|
z%TestMerge.test_merge_inner_join_emptyc C s, t ||}t ||ddgd}t ||¡ d S )NrV rW ©Úon©r rr rs )r\ r] rS ÚjoinedÚexpr# r# r$ Útest_merge_common s
|
zTestMerge.test_merge_commonc C sZ t ddddgddddgddddgddddgd}| t¡}|}t||}t ||¡ d S )Nr& r rG rM )r r& rG rM )r
|
Úastyper3 r rr rs ©r\ ri rm rv ru r# r# r$ Ú test_merge_non_string_columns¡ s ,ÿ
|
|
z'TestMerge.test_merge_non_string_columnsc C sF | d¡}| d¡}t||dd}t||dd d¡}t ||¡ d S )NrV rx )Ú set_indexr rr rs )r\ r] rS ri rm ru rv r# r# r$ Útest_merge_index_as_on_arg s
|
|
|
z$TestMerge.test_merge_index_as_on_argc C sÐ t dddddddgtj d¡d}t dtj d ¡iddddgd
|
}t||d d d dd}t||d d ddd}t ||jd d
|jf ¡ t||d d d d d}t||d d dd d}t ||jd d
|jf ¡ d S )NrH rI ra rb rc rd re rj rN rk rf Tri F)Úleft_onrp ÚhowÚsortrm )Úright_onro r r
) r
|
r r r[ r rr rs ÚlocÚcolumns)r\ ri rm Zmerged1Úmerged2r# r# r$ Ú(test_merge_index_singlekey_right_vs_left¶ sL ÿ ÿ ÿ ÿ ÿz2TestMerge.test_merge_index_singlekey_right_vs_leftc C sÀ t dddddddgtj d¡d}t dtj d ¡iddddgd
|
}t||d d d d}|j|d dj|j }t ||¡ t||d d d d}|j|d dj|j }t ||jd d
|j
|
f ¡ d S )NrH rI ra rb rc rd re rj rN rk rf TÚinner©r rp r rx )r ro r ) r
|
r r r[ r Újoinr rl rr rs r ©r\ ri rm ru rv r# r# r$ Ú test_merge_index_singlekey_innerÌ s ÿ z*TestMerge.test_merge_index_singlekey_innerc C sÊ d}t jtjj|d t||dd W 5 Q R X d}t jtjj|d t||dd W 5 Q R X d}t jtjj|d t||ddd W 5 Q R X d
|
}t jt|d t||d gd d gd W 5 Q R X d S )Nz&Must pass right_on or right_index=True©ÚmatchT)ro z$Must pass left_on or left_index=True)rp zRCan only pass argument "on" OR "left_on" and "right_on", not a combination of bothrf )r ry z)len\(right_on\) must equal len\(left_on\)rV rW ©r r )ÚpytestÚraisesr0 Úerrorsr r Ú
|
ValueError)r\ r] rS ri rm Úmsgr# r# r$ Útest_merge_misspecifiedÛ s ÿz!TestMerge.test_merge_misspecifiedc C s d}t jt|d t||ddddgd W 5 Q R X d}t jt|d t||dddgdd W 5 Q R X t jt|d t||dddgddgd W 5 Q R X d S ) Nz>right_index parameter must be of type bool, not <class 'list'>r ri FrV rW ©r ro rp z=left_index parameter must be of type bool, not <class 'list'>©r r r r )r\ r] rS r r# r# r$ Ú&test_index_and_on_parameters_confusionî s4 ûûûz0TestMerge.test_index_and_on_parameters_confusionc C sN t ||dd}|d ¡ d ¡ }t||ks2td|ks>td|ksJtd S )Nrf rx rG Zv1_xZv1_y)r Z value_countsÚsumr ÚAssertionError)r\ ri ÚmergedZexp_lenr# r# r$ Útest_merge_overlap
|
s
|
z TestMerge.test_merge_overlapc C s t ddddgddddgd}t ddd dgd
|
d d d gd}|j|ddddd}tddddddtjgdd}t |d |¡ tdtjddddd gdd}t |d |¡ tddddddtjgdd}t |d |¡ td tjd
|
d d
|
d d gdd}t |d |¡ d S )Nr+ ÚbarZbazr& rG rM rN )Úlkeyr_ ZquxrO é rd r )Úrkeyr_ r¡ r£ ÚouterT)r r r r
©r9 Úvalue_xÚvalue_y)r
|
r r r r5 rr Úassert_series_equal)r\ ri rm r r| r# r# r$ Ú%test_merge_different_column_key_names s" ÿ z/TestMerge.test_merge_different_column_key_namesc C sz t dddtdd}t dddtdd}t||d d d d
|
}d |d < |d dk ¡ sZtd |d< |d dk ¡ svtd S )Nr r& ©rH rI r^ rk r+ r ©ra rb T©ro rp Úcopyr¢ rH Zpeekaboorb )r
|
r r Úallr )r\ ri rm r r# r# r$ Útest_merge_copy% s zTestMerge.test_merge_copyc C sz t dddtdd}t dddtdd}t||d d d
|
d }t |d j|d j¡sZtt |d j|d j¡svtd S )Nr r& rª r^ rk r+ r r« TFr¬ rH rb )r
|
r r r Z shares_memoryZ_valuesr )r\ Úusing_array_managerri rm r r# r# r$ Útest_merge_nocopy1 s
|
zTestMerge.test_merge_nocopyc C sÄ t dddddgttddddgd}t ddddd dgttd
|
d }t||dd d }t dddddddd dg t ddddddd tjtjg ¡dddddddd dg ddddgd}t ||¡ d S )Nr& rG rM rO ©rf r_ r_ rf ©r rN r¢ )rf Úrvaluer¤ ©ry r r )rf r_ r´ r´ ) r
|
r r r r Úarrayr5 rr rs )r\ ri rm r{ rv r# r# r$ Ú"test_intelligently_handle_join_key: s ÿ" ýúz,TestMerge.test_intelligently_handle_join_keyc C sæ t dgdgd}t dgdgd}t||dd}|d jd ksDtt d
|
gdgd}t d gd gd}t||dd}|d jd kstt ddgi}t ddgi}t dg¡}t dg¡}t||||dd}|d jtjksâtd S )Nr& r^ re rG é r¤ ©r rf r' TFr ÚboolÚval©r r r Úkey_0)r
|
r r) r r r¶ Úint_)r\ rR rS r] r¡ r£ r# r# r$ Ú test_merge_join_key_dtype_castM s z(TestMerge.test_merge_join_key_dtype_castc
|
C s¾ t dddddgt d¡dddgdd }t d
|
t d ¡idd }tjddddd dgdd }t||d|dd}t|||ddd}t |d |d ¡ |d ¡ ¡ s¤t |d ¡ ¡ s¸t t dt d¡idgd}t d
|
t d ¡i}t dddddg¡}t ddddd dg¡}t||||dd}t
|
dddddddd dg tj dd}t |d |¡ t dt d¡i}t d
|
t d ¡i}tjddddddgtj d }t||d|dd}t |d t
|
|dd¡ d S )Nr& rG rM rO r² r_ rf r' ©r r) r´ r¢ r( rN r¤ r¼ r³ r½ r8 r T)ro r r r¥ ) r
|
r Úaranger¶ r rr r¨ Znotnar® r r r¾ r' ) r\ ri rm rf r r r¡ r£ rv r# r# r$ Ú test_handle_join_key_pass_arrayd s0 ý" z)TestMerge.test_handle_join_key_pass_arrayc C sx t ¡ }tddgi|gd}tdddgi||gd}dd dd d d
|
d d
|
}tjt|d t|| W 5 Q R X d S ) Nr; rH rk ÚyrI ra z>No common columns to perform merge on. Merge options: left_on=z , right_on=z , left_index=Fz, right_index=r )r Únowr
|
r r r r )r\ ÚdtrR rS r r# r# r$ Ú&test_no_overlap_more_informative_error s ÿz0TestMerge.test_no_overlap_more_informative_errorc C sò t ddd}t ddd}t ddd}t ddd}tddgi|gd }td
|
d d gi||gd }t|| tddd d gi|||gd }td
|
d dddddgi||||||gd }t|| tddd gi||gd }td
|
d dgi||gd }t|| d S )NéÜ rO r& rG rM rN r; rH rk rà rI ra Úqrb rc ÚfÚgÚh©r r
|
Ú _check_merge)r\ rÅ Údt2Údt3Zdt4rR rS r# r# r$ Ú test_merge_non_unique_indexes s
|
ÿ
|
z'TestMerge.test_merge_non_unique_indexesc C sv t ddd}t ddd}t ddd}tdddd d
|
gi||||gd }td d ddddgi|||||gd }t|| d S )NrÇ rO r& rG rM r; rH rI ra rb rk rà rc rÉ rÊ z hÚirÌ )r\ rÅ rÎ rÏ rR rS r# r# r$ Ú(test_merge_non_unique_index_many_to_many¨ s ÿz2TestMerge.test_merge_non_unique_index_many_to_manyc C sZ t dgdgd}t dg i}t||ddd}t ||¡ t||ddd}t ||¡ d S )Nr& rG r² rf ri rµ rm rq )r\ ri rm ru r# r# r$ Ú test_left_merge_empty_dataframe² s z)TestMerge.test_left_merge_empty_dataframeÚkwargTrn r; ©ro r rH ©r rp r c C s^ t dddgd}t dddgd}t ddddddgtd}t||fd |i|}t ||¡ d S )
|
NrH rI ra r³ r; rà ÚzrÀ r )r
|
r4 r rr rs )r\ Ú join_typerÔ ri rm Úexp_inru r# r# r$ Ú!test_merge_left_empty_right_empty¼ s
|
z+TestMerge.test_merge_left_empty_right_emptyc sf t dddgd t dddgdd d
|
gd d d ggdddgdt tjtjgd tdtjtjgd tdtjtjgd tdddd gdd d gdd
|
d gdddddddgd}|dd
} fdd} fdd}ddddddfD ]}||| ||| qäddd}||| dddg|d< ||| ddd }||| tjtjgd td|d< ||| d S ) NrH rI ra r³ r& rG rM rN rO r¢ rd r é r; rà r× r( ©rH rI ra r; rà r× r c sH t fddi|}t || ¡ t fddi|}t || ¡ d S )Nr r ri rz ©r| rÔ ru ©ri rm r# r$ Úcheck1á s z>TestMerge.test_merge_left_empty_right_notempty.<locals>.check1c sH t fddi|}t || ¡ t fddi|}t || ¡ d S )Nr rm r¤ rz rÝ rÞ r# r$ Úcheck2ç s z>TestMerge.test_merge_left_empty_right_notempty.<locals>.check2Trn rÕ rÖ r )r
|
r r¶ r5 r4 )r\ Úexp_outrÙ rß rà rÔ r# rÞ r$ Ú$test_merge_left_empty_right_notemptyÏ s8 *ú÷ þ
|
|
|
|
|
|
z.TestMerge.test_merge_left_empty_right_notemptyc
|
sê t dddgdddgddd ggd
|
d d gd t dddgd t dddgdddgddd gtjtjgd tdtjtjgd tdtjtjgd tddd
|
d d dddgd dd
j t¡_fdd fddd S )Nr& rG rM rN rO r¢ rd r rÛ rH rI ra r³ r; rà r× r( rÜ r c sH t fddi|}t || ¡ t fddi|}t || ¡ d S )Nr r rm rz rÝ rÞ r# r$ rß s z>TestMerge.test_merge_left_notempty_right_empty.<locals>.check1c s t fddi|}t || ¡ t fddi|}t || ¡ ddddddddd ddd
|
fD ]} | | qhd S ) Nr ri r¤ Trn r; rÕ rH rÖ r rz rÝ ©rß rà rÙ rá ri rm r# r$ rà s ü
|
z>TestMerge.test_merge_left_notempty_right_empty.<locals>.check2)r
|
r r¶ r5 r4 rl r~ r` r# rã r$ Ú$test_merge_left_notempty_right_emptyþ s *ú÷ z.TestMerge.test_merge_left_notempty_right_emptyc C s~ t ||dddgd}|d d
}t t|jd dt|jd dt|jd ddddd gd}|j|dd
|
}t ||¡ d S ) Nr² rf r_ r³ r r( )r¦ rf r§ r¦ r§ rx )r
|
r Údtypesr rr rs )r\ rD rE r] rt rv Úactualr# r# r$ Útest_merge_empty_frame( s þ ýúz TestMerge.test_merge_empty_framec C sd t ||dddgd}t ||dddgd}t |||ddddgd}|j|dd}t ||¡ d S ) Nr² rf r_ r³ ©rf r¦ r§ r¦ r§ rx rq )r\ rD rF Zdf_leftZdf_rightrv ræ r# r# r$ Útest_merge_all_na_column: s" þþýúz"TestMerge.test_merge_all_na_columnc C s t jjddddt jjddddtdddtddd td
|
dd tdddtddd tdd d tdd d td d dtd
|
dd tddd g
|
d}t |¡}|j ¡ }t |¡}t |t j d¡d¡}|j |ddd}t ||ddd}t
|
||¡ |j ¡ |j ¡ k ¡ s t d S )Nr r^ ©ÚsizerÇ r& é iÛ rG rN iÚ rM iØ rO )Zvar1Zvar2Úvar3rd )rí Zvar8rí F©ry r
)r r Úrandintr r
|
Ú from_dictrí Úuniquer
r rr rs r® r )r\ rb r] rí Únewru r| r# r# r$ Útest_merge_nosortO s,
|
|
|
|
|
|
|
|
|
|
öý
|
|
|
zTestMerge.test_merge_nosort)r
ÚvaluesFr& r r ri rm c C sB t ddddgi}|j|d||d}t |dgd}t ||¡ d S )NrH r& r )ry r r
r³ rq )r\ r
rô r r] ru rv r# r# r$ Ú test_merge_same_order_left_rightm s z*TestMerge.test_merge_same_order_left_rightc C s t ddgddgd}t dgdgd}|j|ddd}t ddd ddd dtjd dtjd ddd d
|
d ¡ ¡ dd d d g }tj||dd d S )Nr r& ©Úi1Úi2©r÷ Úi3r÷ Ú_)ry Úrsuffixç ©r r& )r÷ rø Úi1_rú Nrø rÿ rú F)Z check_dtype)r
|
r r r5 r Ú reset_indexrr rs ©r\ rR rS ru rv r# r# r$ Útest_merge_nan_righty s"
|
|
ûÿ ÷
|
|
öÿ z TestMerge.test_merge_nan_rightc C s t ddgddgd}t dgdgd}|j|dd d
|
}t ddd d tjd ddd dtjd d d dddg }t ||¡ d S )Nr r& g à?ç ø?rö gffffffæ?rù rû r÷ )rü ry rþ rý )r÷ rÿ rø rú rø rÿ rú )r
|
r r r5 rr rs r r# r# r$ Útest_merge_nan_right2 s
|
|
üÿ
|
ùz TestMerge.test_merge_nan_right2c s< G fdddt |}|j|dd}t| s8td S )Nc s e Zd Ze fddZdS )z0TestMerge.test_merge_type.<locals>.NotADataFramec s S r r# r` ©Ú NotADataFramer# r$ Ú _constructor s z=TestMerge.test_merge_type.<locals>.NotADataFrame._constructorN)Ú__name__Ú
|
__module__Ú __qualname__Úpropertyr r# r r# r$ r s r rV rx )r
|
r Ú
|
isinstancer )r\ r] rS Znadru r# r r$ Útest_merge_type s zTestMerge.test_merge_typec C s¬ t tdddddgtddgd¡}t tdd }t||gd
|
d }t||gd
|
d }t tdddddtdddddgtddtddgd}|r| t¡}t ||¡ d S ) NiÝ é rO é8 r iäW )rb ÚtrÅ r³ T)Z ignore_index)
|
r
|
rð r r r r r~ r4 rr rs )r\ r° rb r] ru rv r# r# r$ Útest_join_append_timedeltas¥ s ÿ þÿ
|
z%TestMerge.test_join_append_timedeltasc C s t d¡}tt||gddgd}tt|gdgd}|j|ddd}tt||gtddt|tjgtddd }t ||¡ d S )
|
Ni £áÚAÚBrk Úrri )rü r ZAB)Ú0Z0r)
|
r Z timedelta64r
|
r r r r0 r6 rr rs )r\ ÚtdÚlhsÚrhsru rv r# r# r$ Ú test_join_append_timedeltas2º s
|
þÿz&TestMerge.test_join_append_timedeltas2ÚunitÚDrË ÚmÚsÚmsÚusÚnsc
|
C sÚ t dddgi}td d gddgdd}d| d}|dkr@d }n|}| |¡ d¡}|d j|ksft|j|dd
|
d }tjd d g|d }t j
|
j j j ||jd }t ddg|dddgd} | d j|ksÊtt || ¡ d S )NÚ entity_idée éf Údays©rl r9 z datetime64[ú]©r rË r z datetime64[s]TrÖ Únatr( ©r! r$ r³ )r
|
r r~ Úto_framer) r r r r¶ r0 ÚcoreÚarraysZ DatetimeArrayZ _simple_newrr rs )
|
r\ r rR Úserr) Z exp_dtyperS ru r$ r| r# r# r$ Útest_other_datetime_unitÊ s& þûz"TestMerge.test_other_datetime_unitc C sÞ t dddgi}td d gddgdd}d| d}|dkrtd }tjt|d
|
| |¡ W 5 Q R X | d ¡ d¡}n"| |¡ d¡}|d j|kst|j |dd d }t ddgt
|
j ddg|ddddgd}t ||¡ d S )Nr! r" r# r$ r% zm8[r& r' z/Supported resolutions are 's', 'ms', 'us', 'ns'r zm8[s]TrÖ r( r( r) r³ )r
|
r r r r r~ r* r) r r r r¶ rr rs ) r\ r rR r- r) r rS ru r| r# r# r$ Útest_other_timedelta_unitè s þz#TestMerge.test_other_timedelta_unitc C sè t dddgdddgddd gd
|
}t dddgdddgddd gd
|
}d d d g|_d d d g|_t dddgdddgddd gdddgddd gd}d d d d d g|_t t|||¡ dd d g|_d}tjt|d t|| W 5 Q R X d S )Nr& rG rM rN rO r¢ rd r rÛ ©rf rg rj rf r+ r )rf rg rj Zv3Zv4rV z;Data columns not unique: Index\(\['foo'\], dtype='object'\)r )r
|
r rr rs r r r r )r\ r] rS rv r r# r# r$ Ú&test_overlapping_columns_error_message s$ "" ûÿ z0TestMerge.test_overlapping_columns_error_messagec C s t tjddddddgd}t tjdddddddgd}t tjdd ddddtjtjgtjdddgd
|
}t||d d d }t ||¡ d S )NÚ20151010rG r- ©Úperiodsr/ r& r² Ú20151011rM rN rè rf r¤ rµ )r
|
r0 Ú
|
date_ranger r5 r rr rs r r# r# r$ Útest_merge_on_datetime64tz s$ þÿþÿ ýÿz$TestMerge.test_merge_on_datetime64tzc
|
C sÆ t ddgtjddddd}t ddgtjddddd}t dddgttjddddtjg tjgttjdddd d }t||d
|
d d }t ||¡ |d jdks°t |d jdksÂt d S )Nr& rG r2 r- r3 r² rM r5 rè rf r¤ rµ r¦ zdatetime64[ns, US/Eastern]r§ )
|
r
|
r0 r6 r r6 r rr rs r) r r r# r# r$ Ú test_merge_datetime64tz_values3 s0 þÿþÿÿÿüÿ z(TestMerge.test_merge_datetime64tz_valuesc C s® t jdd}tt jd|jdgdgt jd|jdgdddd gd
|
}|d d
}|j|dd }tttd t|d t|d ttd t|d ddddddgd
|
}t ||¡ d S )NÚUTCr. Z2018g @Z2019)r r_ Údate2r r_ r: r³ r rx r( )r¦ Údate2_xr r§ Údate2_yr¦ r; r§ r< )
|
r0 ZDatetimeTZDtyper
|
r1 r/ r r r3 rr rs )r\ Zdtzrm ri ru rv r# r# r$ Ú test_merge_on_datetime64tz_emptyN s( ýú û ø
|
z*TestMerge.test_merge_on_datetime64tz_emptyc C s¼ t tjddddddgd}d|d < t t d
|
d d g¡d d}|d j d¡j d¡|d< t||ddd}t tjddddddgd tj gd tj gd d gd d}t
|
||¡ d S )Nz2017-10-29 01:00rN ÚHz Europe/Madrid)r4 Úfreqr/ r r³ r& r_ z2017-10-29 03:00:00z2017-10-29 04:00:00z2017-10-29 05:00:00rG )r r_ r9 r¤ ©r ry rd rM )r r¦ r§ ) r
|
r0 r6 Ú to_datetimerÅ Z tz_localizeZ
|
tz_convertr r r5 rr rs r r# r# r$ Ú+test_merge_datetime64tz_with_dst_transitiong s: þýÿøÿ ÿûÿ z5TestMerge.test_merge_datetime64tz_with_dst_transitionc C s t jdddd}tttt||dgd}t||g}|j|dddd }tt tj
|
dtj d
|
d ¡ d d ¡d ¡ddg| ¡ jd}t ||¡ d S )Nz
|
2016-01-01é ÚM©r4 r? Zpnum©rl r Tr )ro rp r r( rG éÿÿÿÿr& Zpnum_xZpnum_y©r rl )r0 Ú period_ranger
|
r r r r r r r rÁ r' ÚrepeatÚreshapeÚ
|
sort_indexrl rr rs )r\ rl r] rS ru rv r# r# r$ Ú"test_merge_non_unique_period_index s $ýz,TestMerge.test_merge_non_unique_period_indexc C s t tjddddddgd}t tjdddddddgd}t tjdd ddddtjtjgtjdddgd
|
}t||d d d }t ||¡ d S )Nr2 rG r rE r& r² r5 rM rN rè rf r¤ rµ )r
|
r0 rI r r5 r rr rs r r# r# r$ Útest_merge_on_periods s ÿþÿ ýÿz TestMerge.test_merge_on_periodsc C sÎ t ddgtjddddd}t ddgtjddddd}tjdddd}tjdddd}t dddgt|tjg tjgt| d }t||d
|
d d }t ||¡ |d jdks¸t |d jdksÊt d S )Nr& rG r2 r rE r² rM r5 rè rf r¤ rµ r¦ z Period[D]r§ )
|
r
|
r0 rI r r6 r rr rs r) r )r\ ri rm Zexp_xZexp_yrv ru r# r# r$ Útest_merge_period_values© s$ ÿÿýÿ z"TestMerge.test_merge_period_valuesc
|
C sZ |\}}| ¡ }| ¡ }tddddddgddtjtjtjtjgddtjtjtjtjgtjdddddgtjdddddgd }td
|
d d d d d gd
|
d d gd |d< |ddddddg }t||dddd}t ||¡ |j|dddd}t ||¡ t ||¡ t ||¡ |}|jddid}t||dddd} t | |¡ |j|dddd} t | |¡ d S )Nr r& rG rM rN rO rH rI )rJ Úcol_conflict_xrL Úcol_conflict_yrP Ú left_onlyÚbothÚ
|
right_only©Ú
|
categoriesÚ_mergerJ rP rL rQ rP r¤ T©ry r Z indicatorZ custom_namer³ ) r r
|
r r5 r r rr rs Úrename)
|
r\ rT rR rS Údf1_copyÚdf2_copyZ df_resultÚtestZdf_result_custom_nameZtest_custom_namer# r# r$ Útest_indicator¿ sn ûÿ ú÷
|
úÿ ÿ ÿ ÿzTestMerge.test_indicatorc C sh |\}}d}t jt|d t||dddd W 5 Q R X t jt|d |j|dddd W 5 Q R X d S )Nz<indicator option can only accept boolean or string argumentsr rJ r¤ rO rX r )r\ rT rR rS r r# r# r$ Ú#test_merge_indicator_arg_validationþ s z-TestMerge.test_merge_indicator_arg_validationc C sä |\}}t ||dddd}|jdk ¡ s,t|j |dddd}|jdk ¡ sPtt ||dddd}|jdk ¡ stt|j |dddd}|jdk ¡ stt ||dddd}|jd k ¡ s¼t|j |dddd}|jd k ¡ sàtd S )
|
NrJ ri TrX rT rm rR r rS )r rW r® r )r\ rT rR rS Ztest2Ztest3Ztest4r# r# r$ Ú%test_merge_indicator_result_integrity s z/TestMerge.test_merge_indicator_result_integrityc
|
C s |\}}dD ]}t dddg|ddgi}d| d}tjt|d t||ddd d
|
W 5 Q R X tjt|d |j|ddd d
|
W 5 Q R X q t ddgddgd }d }tjt|d t||ddd d
|
W 5 Q R X tjt|d |j|ddd d
|
W 5 Q R X d S )N)Z_right_indicatorZ_left_indicatorrW rJ r& rG zECannot use `indicator=True` option when data contains a column named z;|Cannot use name of an existing column for indicator columnr r¤ TrX )rJ Úcustom_column_namez:Cannot use name of an existing column for indicator columnr` ©r
|
r r r r )r\ rT rR rû rÑ Z df_badcolumnr r# r# r$ Ú test_merge_indicator_invalid s4
|
ÿ û ÿz&TestMerge.test_merge_indicator_invalidc C s´ t ddgddgd}t dddgdddgd}t ddddgddddgd}td d
|
d d gd d d
|
gd |d < t||ddgddd}t ||¡ |j|ddgddd}t ||¡ d S )Nr r& rH rI )rJ Úcol2rM r; rà rR rS rT rU rW rJ rc r¤ TrX ©r
|
r r rr rs )r\ Zdf3Zdf4Zhand_coded_resultZtest5r# r# r$ Ú%test_merge_indicator_multiple_columns= s ÿ
|
þ
|
z/TestMerge.test_merge_indicator_multiple_columnsc C sb t ddddgddddgd td
|
d }t ddddd gd ddddgdtdd }| ¡ }| ¡ }t||dddd}t ||¡ t ||¡ t ddddgddddgddddgd dddgdtd
|
ddddgd}t||dddd}t ||¡ t ddddgddddgd dddgd td
|
d }t||ddd }t ||¡ t ||¡ t ||¡ t||ddd }t ||¡ t ddddgddddgd dddgd dddgtd
|
d }| d¡} t| |dddd }t ||¡ t|t d gd!gdd
|
gd g}
|
t||
|
ddd"d d#} tj t
|
| d$ t||
|
dddd W 5 Q R X tj t
|
| d$ t||
|
ddd W 5 Q R X t|t dgd%gdd&gd gdd'} t| |ddd(d d)} tj t
|
| d$ t| |dddd W 5 Q R X tj t
|
| d$ t| |ddd W 5 Q R X t| |
|
dd*d d+} tj t
|
| d$ t| |
|
ddd(d W 5 Q R X d,} tj t
|
| d$ t| |
|
dd"d W 5 Q R X d-} tj t | d$ t||dd.d W 5 Q R X t ddddgd/d0d/d0gddddgd td
|
d }t dddgd/d0d/gd ddgd1td&d }t dddgd/d0d/gdddgd ddgd2td&d } d3} tj t
|
| d$ t||ddd W 5 Q R X t||ddgdd }t || ¡ d S )4NrH rI ra rb ÚcatÚdogZweaselZhorserª rN rk rc ZmeowZbarkzum... weasel noise?ZnayZchirp©rH ra rO Tz1:1)ro rp Úvalidate)Úa_xrI Úa_yra rj rk rF Z
|
one_to_one©rH rI ra )ry ri )rI rH ra rH )ro r ri ZmooZ one_to_manyzBMerge keys are not unique in right dataset; not a one-to-one merger ZcowrM )r
Z many_to_onezAMerge keys are not unique in left dataset; not a one-to-one mergeZ many_to_manyzCMerge keys are not unique in right dataset; not a many-to-one mergezBMerge keys are not unique in left dataset; not a one-to-many mergez"jibberish" 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"Z jibberishr r& )rH rI rb ©rH rI ra rb zQMerge keys are not unique in either left or right dataset; not a one-to-one merge) r
|
r r r rr rs r r r r r r )r\ ri rm Z left_copyZ
|
right_copyru rv Z
|
expected_2Z
|
expected_3Zleft_index_resetZ right_w_dupsr Z left_w_dupsZexpected_multir# r# r$ Útest_validationP s( þ þû
|
|
|
|
ü
|
ø ÿ
|
|
|
ýú
|
|
|
ýù
|
|
û ûû ÿûûûÿ
|
|
|
ýú ýú üù ÿzTestMerge.test_validationc C s: t g g g d}tjdd t||dd W 5 Q R X d S )Nrl Úraise)Údividerª rx )r
|
r Zerrstater )r\ rH r# r# r$ Ú)test_merge_two_empty_df_no_division_error s z3TestMerge.test_merge_two_empty_df_no_division_errorr¤ zindex,expected_indexrG rN Nú
|
2001-01-01z
|
2002-02-02z
|
2003-03-03c
|
C s8 g | ]0}t d ddg|dt d dddddgtjdfqS )r& rG rM r( N)r r Úfloat64©Ú.0Zdtypr# r# r$ Ú
|
<listcomp>) s ýþzTestMerge.<listcomp>)r& rG )rG rM )rM rN ú
|
2001-01-02ú
|
2001-01-03)r? Z1dZ2dZ3dc C s® t dddgdddgd|d}t ddddddd gi}|j|d
|
d |d }t dddgdddgdddgtjddgtjddgtjd d ggd d
|
dgd}|j|d d t ||¡ d S )Nr r& rG ©rH rf rk rI rM rN rO rf Tr rH r³ )Úinplace)r
|
r r r5 r rr rs )r\ r rl Úexpected_indexrR rS ru rv r# r# r$ Ú$test_merge_on_index_with_more_values s ,
|
|
|
ú÷ z.TestMerge.test_merge_on_index_with_more_valuesc C s t dddgdddgd}t ddddgi}t dddd gddddgddddgddd dgdddtjgd
|
}|j|d d d d }t ||¡ d S )Nr& rG rM r ry rI ©rH rf rI rH rf rH Trm r ©r
|
r r5 r rr rs r r# r# r$ Ú test_merge_right_index_rightX s " ýz&TestMerge.test_merge_right_index_rightc C s t ddgddgd}t ddgddgd}|j|dd g|d
|
}|d kr`t ddgddgd}nt ddgddgd}t ||¡ d S ) Nrg Zpigé( r )ÚanimalÚ max_speedZquetzaléP r r rµ rm rq )r\ r Úleft_dfÚright_dfru rv r# r# r$ Ú test_merge_preserves_row_orderg s z(TestMerge.test_merge_preserves_row_orderc C s´ t dddgtdddgtddd}t ddddgitddd gd
|
}|j|d d d d}t dddd gtdddd gddddgddddtjgd
|
}|jdd dgd}t ||¡ d S )Nr& rG rM rH rI ÚabcrU ry ra rk rf Trm r r} r r³ )
|
r
|
r r r r r r5 Zreindexrr rs r r# r# r$ Ú8test_merge_take_missing_values_from_index_of_other_dtypes s þÿ
|
|
ý úzBTestMerge.test_merge_take_missing_values_from_index_of_other_dtypec C sn t t d¡ d¡d dddddgd }t t d¡ d
|
¡d ddd d gd }|jjD ] }d |j_qR| |¡ d S )Nr¸ )rN rO r& rH rI ra rb rc r³ )rO rN r; rà F) r
|
r rÁ rK Z_mgrr, ÚflagsZ writeabler )r\ rX rY r" r# r# r$ Útest_merge_readonly s ÿ
|
ÿ
|
z TestMerge.test_merge_readonly)Lr r r
|
r Úfixturer] rS ri rm rw r} r r r r r r r r© r¯ r± r· r¿ r rÆ rÐ rÒ rÓ ÚmarkÚ parametrizerÚ râ rä rç ré ró rõ r r r r r r. r/ r1 r7 r8 r= rB rM rN rO r] r^ r_ rb re rn rq r r r0 r6 rr ÚALL_REAL_NUMPY_DTYPESr Ú from_tuplesr r5 r r r| r r r r r# r# r# r$ rU p sÚ
|
|
|
|
|
|
üþ
|
|
/* "ÿ
|
|
?
|
" F þ ÿþû ûÿþþþ þáþ'
|
rU c C sN dD ]D}| j ||d}t| ¡ | ¡ |dd}| d¡}tj||dd qd S )N)r ri r¤ r¹ T)r r
rl F)Z check_names)r r r r rr rs )r; rà r ru rv r# r# r$ rÍ s
|
|
rÍ c @ s| e Zd Zej dddgeddg d¡g¡dd Zej de j
|
e j e j e j e jg¡ej de j
|
e je je jg¡d d
|
Zej d d d dgdddgd d dgdddgdfd d dgddgd dgddgdfd d gdddgd d gddgdfg¡dd Zdd Zdd Zdd Zdd Zej d d d d ged d d g d¡fd!ddged d d g d¡fd d ged"d#ged$fd d ged"d#ged$fg¡d%d& Zej d ed d gd'd$d d d(gfed d gd)d$d d d(gfd d d gd*d+d,gfd!ddgd*d+d,gfd d d gd*d+d,gfej d-d d.d/d0d1gfej d-d d.d/d d gfej d-d d.d/d!dgfej d2dd3ej d2dd4d5fg ¡d6d7 Z ej d8d d gd9fg d:fd gd;fd gd<fg¡d=d> Z ej d8d d gd9fg d:fd gd;fd gd<fg¡d?d@ Z ej dAd:d#d dBgd"dCdggfd9d#d dBgd"dCdggfd<d#d dBgd"dCdggfd;d"dCdgd#d dBggfg¡dDdE Z dFdG Z!ej dHdIdJdKdLdMdNdOdPdQdRg
|
¡dSdT Z"dUS )VÚTestMergeDtypesÚ
|
right_valsr+ r Úcategoryc
|
C sv t ddgtddg d¡ddgddgtddgdd tddgd
|
d d }t d |i}t||d d }t|jjsrtd S )Nr+ r r r& rG ç ð?ç @Úuint64r( Úint32)r r ÚCr ÚEÚFr rx )r
|
r r~ r r r r) r )r\ r ri rm ru r# r# r$ Útest_different¤ s úÿ
|
z TestMergeDtypes.test_differentÚd1Úd2c
|
C s@ t |¡}t |¡}tt jdddgd |dddgd t jt d ¡t jdd
|
}t d d g¡}td t jddg|di|d}|j|ddgd}| ¡ } |j
|
dkr¬t d¡}t jt j |d| d < d| j | j dk| jdk@ d f< d| j | j dk| jdk@ d f< t || ¡ |j|ddgdd}| jddgddd t || ¡ d S )Nr r& rG r r( r+ r rì é )Úk1Úk2Úv)rG r )r& r+ rj rO rd rk r r rx rÑ rs Trî Z mergesort)Úkindrz )r r) r
|
r¶ rÁ r' r r r r r¡ r5 r r r rr rs Z sort_values)
|
r\ r r Zdtype1Zdtype2ri rl rm ru rv r# r# r$ Útest_join_multi_dtypesº s*
|
|
|
ýÿ
|
|
z&TestMergeDtypes.test_join_multi_dtypesz int_vals, float_vals, exp_valsr& rG rM r r ç @©ÚXÚYc C sd t d|i}t d|i}t |}|j|ddd}t ||¡ |j|ddd}t ||ddg ¡ d S )Nr¥ r¦ r rq )r\ Zint_valsZ
|
float_valsZexp_valsr r rv ru r# r# r$ Útest_merge_on_ints_floatsÛ s z)TestMergeDtypes.test_merge_on_ints_floatsc C s t ddgddgdddgd}t d gd
|
gd dd gd}|j|dd d}t ddgddgtjdgdddd gd}t ||¡ d S )Nr r r^ r¸ re rf rg r³ rG éÈ )rf rj rj ri rµ g i@r0 )r
|
r r r5 rr rs r r# r# r$ Útest_merge_key_dtype_castñ s þz)TestMergeDtypes.test_merge_key_dtype_castc C sö t ddddgi}t ddddgi}t dgdgd }t t¡" |j|ddd
|
}t ||¡ W 5 Q R X t t¡* |j|ddd
|
}t ||ddg ¡ W 5 Q R X t dtjtjdgi}t d ¡* |j|ddd
|
}t ||ddg ¡ W 5 Q R X d S ) Nr¥ r& rG rM r¦ çñ?g @r£ r¤ r )r
|
rr Zassert_produces_warningÚ UserWarningr rs r r5 )r\ r r rv ru r# r# r$ Ú!test_merge_on_ints_floats_warningü s z1TestMergeDtypes.test_merge_on_ints_floats_warningc C st t dtddgtdi}t dddgi}t dddgitd}t||dd}t ||¡ t||dd}t ||¡ d S ©Nrf TFr( rx )r
|
r r4 r rr rs ©r\ rR rS rv ru r# r# r$ Ú(test_merge_incompat_infer_boolean_object s z8TestMergeDtypes.test_merge_incompat_infer_boolean_objectc C sx t dtddtjgtdi}t dddgi}t dddgitd}t||dd}t ||¡ t||dd}t ||¡ d S r )r
|
r r r5 r4 r rr rs r® r# r# r$ Ú5test_merge_incompat_infer_boolean_object_with_missing s zETestMergeDtypes.test_merge_incompat_infer_boolean_object_with_missingzdf1_vals, df2_valsr rH rI rý FTr( c C s\ t d|i}t d|i}t||dgd}t|jjs8tt||dgd}t|jjsXtd S )Nr rx )r
|
r r r r) r )r\ Údf1_valsÚdf2_valsrR rS ru r# r# r$ Ú!test_merge_incompat_dtypes_are_ok+ s z1TestMergeDtypes.test_merge_incompat_dtypes_are_okr ra r r Ú1Ú2z1/1/2011r rE z
|
2011-01-01z
|
2011-01-02Z20130101)r4 r- r3 c C sÀ t d|i}t d|i}d|d j d|d j d}t |¡}tjt|d t||dgd W 5 Q R X d|d j d|d j d}t |¡}tjt|d t||dgd W 5 Q R X d S )Nr zYou are trying to merge on z and z9 columns. If you wish to proceed you should use pd.concatr rx )r
|
r) ÚreÚescaper r r r )r\ r± r² rR rS r r# r# r$ Ú test_merge_incompat_dtypes_errorC s ÿ
|
ÿ
|
z0TestMergeDtypes.test_merge_incompat_dtypes_errorzexpected_data, howr¤ r rm ri c C s` t dgdg|d}t dgdg|d}t|||d}tt|}t ||dg|d}t ||¡ d S )N)r& ÚidrÀ ©rG r¹ ©rl r r) ©r
|
r r r rr rs ) r\ Úany_numeric_ea_dtyper Ú expected_datar r ru Z exp_indexrv r# r# r$ Útest_merge_EA_dtypes s ÿz#TestMergeDtypes.test_merge_EA_dtypec C s` t dgdg|d}t dgdg|d}t|||d}tt|}t ||dg|d}t ||¡ d S )N)rH r¹ rÀ )rI r¹ r» r¼ ) r\ r r¾ Zany_string_dtyper r ru Zexp_idxrv r# r# r$ Útest_merge_string_dtype s ÿz'TestMergeDtypes.test_merge_string_dtypezhow, expected_datarN rO c C s\ t ddgddgd}t ddgddgd}t|||d }t |d
|
d d gd }t ||¡ d S )NTFr& rO ©r r rM rN ©r r r¹ r r r r³ rq )r\ r r¾ rR rS ru rv r# r# r$ Útest_merge_bool_dtype s
|
z%TestMergeDtypes.test_merge_bool_dtypec C sÎ t ddddd gdddddgd t ¡ d
|
}| ¡ }t dddtjdgddd tjd gd |d
|
}| ¡ }t||dg|d}t ||¡ t ||¡ tt
|
d¡t ¡ t
|
d¡gt dddg¡d}t |j|¡ d S )Nr´ rµ Ú3Ú4Ú5Ú6Ú7Ú8)©Úlvl0zlvl1-a©rË zlvl1-b)Údatar) Ú9Z11)rÊ ©rË zlvl1-crÊ )ri rm ry r ÚOrÌ rÏ rk )r
|
r0 Z StringDtyper ZNAr rr rs r r r) r r r¨ rå ) r\ rØ Z string_dtyperR rZ rS r[ r rv r# r# r$ Útest_merge_ea_with_string¬ s0 þûþû ÿþz)TestMergeDtypes.test_merge_ea_with_stringzleft_empty, how, exp)Fri ri )Frm Úempty)Fr rÒ )Fr¤ ri )FÚcrossÚ empty_cross)Tri rÒ )Trm rm )Tr rÒ )Tr¤ rm )TrÓ rÔ c C sú t ddgddgd}t dgdgddd }|r<| d
|
¡}n
|
| d
|
¡}|j||d }|d kr~t ddgddgtjtjgd }nl|dkr t tjgdgdgd}nJ|dkrÎt dddgdd}|rê|dddg }n |dkrêt ddddgdd}t ||¡ d S )NrG r& rM rN rÁ rO r r' r( r r¹ ri )r r r rm )r r r rÒ r r r rÀ rÔ ZA_xZA_y)r
|
Úheadr r r5 rr rs )r\ Z
|
left_emptyr r| ri rm ru rv r# r# r$ Útest_merge_emptyÍ s"
|
"z TestMergeDtypes.test_merge_emptyN)#r r r
|
r r r r r~ r r r' r Úint16Zint8Zuint8rs Úfloat32Úfloat16r¢ r§ r© r¬ r¯ r° r4 rº r³ r0 r6 r¸ r¿ rÀ rà rÑ rÖ r# r# r# r$ r £ s´ ÿ
|
& ýþ
|
úþ
|
þ þóþ
|
|
üþ
|
|
üþ
|
üþ
|
!öþr c C sN t j d¡ ttt jjddgdd tddg¡t jjdddgddd S )
|
NéÒ r+ r )r^ rê ÚoneÚtwoZthreer¤ )r r Úseedr
|
r Úchoicer~ ÚCDTr# r# r# r$ ri õ s
|
ÿüÿri c C s4 t j d¡ ttddg tddg¡ddgdS )NrÚ r+ r r& rG )r¥ ÚZ)r r rÝ r
|
r r~ rß r# r# r# r$ rm s "ÿrm c
|
@ sþ e Zd Zdd Zdd Zdd Zdd Zej d d
|
d g¡d d Z
|
dd Z ej ddd dd dd g¡dd Z dd Z dd Zej d d
|
d g¡ej dd d
|
d
|
d gd
|
d gd
|
d gfd d d d gd d gd d gfd d d d gd d gd d gfg¡d d! Zd"d# Zd$S )%ÚTestMergeCategoricalc C sV t ||dd}|j ¡ }ttddgdt d¡t d¡gdddgd }t ||¡ d S )
|
Nr¥ rx r+ r rU rÐ ZY_xZY_yrk © r rå rL r r r r) rr r¨ )r\ ri r ru rv r# r# r$ Útest_identical s
|
þz#TestMergeCategorical.test_identicalc C sV t ||dd}|j ¡ }ttddgdt d¡t d¡gddd gd
|
}t ||¡ d S ) Nr¥ rx r+ r rU rÐ r' r¦ rà rk râ ©r\ ri rm r ru rv r# r# r$ Ú
|
test_basic s
|
ýúz TestMergeCategorical.test_basicc C sD t ddddddddddddd}t d d d d d dd
|
d
|
d
|
d
|
d
|
dd }t||d ddd }| ¡ }| ¡ }|d d¡|d< t||d ddd }|d tdg¡|d< t ||¡ | ¡ }|d d¡|d< t||d ddd }t ||¡ | ¡ }|d d¡|d< | ¡ }|d d¡|d< t||d ddd }t ||¡ d S )NrH rI ra rb rc )r r& rG rM rN Únullr« rÉ rÊ rª ri )r r r r )r
|
r r r~ r rr rs )r\ rm ri r] rv Zcrightru Zcleftr# r# r$ Útest_merge_categorical$ s6 þÿþÿ z+TestMergeCategorical.test_merge_categoricalc C s t tdddgdddgddddgd}t tdddgdddgdd d
|
d gd }t||d gd}t tdddgdddgd d
|
d gd}t ||¡ d S )Nr r r rU ZA0ZB0ZC0)ÚFooÚLeftZC1ÚB1ÚA1)rè ÚRightrè rx )rè ré rì rd r r# r# r$ Ú'tests_merge_categorical_unordered_equalO s$ þÿþÿ ýÿz<TestMergeCategorical.tests_merge_categorical_unordered_equalÚorderedTFc C sØ t ddg|d}tdddgtdddg|dddd gd
|
d d g¡}td ddgtdddg|ddddgd d d g¡}t||dddd}tdddgtdddg|dddd gddtjgd d d g¡}t ||¡ d S )NZP2ZP1©rV rî r r r( r r& rG )r¹ ÚprH r¹ rð r r^ r rì )r¹ rð r ri Tr g &@g (@)r¹ rð rH r ) r r
|
r r r r r5 rr rs )r\ rî ZpcatrR rS ru rv r# r# r$ Ú5test_multiindex_merge_with_unordered_categoricalindexh s8 ýÿúýÿú
|
üÿùzJTestMergeCategorical.test_multiindex_merge_with_unordered_categoricalindexc C s |j |j d¡d}t||dd}|j ¡ }ttddgdt d¡td d
|
gdgdd d gd }t
|
||¡ |j j |j j ¡st|jj |jj ¡std S )Nr )rà r¥ rx r+ r rU rÐ r& rG r¦ rà rk )Úassignrà r~ r rå rL r r r r) rr r¨ r¥ rô Z#_categories_match_up_to_permutationr rä r# r# r$ Útest_other_columns
s
|
ýú z'TestMergeCategorical.test_other_columnsÚchangec C s | S r r# r: r# r# r$ r< r= z TestMergeCategorical.<lambda>c C s | tdddg¡S )Nr+ r Zbah©r~ rß r: r# r# r$ r< r= c C s | tdd¡S )NT©rî rõ r: r# r# r$ r< r= c C s ||j d¡}|j|d}t|j jjs.tt||d|d}|j ¡ }t
|
t d¡t d¡t d¡gdddgd }t ||¡ d S )
|
Nr4 )r¥ r¥ rµ rÐ r' r¦ rà rk )r¥ r~ rò r rô r) r r rå rL r r rr r¨ ) r\ rô rØ ri rm r¥ r ru rv r# r# r$ Ú test_dtype_on_merged_different s
|
ÿz3TestMergeCategorical.test_dtype_on_merged_differentc s d t dddddddd d
|
d g
|
d d dddgd fdddD fdddD d}| dd ¡}t||t|jd}t ||¡ d S )NrO rH rI ra rb rc rÉ rÊ rË rÑ Újr Úwr; rà r× rG c s$ g | ] }|gd D ]}|qqS rº r# ©ru ZeachÚletter©r r# r$ rv º s þzKTestMergeCategorical.test_self_join_multiple_categories.<locals>.<listcomp>)r r! Úurð Úoc s g | ]}|g D ]}|qqS r# r# rú rü r# r$ rv ¿ s ó)
|
ZaaZbbÚccÚddÚeeÚffZggÚhhÚiiZjjrm c S s
|
| d¡S )Nr )r~ r: r# r# r$ r< Ó r= zITestMergeCategorical.test_self_join_multiple_categories.<locals>.<lambda>rx )r
|
Úapplyr r r rr rs )r\ r] ru r# rü r$ Ú"test_self_join_multiple_categories² s
|
þ
|
þøÿ z7TestMergeCategorical.test_self_join_multiple_categoriesc C s, t tddddgtddddggddgd}|d d ¡|d< t tddddgtddd
|
d ggdd gd}|d d ¡|d< t t d ¡ ¡ ddgt d¡ ¡ dtjgt d¡ ¡ tjd ggddd gd}t||ddgd}t ||¡ t t d ¡ ¡ ddggddd gd}t||ddgd}t ||¡ d S )NiÑ r& rª rG çÍÌÌÌÌÌô?r Znum2r³ r rM çffffffö?Znum4rr rw rx r¤ r@ r )
|
r
|
r r~ r0 r1 r r5 r rr rs )r\ r] rS Zexpected_outerZ result_outerZexpected_innerZ result_innerr# r# r$ Ú test_dtype_on_categorical_datesÚ s2 ÿ ÿýú þz4TestMergeCategorical.test_dtype_on_categorical_datesz.category_column,categories,expected_categoriesrG r& ÚFalseÚTruec C s t ddddg|d}|d t||d¡|d< t ddgddgd }| |¡}t ddg|ddgd
|
}|d t||d¡|d< t ||¡ d S ) Nr& rG rM rN )r¹ rf rf rö rÛ )r¹ Únum)r¹ rf r )r
|
r~ rß r rr rs ) r\ Zcategory_columnrV Zexpected_categoriesrî rR rS ru rv r# r# r$ Ú/test_merging_with_bool_or_int_cateorical_columnú s
|
zDTestMergeCategorical.test_merging_with_bool_or_int_cateorical_columnc C s\ t tddtjgdddd}t||dd}t tddtjgddddd}t ||¡ d S ) Nr& rG r* r( rÁ r rx )r ÚB_xZB_y)r
|
r r r5 r rr rs )r\ r] ru rv r# r# r$ Útest_merge_on_int_array s ÿz,TestMergeCategorical.test_merge_on_int_arrayN)r r r
|
rã rå rç rí r r r rñ ró r÷ r r r r r# r# r# r$ rá
|
s6
|
+
|
ýþ
|
( ýþ rá c C s t ddddgidddgdS )NrH r¸ r^ r rG r& rk rQ r# r# r# r$ r s r c C s t ddddgidddgdS ) NrI é, éd r¨ rM r& rG rk rQ r# r# r# r$ r
s r
c @ sr e Zd Zej dddeddgddgdd d
|
gd fdd eddgddgdd
|
d gd fd dedddgddejgdd d
|
dgd fd d edddgejddgddd
|
d gd fddeejddgdddgddd
|
d gd fdd eddejgdddgdd
|
d dgd fddedddejgejdddgddd
|
d dgd fdd edddejgejdddgddd
|
d dgd fg¡dd Z dS )ÚTestMergeOnIndexeszhow, sort, expectedr Fr¸ r^ r¨ r rª rG r& rk Tri r rm r rM r¤ c C s$ t ||dd||d}t ||¡ d S )NT)ro rp r r
rz )r\ r r
r r
rv ru r# r# r$ Útest_merge_on_indexes% s 0 ÿz(TestMergeOnIndexes.test_merge_on_indexesN)
|
r r r
|
r r r r
|
r r5 r r# r# r# r$ r $ sV ""$ý$ý ÿý ÿý
|
þý
|
þýÝþ/r rl c C s g | ]}t d dg|ddqS )r& rG Ú index_colr8 )r rt r# r# r$ rv ] s rv r r r )rV r9 rG )ÚstartÚstopr9 r, z
|
2018-01-02r¥ c C s t | j d| j dS )Nú[r& )Útyper r) r: r# r# r$ r< c r= )r? c C sb t dddgi| d}t dddgi| d}|j|dgd }t ddgddgd
|
| d}t ||¡ d S ) NÚ left_datar& rG rk Ú
|
right_datar r r rx )r r rq )rl ri rm ru rv r# r# r$ Útest_merge_index_types[ s
|
r z-on,left_on,right_on,left_index,right_index,nmr¤ r F)NNNTTr T)NNNTTNc
|
C sø t dddddgitjddgddggd d
|
gd d }tddddgtjddgddggd d
|
gd |d }t ddgddgdtjddgdggd d
|
gd d }|d k rÀt||| ||||d} t | |¡ n4d}
|
tjt |
|
d t||| ||||d} W 5 Q R X d S )Nr r& rG rM rN rH rI r r¤ r ©Únamesrk r% rÁ )ry r r ro rp z$Cannot merge a Series without a namer )
|
r
|
r Ú from_productr r rr rs r r r ) ry r r ro rp ÚnmrH rI rv ru r r# r# r$ Útest_merge_seriesq sF þ
|
ýþù ùr c C s t dddddgitjddgddggd d
|
gd d } tddddgtjddgddggd d
|
gd d d}tjtdd t| |d d
|
gd W 5 Q R X d S )Nr r& rG rM rN rH rI r r¤ r r rk )r r r% z-Not allowed to merge between different levelsr rx )r
|
r r r r r r r rª r# r# r$ Ú test_merge_series_multilevel¦ s þ
|
ý ÿr! z!col1, col2, kwargs, expected_colsÚsuffixes)Ú Ú_dupr Z0_dup)Nr$ )Ú_xÚ_yZ0_xZ0_yr% r& rH )Nr& rý )r% Nz0.0_xrI Úb_yrj )Nr% )Z_aNZ0_ark c C s t | dddgi}t |dddgi}t ddgddgddgg|d}|j|fddd |}t ||¡ t||fddd |}t ||¡ d S )
|
Nr& rG rM rN rO r¢ r³ Trn rq )rJ rc ÚkwargsZ expected_colsrH rI rv ru r# r# r$ Útest_merge_suffix¸ s r) z how,expectedr r¨ r é< éF éX é¼ é )r rê ZB2r& r c C sb t dddgdddgd}t dddgd d
|
d gd}t||d | d d}d ddg|_t ||¡ d S )Nr r¨ r& r* r+ r rÁ r r, r- r. r )r% r% )ry r r" r )r
|
r r rr rs )r rv r r
ru r# r# r$ Útest_merge_duplicate_suffixØ s
|
r/ zcol1, col2, suffixes)rH rH )NN)rH rH )r# N)r r )Nr# c C sX t | dddgi}t |dddgi}d}tjt|d t||dd|d W 5 Q R X d S )
|
Nr& rG rM rN rO z'columns overlap but no suffix specifiedr T©ro rp r" ra )rJ rc r" rH rI r r# r# r$ Útest_merge_suffix_errorö s
|
r1 rÞ c C sT t ddddgi}t ddddgi}tjtdd t||d
|
d
|
| d W 5 Q R X d S ) NrH r& rG rM rI rN rO zPassing 'suffixes' as ar Tr0 )r
|
r r Ú TypeErrorr )r" rH rI r# r# r$ Útest_merge_suffix_raises s r3 zcol1, col2, suffixes, msg)rH rH rl z(too many values to unpack \(expected 2\)z1not enough values to unpack \(expected 2, got 1\)c C sT t | dddgi}t |dddgi}tjt|d t||dd|d W 5 Q R X d S ) Nr& rG rM rN rO r Tr0 ra )rJ rc r" r rH rI r# r# r$ Ú test_merge_suffix_length_error s r4 Ú cat_dtyperÛ rÜ Úreversec C sô t dddgddt dddgddd}ttdddg |d ¡dd d
|
gd d ¡}dddg}dd d
|
g}|r|| ¡ | ¡ tt| || ¡|d d ¡}|j|ddd}tdd d
|
gdd d
|
gtdddg |d ¡d d ¡}t ||¡ d S )NrH rI ra Frï )rÛ rÜ rÛ r& rG rM ©r+ ri r+ ©r+ rm Trn ©ri rm r+ ) r r
|
r r~ r r6 r rr rs ) r5 r6 Z
|
cat_dtypesrR Zdata_fooZ
|
data_rightrS ru rv r# r# r$ Útest_merge_equal_cat_dtypes s8 þ"ÿþ
|
|
ÿþýÿúr: c C s® t dddgdd} ttddg | ¡ddgd d ¡}ttdddg | ¡d
|
ddgd d ¡}|j|d d d }tddgd
|
dgtddg | ¡d d ¡}t ||¡ d S )NrH rI ra Frï r& rG r7 r+ rM r8 Trn r9 )r r
|
r r~ r r rr rs )r5 rR rS ru rv r# r# r$ Ú test_merge_equal_cat_dtypes2B s$ ÿþ ÿþ ÿþr; c C sj t dtt dd¡t dd¡gddi} | ¡ }|d d¡|d< t|| ddd }| ¡ }t ||¡ d S )
|
NrH r r& rG Úintervalr( r r r@ ) r
|
r r0 ZIntervalr r~ r rr rs )rm ri ru rv r# r# r$ Ú test_merge_on_cat_and_ext_arrayX s "ÿr= c sÈ ddddg} dddg}t j| |fdd gd
|
}t|d }d |d < t|d }d |d < d d|j|d fd} fdd| D fdd| D }t j||gdd gd
|
}t|d }d |d < t ||¡ d S )NrH rI ra rb r´ rµ rÄ r¤ r r r³ r# r¹ r% r& ©ry r" c s g | ] }| qS r# r# ©ru rû )Úl_sufr# r$ rv z s z1test_merge_multiindex_columns.<locals>.<listcomp>c s g | ] }| qS r# r# r? )Úr_sufr# r$ rv z s )r r r
|
r rr rs ) ÚlettersZnumbersrl Zframe_xZframe_yru Zexpected_labelsr{ rv r# )r@ rA r$ Ú test_merge_multiindex_columnsf s(
|
|
|
ÿ ÿ
|
rC c C s t dddgdddgd} t dddgt d d
|
d g¡d }t| |d dd}t dddgdddgt d d
|
dg¡d}t ||¡ d S )NrH rI ra r´ rµ rÅ )r; rà rÄ Ú2000Z2001Z2002)rà r× ri rà r@ r6 )r; rà r× )r
|
r0 rA r rr rs ©rR rS ru rv r# r# r$ Ú test_merge_datetime_upcast_dtype s ÿýÿrF Ú n_categoriesrO é c C s t dgtt| }tt| d dg|d}tdggdgt dgtt| dd}t||dddd }td
|
d t| d D d d g|d}t ||¡ d S )Nr r& r_ rH r¢ rU ri Tr c S s& g | ] }|d k r|dgn|t jgqS )rG g @)r r5 )ru rÑ r# r# r$ rv ¦ s z9test_categorical_non_unique_monotonic.<locals>.<listcomp>r¦ r§ )r r r r
|
r rr rs )rG ro rR rS ru rv r# r# r$ Ú%test_categorical_non_unique_monotonic s ýýrI c C s¸ t ddddddgdddgddddddgd} t| } t ddddddgdddgddddddgddd d
|
d d gd }t| ddg¡d }t| | ¡ ddgddgdd}|jddgdd}| j|ddgd}t ||¡ t ddddddgdddgddddddddgd} t| } t ddddddgdddgddddddddgddd d
|
d d gd }t| ddg¡d }t| | ¡ ddgddgdd}|jddgdd}| j|ddgd}t ||¡ d S )NrH rI ra r r& )ÚCat1ÚInt1rª g333333ó?r r r gù?)ÚCatÚIntÚFactorrL rM rN rJ rK ri r¼ )Zaxisrx Trö ) r r
|
r r r Zdropr rr rs )rH rI rv ru r# r# r$ Ú&test_merge_join_categorical_multiindex s\ þýû ÿü ÿûûrO Úfuncr Z
|
merge_asof)r( Úerr_msg)r ro r ro )r rp r rp c C sx t ddgddgd}t ddgddgd}d |d
|
d |d d }tjt|d tt| ||f| W 5 Q R X d S )Nr& rG rM rN rª rO r¢ rh zCan only pass argument "r ú" OR "z " not both\.r ©r
|
r r r Úgetattrr0 ©rP r( rQ ri rm r r# r# r$ Ú/test_merge_join_cols_error_reporting_duplicateså s
|
|
rV c C sx t ddgddgd}t ddgddgd}d |d
|
d |d d }tjt|d tt| ||f| W 5 Q R X d S )Nr& rG rM rN rª rO r¢ rh z Must pass "r rR z"\.r rS rU r# r# r$ Ú,test_merge_join_cols_error_reporting_missingö s
|
|
rW r( c C sj t ddgddgd}t ddgddgd}d }tjt|d
|
" tt| ||fd d i| W 5 Q R X d S ) Nr& rG rM rN rª rO r¢ rh zZCan only pass argument "on" OR "left_index" and "right_index", not a combination of both\.r ry rH rS )rP r( ri rm r r# r# r$ Ú1test_merge_join_cols_error_reporting_on_and_index
|
s
|
ÿrX c C sv t ddgddgd} t ddgddgd}t| |dddd}t ddgddgddgddgddgd}t ||¡ d S ) Nr& r+ )r; r× rm Tr; )r ro r )r; Zx_xZz_xZx_yZz_yrq ©ri rm ru rv r# r# r$ Útest_merge_right_left_index
|
s ûÿ rZ c C s t dgdgd ddg¡} t ddgi dg¡}t g g dtjd ddg¡}t| |dgdd}t ||¡ t|| ddgd }t ||¡ d S )
|
Nr& rG rª rH rI r( TrÖ rÕ )r
|
r r r' r rr rs )rR rS rv ru r# r# r$ Ú$test_merge_result_empty_index_and_on,
|
s r[ c C s t dddgddd} t dddgdd}tjtdd t| |dd W 5 Q R X tjtdd t|| dd
|
d W 5 Q R X d S ) Nr& rG rM )rH rI Úb_xrª ú(Passing 'suffixes' which cause duplicater rH rx )r& r% r> ©r
|
r r r r rÞ r# r# r$ Ú.test_merge_suffixes_produce_dup_columns_raises8
|
s r_ c C sp t dddgdddggdddgd} t ddgdd}t| |dd}t ddddggdd d d
|
gd}t ||¡ d S ) Nr& rG rH rI r³ rM rª rx r\ r' rq rY r# r# r$ Ú3test_merge_duplicate_columns_with_suffix_no_warningD
|
s
|
" r` c C sf t ddddgddddggddddgd} t ddgdd}tjtd d
|
t| |dd W 5 Q R X d S ) Nr& rG rH rI r\ r³ rM rª r] r rx r^ rÞ r# r# r$ ÚItest_merge_duplicate_columns_with_suffix_causing_another_duplicate_raisesN
|
s (ra c C s t ddgddggtddgd} t dd gd
|
d ggd d gd}t|| dddd}t dd ddgd
|
d ddggtd d ddgd}t ||¡ d S )Nr& rG rM rN rH g \@r³ rÛ r^ r rì r; rà r Tr ©r
|
r r rr rs rE r# r# r$ Ú%test_merge_string_float_column_resultW
|
s ÿrc c C sX t dgdgdgd} t dgdgdgd}tjtdd t| |dgdd W 5 Q R X d S )
|
Nr¥ r é )rÍ r rl iç zCan only pass argumentr T)ry ro r^ )Zdf_1Zdf_2r# r# r$ Ú/test_mergeerror_on_left_index_mismatched_dtypesb
|
s re c C sb t td} t| tdd}tdtddi}t||| dd}t||| jdd}t ||¡ d S )NrM rÁ r r¢ r r )r r r
|
r Ú_datarr rs )Úcirm ri Úresrv r# r# r$ Ú#test_merge_on_left_categoricalindexj
|
s ri r) r* c C sä t ddgddgd| d}t tjtjgddgd| d}t||dd d
|
}t ddtjtjgddtjtjgtjtjddgd | d}t ||¡ t||dd d
|
}t tjtjddgddtjtjgtjtjddgd | d}t ||¡ d S ) Nr& rG )rf rJ r( rM rN )rf rc rf r¤ rµ )rf rJ rc )rf rc rJ r~ )r) ri rm ru rv r# r# r$ Útest_merge_outer_with_NaNv
|
s( ýú ýúrj c C sh t ddgitdgddd} t ddgitdgddd}t| |ddd}t dgdd}t ||¡ d S ) NrH r& ra r¥ rk rb r )rj rk rb rY r# r# r$ Ú test_merge_different_index_names
|
s
|
rk c C sh t dddgdd| d}t dddgdd| d}|j||d}t dddgddd| d}t ||¡ d S © Nr& rG rM rª r( rh r¹ rl rq ©r½ rØ ri rm ru rv r# r# r$ Ú test_merge_ea
|
s
|
rn c C s t dddgdd| d}t dddgdd| ¡ d}|j||d}t tdddg| dtdddg| dtdddg| ¡ dd}t ||¡ d S rl )r
|
Úlowerr r rr rs rm r# r# r$ Útest_merge_ea_and_non_ea¥
|
s ýÿrp r' úint64[pyarrow]c C sr t d¡ tdddgi| d}tdddgidd}| |¡}| ¡ }t ||¡ | |¡}| ¡ }t ||¡ d S )NZpyarrowrH r& rG r( rq )r Z importorskipr
|
r r rr rs )r) r] rS ru rv r# r# r$ Ú!test_merge_arrow_and_numpy_dtypes´
|
s
|
|
|
rr )r r )Xr r r r¶ Únumpyr r Zpandas.core.dtypes.commonr r Zpandas.core.dtypes.dtypesr Zpandasr0 r r r
|
r r r r r r r r Zpandas._testingZ_testingrr Zpandas.api.typesrß Zpandas.core.reshape.concatr Zpandas.core.reshape.merger r r% r2 r7 r rD rE rF rT rU rÍ r ri rm rá r r
r r r r r r r! r) r5 r/ r1 r3 Útupler4 r: r; r= rC rF rI rO rV rW rX rZ r[ r_ r` ra rc re ri rj rk rn rp rr r# r# r# r$ Ú<module> sZ 4
|
|
|
|
|
2 T
|
|
|
|
7 ýÿø
|
|
øþ
|
(óþ
|
ÿþ
|
ýÿþùþ
|
þ
|
|
þþ
|
#
|
8þþ þþ þþ
|
|
|
