U
|
®ý°dG ã @ sT d Z ddlZddlZddlZddlmZ ej ej
|
ej j ej j dZG dd dZdS )zX
|
Tests for Series cumulative operations.
|
|
See also
|
--------
|
tests.frame.test_cumulative
|
é N)ÚcumsumÚcumprodÚcumminÚcummaxc @ sð e Zd Zej dejejg¡dd Z ej dddg¡dd Z
|
ej d e d
|
¡e d ¡e d ¡ d ¡g¡ej d ddddddddggddddddddggddddddddggddddddddggg¡dd Zej dde jdddfde jdddfg¡dd Zej d dddddddgdddddddgg¡ejjdd d d d gd!d"gd#ej de ¡ ¡d$d% Zej d&d'e jd
|
d(ejd(ged)gd*e dd
|
ejd
|
g¡gde ddejdg¡gde ddejdg¡gg¡d+d, Zd-d. Zd/S )0ÚTestSeriesCumulativeOpsÚfuncc C sr t j||j|t |¡dd | ¡ }tj|d d d
< ||dd d
}|t | ¡ ¡}t j|j|dd d S )NT)Z check_dtypeé é F)ÚtmÚassert_numpy_array_equalÚvaluesÚnpÚarrayÚcopyÚNaNÚdropna)ÚselfÚdatetime_seriesr ÚtsÚresultÚexpected© r úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/test_cumulative.pyÚtest_datetime_series s ýz,TestSeriesCumulativeOps.test_datetime_seriesÚmethodr r c C s t | }t|| j}|t |¡}t ||¡ | ¡ }tj|d d d
< t|| dd d
}|| ¡ }|j
|
d ¡|_
|
t ||¡ d S )Nr r ) ÚmethodsÚgetattrr r r r
|
r r r r ÚindexZ
|
_with_freqÚassert_series_equal)r r r Úufuncr r r r r r Útest_cummin_cummax) s z*TestSeriesCumulativeOps.test_cummin_cummaxr r z
|
1999-12-31z
|
US/Pacificzmethod, skipna, exp_tdiTÚNaTú2 daysú3 daysú1 daysFc C s\ t ddddddg¡}t || ¡}t |¡}t || ¡}t|||d}t ||¡ d S )Nr! r" r$ r# ©Úskipna)ÚpdZ to_timedeltaÚSeriesr r
|
r ) r r r r& Zexp_tdiZtdiÚserr r r r r Ú test_cummin_cummax_datetimelike9 s
|
z7TestSeriesCumulativeOps.test_cummin_cummax_datetimelikez func, expú2012-1-1ÚD©Úfreqú2012-1-2c C s t t jdddt jt jdddg¡}t||dd}t t jdddt jt jg¡}t ||¡ t||dd}t t jdddt j|g¡}t ||¡ d S )Nr+ r, r- r/ Fr% T)r' r( ÚPeriodr! r r
|
r )r r Úexpr) r r r r r Útest_cummin_cummax_period^ s ÿ z1TestSeriesCumulativeOps.test_cummin_cummax_periodÚargc C s | S ©Nr ©Úxr r r Ú<lambda>z ó z TestSeriesCumulativeOps.<lambda>c C s | S r4 r r5 r r r r7 z r8 ÚidentityZinverse)Úidsc C sF |t |¡}t| }||j}t |¡}t|| }t ||¡ d S r4 )r' r( r r r r
|
r ) r r3 r r r) r Zexp_valsr r r r r Útest_cummethods_boolr s
|
|
z,TestSeriesCumulativeOps.test_cummethods_boolzmethod, expectedr r )Zdtyper c C s0 t ddtjdg¡}t|| }t ||¡ d S )NFT)r' r( r Únanr r
|
r )r r r r) r r r r Ú$test_cummethods_bool_in_object_dtype s
|
z<TestSeriesCumulativeOps.test_cummethods_bool_in_object_dtypec C sD t t jddt jddg¡}tjtdd | ¡ W 5 Q R X d S )Nr )Údaysé z#cumprod not supported for Timedelta)Úmatch)r' r( Ú TimedeltaÚpytestZraisesÚ TypeErrorr )r r) r r r Útest_cumprod_timedelta s z.TestSeriesCumulativeOps.test_cumprod_timedeltaN)Ú__name__Ú
|
__module__Ú __qualname__rB ÚmarkZ parametrizer r r r r r' rA Ú TimestampZ tz_localizer* r0 r2 r Úkeysr; r( r< Úobjectr= rD r r r r r sp
|
|
ýþýýøþ þþ
|
þþ ÿüþ
|
r )Ú__doc__Únumpyr rB Zpandasr' Zpandas._testingZ_testingr
|
r r ZminimumÚ
|
accumulateÚmaximumr r r r r r Ú<module> s ü
|
