U
|
ý°dé
|
ã @ sN d dl Zd dlZd dlmZmZmZmZmZm Z d dl
|
m Z G dd dZ dS )é N)Ú DataFrameÚ DatetimeIndexÚ PeriodIndexÚSeriesÚ
|
date_rangeÚ period_rangec @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) Ú TestToPeriodc C s d}t dddd}ttj t||¡|ddddd gd
|
}d |d < t ||¡}| ¡ }| ¡ }t
|
dd|_ t ||¡ | d ¡}|j d ¡|_ t ||¡ d S )Né ú1/1/2000ú1/1/2001ÚD©ÚfreqÚAÚBÚCÚE©ÚindexÚcolumnsÚaÚmixÚM)r r ÚnpÚrandomÚrandnÚlenÚtmÚget_objÚ to_periodÚcopyr r Ú assert_equalÚasfreq)ÚselfÚframe_or_seriesÚKÚdrÚobjÚptsÚexp© r* ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_to_period.pyÚtest_to_period s ÿ
|
zTestToPeriod.test_to_periodc C s t ddddg}tddddgdd}ttj dd¡||d}t ||¡}| ¡ }||_ t
|
| ¡ |¡ |tkr| ¡ }||_ t |j d d
|
|¡ d S ) Nz
|
2011-01-01z
|
2011-01-02z
|
2011-01-03z
|
2011-01-04r r é r é ©Zaxis)r r r r r r r r r r r! r r Úassert_frame_equal)r# r$ ÚidxZexp_idxr' Úexpectedr* r* r+ Útest_to_period_without_freq# s
|
ÿ z(TestToPeriod.test_to_period_without_freqc C s t dd}ttj t|d¡|d}d|d< |j}|jdd}| ¡ }t dd|_
|
t ||¡ |jd dd}t |j
|
|j
|
d ¡¡ d S )
|
Nr
|
r r ©r r r r. r/ r )r r r r r r ÚTr r r r r r0 Zassert_index_equalr" )r# r& Údfr( r) r* r* r+ Útest_to_period_columns5 s
|
z#TestToPeriod.test_to_period_columnsc C sZ t dd}ttj t|d¡|d}d|d< d}tjt|d |j d d
|
W 5 Q R X d S ) Nr
|
r r r4 r r z)No axis named 2 for object type DataFrame©Úmatché r/ )
|
r r r r r r ÚpytestÚraisesÚ
|
ValueErrorr )r# r& r6 Úmsgr* r* r+ Útest_to_period_invalid_axisC s
|
z(TestToPeriod.test_to_period_invalid_axisc C s\ t |td}|tkr | ¡ }t|tsXdt|j }tj t
|
|d | ¡ W 5 Q R X d S )N)r Zdtypezunsupported Type r8 ) r Úobjectr Zto_frameÚ
|
isinstancer ÚtypeÚ__name__r; r< Ú TypeErrorr )r# r r$ r' r> r* r* r+ Útest_to_period_raisesL s
|
z"TestToPeriod.test_to_period_raisesN)rC Ú
|
__module__Ú __qualname__r, r3 r7 r? rE r* r* r* r+ r s
|
r )Únumpyr r; Zpandasr r r r r r Zpandas._testingZ_testingr r r* r* r* r+ Ú<module> s
|
