U
|
ý°dD ã @ sx d dl mZ d dlZd dlZd dlmZ d dlZd dlm Z m
|
Z
|
m Z m Z m Z d dlmZ G dd dZG dd dZdS ) é )ÚdeepcopyN)ÚPerformanceWarning)Ú DataFrameÚIndexÚ
|
MultiIndexÚSeriesÚconcatc
|
@ s` e Zd Zdd Zej ddddddd d
|
g¡d d Zd d Zdd Z dd Z
|
dd Z dd Z dS )ÚTestIndexConcatc C sÔ t dddgdddgddd gd
|
}t d d d dgi}tdddg|_tdddg|_t||gdd|d}tj}t |||dgdddd gdddd gddd |ggtddddgd}|sÄ|jddddg }t ||¡ d S )NÚaÚbÚcé é é g @g @g333333ó?)Ztest1Ztest2Útest3r gÍÌÌÌÌÌ@g@g333333@ÚxÚyÚzÚqT)ÚaxisÚ ignore_indexÚsort©Úindex) r r r r ÚnpÚnanÚlocÚtmÚassert_frame_equal)Úselfr Zframe1Zframe2Zv1r Úexpected© r! ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/reshape/concat/test_index.pyÚtest_concat_ignore_index s& ÿ
|
|
|
|
üù z(TestIndexConcat.test_concat_ignore_indexz#name_in1,name_in2,name_in3,name_out)Úidxr$ r$ r$ )r$ r$ NN)r$ NNN)Úidx1Úidx2NN)r% r% r& N)r% r& Zidx3N)NNNNc
|
C s¾ t dddg|dt dddg|dt dddg|dg}dd t|d d
|
d gD }t|d d }t dddddg|d}tdd dtjtjgtjdd dtjgtjtjdd dgd|d} t || ¡ d S )Nr
|
r r ©ÚnameÚdÚec S s& g | ] \}}t |d ddgi|dqS )r r r r )r )Ú.0Úir r! r! r" Ú
|
<listcomp>@ s z@TestIndexConcat.test_concat_same_index_names.<locals>.<listcomp>r r r r ©r r r )r r r r )r Úzipr r r r r r )
|
r Zname_in1Zname_in2Zname_in3Zname_outÚindicesÚframesÚresultZexp_indr r! r! r" Ú test_concat_same_index_names- s" ýÿ ýú z,TestIndexConcat.test_concat_same_index_namesc C sÊ t tj dd¡tdttdddd}t tj dd¡tdttdddd}t||gdd gd
|
d gd }t||gdd gd
|
gd }t|jj}d |d < |jj |dd t
|
||¡ |jj|jjksÆt d S )Nr ÚABCÚabcZindex_ar' )Úcolumnsr Zindex_bÚkey0Úkey1Zlvl0Zlvl1©ÚkeysÚnamesr T)Zinplace) r r ÚrandomZrandÚlistr r r r; Z set_namesr r ÚAssertionError)r r
|
r r2 Úexpr; r! r! r" Útest_concat_rename_indexQ s" ý ý z(TestIndexConcat.test_concat_rename_indexc C sP t ddg}t||g|dd}|r*|dkr<|j|jk sLtn|j|jksLtd S )Nr r T©r Úcopy©r r )r r r r> )r r Úusing_copy_on_writeZserÚcombr! r! r" Ú test_concat_copy_index_seriesg s
|
z-TestIndexConcat.test_concat_copy_index_seriesc C sª t ddgddggddgd}t||g|dd }|sT|j|jk sBt|j|jk s¦tnR|d
|
kr~|j|jk slt|j|jks¦tn(|d kr¦|j|jkst|j|jk s¦td S ) Nr r r é r
|
r ©r6 TrA rC )r r6 )r r r r> r6 )r r rD ÚdfrE r! r! r" Ú test_concat_copy_index_framep s z,TestIndexConcat.test_concat_copy_index_framec C s¨ t dddgdd}t dddgd d}t||gdd
|
d }t|jtjsHttddgddgddgg}tj ||d
|
d
|
d t dddg}t dddg}t||gdd d }t|jtjs´ttddgddgddgg}t d¡|_tj ||d
|
d
|
d tddgddgd}tddgddgd}t||gdd
|
d }tddgddgddgddggddgd}tj ||d
|
d
|
d t||gdd
|
d }tddddgddddgg}tj ||d
|
d
|
d d S )Nr r r r r' rG é é r T)r r ©Zcheck_index_typeZcheck_column_typeF)ÚAÚBé é r rN rO rH )
|
r r Ú
|
isinstancer6 ÚpdÚ
|
RangeIndexr> r r r )r Ús1Ús2Úresr? Údf1Údf2r! r! r" Útest_default_index~ s* ( z"TestIndexConcat.test_default_indexc C s$ t tjjdddd dd¡ddddgd}t||gd d
|
}t |jd d
d d
f |¡ t |jd d
dd
f |¡ t||gdd
|
}t |jd d
|¡ t |jdd
|¡ tt tj dd¡ddd d gdt tjjddd d dd ¡ddgdgd d
|
}t||gd d
|
}t |jd d
d d
f |¡ t |jd d
dd
f |¡ t||gdd
|
}t |jd d
|¡ t |jdd
|¡ |jdd
d d
f
|
|jdd
¡}t ||¡ |jdd
d d
f
|
|jdd
¡
|
|jdd
¡}t ||¡ t||gdd
|
}|
|
|¡}t ||¡ d S )Nr é
|
é( )ÚsizerG rN ÚCrH r r. rO é r rL rQ é ) r r r< ÚrandintZreshaper r r ÚilocÚrandnÚ_append)r rI r2 r r! r! r" Útest_dups_index s@
|
þ ÿþù
|
& 6
|
z TestIndexConcat.test_dups_indexN) Ú__name__Ú
|
__module__Ú __qualname__r# ÚpytestÚmarkÚ parametrizer3 r@ rF rJ rZ re r! r! r! r" r s$ ùþ
|
r c @ sL e Zd Zdd Zdd Zdd Zdd Zej d d
|
ge
|
d gd ge d d¡gdgddggdge jddd dgg¡ej dd
|
ge
|
d gd ge d d¡gdgddggdge jddd dgg¡eedddZdd Zd d Zd d Zej d d!d"ggd!d"d"ggg¡d#d$ Zd%d& Zd'd( Zd)d* Zej d+d,d-d.g¡d/d0 Zd1d2 Zd3S )4ÚTestMultiIndexConcatc C sp |}|j }t||gddgdgd}|j jd|j ks8tt |jd |¡ t |jd |¡ |j jdksltd S )Nr r Ú iterationr9 )rm r )r r r; r> r r r Znlevels)r Z multiindex_dataframe_random_dataÚframer r2 r! r! r" Ú test_concat_multiindex_with_keysÏ s z5TestMultiIndexConcat.test_concat_multiindex_with_keysc C s: t jdgtdgdd gd}tdtdi|tjd}t||gddgdgd }t jddgdgtdgddd gd}tdttdd i|tjd}t ||¡ t||d d
gddgdgd }dgd dgd }dgd
|
}ttdttd }tt
|
|||}t j |ddd gd}td|i|tjd}t ||¡ d S ) Nr rK Úlevel1©r; Úcol)r Údtyper Úlevel2r9 rP ) r Ú from_productÚranger r Úint32r r= r r r/ Ú from_tuples) r r rI r2 r rt rp Zno_nameZtuplesr! r! r" Ú/test_concat_multiindex_with_none_in_index_namesÙ s" ÿ
|
zDTestMultiIndexConcat.test_concat_multiindex_with_none_in_index_namesc C sª t tj dd¡}tt d¡t d¡gt t d¡d¡t t d¡d¡gd|_
|
t |j dddgd d
f |j dgd d
f g}|j ddddgd d
f }t ||¡ d S )Nr` r r ©ÚlevelsÚcodesrG rK )r r r< rc r rS rT ÚrepeatÚarangeZtiler r rb r r )r rI rW r? r! r! r" Ú!test_concat_multiindex_rangeindexî s "þ0z6TestMultiIndexConcat.test_concat_multiindex_rangeindexc
|
C sÐ t dgdgg¡}tdg|d}t dgdgg¡}tdg|d}||d}t dd gdgddggddgddgddggd
|
d d gd }tdgdgg|d}tt|d
|
gd }t ||¡ t|d
|
gd } t | |¡ d S ) Nr
|
r r r r r )rU rV rU rV Ztestname)r{ r| r; rq )r ru r r r r r )
|
r Zexample_multiindex1Zexample_dataframe1Zexample_multiindex2Zexample_dataframe2Z example_dictÚexpected_indexr Z result_copyZresult_no_copyr! r! r" Ú(test_concat_multiindex_dfs_with_deepcopyý s
|
ý z=TestMultiIndexConcat.test_concat_multiindex_dfs_with_deepcopyÚmi1_listr
|
r r g @g @r rN rO r) Z2017Z2018)ÚstartÚendZperiodsÚmi2_list)r r
c
|
C s t |¡}t |¡}tt dt|f¡|d}tt dt|f¡|d}|d |d krt |d t|d gddddgddddggd}nDt |d |d t|d t|d gddddgddddggd}tt dt|f¡|d}t d ¡ t ||fdd} W 5 Q R X t
|
|| ¡ d S )Nr rH r rz r r r. ) r ru r r ÚzerosÚlenr= r Úassert_produces_warningr r )
|
r r r
Zmi1Zmi2rX rY Z expected_miÚ expected_dfÚ result_dfr! r! r" Ú*test_concat_with_various_multiindex_dtypes s&
|
|
þþû z?TestMultiIndexConcat.test_concat_with_various_multiindex_dtypesc C s t ddddgidddgd}t|gdgd }dgdddgg}|j}t |¡}t ||¡ |}t ddddgit |¡d}t ||¡ d S )
|
Nrr r
|
r r Ú1Ú2r ÚX©r: )r r r r ru r Úassert_index_equalr )r rI Ú iterablesZ result_indexr r r r! r! r" Útest_concat_multiindex_B s
|
ÿz,TestMultiIndexConcat.test_concat_multiindex_c C s¶ t ddgi}t ddgi}t ddgi}t|||gdddgd}tjtdd |jd
|
d d
f }W 5 Q R X t ddddgitd
|
d d
|
gd }tjtdd |jd
|
}W 5 Q R X t ||¡ t dd d dgi}t dd dgi}t dddgi}t|||gdddgd}tjtdd |jd
|
d d
f }W 5 Q R X t dddddddgdddddddgtdd d dg¡}d d g|j
|
_ tjtdd |jd
|
d d
f }W 5 Q R X t ||¡ d S )Nr( r r r r r r zindexing past lexsort depth©Úmatch)r r )r r r r
|
r r r) r Zaababcd)r
|
r r( ) r r r r r r r r r= Z set_indexr r; )r rX rY Zdf3Zdf_aZout_aZdf_bZout_br! r! r" Ú test_concat_with_key_not_uniqueS sV ÿ ÿ ÿ ÿ ýÿú ÿ z4TestMultiIndexConcat.test_concat_with_key_not_uniquec C sh t ddgidgd}t ddgidgd}d}tjt|d$ t||gddgdddggd W 5 Q R X d S ) NrN r r r r z*Level values not unique: \['x', 'y', 'y'\]r )r: r{ ©r ri ZraisesÚ
|
ValueErrorr )r rX rY Úmsgr! r! r" Ú"test_concat_with_duplicated_levels s
|
z7TestMultiIndexConcat.test_concat_with_duplicated_levelsr{ r r c C sZ t ddgidgd}t ddgidgd}d}tjt|d t||g|d W 5 Q R X d S ) NrN r r r r z+levels supported only when keys is not Noner )r{ r )r r{ rX rY r r! r! r" Ú&test_concat_with_levels_with_none_keys s
|
z;TestMultiIndexConcat.test_concat_with_levels_with_none_keysc C sv t dddgi}t dddgi}t||gddd}t ddgddgd}t ||¡ t dd¡}tj|j|dd d S )
|
Nr
|
r r r T)r r )r
|
r r )Úexact)r r r r rS rT r r )r rX rY r2 r r r! r! r" Ú test_concat_range_index_result s z3TestMultiIndexConcat.test_concat_range_index_resultc C s t dddggtdddgddd}t ddggtddgddd}t||gdd dd
|
}t ddd gddtjggtdddgddd}t ||¡ d S ) Nr r r r Úobject©rs rH TÚouter©r Újoinr ç ð?©r r r r r r r )r rX rY r2 r r! r! r" Ú test_concat_index_keep_dtype s " ÿz1TestMultiIndexConcat.test_concat_index_keep_dtypec C s t dddggtdddg|dd}t ddggtddg|dd}t||gdddd }t ddd
|
gddtjggtdddg|dd}t ||¡ d S ) Nr r r r r rH Tr r r¢ r£ )r Zany_numeric_ea_dtyperX rY r2 r r! r! r" Ú'test_concat_index_keep_dtype_ea_numeric¦ s
|
ÿ þz<TestMultiIndexConcat.test_concat_index_keep_dtype_ea_numericrs ZInt8ZInt16ÚInt32c C s t dddggtdddg|dd}t ddggtddgddd}t||gdd dd
|
}t ddd gddtjggtdddgddd}t ||¡ d S ) Nr r r r r rH r¦ Tr r r¢ r£ )r rs rX rY r2 r r! r! r" Ú test_concat_index_find_common³ s " ÿz2TestMultiIndexConcat.test_concat_index_find_commonc C s¶ t dddg}t ddg}t ddddg}t g td}t||||gddddd }tdgd
|
tjg dgd
|
tjg dtjgd tjgd dg tjg gtd}tj||d d d d S )Nr
|
r r r) r Fr r )r r¡ r r r r TrM )r r r r r r r r )r rU rV Zs3Zs4r2 r r! r! r" Ú(test_concat_axis_1_sort_false_rangeindex¾ s2
|
ÿ üù ÿz=TestMultiIndexConcat.test_concat_axis_1_sort_false_rangeindexN)rf rg rh ro ry r r ri rj rk rv r r~ rS Z
|
date_ranger= r r r r r r r¤ r¥ r§ r¨ r! r! r! r" rl Î sD
|
üþ üþ
|
ÿ .
|
|
|
|
rl )rB r Únumpyr ri Z pandas.errorsr ZpandasrS r r r r r Zpandas._testingZ_testingr r rl r! r! r! r" Ú<module> s <
|
