U
|
ý°d' ã @ sZ d dl Zd dlZd dlmZmZ d dlmZ G dd dZ ej
|
ddddg¡dd Z dS )
|
é N)Ú DataFrameÚSeriesc @ sd e Zd Zej ddddejejejdddddgfdddejejejddddejgfdddejdddddejejgfdddejejddddejejgfdddejejd d d
|
ddejgfdddejejd d d
|
dejejgfdddddd d d
|
ejejejgfdddejdd d d
|
ejejejgfg¡d d Zd d Z dd Z
|
dd Z ej dddddddg¡dd Z dS ) ÚTestDataFramePctChangez periods,fill_method,limit,expé ÚffillNç ø?r Úbfilléÿÿÿÿg à¿g333333ã¿c C sH t jt jddddt jt jg}||}|j|||d}t |||¡ d S )Nr é é é
|
)ÚperiodsÚ fill_methodÚlimit)ÚnpÚnanÚ
|
pct_changeÚtmZ assert_equal) Úselfr r r ÚexpZframe_or_seriesÚvalsÚobjÚres© r úad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_pct_change.pyÚtest_pct_change_with_nas s z/TestDataFramePctChange.test_pct_change_with_nasc C s¦ t t ddd¡t ddd¡t ddd¡g tj¡}tj|jd< tj|jd< d|jd< tdD ]B}|j|d |j|d j |d d
|
}|j
|
|d d }t ||¡ q^d S ) Nr é( r )r r )r r é< )r
|
é r
|
)Úaxisr Úpad)r r ) r r ZarangeZastypeÚfloat64r ZiatÚranger Úshiftr r Úassert_frame_equal)r Zpnlr ÚexpectedÚresultr r r Útest_pct_change_numeric" s &ÿþ
|
$z.TestDataFramePctChange.test_pct_change_numericc C sÊ |j d d}t ||| d¡ d ¡ | d¡}|jdd}t ||| d¡ d ¡ |j ddd}|jddd}t ||| d¡ d ¡ |j d d
|
}|jdd}t |||jd d
|
d |¡¡ d S ) N©r r r
|
r )Úmethodr ©r r )r) r Z5D)Úfreq)r r r$ r# ZfillnaZ reindex_like)r Údatetime_frameÚrsZfilledr r r Útest_pct_change1 s
|
ÿz&TestDataFramePctChange.test_pct_changec C s\ t ddtjddg}t||d}| ¡ }t tjdddd g}t||d}t ||¡ d S )
|
Nç ð?r g @g @)ÚaÚbç à?ç gVUUUUUå?gÉ?)r r r r r r r$ )r ÚsZdfZchgr% Zedfr r r Ú test_pct_change_shift_over_nasC s z5TestDataFramePctChange.test_pct_change_shift_over_nasz!freq, periods, fill_method, limit)Z5Bé NN)Ú3Br NN)r7 r r N)Ú7Bé r r )r8 r9 r r )Z14Bé NNc C sl |j |||d}|j |||d}t ||¡ t|j|jd}|j |||d}|j |||d}t ||¡ d S )N)r+ r r r* )ÚindexÚcolumns)r r r$ r r; r< ) r r, r+ r r r Zrs_freqZ
|
rs_periodsZempty_tsr r r Ú test_pct_change_periods_freqM s ÿ ÿ z3TestDataFramePctChange.test_pct_change_periods_freq) Ú__name__Ú
|
__module__Ú __qualname__ÚpytestÚmarkÚ parametrizer r r r' r. r5 r= r r r r r s6 " " " "øþ
|
|
úþ r r r r c C s² t tjdddddgddtjdddgddd gd d
|
}|j| d }| d krdtjtjtjtjd d g}ntjtjdd d d g}t tjtjd dd d g|ddd gd d
|
}t ||¡ d S )Nr r
|
r é é r )r r r0 r1 )r; r( g @r/ r3 r2 )r r r r Úinfr r$ )r Údatar& Z second_columnr% r r r Ú'test_pct_change_with_duplicated_indicesj s $
|
ÿ
|
þrH ) Únumpyr rA Zpandasr r Zpandas._testingZ_testingr r rB rC rH r r r r Ú<module> s _
|
