U f/e:&@sdddlZddlZddlZddlmZmZmZmZm Z m Z ddl m Z GdddZGdddZdS)N) DataFrame DatetimeIndexIndexSeries Timestamp date_rangec@s<eZdZddZddZddZddZd d Zd d Zd S)TestSeriesAppendcCs.|dd|dd}|j|jks*tdS)N)appendnameAssertionError)selfdatetime_seriesresultrK/tmp/pip-unpacked-wheel-tiezk1ph/pandas/tests/series/methods/test_append.pytest_append_preserve_namesz*TestSeriesAppend.test_append_preserve_namec Cs||}|D]H\}}||jkr6|||ksZtq||jkrR|||ksZtqtdqd}tjt|d|j|ddW5QRXdS)Nzorphaned index! Indexes have overlapping values:matchTZverify_integrity)r itemsindexr pytestraises ValueError)r rZ string_seriesZ object_seriesZappended_seriesidxvaluemsgrrr test_appends    zTestSeriesAppend.test_appendcCsH|dd|dd|ddg}|d|dd}t||dS)Nr r)r tmassert_series_equal)r rpiecesrrrrtest_append_many#s"z!TestSeriesAppend.test_append_manyc Cstdddg}tdddg}tddddddgddddddgd}t|||tt||g|tddddddg}tj|j|d d |d d tjtj||gd d |d d d }tjt|d |j|d dW5QRXtjt|d tj||gd dW5QRXdS)Nr!r rrT)Z ignore_index)Zcheck_index_typerrr) rr"r#r pdconcatrrr)r s1s2exprrrrtest_append_duplicates)s*$ z'TestSeriesAppend.test_append_duplicatescCsBtdddg}||g}||f}||}||}t||dS)Nr!r&r')rr r"r#)r sZ list_inputZ tuple_inputexpectedrrrrtest_append_tuples@s   z#TestSeriesAppend.test_append_tuplesc Csltddgddgd}d}tjt|d|j|W5QRXtjt|d|j|gW5QRXdS)Nr!r&r'r()ABzCto_append should be a Series or list/tuple of Series, got DataFramer)rrr TypeErrorr4r )r dfrrrrtest_append_dataframe_raisesKs z-TestSeriesAppend.test_append_dataframe_raisesN) __name__ __module__ __qualname__rrr%r0r3r8rrrrrs  rc@sDeZdZddZddZddZddZd d Zd d Zd dZ dS)!TestSeriesAppendWithDatetimeIndexc Cs tdddd}ttjt||}ttjt|d|d}||}||}tt |j d}t |j |t |j |||}t |||||g}tt |j d}t |||}|} d |_d | _||jd kst|| jdkstdS) N 5/8/2012 1:45r 5Tperiodsfreqr(r*r&r'Zfoobar)rrnprandomrandnlenrr rZtilevaluesr"assert_index_equalrcopyr r ) r rngtsr7r result_dfZex_indexappendedrng1rng2rrrrWs&     z-TestSeriesAppendWithDatetimeIndex.test_appendc Cstddddd}tddddd}tddddd}ttjt||}ttjt|d|d }ttjt||}ttjt|d|d }||}||} t |j |t | j |||} t | |dS) Nr=r r> US/Easternr@rAtz 5/8/2012 2:35r(r* rrrCrDrErFrr r"rHr r rJrOrng3rKr7ts2df2rrLrMrrrtest_append_tzqs   z0TestSeriesAppendWithDatetimeIndex.test_append_tzc Csddlm}tddd|dd}tddd|dd}tdd d|dd}ttjt||}ttjt|d |d }ttjt||}ttjt|d |d }| |} | |} t | j |t | j || |} t | |dS) Nr)timezoner=r r>rPrQrSrTr(r*) Zpytzr[rrrCrDrErFrr r"rHr) r r[rJrOrWrKr7rXrYrrLrMrrrtest_append_tz_explicit_pytzs:    z>TestSeriesAppendWithDatetimeIndex.test_append_tz_explicit_pytzc Cstddddd}tddddd}tddddd}ttjt||}ttjt|d|d }ttjt||}ttjt|d|d }||}||} t |j |t | j |||} t | |dS) Nr=r r>zdateutil/US/EasternrQrSrTr(r*rUrVrrrtest_append_tz_dateutils8   z9TestSeriesAppendWithDatetimeIndex.test_append_tz_dateutilc Cstddddd}tddddd}tdg|d}tdg|d}||}td d gddd }tddg|d}t|||jj|jksttdddd d}tdddd d}tdg|d}tdg|d}||}td d gd dd }tddg|d}t|||j}||jjksttddddd}tdddd d}tdg|d}tdg|d}||}t t dddt dd dg}tddg|d}t||dS)N1/1/2011 01:00r!HrPrQ1/1/2011 02:00r*r&z2011-01-01 01:00z2011-01-01 02:00)rRrAUTCz US/CentralrR) rrr rr"r#rrRr rr) r rNrOser1ser2 ts_result exp_indexr/utcrrrtest_series_append_awaresN       z:TestSeriesAppendWithDatetimeIndex.test_series_append_awarecCstdddd}tddddd}ttjt||d}ttjt||d}||}|jt |jt }|j |st tdddd}t d }ttjt||d}ttjt||d}||}|jt |j}|j |st dS) Nr^r!r_r?r`rPrQr*d) rrrCrDrErFr rZastypeobjectequalsr range)r rNrOrcrdrer2rrrtest_series_append_aware_naives  z@TestSeriesAppendWithDatetimeIndex.test_series_append_aware_naivecCstddddd}tddddd}tdddg|d }td d d g|d }||}td dddddgdd}tdddd d d g|d }t|||jj|jkstdS)Nz1/1/2016 01:00r'r_rPrQz8/1/2016 01:00r!r&r*r z2016-01-01 01:00z2016-01-01 02:00z2016-01-01 03:00z2016-08-01 01:00z2016-08-01 02:00z2016-08-01 03:00rb) rrr rr"r#rrRr )r rNrOrcrdrerfr/rrrtest_series_append_dsts$   z8TestSeriesAppendWithDatetimeIndex.test_series_append_dstN) r9r:r;rrZr\r]rhrmrprrrrr<Vs,r<)ZnumpyrCrZpandasr+rrrrrrZpandas._testingZ_testingr"rr<rrrrs   F