U
|
ý°dK ã @ sª d dl Zd dl m Z d dlZd dlZd dlZd dlZd dlmZm Z m
|
Z
|
m Z m Z m Z mZmZmZ d dlmZ G dd dZG dd dZG dd d Zd
|
d Zd d ZdS )é N)Údatetime) Ú DataFrameÚ DatetimeIndexÚIndexÚ
|
MultiIndexÚSeriesÚ TimestampÚconcatÚ
|
date_rangeÚ to_timedeltac @ sò e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Ze j
|
d ddg¡dd Z e j
|
dddg¡e j
|
dddg¡e j
|
de jedg¡dd Ze j
|
dddg¡e j
|
dddg¡dd Ze j
|
dddg¡e j
|
dddg¡dd ZdS ) ÚTestDatetimeConcatc C sd t ddd}td|i}t||g}|jd d
d |k ¡ sBt|jdd
d |k ¡ s`td S )Nz1/1/2000é
|
)ÚperiodsÚtime)r
|
r r ÚilocÚallÚAssertionError)ÚselfÚrngÚdfÚresult© r úad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/reshape/concat/test_datetimes.pyÚ test_concat_datetime64_block s
|
z/TestDatetimeConcat.test_concat_datetime64_blockc C sr g }| tddddg¡ | tddddg¡ tj|ddgd}tdd d
|
d }t|td
|
d }t||g d S ) NiÚ é é ÚhiÚdateÚtest©Úcolumnsz2000/1/1ÚDr )ÚstartÚfreqr )r r )Úappendr r Z from_recordsr
|
Úranger )r ÚrowsZdf2_objÚindÚdf1r r r Ú%test_concat_datetime_datetime64_frame! s z8TestDatetimeConcat.test_concat_datetime_datetime64_framec
|
C s¸ t ddddd}t |d |d dd}td d
|
d dgi|d }td d
|
d dgi|d }t||gd
|
d}tdddgdd d¡ d¡}td
|
d
|
gd d gddgg|d d gd}t ||¡ t ddddd}td d
|
d dgi|d } t|| gd
|
d}tddddddg}ttjd
|
gtjd gtjdgd
|
tjgd tjgdtjgg|d d gd}t ||¡ t| d¡
|
¡ | d¡
|
¡ gd d }td
|
d dgtjgd tjgd d
|
d dg d | |¡d }t ||¡ d S ) Nú
|
2011-01-01é ÚHz Europe/Paris©r r# Útzr éÿÿÿÿ)r" Úendr# Úar r ©ÚindexÚb©Zaxisz2011-01-01 00:00:00+01:00z2011-01-01 01:00:00+01:00z2011-01-01 02:00:00+01:00©r# ÚUTC©r3 r z
|
Asia/Tokyoz2010-12-31 15:00:00+00:00z2010-12-31 16:00:00+00:00z2010-12-31 17:00:00+00:00z2010-12-31 23:00:00+00:00z2011-01-01 00:00:00+00:00z2011-01-01 01:00:00+00:00T)Úsort©r1 r4 ) r
|
r r r Z
|
tz_convertÚtmÚassert_frame_equalÚnpÚnanZresampleZmeanr$ )
|
r Zidx1Zidx2r( Údf2r Úexp_idxÚexpectedZidx3Údf3r r r Ú test_concat_datetime_timezone/ sj ýúø ÷ÿ ÿ úÿ úö $,þz0TestDatetimeConcat.test_concat_datetime_timezonec C s¼ t ddddd}ttd}t||d}t|d d
|dd
g}t ||¡ t|dd
|d d
g}t|dd
|d d
|dd
|d d
¡d}d |jj _
|
t ||¡ d S )Nz 01-Jan-2013éd Z50Lr7 r- r2 é2 ) r
|
Úlistr% r r r; r< r$ r3 Ú_datar# )r ZdrÚdatarA r r r r Ú test_concat_datetimeindex_freqp s 6
|
z1TestDatetimeConcat.test_concat_datetimeindex_freqc C s t t ddd¡t ddd¡t ddd¡gdd}tddgtjdd g|d d
|
gd d gd d}tddgtjdd g|d d d
gd d gd d}tjdd d g|gd d gd }|jd jtks¾t t
|
ddgdt j gt j dgg|d}t ||gdd}t ||¡ d S )NiÝ r éÞ iß Úobject©Údtyper1 r4 r r/ ÚfirstÚsecond)Únamesr2 r5 )r Údtr r r Ú from_arraysÚlevelsrM rK r r r= r> r r; r< )r ÚidxÚsÚs2ÚmirA r r r r Ú,test_concat_multiindex_datetime_object_index s> &þ
|
þûþ
|
þþ þþz?TestDatetimeConcat.test_concat_multiindex_datetime_object_indexc C s< t tddddd}t tjddgdd }t |d |d tjtjg}t||gd
|
d }t ||¡ t tjtd dd }t||gd
|
d }t ||¡ t tdddd }t tdddd }tj|d d
< t |d |d tjtjg}t||gd
|
d }t ||¡ tj|d d
< t tjtd dd }t||gd
|
d }t ||¡ d S )Nú20151124 08:00ú20151124 09:00Ú1hú
|
US/Eastern©r# r. r r zdatetime64[ns, US/Eastern])r3 rM T©Ú ignore_indexé r6 z20151124 10:00z20151124 11:00zdatetime64[ns])r r
|
ÚpdÚNaTr r; Úassert_series_equalr% ©r ÚxÚyrA r r r r Útest_concat_NaT_series¤ s( ÿ z)TestDatetimeConcat.test_concat_NaT_seriesr. Nr7 c s t tjgtjgg}| fdd¡}t td dgtd dggddgd}t tjtjtd dtd dg}t||gd d
|
}t ||¡ d S ) Nc s | j ¡S ©N©rQ Ú tz_localize©re ©r. r r Ú<lambda>È ó z?TestDatetimeConcat.test_concat_NaT_dataframes.<locals>.<lambda>ú
|
2015/01/01rl ú
|
2016/01/01r r+ r2 r r5 )r ra rb Úapplyr r r; r< )r r. rN rO rA r r rl r Útest_concat_NaT_dataframesà s þ
|
|
üÿ z-TestDatetimeConcat.test_concat_NaT_dataframesÚtz1Útz2rU Z20150101c s t tjgtjgg fdd¡}t |g fdd¡}t||gdd}t ttjtj|gdddgd}| fdd¡} kr| t¡}t ||¡ d S ) Nc s | j ¡S rh ri rk )rs r r rm à rn zNTestDatetimeConcat.test_concat_NaT_dataframes_all_NaT_axis_0.<locals>.<lambda>c s | j ¡S rh ri rk ©rt r r rm á rn r r5 r r2 c s | j ¡S rh ri rk ru r r rm å rn )
|
r ra rb rq r r ÚastyperK r; r< )r rs rt rU rN rO r rA r )rs rt r Ú)test_concat_NaT_dataframes_all_NaT_axis_0Ù s "
|
z<TestDatetimeConcat.test_concat_NaT_dataframes_all_NaT_axis_0c C s t ttjtjgj |¡}t ttjgj |¡dgd}t ttjtjgj |¡ttjtjgj |¡d}t||gdd}t ||¡ d S )Nr r )r r r5 ) r r ra rb rQ rj r r; r< ©r rs rt rN rO rA r r r r Ú)test_concat_NaT_dataframes_all_NaT_axis_1ë s þÿz<TestDatetimeConcat.test_concat_NaT_dataframes_all_NaT_axis_1c C s t tjtjgj |¡}ttd|dgtd|dggddgd}ttjtjtd|dtd|dg}||krv| t¡}t ||g}t
|
||¡ d S )Nro rl rp r r+ r2 ) r ra rb rQ rj r r rv rK r r; r< rx r r r Ú(test_concat_NaT_series_dataframe_all_NaTû s$ þû
|
|
üÿ
|
z;TestDatetimeConcat.test_concat_NaT_series_dataframe_all_NaT)Ú__name__Ú
|
__module__Ú __qualname__r r) rC rI rX rg ÚpytestÚmarkÚ parametrizerr ra rb r rw ry rz r r r r r s$ A%
|
r c @ s| e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z e
|
j dde
|
j eje
|
j jdddg¡dd Zdd ZdS )ÚTestTimezoneConcatc C s t tddddd}t tdd}t |d |d |d |d gd
|
d }t||gd d }t ||¡ t tddddd}t ddg}t |d |d |d |d gd
|
d }t||gd d }t ||¡ ttdd d gg}|d j d¡|d< ttdd dgg}|d j d¡|d< t||g}|d j dks2t
|
ttdd d gg}|d j d¡|d< ttdd dgg}|d j d¡|d< t||g}|d j dks¢t
|
ttdd d gtdd dgg}|d j d¡|d< ttdd dgg}|d j d¡|d< t||g}|d j dks t
|
ttdd d gg}|d j d¡|d< ttdd dgtdd dgg}|d j d¡|d< t||g}|d j dkst
|
d S )NrY rZ r[ r7 r] ú
|
2012-01-01z
|
2012-01-02r r rK rL Tr^ r1 r4 ià r zdatetime64[ns, UTC]z Europe/Londonz datetime64[ns, Europe/London]r+ ) r r
|
r r; rc r r rQ rj rM r )r re rf rA r rN rO r r r Útest_concat_tz_series sD $ $ z(TestTimezoneConcat.test_concat_tz_seriesc C s t dtj ¡ dt dtj ¡ dg}t dtj ¡ dt dtj ¡ dg}tt|t|gdd}t |t|| ¡ |jdkst d S ) Nr* rl ú
|
2011-02-01r z
|
2012-02-01Tr^ zdatetime64[ns, tzlocal()])
|
r Údateutilr. Ztzlocalr r r; rc rM r ©r re rf r r r r Ú test_concat_tz_series_tzlocalR s þþz0TestTimezoneConcat.test_concat_tz_series_tzlocalc C s¬ t dddt dddg}t d¡t d¡g}tt|t|gdd}t |t|| d d
|
¡ tjd d d tjdd d g}tt|t|gdd}t |t|| d d
|
¡ d S )Nr* r\ rl r z1 dayz2 dayTr^ rK rL z2011-03ÚMr6 z2011-04)r ra Z Timedeltar r r; rc ZPeriodr r r r Ú'test_concat_tz_series_with_datetimelikea s
|
|
þ z:TestTimezoneConcat.test_concat_tz_series_with_datetimelikec C sR t tdddtddddtdd}t|j ¡ |j ¡ gd d
|
}t ||¡ d S ) NZ20130102r\ rl Z20130603ZCET©ÚAÚBé r2 r r5 ) r r r% r r Zto_framer r; r< )r r? rB r r r Útest_concat_tz_framep s
|
|
þû z'TestTimezoneConcat.test_concat_tz_framec C sê t dd d}t ddd}t ddd}td|gi}td|gi}td|gi}t||gjdd}td||gitd}t ||¡ t||gjdd}td||gitd}t ||¡ t||gjdd}td||gi}t ||¡ d S ) Nú
|
2015-01-01rl r7 ÚESTr T)ZdroprL )r r r Z reset_indexrK r; r< ) r Úts1Úts2Úts3r( r? rB ÚresultsrA r r r Útest_concat_multiple_tzs} s z+TestTimezoneConcat.test_concat_multiple_tzsc C sê t tdddtdddtdddgdddgdddgdd d
|
gd }|d d d ¡|d < | d dg¡}tdddgd dd d}tdddgd dd}t ||g¡}t dddgd dd d
|
gd d|ddgd}t||g}t
|
||¡ d S )NrJ r r r+ r r ÚCr` r é )rQ r4 ÚcÚdrQ c S s t | ddS )Nú
|
US/Pacificrl )r )r r r r rm ¢ rn zCTestTimezoneConcat.test_concat_multiindex_with_tz.<locals>.<lambda>r4 z
|
2014-01-01z
|
2014-01-02z
|
2014-01-03r )r. Úname)r )r r r r r8 ) r r rq Z set_indexr r r rR r r; r< )r r Zexp_idx1Zexp_idx2r@ rA r r r r Ú test_concat_multiindex_with_tz s4
|
|
|
ýøÿ ÿ ÿ z1TestTimezoneConcat.test_concat_multiindex_with_tzc C sz t ddg¡ d¡}td|i}t||d}t||gddd}tt|t| t jt jgt| d}t ||¡ d S )Nr r r7 r r T)r9 r_ ) ra Ú to_datetimerj r r rF rb r; r< )r Útsr1 r4 r rA r r r Útest_concat_tz_not_aligned± s $ÿz-TestTimezoneConcat.test_concat_tz_not_alignedÚt1r z*GH23037 incorrect dtype when concatenating)Úreason)Zmarksc C sx t |dd}t ddd}t ddd}t||gg}t|gg}t||g}t||g|tjggddgd}t ||¡ d S )Nr7 rl r r r2 )r r r ra rb r; r< ) r r r r r r( r? r rA r r r Útest_concat_tz_NaT¼ s z%TestTimezoneConcat.test_concat_tz_NaTc C s> t ttddddt g}ttdddd}t ||¡ d S )NÚ2000r r7 )r r. )r r r
|
r; r< )r r rA r r r Útest_concat_tz_with_empty× s
|
ÿz,TestTimezoneConcat.test_concat_tz_with_emptyN)r{ r| r} r r r r r r r r~ r r Úparamra rb Zxfailr¢ r¤ r r r r r s( 8 ÿþþþ
|
r c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestPeriodConcatc C sp t tjddgdd}t tjddgdd}t |d |d |d |d gd d
|
}t||gd d }t ||¡ d S ) Nú
|
2015-11-01ú
|
2015-12-01r! r6 ú
|
2015-10-01ú
|
2016-01-01r r z Period[D]rL Tr^ )r ra Ú PeriodIndexr r; rc rd r r r Útest_concat_period_seriesá s
|
$z*TestPeriodConcat.test_concat_period_seriesc C s~ t tjddgdd}t tjddgdd}t |d |d |d |d gd
|
d }t||gd d }t ||¡ |jd
|
ksztd S )Nr§ r¨ r! r6 r© rª r r r rK rL Tr^ )r ra r« r r; rc rM r rd r r r Ú'test_concat_period_multiple_freq_seriesè s $ z8TestPeriodConcat.test_concat_period_multiple_freq_seriesc C sd t tjddgdd}t tjddgdd}t |d |d |d |d gdd }t||gd
|
d }t ||¡ |jdksztt tjddgdd}t tddg}t |d |d |d |d gdd }t||gd
|
d }t ||¡ |jdksîtt tjddgdd}t d d g}t |d |d |d |d gdd }t||gd
|
d }t ||¡ |jdks`td S )Nr§ r¨ r! r6 r r r rK rL Tr^ r r ) r ra r« r r; rc rM r r rd r r r Ú test_concat_period_other_seriesð s$ $ $ $ z0TestPeriodConcat.test_concat_period_other_seriesN)r{ r| r} r¬ r r® r r r r r¦ à s r¦ c C sZ t t d¡dd} td| i}t||g}t |jd d
|¡ t |jdd
|¡ d S )Nr rU )Úunitr )r r= Zaranger r r; r< r )r r r r r r Ú test_concat_timedelta64_block s
|
r° c C s t ddit dtjfg¡d} t ddit dtjfdtjfg¡d}t| |gdd}t dtjgdd t dtjfdtjfg¡}t ||¡ d S )
|
Nr1 r r2 r4 r r r5 g ð?r: )
|
r r Ú from_tuplesra rb r r= r> r; r< )ÚleftÚrightr rA r r r Ú#test_concat_multiindex_datetime_nat s ÿ ÿr´ )r rQ r
Únumpyr= r~ Zpandasra r r r r r r r r
|
r Zpandas._testingZ_testingr; r r r¦ r° r´ r r r r Ú<module> s , H(
|
