U
|
ý°dÁm ã
|
@ s d dl Zd dlZd dlZd dlmZmZmZmZ d dl m
|
Z e d¡e d¡e d¡gZ ej dddej dddej dddgZe d¡e d ¡e d
|
¡gZejd d d ejdd d ejdd d gZdddgdddgdejdgedddgddd ge eeed ZG d d d ZdS ) é N)Ú CategoricalÚ DataFrameÚIndexÚSeriesú
|
2011-01-01ú
|
2011-01-02z
|
2011-01-03ú
|
US/Eastern©Útzú1 daysú2 daysz3 daysú2011-01ÚM©Úfreqú2011-02z2011-03TFé é é gñ?gffffff
|
@ÚXÚYÚZÚaÚbÚc) ÚboolÚint64Úfloat64ÚcategoryÚobjectzdatetime64[ns]údatetime64[ns, US/Eastern]útimedelta64[ns]z period[M]c @ sÞ e Zd ZdZejee ¡ ddd Z e Z
|
dd Z dd Z d d
|
Z d d Zd d Zej dddddg¡dd Zdd Zdd Zdd Zd d Zd d Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Z d*d+ Z d,d- Z d.d/ Z d0S )1ÚTestConcatAppendCommonzE
|
Test common dtype coercion rules between concat and append.
|
)Úparamsc C s |j }|t| fS )N)ÚparamÚ data_dict)ÚselfÚrequestÚkey© r) úed:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/reshape/concat/test_append_common.pyÚitem3 s zTestConcatAppendCommon.itemc C s` |\}}||}t |tr*|j|ks\tn2t |tr\| d¡rN|jdks\tn|j|ks\td S )NZperiodz Period[M])Ú
|
isinstancer ÚdtypeÚAssertionErrorr Ú
|
startswith)r& r+ Zindex_or_seriesÚtypÚvalsÚobjr) r) r* Ú test_dtypes: s
|
|
|
z"TestConcatAppendCommon.test_dtypesc C s² |\}}|}|}|dkrJt t|t| }t t|t| t| }n|| }|| | }t| t|¡}t|} t || ¡ t| t|t|g¡}t|} t || ¡ t|dd}
|
t|dd} |
|
| ¡}t|} t || ¡ t|dd}
|
t|dd} |
|
| ¡}t|dd} t || ¡ tjtdd t| |¡ W 5 Q R X tjtdd t| t||g¡ W 5 Q R X t |j
|
t |dd}t |} tj || dd t j t |t |gdd}tj || dd t |j
|
t |t |gdd}t |} t || ¡ t j t |t |t |gdd}t || ¡ t |dd} t |dd} | j
|
| dd}t |} tj || dd t j | | gdd}tj || dd t |dd} t |dd} | j
|
| dd}t |dd} tj || dd t j | | gdd}tj || dd d
|
}tjt|d t |
|
|¡ W 5 Q R X tjt|d t |
|
t ||g¡ W 5 Q R X tjt|d t t ||g¡ W 5 Q R X tjt|d t t |t ||g¡ W 5 Q R X d S ) Nr Úx)ÚnameÚyzall inputs must be Index)ÚmatchT©Z ignore_index©Zcheck_index_typezPcannot concatenate object of type '.+'; only Series and DataFrame objs are valid)r Úlistr ÚappendÚtmÚassert_index_equalÚpytestZraisesÚ TypeErrorr Ú_appendÚassert_series_equalÚpdÚconcat)r& r+ Útyp1Úvals1Úvals2Úvals3Úexp_dataÚ exp_data3ÚresÚexpÚi1Úi2Ús1Ús2Úmsgr) r) r* Útest_concatlike_same_dtypesF s
|
|
þ ÿ z2TestConcatAppendCommon.test_concatlike_same_dtypesc C sö |\}}|\}}|}d } d }
|
||kr(d S |dks8|dkr<d S |dkrl|dkrl|}
|
t jjdd} |j | ¡ nX|dkr|dkr|}
|
t jjdd} |j | ¡ n(|dks¼|dks¼|dks¼|dkrÄt} t}
|
|| } || | } t| t|¡}t| | d}t ||¡ t| t|t|g¡}t| | d}t ||¡ t
|
|j t
|
|d d
|
}t
|
| |
|
d}tj ||d d t jt
|
|t
|
|gd d
|
}tj ||d d t
|
|j t
|
|t
|
|gd d
|
}t
|
| |
|
d}t ||¡ t jt
|
|t
|
|t
|
|gd d
|
}t ||¡ d S ) Nr r )r r zGH#39187 casting to object)Úreasonr r! ©r- Tr8 r9 )r> ÚmarkZxfailÚnodeZ
|
add_markerr r r; r< r= r r@ rA rB rC )r& r+ Úitem2r' rD rE Ztyp2rF rG Zexp_index_dtypeZexp_series_dtyperT rH rI rJ rK r) r) r* Ú test_concatlike_dtypes_coercion° s` ÿþýü þz6TestConcatAppendCommon.test_concatlike_dtypes_coercionc C sB t ddg¡}t ddg¡}tt d¡t d¡t d¡t d¡g}| |¡}t ||¡ t |d t jsnt
|
t |d t jst
|
t |}t |}| |¡}t |t |ddddgd¡ t |jd t jsÎt
|
t |jd t jsät
|
t ||g¡}t |t |ddddgd¡ t |jd t js&t
|
t |jd t js>t
|
d S ) Nr r r r r éÿÿÿÿr ©Úindex)rB Ú DatetimeIndexÚTimedeltaIndexr Ú TimestampÚ Timedeltar; r< r= r, r. r r@ rA ZilocrC )r& ZdtiÚtdirK rJ ZdtsÚtdsr) r) r* Ú.test_concatlike_common_coerce_to_pandas_object s. üÿ
|
|
zETestConcatAppendCommon.test_concatlike_common_coerce_to_pandas_objectc C s¸ |}t jddg|d}t jddg|d}t jddddg|d}| |¡}t ||¡ t|}t|}| |¡}t |t|ddddgd¡ t ||g¡}t |t|ddddgd¡ d S ) Nr r r ú
|
2012-01-01ú
|
2012-01-02r r rY ) rB r[ r; r< r= r r@ rA rC ) r& Útz_aware_fixturer
|
Údti1Údti2rK rJ Údts1Údts2r) r) r* Útest_concatlike_datetimetz! s
|
ÿ
|
|
z1TestConcatAppendCommon.test_concatlike_datetimetzr
|
ÚUTCr z
|
Asia/TokyoZEST5EDTc C s t jddd|d}t jddg|d}td|d d
|
gd }td|d d
|
gd }t jdd dddg|d}td|d d
|
gd }t | |¡|¡ t t ||g¡|¡ d S ) Nz
|
2014-07-15z
|
2014-07-17ÚD)ÚstartÚendr r
|
z
|
2014-07-11z
|
2014-07-21r r ÚAÚB)rZ Úcolumnsz
|
2014-07-16)rB Z
|
date_ranger[ r r< Úassert_frame_equalr@ rC )r& r
|
Zix1Zix2Zdf1Zdf2Zexp_idxrK r) r) r* Ú test_concatlike_datetimetz_short6 s þz7TestConcatAppendCommon.test_concatlike_datetimetz_shortc C s |}t jddg|d}t ddg¡}tt jd|dt jd|dt d¡t d¡gtd}| |¡}t ||¡ t|}t|}| |¡}t
|
|t|ddddgd ¡ t ||g¡}t
|
|t|ddddgd ¡ t jddgd
|
d} tt jd|dt jd|dt jdd
|
dt jdd
|
dgtd}| | ¡}t ||¡ t|}t| }
|
| |
|
¡}t
|
|t|ddddgd ¡ t ||
|
g¡}t
|
|t|ddddgd ¡ d S ) Nr r r rb rc rS r r rY z
|
US/Pacific) rB r[ r r] r r; r< r= r r@ rA rC ) r& rd r
|
re rf rK rJ rg rh Zdti3Zdts3r) r) r* Ú$test_concatlike_datetimetz_to_objectG sH üù
|
|
|
üù
|
|
|
z;TestConcatAppendCommon.test_concatlike_datetimetz_to_objectc C s´ t jddgdd}t jddgdd}t jddddgdd}| |¡}t ||¡ t|}t|}| |¡}t |t|ddddgd ¡ t ||g¡}t |t|ddddgd ¡ d S )
|
Nr r r r z2012-01z2012-02r r rY ) rB Ú PeriodIndexr; r< r= r r@ rA rC ©r& Úpi1Zpi2rK rJ Úps1Zps2r) r) r* Ú test_concatlike_common_period| s
|
|
z4TestConcatAppendCommon.test_concatlike_common_periodc C sÚ t jddgdd}t jddgdd}tt jdddt jdddt jdddt jdddgtd}| |¡}t ||¡ t|}t|}| |¡}t
|
|t|d d
|
d d
|
gd ¡ t ||g¡}t
|
|t|d d
|
d d
|
gd ¡ d S ) Nr r r r rb z
|
2012-02-01rk rS r r rY ) rB rt r ÚPeriodr r; r< r= r r@ rA rC ru r) r) r* Ú1test_concatlike_common_period_diff_freq_to_object s$ üù
|
|
|
zHTestConcatAppendCommon.test_concatlike_common_period_diff_freq_to_objectc C sx t jddgdd}t ddg¡}tt jdddt jdddt d¡t d¡gtd}| |¡}t ||¡ t
|
|}t
|
|}| |¡}t |t
|
|dd dd gd
|
¡ t ||g¡}t |t
|
|dd dd gd
|
¡ tt d¡t d¡t jdddt jdddgtd}| |¡}t ||¡ t
|
|}t
|
|}| |¡}t |t
|
|dd dd gd
|
¡ t ||g¡}t |t
|
|dd dd gd
|
¡ d S ) Nr r r r r r rS r r rY )rB rt r\ r ry r^ r r; r< r= r r@ rA rC )r& rv r_ rK rJ rw r` r) r) r* Ú0test_concatlike_common_period_mixed_dt_to_object¨ sD üù
|
|
|
üù
|
|
|
zGTestConcatAppendCommon.test_concatlike_common_period_mixed_dt_to_objectc C sN t ddtjgdd}t dddgdd}t ddtjdddgdd}t tj||gdd|¡ t |j|dd|¡ t ddgdd}t ddgdd}t ddddg}t tj||gdd|¡ t |j|dd|¡ t dd tjgdd}t tjdddgdd}t dd tjtjdddgtjd}t tj||gdd|¡ t |j|dd|¡ d S )
|
Nr r r rS Tr8 r é
|
é ) r ÚnpÚnanr< rA rB rC r@ r ©r& rN rO rK r) r) r* Útest_concat_categoricalØ s z.TestConcatAppendCommon.test_concat_categoricalc C s| t tdddgdddgd}t tdddgdddgd}tj||gdd}t tddddddgdddgd}t ||¡ d S )Nr r r ©Ú
|
categoriesTr8 )r r rB rC r< rA )r& r r ÚresultÚexpectedr) r) r* Ú6test_union_categorical_same_categories_different_orderó s ÿzMTestConcatAppendCommon.test_union_categorical_same_categories_different_orderc C sv t ddtjgdd}t dddg}t ddtjdddgtjd}t tj||gdd|¡ t |j|dd|¡ t dddddtjgtjd}t tj||gdd|¡ t |j|dd|¡ t ddgdd}t ddg}t ddddg}t tj||gdd|¡ t |j|dd|¡ t ddddg}t tj||gdd|¡ t |j|dd|¡ t dd tjgdd}t dddg}t dd tjdddgtjd}t tj||gdd|¡ t |j|dd|¡ t ddddd tjgtjd}t tj||gdd|¡ t |j|dd|¡ t dd tjgdd}t d
|
d d g}t dd tjd
|
d d g}t tj||gdd|¡ t |j|dd|¡ t d
|
d d dd tjg}t tj||gdd|¡ t |j|dd|¡ t dd gdd}t tjtjtjg}t dd tjtjtjg}t tj||gdd|¡ t |j|dd|¡ t tjtjtjdd g}t tj||gdd|¡ t |j|dd|¡ d S ) Nr r r rS Tr8 r r| r} r r r ) r r~ r r r< rA rB rC r@ r r) r) r* Ú test_concat_categorical_coercioný sP z7TestConcatAppendCommon.test_concat_categorical_coercionc C s t ddtjgdd}t dddgdd}t ddddtjg}t ddtjdddddddtjg dd}t tj|||gdd|¡ t |j||gdd|¡ t ddddtjddtjdddg dd}t tj|||gdd|¡ t |j||gdd|¡ t dd d
|
gdd}t ddd gdd}t dd dg}t dd d
|
ddd dd dg }t tj|||gdd|¡ t |j||gdd|¡ t dd ddd d
|
ddd g }t tj|||gdd|¡ t |j||gdd|¡ t dd d
|
gdd}t ddd gdd}t d d dg}t dd d
|
ddd d d dg }t tj|||gdd|¡ t |j||gdd|¡ t d d ddd d
|
ddd g }t tj|||gdd|¡ t |j||gdd|¡ d S )Nr r r rS ÚfloatTr8 é é é r r| r} é )r r~ r r< rA rB rC r@ )r& rN rO Ús3rK r) r) r* Ú&test_concat_categorical_3elem_coercion= s6 & & z=TestConcatAppendCommon.test_concat_categorical_3elem_coercionc C s< t ddgdd}t ddgdd}t ddg}t ddgdd}t dtjg}t dddgdd}t dddddddddtjdddg }tj||||||gdd}t ||¡ |j|||||gdd}t ||¡ t ddddtjddddddddg }tj||||||gdd}t ||¡ |j|||||gdd}t ||¡ d S ) Nr r r rS r r Tr8 )r r~ r rB rC r< rA r@ ) r& rN rO r Zs4Zs5Zs6rK rJ r) r) r* Ú&test_concat_categorical_multi_coerciong s $ $ z=TestConcatAppendCommon.test_concat_categorical_multi_coercionc C sÜ t tddtjgdd}t tdddgdd}t tddtjdddgdd}t tj||gdd|¡ t |j|dd|¡ t tddtjdddddtjg dd}t tj|||gdd|¡ t |j||gdd|¡ d S )Nr r T)Zorderedr8 ) r r r~ r r< rA rB rC r@ r r) r) r* Ú test_concat_categorical_ordered~ s & z6TestConcatAppendCommon.test_concat_categorical_orderedc C sô t tjtjtjgtjddd}t tjdg}t tjtjtjdg}t tj||gdd|¡ t |j |dd|¡ t dtjgdd}t tjtjg}t dtjtjtjgdd}t tj||gdd|¡ t |j |dd|¡ t tjtjgdd}t tjtjg}t tjtjtjtjg}t tj||gdd|¡ t |j |dd|¡ t tj||gdd|¡ t |j |dd|¡ t tjtjgdd}t tjtjgdd}t tjtjtjtjgdd}t tj||gdd|¡ t |j |dd|¡ d S )NrS r r Tr8 r )
|
r r~ Úarrayr r r< rA rB rC r@ r r) r) r* Ú$test_concat_categorical_coercion_nan s, z;TestConcatAppendCommon.test_concat_categorical_coercion_nanc C sÌ t g dd}t ddgdd}t tj||gdd|¡ t |j|dd|¡ t tj||gdd|¡ t |j|dd|¡ t g dd}t g dd}t tj||gdd|¡ t |j|dd|¡ t g dd}t g dd}t tj||gdd|¡ t |j|dd|¡ t tj||gdd|¡ t |j|dd|¡ t g dd}t tjtjg}t tjtjg}t tj||gdd|¡ t |j|dd|¡ t tj||gdd|¡ t |j|dd|¡ d S )Nr rS r r Tr8 r )r r< rA rB rC r@ r~ r r r) r) r* Ú test_concat_categorical_empty² s. z4TestConcatAppendCommon.test_concat_categorical_emptyc C s t ddgddgd}ddg}t||d}t ddddgddgd}ddddg}t||dtddddgd}t t ||g¡|¡ t | |¡|¡ t ddgddd gd}ddg}t||d} tj|| gd
|
d }
|
ttd ddddgd}t |
|
|¡ |j| d
|
d }
|
t |
|
|¡ d S ) Nr r r r r )Zcatsr1 r rY r Tr8 Zabab) r r r r< rq rB rC r@ r: ) r& Úcatr1 ZdfZcat2rF rK Zcat3rG Zdf_different_categoriesrJ r) r) r* Ú test_categorical_concat_appendØ s z5TestConcatAppendCommon.test_categorical_concat_appendN) Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r> ZfixtureÚsortedr% Úkeysr+ rV r3 rQ rW ra ri rT Z parametrizerr rs rx rz r{ r r r r r r r r r r) r) r) r* r" . s0
|
jQ
|
50
|
@*&&r" )Únumpyr~ r> ZpandasrB r r r r Zpandas._testingZ_testingr< r] Zdt_dataZtz_datar^ Ztd_datary Z period_datar r% r" r) r) r) r* Ú<module> s> ý ýý ý
|
÷
|
