U
|
ý°dp ã @ sf d dl m Z d dlZd dlZd dlmZmZmZmZm Z m
|
Z
|
d dl m Z d dlmZ G dd dZdS )é )ÚdatetimeN)Ú DataFrameÚ DatetimeIndexÚSeriesÚ
|
date_rangeÚ period_rangeÚ to_datetime)Úoffsetsc @ sª e Zd Zejddddgddd Zdd Zd
|
d Zej d d dg¡dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'S )(Ú
|
TestAsFreqÚsÚmsÚusÚns)Úparamsc C s |j S )N)Úparam)ÚselfÚrequest© r ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_asfreq.pyÚunit s zTestAsFreq.unitc C s( |dddgt tdddtdddtddd gd
|
d d }| d ¡}| d
|
¡}t ||¡ |jd dd}| d
|
¡}t ||¡ | t ¡ ¡}| t ¡ ¡}t ||¡ |d d
d¡}t|dksÆt ||k sÒt |t
|
kr$|jddd}| ¡ ¡ }t
|
ddddgddddgdd ¡ }t ||¡ d S )Ng g ð?g @iÙ é
|
é é é é ÚBM©Úfreq©ÚindexÚBÚpad©Úmethodr ÚMÚDéÿÿÿÿ)Ú
|
fill_valueé< é g ð¿Úcount)r Úname)r r ÚasfreqÚtmÚ assert_equalr ZBDayÚ BMonthEndÚlenÚAssertionErrorr Z value_countsZ
|
sort_indexÚassert_series_equal)r Úframe_or_seriesÚtsZdaily_tsZ
|
monthly_tsÚresultÚexpectedr r r Ú test_asfreq2 s>
|
|
|
ýúþ
|
|
|
|
|
|
ÿ
|
zTestAsFreq.test_asfreq2c C sH t dg}||td d¡}|dg| ¡ d d¡}t |j|j¡ d S )Nz2016-09-29 11:00©r ZdtypeÚHé r )r Úobjectr, Úcopyr- Úassert_index_equalr )r r3 r r6 r5 r r r Ú test_asfreq_datetimeindex_empty= s
|
z*TestAsFreq.test_asfreq_datetimeindex_emptyÚtzz
|
US/Easternzdateutil/US/Easternc C s6 t ddd|d}|tj t|¡|d}| d¡ d S )Nz
|
2011-12-01z
|
2012-07-20r% )r r? r ÚT)r ÚnpÚrandomÚrandnr0 r, )r r? r3 ZdrÚobjr r r Útest_tz_aware_asfreq_smokeD s z%TestAsFreq.test_tz_aware_asfreq_smokec C st t ddd}t ddd}tj dd¡}t||d}t||d}|tkrV|d }|d }|jdd d
|
}t ||¡ d S ) Nz1/1/2000 09:30é ©Úperiodsú1/1/2000r: r r r% T)Ú normalize) r rA rB rC r r r, r- r. )r r3 ÚrngZnormÚvalsrD r6 r5 r r r Útest_asfreq_normalizeM s z TestAsFreq.test_asfreq_normalizec C s` d}t dd|d}tttddg|d}t ||¡}||jjksFt|| d¡jjks\td S )NÚbarZ20130101rF )rH r+ Zfoo)Úcolumnsr Z10D)
|
r r ÚlistÚranger- Úget_objr r+ r1 r, )r r3 Z
|
index_namer rD r r r Útest_asfreq_keep_index_name\ s z&TestAsFreq.test_asfreq_keep_index_namec C s´ t dddd}ttj t|d¡|d}t ||¡}|jddd }|jddd }t|t|ksdt t
|
|j |¡ |jdd
|
d }|jdd
|
d }t|t|ks¢t t
|
|j |¡ d S ) NÚAz1/1/2001z
|
12/31/2010)r ÚstartÚendr: r r% rV )ÚhowrU ) r r rA rB rC r0 r- rR r, r1 r= r )r r3 r rD r5 Z exp_indexr r r Útest_asfreq_tsf s zTestAsFreq.test_asfreq_tsc C s t dddg}tddddgi|d}t ||¡}|jjd ks@t|jjd ksPt| d ¡jjd ksft| d ¡ ¡ jjd kstd S )
|
Nz
|
2012-01-01z
|
2012-01-02z
|
2012-01-03Úcolr) é r: r r% )
|
r r r- rR r r r1 Z inferred_freqr, Zresample)r r3 ZdtirD r r r Ú%test_asfreq_resample_set_correct_frequ s z0TestAsFreq.test_asfreq_resample_set_correct_freqc C s$ | g ¡}| d¡}||k s td S )Nr )Úreindexr, r1 )r Údatetime_frameZ zero_lengthr5 r r r Útest_asfreq_empty s
|
|
z TestAsFreq.test_asfreq_emptyc C sD | t ¡ ¡}| d¡}t ||¡ |j ddd}|j ddd}d S )Nr r r! r" )r, r r/ r- Úassert_frame_equal)r r] Zoffset_monthlyZ rule_monthlyZfilledZ
|
filled_depr r r Ú test_asfreq s
|
|
zTestAsFreq.test_asfreqc C sp t ddddgitdddtdddtdddgd}| d¡}t|jtsNt|d d¡}t|jtsltd S ) NrT r) rZ r: iÛ r r r )r r r, Ú
|
isinstancer r r1 )r Údfr4 r r r Útest_asfreq_datetimeindex s þ
|
z$TestAsFreq.test_asfreq_datetimeindexc C s¢ t dddd}tt t|¡|dd}td|i}d |jd< |jd d
|
d }|jd d d
|
¡}d |jd< t
|
||¡ |jd d d
|
¡}|jd d
|
d }t ||¡ d S ) Nz1/1/2016r Z2S)rH r Úfloatr8 Úone)z2016-01-01 00:00:08re Z1Sg "@)r r' r ) r r rA Zaranger0 r Úlocr, Zfillnar- r_ r2 )r rK r4 rb Z actual_dfZ expected_dfZexpected_seriesZ actual_seriesr r r Útest_asfreq_fillvalue£ s
|
|
z TestAsFreq.test_asfreq_fillvaluec C sf t ddd}|tj d¡|d}| ¡ }dd |jD |_|jddd }|jddd }t ||¡ d S )
|
NrI rF rG r c S s g | ] }| ¡ qS r )Údate)Ú.0Úxr r r Ú
|
<listcomp>½ s zATestAsFreq.test_asfreq_with_date_object_index.<locals>.<listcomp>Z4HZffillr" ) r rA rB rC r< r r, r- r. )r r3 rK r4 Zts2r5 r6 r r r Ú"test_asfreq_with_date_object_index¸ s z-TestAsFreq.test_asfreq_with_date_object_indexc C sV t ddddg}|td|d}| t|¡}|j d¡|_| d¡}t ||¡ d S ) Nz
|
2021-01-04z
|
2021-01-02z
|
2021-01-03z
|
2021-01-01é r Zinferr% ) r rQ r\ Úsortedr Z
|
_with_freqr, r- r. )r r3 r r5 r6 r r r Ú test_asfreq_with_unsorted_indexà s
|
z*TestAsFreq.test_asfreq_with_unsorted_indexc C sD t tddd |¡ ¡ dd}t ddgdd |¡}t ||¡ d S )NÚ2000rZ rG r% r z
|
2000-01-01z
|
2000-01-02)r r Zas_unitrJ r- r= )r r r5 r6 r r r Útest_asfreq_after_normalizeÏ s ÿz&TestAsFreq.test_asfreq_after_normalizeN)Ú__name__Ú
|
__module__Ú __qualname__ÚpytestZfixturer r7 r> ÚmarkZ parametrizerE rM rS rX r[ r^ r` rc rg rl ro rq r r r r r
|
s"
|
%
|
|
r
|
)r ÚnumpyrA ru Zpandasr r r r r r Zpandas._testingZ_testingr- Zpandas.tseriesr r
|
r r r r Ú<module> s
|
