U
|
ý°d ã @ sR d dl Zd dlZd dlmZmZmZmZmZm Z m
|
Z
|
d dl m Z G dd dZdS )é N)Ú DataFrameÚ DatetimeIndexÚIndexÚ
|
MultiIndexÚSeriesÚconcatÚ
|
date_rangec @ sx e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z e
|
j de d¡dfdg¡dd Zdd Zdd ZdS )ÚTestSeriesConcatc C sÞ t ¡ }d|_|d d
|dd
|dd
g}t|}t ||¡ |j|jksTtt|dddgd}| ¡ }ttj |j
|
j dd |_
|
t dddgd
|
d |D ¡t t|¡g}tdddg|j
|
g|d }||_
|
t ||¡ d S ) NÚfooé é r é é )ÚkeyszM8[ns]©Zdtypec S s g | ] }t |qS © )Úlen)Ú.0Úxr r ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/reshape/concat/test_series.pyÚ
|
<listcomp> s z7TestSeriesConcat.test_concat_series.<locals>.<listcomp>)ÚlevelsÚcodes)ÚtmÚmakeTimeSeriesÚnamer Úassert_series_equalÚAssertionErrorÚcopyr ÚnpÚarrayÚindexÚvaluesÚrepeatZaranger r )ÚselfÚtsÚpiecesÚresultÚexpectedZ exp_codesZ exp_indexr r r Útest_concat_series s " *z#TestSeriesConcat.test_concat_seriesc C s6 t dg}t g td}|}t||g}t ||¡ d S )Nr r )r Úobjectr r r )r$ Ús1Ús2r( r' r r r Ú1test_concat_empty_and_non_empty_series_regression% s
|
|
zBTestSeriesConcat.test_concat_empty_and_non_empty_series_regressionc C s t ¡ }|d d
|dd
|dd
g}t|dd}t|j}t ||¡ t|dddgdd}t|dddgd j}t ||¡ d S )
|
Néþÿÿÿr r ©ÚaxisÚAÚBÚC)r r0 ©r! )r r r r ÚTÚassert_frame_equal)r$ r% r& r' r( r r r Útest_concat_series_axis1. s "
|
z)TestSeriesConcat.test_concat_series_axis1c C s t tj d¡dd}t tj d¡dd}t||gdd}t||d}t ||¡ d |_t||gdd}t |j
|
t ddgd d
|
¡ d S ) Nr r1 ©r r2 r r/ ©r1 r2 r r* r ) r r ÚrandomÚrandnr r r r6 r Úassert_index_equalÚcolumnsr ©r$ Úsr, r' r( r r r Ú/test_concat_series_axis1_preserves_series_names; s z@TestSeriesConcat.test_concat_series_axis1_preserves_series_namesc C s t tj d¡dddgdd}t tj d¡ddddgd d}t||gd
|
|d }t||d ddddgd }|rr| ¡ }t ||¡ d S )Né ÚcÚaÚbr1 ©r! r é Údr2 r )r0 Úsortr9 r4 ) r r r: r; r r Z
|
sort_indexr r6 )r$ rH r? r, r' r( r r r Ú%test_concat_series_axis1_with_reindexH s z6TestSeriesConcat.test_concat_series_axis1_with_reindexc C sÈ t dddg}t dddg}t||gdddgd gd
|
}tddgddgddggtddgd d d }t ||¡ t||gdd dgd dgd
|
}tddgddgddggtjd dgd dgdd }t ||¡ d S )Nr r rA rF r é rC rD r1 )r0 r Únamesr8 ©r= )rC r )rD r r2 )rK )r r r r r r6 r Ú from_tuplesr> r r r Ú&test_concat_series_axis1_names_appliedR s ÿ þz7TestSeriesConcat.test_concat_series_axis1_names_appliedc C s~ t dddddd
}ttj t|¡|dd}ttj t|¡|dd}t||gd d
|
d }ttd }t j
|
|j |d
|
d d S )Nz 01-Jan-2013z 01-Jan-2014ZMS)Úfreqr éÿÿÿÿÚvaluerE r T©r0 Z ignore_indexr )Úexact) r r r r: r; r r r Úranger r< r= )r$ Údatesr+ r, r' r( r r r Ú0test_concat_series_axis1_same_names_ignore_indexc s zATestSeriesConcat.test_concat_series_axis1_same_names_ignore_indexz s1name,s2nameé¾ ©é+ r )rW rX c C sR t ddd|d}t ddd|d}t||g}t ddddd}t ||¡ d S ) Nr r )rC rD r8 r rJ )rB rG )rC rD rB rG )r r r r )r$ Zs1nameZs2namer+ r, r' r( r r r Ú&test_concat_series_name_npscalar_tuplem s
|
z7TestSeriesConcat.test_concat_series_name_npscalar_tuplec C sø t ddgdd}t ddg}t ddg}t|||gdd}tddgddgddgddd dgd
|
}t ||¡ t|||gdd d d gd}tddgddgddgdd d d gd
|
}t ||¡ t|||gddd}tddgddgddgd}t ||¡ d S )Nr r r
|
r8 rF r r/ )r
|
r r r rL ÚredÚblueÚyellow)r0 r )r[ r\ r] TrR )r r r )r r r r r6 )r$ Z named_seriesZunnamed_series1Zunnamed_series2r' r( r r r Ú(test_concat_series_partial_columns_namesx s4 ÿ ýþ ÿ z9TestSeriesConcat.test_concat_series_partial_columns_namesc C s0 |dg}t |jd d d
g}t ||¡ d S )Néd rP )r Zilocr Z assert_equal)r$ Zframe_or_seriesÚobjr' r r r Ú&test_concat_series_length_one_reversed s
|
z7TestSeriesConcat.test_concat_series_length_one_reversedN)Ú__name__Ú
|
__module__Ú __qualname__r) r- r7 r@ rI rN rV ÚpytestÚmarkZ parametrizer Úint64rZ r^ ra r r r r r s
|
|
ÿ
|
r )Únumpyr re Zpandasr r r r r r r Zpandas._testingZ_testingr r r r r r Ú<module> s $
|
