U f/e @sZddlmZddlZddlmZmZmZmZmZddl m Z ddl m Z GdddZdS))datetimeN) DataFrame DatetimeIndexSeries date_range to_datetime)offsetsc@s<eZdZddZddZddZddZd d Zd d Zd S) TestAsFreqcCs~tdddgdddgd}|t|j}|jjdks:t|jjdksJt|djjdks`t| djjdksztdS) Nz 2012-01-01z 2012-01-02z 2012-01-03)datecolD) rZ set_indexrr indexfreqAssertionErrorZ inferred_freqasfreqZresample)selfdfrJ/tmp/pip-unpacked-wheel-tiezk1ph/pandas/tests/frame/methods/test_asfreq.py%test_asfreq_resample_set_correct_freqsz0TestAsFreq.test_asfreq_resample_set_correct_freqcCsl|t}|d}t|d|d|jddd}|jddd}|g}|d}||k shtdS)NZBMABpadmethod)rrZ BMonthEndtmZassert_almost_equalreindexr)rZdatetime_frameZoffset_monthlyZ rule_monthlyZfilledZ filled_depZ zero_lengthresultrrr test_asfreq$s   zTestAsFreq.test_asfreqcCsptddddgitdddtdddtdddgd}|d}t|jtsNt|dd}t|jtsltdS) Nrr r r i rr)rrr isinstancerrr)rrtsrrrtest_asfreq_datetimeindex5s   z$TestAsFreq.test_asfreq_datetimeindexcCstdddd}ttt||d}td|i}d|jd<|jdd d }|jdd d }d|jd<t |||jdd d }|jdd d }t ||dS) Nz1/1/2016 Z2S)periodsrr#one)z2016-01-01 00:00:08r)Z1Sg"@)rZ fill_value)r) rrnpZarangelenrlocrZfillnarZassert_frame_equalZassert_series_equal)rrngr%rZ actual_dfZ expected_dfZexpected_seriesZ actual_seriesrrrtest_asfreq_fillvalue@s    z TestAsFreq.test_asfreq_fillvaluecCsftddd}|tjd|d}|}dd|jD|_|jddd }|jddd }t||dS) Nz1/1/2000)r(r#cSsg|] }|qSr)r ).0xrrr YszATestAsFreq.test_asfreq_with_date_object_index..Z4HZffillr) rr*randomZrandncopyrrr assert_equal)rframe_or_seriesr-r%Zts2r expectedrrr"test_asfreq_with_date_object_indexTs z-TestAsFreq.test_asfreq_with_date_object_indexcCsVtddddg}|td|d}|t|}|jd|_|d}t||dS) Nz 2021-01-04z 2021-01-02z 2021-01-03z 2021-01-01r#Zinferr) rrangersortedrZ _with_freqrrr5)rr6rr r7rrrtest_asfreq_with_unsorted_index_s  z*TestAsFreq.test_asfreq_with_unsorted_indexN) __name__ __module__ __qualname__rr!r&r.r8r<rrrrr s   r )rZnumpyr*ZpandasrrrrrZpandas._testingZ_testingrZpandas.tseriesrr rrrrs