U
|
ý°db' ã @ sR d dl Zd dlZd dlZd dlmZmZmZmZm Z d dl
|
m Z G dd dZ dS )é N)Ú DataFrameÚ
|
RangeIndexÚSeriesÚconcatÚ
|
date_rangec @ s^ e Zd Zdd Zdd Zej dddg¡ej dg d d
|
d gg¡d d Zej de j
|
e j e j fe j
|
e j e j fde j
|
e j fde je j fde j
|
e j fde je j fddg¡dd Zej dddddddg¡dd Zej ddddddg¡ej d dddddg¡d d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 ZdS )3ÚTestEmptyConcatc C s6 t tj dd¡tdd}|d d
¡ }d|d< |dd
}||||dd
g}t|d|d }|jd
|
d d d dgd}|d d¡|d< d|j dd
df< t
|
||¡ t dt dit ddddd}t }t||gdd}t
|
||¡ t||gdd}t
|
||¡ t||g}t
|
||¡ t||g}t
|
||¡ d S )Né
|
é Zabcd©Úcolumnsé ÚbarZfoor )ÚaxisÚsortÚaÚbÚcÚdÚOÚAi' Z20130101Ús)ZperiodsÚfreq©Úindexé ©r )r ÚnpÚrandomZrandnÚlistÚcopyr ZreindexÚastypeÚlocÚtmÚassert_frame_equalÚranger ) Úselfr ÚdfZdfcopyÚemptyÚframesZ concattedÚexpectedÚresult© r+ ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/reshape/concat/test_empty.pyÚtest_handle_empty_objects s.
|
ÿ z)TestEmptyConcat.test_handle_empty_objectsc C s t dddgdd}t ddd}t||gdd }tdddgtjtjtjgd
|
tdd }t ||¡ t dddgdd}t ddd}t||gd d }t dddg}t ||¡ t dddgdd}t d dd}t||gdd }tdddgtjtjtjgd dd gtdd}t ||¡ d S )Nr é é Úx)ÚnameÚyÚfloat64)r1 Údtyper )r0 r2 r r )r0 r )r r ) r r r r Únanr r" r# Úassert_series_equal)r% Ús1Ús2ÚresÚexpr+ r+ r, Útest_concat_empty_series0 s, þ ýz(TestEmptyConcat.test_concat_empty_seriesÚtzNÚUTCÚvaluesr r. r/ c C sx t g ddj |¡}|r d ntj}t ||d}tt tjgt| ddj |¡|d}t ||gdd}t
|
||¡ d S )NúM8[ns]©r4 )r r r r ) r ÚdtZ tz_localizer r3 r ÚpdZNaTÚlenr r" r# )r% r< r> Úfirstr4 Úsecondr) r* r+ r+ r, Ú!test_concat_empty_series_timelikeM s þÿz1TestEmptyConcat.test_concat_empty_series_timelikezleft,right,expectedzm8[ns]r? )ÚcategoryrG rG )rG ÚobjectrH c C s* t t|dt|dg}|j|ks&td S ©Nr@ )r r r4 ÚAssertionError)r% ÚleftÚrightr) r* r+ r+ r, Ú test_concat_empty_series_dtypes_ s z/TestEmptyConcat.test_concat_empty_series_dtypesr4 r3 Zint8Zuint8Úboolc C sR t |¡}tt|dg}|j|ks(ttt|dt|dg}|j|ksNtd S rI )r r4 r r rJ )r% r4 r* r+ r+ r, Ú0test_concat_empty_series_dtypes_match_roundtripst s
|
|
z@TestEmptyConcat.test_concat_empty_series_dtypes_match_roundtripsÚdtype2c st ||kr d S dd dd fdd}t |¡}t |¡}|||}tt|dt|dgj}|j|ksptd S )Nc S sb | j |j h}t|dddh s6| j dks2|j dkr6dS t|ddh s^| j dksZ|j dkr^dS d S )NÚiÚur ©ÚkindrC ©r4 rP Ztypsr+ r+ r, Úint_result_type s ÿÿÿÿzSTestEmptyConcat.test_concat_empty_series_dtypes_roundtrips.<locals>.int_result_typec S s: | j |j h}t|dddh s6| j dks2|j dkr6dS d S )NÚfrQ rR rS rU r+ r+ r, Úfloat_result_type s ÿÿzUTestEmptyConcat.test_concat_empty_series_dtypes_roundtrips.<locals>.float_result_typec s0 | |}|d k r|S | |}|d k r,|S dS )Nr r+ )r4 rP r* ©rX rV r+ r, Úget_result_type s
|
|
zSTestEmptyConcat.test_concat_empty_series_dtypes_roundtrips.<locals>.get_result_typer@ )r r4 r r rT rJ )r% r4 rP rZ r) r* r+ rY r, Ú*test_concat_empty_series_dtypes_roundtrips s
|
|
|
z:TestEmptyConcat.test_concat_empty_series_dtypes_roundtripsc C s4 t tddttjdttjdgjtjks0td S )Nr? r@ )r r r Úbool_Úint64r4 Úobject_rJ ©r% r+ r+ r, Ú&test_concat_empty_series_dtypes_triple s ÿýÿz6TestEmptyConcat.test_concat_empty_series_dtypes_triplec C s. t tt g ¡ddtddgjdks*td S )NrG r@ r3 )r r r Úarrayr4 rJ r_ r+ r+ r, Ú2test_concat_empty_series_dtype_category_with_arrayµ s ÿýÿzBTestEmptyConcat.test_concat_empty_series_dtype_category_with_arrayc C s¤ t tdd d¡tdd d¡g}|jdks2tt tdd d¡tddg}t tj¡}|j|ksjtt tdd d¡tddg}t d¡}|j|ks td S )Nr3 r@ ZSparsezSparse[float64]rH ) r r r r4 rJ rB Z SparseDtyper r3 )r% r* r) r+ r+ r, Ú&test_concat_empty_series_dtypes_sparse¾ s þÿÿ ÿ
|
z6TestEmptyConcat.test_concat_empty_series_dtypes_sparsec C sT t dddgtjdddgd}t |jd}t||gdd}| t¡}t ||¡ d S )Nr r r. r/ )ZRowZEmptyColZ NumberColr
|
r ) r r r5 r r r rH r" r# )r% Zdf_1Zdf_2r* r) r+ r+ r, Ú!test_concat_empty_df_object_dtypeÓ s
|
|
z1TestEmptyConcat.test_concat_empty_df_object_dtypec C sæ t tdd}|d tj¡|d< |d tj¡|d< |d tj¡|d< t||g}|d jtjksjt |d jtjks~t |d jtjkst t|| tj¡g}|d jtj
|
ksºt |d jtjksÎt |d jtjksât d S )NÚabcr
|
r r r ) r r r r r\ Úint32r3 r r4 rJ r^ )r% r& r* r+ r+ r, Ú"test_concat_empty_dataframe_dtypesÛ s z2TestEmptyConcat.test_concat_empty_dataframe_dtypesc C sr t }t dddgiddgdd}t dg itddd}d|fd|ffD ]&\}}t||gd|d }t ||¡ qFd S )
|
Nr r r. r r] )r r4 ÚinnerÚouter)r Újoin)r r r r" r# )r% Údf_emptyZdf_aZ df_expectedÚhowr) r* r+ r+ r, Ú test_concat_inner_join_emptyë s z,TestEmptyConcat.test_concat_inner_join_emptyc C s\ t dd gdd ggddgd}t dd gdd ggddgd}t||g}|j}t |j|¡ d S )Nr r. r r )Údatar r/ r )r r Zdtypesr" r6 )r% Údf1Údf2r* r) r+ r+ r, Útest_empty_dtype_coerceõ s
|
z'TestEmptyConcat.test_empty_dtype_coercec C s t ddgd}t ddgd}t|||g}t dddgd}t ||¡ t ddgd}t dgd}t||g}t ddgd}t ||¡ d S )Nr r r
|
r )r r r" r# )r% ro rp r* r) Zdf3Zdf4r+ r+ r, Útest_concat_empty_dataframe s z+TestEmptyConcat.test_concat_empty_dataframec C st t dddgdddgd}t ddddgi}t|d d
|d d
g}|d jtjks\t|d jtjksptd S ) Nr r. r/ r r r )r r r )r r r4 r r] rJ r^ )r% ro rp r* r+ r+ r, Ú,test_concat_empty_dataframe_different_dtypes s
|
z<TestEmptyConcat.test_concat_empty_dataframe_different_dtypesc C s^ t dtjg t ¡ di}t dtjdddgt ¡ di}| ¡ }t||g}t ||¡ dS )z<48510 `concat` to an empty EA should maintain type EA dtype.r r@ r r. r/ N)r rB ra Z
|
Int64Dtyper r r" r# )r% rk Zdf_newr) r* r+ r+ r, Útest_concat_to_empty_ea s
|
z'TestEmptyConcat.test_concat_to_empty_ea) Ú__name__Ú
|
__module__Ú __qualname__r- r; ÚpytestÚmarkZ parametrizerF r r\ rf r^ Úfloat32r] rM rO r[ r` rb rc rd rg rm rq rr rs rt r+ r+ r+ r, r sN õþ
|
ÿ
|
þ (
|
r )Únumpyr rx ZpandasrB r r r r r Zpandas._testingZ_testingr" r r+ r+ r+ r, Ú<module> s
|
|
