U
|
ý°dÇ ã @ s6 d dl Z d dlmZmZ d dlmZ G dd dZdS )é N)Ú PeriodIndexÚ period_rangec @ sF e Zd Zdd Zdd Zej dddg¡dd Zd
|
d Z d d Z
|
dS )ÚTestPeriodIndexc C s t dddd}t dddd}t dddd}t dddd}t dddd}t d ddd}t d
|
dd d}| dd
|
¡|ksvt| dd ¡|kst| dd ¡|kst| dd¡|ks²t| dd¡|ksÆt| d d
|
¡|ksÚt| d
|
d
|
¡|ksît| dd
|
¡|kst| dd
|
¡|kst| dd
|
¡|ks0t| dd
|
¡|ksFt| d d
|
¡|ks\t| d
|
d
|
¡|ksrt| dd
|
¡|kst| dd
|
¡|kst| dd
|
¡|ks´t| dd
|
¡|ksÊt| d d
|
¡|ksàt| d
|
d
|
¡|ksöt| dd
|
¡|ks t| dd
|
¡|ks"t| dd
|
¡|ks8t| dd
|
¡|ksNt| d d
|
¡|ksdt| d
|
d
|
¡|kszt| dd
|
¡|kst| dd
|
¡|ks¦t| dd
|
¡|ks¼t| dd
|
¡|ksÒt| d d
|
¡|ksèt| d
|
d
|
¡|ksþt| dd
|
¡|kst| dd
|
¡|ks*t| dd
|
¡|ks@t| dd
|
¡|ksVt| dd
|
¡|kslt| d
|
d
|
¡|kst| dd
|
¡|kst| dd
|
¡|ks®t| dd
|
¡|ksÄt| dd
|
¡|ksÚt| dd
|
¡|ksðt| d d
|
¡|kstd}tjt|d | dd¡ W 5 Q R X | d¡} | d¡}
|
t dddd} t | j| j¡ | j dksrtt |
|
j| j¡ |
|
j dkstd S )NÚAz1/1/2001)ÚfreqÚstartÚendÚQÚMÚDÚHz1/1/2001 00:00ZMinÚSz1/1/2001 00:00:00Úsr ZStarTZbeGINzHow must be one of S or E)ÚmatchÚTZfooÚ3Mz2001-12)
|
r ÚasfreqÚAssertionErrorÚpytestZraisesÚ
|
ValueErrorÚtmZassert_numpy_array_equalZasi8Zfreqstr) ÚselfÚpi1Zpi2Zpi3Zpi4Zpi5Zpi6Zpi7ÚmsgZresult1Zresult2Úexpected© r úfd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/period/methods/test_asfreq.pyÚ test_asfreq sx
|
|
zTestPeriodIndex.test_asfreqc C sD t ddddgdd}|jdd}t dddd gdd}t ||¡ d S )
|
Nú2011-01ú2011-02ÚNaTz2011-04r
|
©r r Z2011Q1Z2011Q2)r r r Úassert_index_equal)r ÚidxÚresultr r r r Útest_asfreq_natQ s z TestPeriodIndex.test_asfreq_natr r Z3Dc C s t ddddgdd}| |¡}t dddd g|d}t ||¡ |j|jksNt|j|d
|
d }t d d ddg|d}t ||¡ |j|jkstd S )Nz2001-01z2001-02r z2001-03Z2Mr! z
|
2001-02-28z
|
2001-03-31z
|
2001-04-30r ©Úhowz
|
2001-01-01z
|
2001-02-01z
|
2001-03-01)r r r r" r r )r r Úpir$ Úexpr r r Útest_asfreq_mult_piW s
|
z#TestPeriodIndex.test_asfreq_mult_pic C s t dddgdd}t dddgdd}tddgd d
|
gD ]2\}}|j||d }t ||¡ |j|jks6tq6d D ]}t dddg|d}| d¡}t d ddgdd}t ||¡ |j|jks¼tt dddg|d}|jdd d }t dddgdd}t ||¡ |j|jksntqnd S )Nz2001-01-01 00:00z2001-01-02 02:00r r r! Z25HÚ1D1HÚ1H1Dr ÚEr& )r+ r, z2001-01-02 00:00z2001-01-03 02:00)r Úzipr r r" r r )r r( r) r r' r$ r r r Útest_asfreq_combined_pie s"
|
z'TestPeriodIndex.test_asfreq_combined_pic C s t dddgdd}t dddgd d}t | d ¡|¡ t | d
|
¡|¡ t dddgd d}t | d ¡|¡ t | d ¡|¡ d S ) Nz
|
2011-01-01z
|
2011-02-01z
|
2011-03-01r r! r r z2011-03r
|
z period[M]r z
|
period[3M])r r r" r Zastype)r r r) r r r Útest_astype_asfreqz s z"TestPeriodIndex.test_astype_asfreqN) Ú__name__Ú
|
__module__Ú __qualname__r r% r ÚmarkZ parametrizer* r/ r0 r r r r r
|
s F
|
r )r Zpandasr r Zpandas._testingZ_testingr r r r r r Ú<module> s
|
