U
|
®ý°dí ã @ s2 d dl m Z mZ d dlZd dlZd dlmZmZmZ d dl m
|
Z
|
m Z m Z m Z mZmZmZmZmZmZ d dlmZ d dlmZ d dlmZ d dlm Z dd Z d
|
d Z ej d d de !dg¡g¡dd Z"ej deddedddeeddde ddg¡dd Z#ej d d d g¡d d! Z$ej deddeddg¡d"d# Z%ej deddedddg¡d$d% Z&d&d' Z'ej d(d)d*d*d+d+d+gd*d*d*d+d+gd+d*d*d*d+gd+d+d*d*d*gd+d+d+d*d*ggfd,d*d+d+d+d+gd*d*d+d+d+gd+d*d*d+d+gd+d+d*d*d+gd+d+d+d*d*ggfd-d*d*d+d+d+gd+d*d*d+d+gd+d+d*d*d+gd+d+d+d*d*gd+d+d+d+d*ggfd.d*d+d+d+d+gd+d*d+d+d+gd+d+d*d+d+gd+d+d+d*d+gd+d+d+d+d*ggfg¡d/d0 Z(ej d1d2d-d3d3d3gfd2d)d3d3d3gfd2d,d3d3d3gfd2d.d3d3d3gfd4d-d3d d gfd4d)d3d3d3gfd4d,d5d3d3gfd4d.d5d d gfg¡d6d7 Z)ej d1d8d-d9d9d9d9d9d9ddgfd8d,ddd9d9d9d9d9d9gfd8d)d9d9d:d:d:d:d9d9gfd8d.ddddddddgfg¡d;d< Z*d=d> Z+d?d@ Z,ej dAd.d,g¡dBdC Z-ej dDdEdFg¡dGdH Z.ej dDdEdFg¡dIdJ Z/ej dKdLdMg¡ej dNdEd-dOd d dddd9dd:dPg
|
fdEd)dOd d d dddd9dd:g
|
fdEd.ej0d d dddd9dd:dPg
|
fdEd,ej0d d d dddd9dd:g
|
fdFd-dOdddd9dd:dPdQdRg
|
fdFd)dOdddd9dd:dPdQdRg
|
fdFd.ej0d dddd9dd:dPdQg
|
fdFd,ej0d dddd9dd:dPdQg
|
fg¡dSdT Z1dUdV Z2ej dNdEd-ej0d d dddd9dej0ej0g
|
fdEd)ej0d d d dddd9dej0g
|
fdEd.ej0ej0d dddd9dej0ej0g
|
fdEd,ej0ej0d d dddd9dej0g
|
fdFd-ej0dddd9dd:d:ej0ej0g
|
fdFd)ej0dddd9dd:d:d:ej0g
|
fdFd.ej0ej0dddd9dd:ej0ej0g
|
fdFd,ej0ej0dddd9dd:d:ej0g
|
fg¡dWdX Z3ej dYd-d dZdddd9dd:dPdQg
|
fd)d dZdd[d\d]d^d_d`dag
|
fd.ej0d dZd[d\d]d^d_d`dag
|
fd,ej0d dZdddd9dd:dPg
|
fg¡dbdc Z4ej dddedg¡dfdg Z5dhdi Z6djdk Z7dldm Z8dndo Z9dpdq Z:drds Z;dtdu Z<dvdw Z=ej dxd*dygdP dzgdR d{g d|gd} d~gd dgd dgd dgd} dgd d
gd dygd d
gdR dgd} d~gd: dg dgd: dgd} dgd dgd dgd d{gd} dzgd dgdP dgd9 dgd9 d~gd} dgd dgd dgd dgd d{gd} d~gd dgdP dg dgd d~gd dgd fd+ej0gd dygd dzgdR d{g d|gd} d~gd dgd dgd dgd} dgd d
gd dygd d
gdR dgd} d~gd: dg dgd: dgd} dgd dgd dgd d{gd} dzgd dgdP dgd9 dgd9 d~gd} dgd dgd dgd dgd d{gd} d~gd dgdP dg dgd d~gd fg¡dd Z>dd Z?dd Z@dd ZAej de
|
dddgd9dd:gddgd9gdd gfddgd9dgdd dgfdddgd9dd:gdd ddgfgddfe
|
dddgd9dd:gddgd9gdd gfddgd9dgdd dgfddgdd:gdddgfgddfe
|
dddgd9dd:gddgd9gdd gfddgd9dgdd dgfddgdd:gdddgfgddfe
|
dddgd9dd:gddgd9gdd gfdgdgddgfdgd:gddgfgddfe
|
dddgd9dd:gddgd9gdd gfdgdgddgfdgd:gddgfgdd fe
|
dgd9gdg ddfe
|
dgd9gdg ddfe
|
i g fgddfe
|
dej0dgej0dd:gdd5gej0gdd gfdej0gej0dgdd dgfdej0dgej0dd:gdd ddgfgddfg ¡dd ZBej ddgd9gdd gfddgd9dgdd dgfddgdd:gdddgfgd8fdgd9gdd gfddgd9dgdd dgfdddgd9dd:gdd ddgfgdfdgd9gdd gfdgdgddgfdgd:gddgfgd fg¡d¡d¢ ZCd£d¤ ZDej d¥e dddgdgd gfddgd dgfdddgd ddgfgddfe dddgdgd gfddgd dgfdddgd ddgfgddfe dddgdgd gfddgd dgfddgddgfgddfe dddgdgd gfddgd dgfddgddgfgddfe dddgdgd gfdgdgfdgdgfgdd fe dddgdgd gfdgdgfdgdgfgddfe ddgdgd gfddgd dgfgdd fe g d¦d§g ddfg¡d¨d© ZEej dªd gdgdgdgd9gged«dd¬ded®dd¬ded¯dd¬ded°dd¬ded±dd¬dgd fd gd dgddgddgdd9gged«dd¬ded«dd¬ded®dd¬ded¯dd¬ded°dd¬dgd8fd gd dgd ddgdddgddd9gged«dd¬ded«dd¬ded«dd¬ded®dd¬ded¯dd¬dgdfg¡d²d³ ZFej d´dµd ie jGd¶d¶d·d·d·gddgd¸fd¹dºie jGd¶d»d¼d½d¾gdºdgd¸fg¡d¿dÀ ZHej dÁdOd g¡dÂdà ZIdÄdÅ ZJdÆdÇ ZKdÈdÉ ZLdÊdË ZMej dÌdÍdÎdÏdÐdÑg¡dÒdÓ ZNdÔdÕ ZOej dÖd×edØedÙg¡dÚdÛ ZPdÜdÝ ZQej dÞedßdàdádâdãfedßdädådâdæfg¡ej dçdEej0d d dddd9dd:g fdFej0d dddd9dd:dPg fdèej0d ddddPdRdédg fg¡dêdë ZRdìdí ZSej jTe re pe dîdïej dðdñddòd dZdZdd gfdóddôd dõdõdöd gfdñddòdZd dZdd gfdóddôdõd dõdöd gfg¡d÷dø ZUdùdú ZVej dûdñdèdüdýdþdEdFg¡dÿd ZWej d1d4d-d5d3dd3gfd4d,dOd5d3dgfd4d)d5d3ddgfd4d.dOd5d d3gfd2d-d5d3ddgfd2d,d5d3ddgfd2d)d5d3ddgfd2d.d5d3ddgfg¡dd ZXej ddd5d5d5d5gfdd d3d3d gfg¡dd ZYej d
|
dñeZd d eZd d ddddgfdüeZd daeZd dddddgfdèeZd deZd dddddgfdýeZd deZd ddd d d gfdþeZd d eZd d d!d"d#d$gfg¡d%d& Z[ej dÞd dddd9gdfed'd¬dd(d8fg¡d)d* Z\d+d, Z]ej dûdýdþg¡d-d. Z^ej d/dýd d0dOd1gfdþd2d3d4d5gfg¡d6d7 Z_d8d9 Z`ej ddd:g¡d;d< Zad=d> Zbd?d@ Zcej dAddBdCdCdDdEgfddFdFdGdHgfg¡dIdJ ZddKdL ZedMdN ZfdOdP ZgdQdR Zhej dddddg¡ej dûdEdFdSg¡ej dTd*d+g¡ej dUd*d+g¡ej dVdWdXdYg¡dZd[ Zid\d] Zjej d^dOd_d`dZdg¡ej dadbdcdddedfg¡ej dgd5d d3d2dddhgdid5d3d2dd ddhgdOej0djej0dkgej0ej0ej0ej0gej0d_ej0dldkdZgdZgej0dmdngg¡dodp Zkdqdr Zldsdt Zmdudv Zndwdx Zodydz Zpd{d| Zqej d}dd~dgdddejrddddgddej0ddd dd ddggdddej0ddddgddej0ddd d dddggej0dd:dPdddgddej0gd ddd
d g gddPdPdPej0ejrd9dddg
|
ddej0gd d
d g ej0gd9 dd g gddPdPdPej0ejrdPdddg
|
ddej0gd d
d g ej0gd9 dd g gddPgd9 ddej0gd d
gd: gddPdej0dPddPgddej0dd
gdgd d
g gg¡dd Zsdd Ztdd Zudd Zvej dd d g¡ej dd*d+g¡dd Zwej dexeyg¡dd Zzej dd d g¡ej dd*d+g¡ej dexeyg¡dd Z{dS ( é )ÚdatetimeÚ timedeltaN)Úis_platform_armÚis_platform_macÚis_platform_power)
|
Ú DataFrameÚ DatetimeIndexÚ
|
MultiIndexÚSeriesÚ TimedeltaÚ TimestampÚ
|
date_rangeÚ period_rangeÚ to_datetimeÚ to_timedelta)Ú BaseIndexer)ÚVariableOffsetWindowIndexer)Ú BusinessDayc C s@ t ddddtjdgi} | | d¡ ¡ | jddd ¡ d S )NÚBr é é é ©Ú min_periods)r ÚnpÚnanÚrollingÚsum©Údf© r úWd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/window/test_rolling.pyÚtest_doc_string" s r" c C sr | t dj}|d |dd |ddd |dddd |ddd d d
|
}tjt|d |d W 5 Q R X d S ) Né r r ©Úwindowr ©r% r T©r% r ÚcenterFz&window must be an integer 0 or greater©Úmatchéÿÿÿÿ)Úranger ÚpytestÚraisesÚ
|
ValueError)Úframe_or_seriesÚcÚmsgr r r! Útest_constructor) s
|
r3 Úwç @Úfoor c C s | t dj}d ddg¡}tjt|d ||d W 5 Q R X d}tjt|d |d|d W 5 Q R X d
|
}tjt|d |dd |d W 5 Q R X d S ) Nr# ú|zwindow must be an integerz=passed window foo is not compatible with a datetimelike indexr) r$ z min_periods must be an integerr r& zcenter must be a booleanr r' )r, r Újoinr- r. r/ )r0 r4 r1 r2 r r r! Útest_invalid_constructor= s þÿr9 r% é )ÚdaysÚ3Dz
|
2015-12-25r# ©Úperiodsr )ÚindexÚoffsetc C sH t t d¡tddddd}tjtdd |jdd d
|
W 5 Q R X d S ) Né
|
ú
|
2015-12-24ÚD©r> Úfreq©r? z,step is not supported with frequency windowsr) r< r: ©Ústep)r r Úaranger r- r. ÚNotImplementedErrorr )r% r r r r! Ú test_freq_window_not_implementedU s þ ÿrK ÚaggÚcovÚcorrc C s@ t tdjddd}tjtdd t|| W 5 Q R X d S )Nr r rG zstep not implementedr) )r r, r r- r. rJ Úgetattr)rL Úrollr r r! Ú&test_step_not_implemented_for_cov_corrl s rQ c C s d}t dt |¡itd|ddd}t ddgt d d
|
d ¡¡}|j| d ¡ }t d|itd|ddd}t ||¡ | d ¡ ¡ }t ||¡ d S )NrA ÚvaluerB rC rD rF ç ç ð?ç @ç ;@r: r$ r< ) r r rI r Úappendr r ÚtmÚassert_frame_equal)r% Únr Ú expected_dataÚresultÚexpectedr r r! Ú&test_constructor_with_timedelta_windowt s þ þ r^ c
|
C s d}t dt |¡itd|ddd}t dt tjdgt dd d
|
¡¡itd|ddd}|j| d d ¡ }|j| d d jt|d }t
|
||¡ t
|
||¡ d S )NrA rR z
|
2017-08-08rC rD rF rT rU rV r: r r& )Úraw) r r rI r rW ÚNaNr r ÚapplyrX rY )r% r_ rZ r r] Zresult_roll_sumZresult_roll_genericr r r! Ú0test_constructor_timedelta_window_and_minperiods s þ þ rb c C s |}t ddddddgi}t ddddddgitddd d
|
}t|jd| dd | }t|jd | dd | jd d}t ||¡ d S )NÚAr r r r: r Ú2020r# r= rF ©Úclosedr Ú2DT©Údrop)r r rO r Ú reset_indexrX rY )rf Úarithmetic_win_operatorsÚ func_nameZdf_fixedÚdf_timer\ r] r r r! Útest_closed_fixed s "þþýrn zclosed, window_selectionsZbothTFÚleftÚrightZneitherc s |t ddddddgitddd d
|
t d fd d |D itddd d
|
}d kr`ddi}ni }t jd| dddf |}tj||dd d S )Nrc rS rT r5 rU ç @rd r# r= rF c s" g | ]}t d j| qS )rc )rO Úiloc)Ú.0Ús©rm rl r r! Ú
|
<listcomp>ã s z9test_datetimelike_centered_selections.<locals>.<listcomp>ÚsemÚddofr rg r T)rf r r( F©Z check_dtype)r r rO r rX rY )rf Zwindow_selectionsrk r] Úkwargsr\ r ru r! Ú%test_datetimelike_centered_selections¬ s& 1
|
ÿ
|
þ
|
þýr{ zwindow,closed,expectedÚ3srU Z2srT c C sX t dt dt dg}|dddg|d}|j| |dd ¡ }|||d}t ||¡ d S )Nz20130101 09:00:01z20130101 09:00:02r rF T©rf r( )r r r rX Ú assert_equal©r% rf r] r0 r? r r\ r r r! Ú,test_datetimelike_centered_offset_covers_allô s ý r rg r é c C s^ t ddddddddg}|dgd |td}|||td}|j| d|d ¡ }t ||¡ d S )
|
Nú
|
2020-01-01ú
|
2020-01-02ú
|
2020-01-03ú
|
2020-01-04r é ©r? ÚdtypeT)r( rf )r Úfloatr r rX r~ r r r r! Ú+test_datetimelike_nonunique_index_centering s øÿ r c C sR t tdtddddd} | jdddd ¡ }t d
|
d d g| jd}t ||¡ d S ) Nr: r rC ©ÚstartrE r> rF rg r Tr' ç à?ç ø?r )r
|
r, r r Úmeanr? rX Úassert_series_equal©rt r\ r] r r r! Ú!test_even_number_window_alignment3 s r c C sº ddddddddg}t d|itddt|dd}| rLdd d
|
d d d d d g}ntjdd dd d
|
d d g}t |dgtddt|dd d d |
}|jt|dd| |d}| ¡ }t ||¡ d S )Nr r Z
|
binary_colr Úminr rF gUUUUUUå?r çÙ?g£å@µmÛ?g._x%Iâ?)Úcolumnsr? ro )r% rf r r( rH ) r r Úlenr r r r rX rY )r( rH Údatar r[ r] r r\ r r r! Ú test_closed_fixed_binary_colA s0 þýü ÿr rf c C s` |}t t d¡tddddd}|jd| d}t|| }t tjgd |jd}t ||¡ d S ) Nr# Ú2000rg rD ©r r? Ú1D©rf rF )
|
r
|
r rI r r rO r r? rX r )rf rk rl ÚserrP r\ r] r r r! Útest_closed_empty[ s r Úfuncr Úmaxc C sJ t dgtdddd}t|jddd| }t |t tjg|jd ¡ d S )
|
Nr r r r= r Ú10Dro r rF ) r
|
r rO r rX r r r r? )r r r\ r r r! Útest_closed_one_entryg s r¢ c C s t dddgdddgdtdddd}t|jdd d
|
d jd d d| }tjdddg|jgdd}tt j
|
dt j
|
g|d d}t ||¡ d S )Nr r r: ©rc r r r= r rc F)Úsortr r¡ ro r )rc N)ZarraysÚnames)r r? Úname) r r rO Úgroupbyr r Z from_arraysr? r
|
r r rX r )r r r\ Zexp_idxr] r r r! Ú test_closed_one_entry_groupbyo s
|
þ ÿr¨ Ú input_dtypeÚintr zfunc,closed,expectedrS é r é c C sT t t d¡ | ¡tdddd}t|jd|d| }t ||jd}t ||¡ d S )NrA r r= r r< r rF )
|
r
|
r rI Úastyper rO r r? rX r )r© r rf r] r r\ r r r! Ú test_closed_min_max_datetime~ s
|
þr® c C sr t t d¡tdddd} | j| jddg d} | jdd d
|
¡ }t tjd d d d dddg| jd}t
|
||¡ d S )NrA r r= r r r# rF r< ro r r r r: r r ) r
|
r rI r ri r? r r r rX r )r r\ r] r r r! Útest_closed_uneven s
|
r¯ c C sn t t d¡tdddd}| d¡}tj||jdd
< t|jdd|d | }t ||jd
|
}t
|
||¡ d S ) NrA r r= r r éýÿÿÿr< r )r rf rF ) r
|
r rI r r r r? rO r rX r )r rf r] r r\ r r r! Útest_closed_min_max_minp£ s
|
r± zclosed,expectedr r g @g @g @g @g @g @c C sd t t d¡tdddd}|jd| d}t ||jd}| ¡ }t ||¡ | d¡}t ||¡ d S ) NrA r r= r r< r rF r )
|
r
|
r rI r r r? ÚmedianrX r Úquantile)rf r] r rP r\ r r r! Útest_closed_median_quantile» s
|
r´ ÚrollerÚ1sc C sX t }t | ¡ ¡ }t ||¡ t tg d}t tg d | ¡ ¡ }t ||¡ d S )NrF )r r r rX rY r )rµ r] r\ r r r! Útests_empty_df_rollingÑ s r· c C sV t tjtjtjg} t t d¡ d¡}| ¡ }t || ¡ | d¡}t || ¡ d S )Nr: r ç¹?) r
|
r r rI r r² rX r r³ )r] rP r\ r r r! Ú!test_empty_window_median_quantileà s
|
r¹ c C sb t tjg} | jddd ¡ }t dg}t ||¡ | jddd ¡ }t tjg}t ||¡ d S )Nr r r rS )r
|
r r r r rX r ©Úxr\ r] r r r! Útest_missing_minp_zeroì s
|
r¼ c C sT t tjgd tddddgd} | jtddd ¡ }t d
|
| jd}t ||¡ d S ) Nr z
|
2017-01-01z
|
2017-01-04z
|
2017-01-06z
|
2017-01-07rF Z2dr r rS )
|
r
|
r r r r r r r? rX r rº r r r! Ú test_missing_minp_zero_variableú s
|
þr½ c C sj t jddgdddggddgd} tt d ¡| d
|
}| d ¡ ¡ }t |j |j ¡ |j
|
j d ddgksft d S ) Nrc r ÚCrC ÚEÚ1Ú2©r¥ )rA r ©r r: ) r Z from_productr r Úonesr rM rX Zassert_index_equalr r? r¥ ÚAssertionError)Úcolsr r\ r r r! Útest_multi_index_names s
|
rÇ c C sz t t d¡}| | ¡}|dkr8t dd tdD }n t tjgd dgd gd }|jd
|
| d ¡ }t ||¡ d S ) N)rA é r c S s$ i | ] }|t jgd dgd qS )r rU r )r r ©rs Úir r r! Ú
|
<dictcomp> s z)test_rolling_axis_sum.<locals>.<dictcomp>rÈ r rU é rA r: ©Úaxis)
|
r r rÄ Ú_get_axis_numberr, r r r rX rY ©Ú
|
axis_framer rÎ r] r\ r r r! Útest_rolling_axis_sum s
|
rÒ c C s t tdtdd}| | ¡}|dkrDt dddgdddgd}nt dddgdddgd}|jd| dd ¡ }t ||¡ d S ) Nr: )r» Úy©r r? rT r5 r r )rÎ r )r r, rÏ r ÚcountrX rY rÐ r r r! Útest_rolling_axis_count s
|
rÖ c C s\ t ddt jddg¡} | jdd t| d¡ ¡ }tt jdt jt jdg}t ||¡ d S )Nr r: r# F)Úwriter r ) r Úarrayr Zsetflagsr
|
r r rX r )Zarrr\ r] r r r! Útest_readonly_array- s
|
rÙ c C s |}t dd tddd|dD }| dkrB|jjd| d ¡ j}n|jd| d ¡ }t d
|
d tdd d|d D d d tddd|dD }t ||¡ d S )Nc S s i | ]}|d gd qS )r r r rÉ r r r! rË : s z)test_rolling_datetime.<locals>.<dictcomp>z 2019-8-01z
|
2019-08-03rC )rE ÚtzrÔ rg rÍ c S s i | ]}|d gd qS )rT r r rÉ r r r! rË B s ÿ r )r> rE rÚ c S s i | ]}|d gd qS )r5 r r rÉ r r r! rË F s ÿ z 2019-8-02z 2019-8-03)r r ÚTr r rX rY )rÑ Ztz_naive_fixturerÚ r r\ r] r r r! Útest_rolling_datetime6 s" ÿþþûÿ rÜ zcenter, expected_datag V@g @X@g X@g ÀX@é g ÀW@é g @W@g @V@g X@g W@é g V@é g ÀV@g ÀU@g W@g ÀT@g U@rA g @U@g S@g @T@é g @S@rÈ c
|
C sª t ¡ }t||td dd}tjjdd}|jddt|d}t ||d }|j
|
d
|
d d |j d dd| dd d¡}| d
|
¡}| d ¡}t||dd} t || ¡ d S )Nim rC )rE i¥ )Úseedr éd )ÚhighÚsize)ÚDateColÚmetricræ T)ZinplaceZ21Dr ro )r% r rf r( rç r )r? r¦ )r Únowr r r ÚrandomZ RandomStateÚrandintr r Z set_indexr rL ÚrenameZ
|
_with_freqr
|
rX r )
|
r( r[ Z
|
date_todayr; Znprr r r\ r? r] r r r! Ú test_rolling_window_as_stringO s [ÿþ
|
|
rì c C sV t dddddgdgd} | d jdddd ¡ }td d
|
d
|
d
|
d gdd }t ||¡ d S ) Nr r r Úarà r: T)r( r rT r5 ©r¦ )r r r r
|
rX r ©r r\ r] r r r! Útest_min_periods1½ s rð c C s@ | t djddd ¡ }| tjtjdddg}t ||¡ d S )Nr# r: r rU )r, r rÕ r r rX r~ )r0 r\ r] r r r! Ú#test_rolling_count_with_min_periodsÅ s rñ c C sT dddt jdddg}ddd dddd g}| |jdd
|
d ¡ }| |}t ||¡ d S ) Nr r r: r r# r rT r5 rU r r )r r r rÕ rX r~ )r0 ÚvaluesZexpected_countsr\ r] r r r! Ú7test_rolling_count_default_min_periods_with_null_valuesÌ s
|
ró z df,expected,window,min_periodsr£ c C s< dd |D }t || j||dD ]\}}t ||¡ q"d S )Nc S s g | ]\}}t ||d qS ©rF )r ©rs rò r? r r r! rv s z/test_iter_rolling_dataframe.<locals>.<listcomp>r )Úzipr rX rY )r r] r% r Úactualr r r! Útest_iter_rolling_dataframe× s F rø zexpected,windowr c sn t dddddgdddddgtd dd
|
d d fd d| D } t| j|ddD ]\} }t || ¡ qTd S )Nr r r: r r# r r« r z
|
2016-01-01rC ©r r> rE ©rc r r¾ c s& g | ] \}}t | j|d f dqS )r¾ rF )r Úlocrõ r r r! rv J s z2test_iter_rolling_on_dataframe.<locals>.<listcomp>r¾ )Úon)r r rö r rX rY )r] r% r÷ r r r! Ú test_iter_rolling_on_dataframe# s ýÿ
|
ÿ rý c sr t dddgdddgd t d¡ d¡} fdd dgddgdgfD }t| |D ]\}}t ||¡ qXd S )
|
Nr» rÓ r r r ©rí Úbrí c s g | ]} j |d gf qS )r )rr )rs Úidxr r r! rv U s z<test_iter_rolling_on_dataframe_unordered.<locals>.<listcomp>)r Úlistr§ r rö rX rY )ÚresultsÚ expectedsr\ r] r r r! Ú(test_iter_rolling_on_dataframe_unorderedQ s
|
r z ser,expected,window, min_periodsÚint64©r c C s< dd |D }t || j||dD ]\}}t ||¡ q"d S )Nc S s g | ]\}}t ||d qS rô ©r
|
rõ r r r! rv } s z,test_iter_rolling_series.<locals>.<listcomp>r )rö r rX r )r r] r% r r÷ r r r! Útest_iter_rolling_seriesZ s # r z expected,expected_index,windowr rC rD r r r
z
|
2020-01-05c C sX t tdtddddd}dd t| |D } t| | |¡D ]\} }t || ¡ q>d S )Nr# r rC rù rF c S s g | ]\}}t ||d qS rô r )rs rò r r r r! rv s z.test_iter_rolling_datetime.<locals>.<listcomp>)r
|
r, r rö r rX r )r] Zexpected_indexr% r r÷ r r r! Útest_iter_rolling_datetime s (ÿr zgrouping,_indexÚlevel)r r )r r r ÚbyÚX)r r )r r )r: r )r r c C s dd }t dtdidddddgd}t ddd d
|
d d gi|d}d | krV|jddd}|jf | d¡j||dd}t ||¡ d S )Nc W s( t | dk rtd| \}}| ¡ | S )Nr z The function needs two arguments)r r/ r )ÚargsrØ Zscaler r r! Ú
|
scaled_sumÉ s z4test_rolling_positional_argument.<locals>.scaled_sumr r# r r r rS r rT r r5 r Úignore)r Úerrors)r )r_ r )r r, ri r§ r ra rX rY )ÚgroupingÚ_indexr_ r r r] r\ r r r! Ú test_rolling_positional_argumentµ s r Úaddc C s t dd| ddgitdtdtdgd}| d¡ ¡ jd d
|
d d ¡ }tdd dd}t dtjtjtjddddgi|d}t
|
||¡ d S )Nrc gVUUUUU%Cç z19700101 09:00:00z19700101 09:00:03z19700101 09:00:06rF r¶ r| ro r: re r« ÚSrD gUUUUUU%Cg Çq Çq Cg Çq Çq CrS ) r r ZresampleZffillr r r r r rX rY )r r r\ Údatesr] r r r! Ú*test_rolling_numerical_accuracy_kahan_meanÙ s0 ýþ ÿùÿ ôr c C sX t ddddddgdgd} | d d¡ ¡ }ttjtjddddgdd}t ||¡ d S ) Ng°rhí|@g'1¬ Zú¿rS r» rà r: gshí|?á?rî )r r r r
|
r r rX r rï r r r! Ú)test_rolling_numerical_accuracy_kahan_sumù s r c C s^ t dddd tdg¡} tj t| ¡}td|i| d}| d¡ ¡ }t
|
||dg ¡ d S )Nr r Z60s©r ÚendrE r r rF ) r rW r r ré Úrandr r r r rX rY )r? r r r\ r r r! Ú$test_rolling_numerical_accuracy_jump s ÿr c C s8 t ddddgtddd} | d¡ ¡ }t || ¡ d S )Ng(¶@S ?ga2U0*©3?r z
|
1999-02-03z
|
1999-02-06r r )r
|
r r r rX r )rt r\ r r r! Ú,test_rolling_numerical_accuracy_small_values s
|
þr c C sp t dddd} ttd| tjd}d|d< | d¡ ¡ }ttjtjtjtjd d d d
|
d d g
|
| d }t ||¡ d S )Nz
|
2015-01-01rA rC rD )r r? r gТ¿»ûÆr r# gÚ¦kÌ/ÖÆç @ç @ç @rF )
|
r r
|
r, r Úfloat64r r r rX r )r Údsr\ r] r r r! Ú(test_rolling_numerical_too_large_numbers s& ö ór$ )r rR )r r5 )r rT )r rT )r rT )r² rT c C sl t dddgdddgd}d|d< t|jdddd| }t ddg||g||gd ddgd
|
}t ||¡ d S ) Nr r rí rÿ r1 )r? r rT ©r% r rÎ ©rí rÿ r1 rF )r rO r rX rY )r rR r r\ r] r r r! Ú test_rolling_mixed_dtypes_axis_1/ s þr' c
|
C s¦ t ddddtjtjtjgdddtjtjtjtjgdddtjdtjdgg} | jdddd ¡ }t dd d
|
d d d d gdd d
|
d
|
d
|
d
|
d
|
gdd d d ddd gg}t ||¡ d S )Nr r r r r« r r% rS rT rU r! r5 rq r )r r r r r rX rY rï r r r! Ú test_rolling_axis_one_with_nan? s ýÿýÿr( rR Útestz
|
2019-12-31z1 days 06:05:01.00003c C sL t dddgi}| |d< |jdddd ¡ }t dddgi}t ||¡ d S )Nrí r r rÿ r% rT r5 )r r r rX rY )rR r r\ r] r r r! Ú&test_rolling_axis_1_non_numeric_dtypesS s
|
r* c C s t dd gddgddgd} t dtjgddgd d
|
gd}| jdd dd ¡ }t ||¡ | jjdd d ¡ j}t ||¡ d S )Nr r r# r« r rú rT r g &@ç *@r )r r% rÎ )r r% )r r r r r rX rY rÛ )r r] r\ r r r! Ú test_rolling_on_df_transposed` s r, )r? r% z2020-01-01 08:00z2020-01-01 08:08rÛ r Z2Tz2020-01-01 12:00Z30TZ1h)r rò r é c
|
C sP t ddddddddd g | d
|
}t|j|d d | }t || d
|
}t ||¡ d S ) Nr r r r: r r# r r« r rF ro r )r
|
rO r rX r )r? r% r rò r# r\ r] r r r! Útest_rolling_period_indexk s r. c C s^ | dddg}|j ddd ¡ }t|tr8t|d j}ttjgdgd }t ||¡ d S )Nr r r r gÍ;f æ?)
|
r rw Ú
|
isinstancer r
|
rò r r rX r )r0 Úobjr\ r] r r r! Útest_rolling_sem s
|
r1 zGH 38921)Úreason)r Ú third_valuerò Úvarg|ØêÐîFÚstdg >g¤ßfoCgPïp æ?g»¹øÛ ö?c C s\ t dd|ddddg}t| d¡| }t tjg| }t ||¡ t |dk|dk¡ d S )Nl ÿ;á r r r: r ©r
|
rO r r r rX r )r r3 rò r# r\ r] r r r! Ú!test_rolling_var_numerical_issues s
|
r7 c C s t tjtjddddddddg
|
tjtjddddddddg
|
dt tdd ftdd
|
ftd d ftd d
|
ftd d ftd d
|
ftd d ftd d
|
ftdd ftdd
|
fg
|
¡d} t dddddgdddddgdtdtd td td tdgd}|jdd ¡ }t | |¡ d S )Ngþÿÿÿÿÿï?g ð¿rT g6µ<AìòÔ¿g ð?gáÿÿÿÿÿï?)r rc z20130101 09:00:00r rc z20130102 09:00:02z20130103 09:00:03z20130105 09:00:05z20130106 09:00:06rF r r r r r: r« r r¬ Z3dr$ )
|
r r r r Ú from_tuplesr r rN rX rY )Úexpr Úresr r r! Ú,test_timeoffset_as_window_parameter_for_corrª sb ö öó
|
|
|
|
|
|
|
|
|
|
öÿå+ ûþ r; Úmethodr ÚskewÚkurtc C s¦ t dt dd¡d i}t d|d ddd
i|jddd
d}t|jdd | }t|jdd | }t |jddd
dd
|
|jd d
¡ ¡ d k s¢t dS ) zf
|
Make sure that decreasing indices give the same results as increasing indices.
|
|
GH 36933
|
rò iñÿÿÿrA r Nr+ rF r# r$ éüÿÿÿr gê-q=)
|
r r rI r? rO r Úabsrò r rÅ )r< r Z
|
df_reverseZ
|
increasingZ
|
decreasingr r r! Ú test_rolling_decreasing_indicesç s
|
*rA r r c C s t dddd}|td|d}|td|ddd
d}|||d}|||ddd
d}|j| |dd ¡ } |j| |dd ¡ }
|
t | |¡ t |
|
|¡ dS )
|
zV
|
Ensure that a symmetrical inverted index return same result as non-inverted.
|
rd r r¶ rD rF Nr+ Tr} )r r, r r rX r~ ) r% rf r] r0 r? Zdf_incZdf_decZ expected_incZ expected_decZ
|
result_incZ
|
result_decr r r! Ú(test_rolling_decreasing_indices_centered÷ s rB zwindow,expectedÚ1nsZ3nsc C sV t dddd}|ddddg|td}|||td}|j| |dd ¡ }t ||¡ d S ) Nrd r rC rD r r Tr} )r r r r rX r~ r r r r! Ú)test_rolling_center_nanosecond_resolution s
|
|
rD zmethod,expectedr g E@g¯µ÷©ª
|
a@g ÀE@gÖÇCßݽW@g Àf@g às@g 5@g¥`UU"@r+ g 1@g >@g K@g a@g/àe²æ?g8j
é{ Ú?göì¹LMï?gô4`ôií?g
|
+TTýë?gùñõê?g %bøîâ¿gOq&à ð¿g zCen@¯¿gk±=ôáCпgÀ~é~é׿gf`8gÕÝ¿c C sx ddddddddg}t dt t|¡d i}G dd dt}|d|d}t| |¡| }t d|i}t ||¡ d S )
|
zÓ
|
Make sure the (rare) branch of non-monotonic indices is covered by a test.
|
|
output from 1.1.3 is assumed to be the expected output. Output of sum/mean has
|
manually been verified.
|
|
GH 36933.
|
TFrò r c @ s e Zd Zdd ZdS )z1test_rolling_non_monotonic.<locals>.CustomIndexerc S sl t j|t jd}t j|t jd}t|D ]:}| j| rLd||< |d ||< q(|||< || j ||< q(||fS )Nr r r )r Úemptyr r, Ú use_expandingÚ window_size) ÚselfZ
|
num_valuesr r( rf rH r r rÊ r r r! Úget_window_boundsk s
|
zCtest_rolling_non_monotonic.<locals>.CustomIndexer.get_window_boundsN)Ú__name__Ú
|
__module__Ú __qualname__rI r r r r! Ú CustomIndexerj s rM r )rG rF N) r r rI r r rO r rX rY )r< r] rF r rM Zindexerr\ r r r! Útest_rolling_non_monotonic) s > rN z
|
2001-01-01)rE r> c C sf t dddddg| d}| ¡ }d|jdd
< | |¡ |¡}t tjtjdddg| d}t ||¡ d S ) Nr r r: r r# rF rS r ) r
|
Úcopyrr r rN r r rX Úassert_almost_equal)r? r% r» rÓ r\ r] r r r! Ú!test_rolling_corr_timedelta_index~ s rQ c
|
C s¬ dt jddt jgdddddgd} t| }|jd d
|
d jddd ¡ }td dddddgitddt jgdddddggdddddgdddddggd d gdd}t ||¡ d S )NZg1Zg2r r r r: r )Úgroupr rR F)Zdropnar r rS r5 rU rT rq r rF ) r r r r§ r r r rX rY )r r r\ r] r r r! Ú!test_groupby_rolling_nan_included s " ýørS c C s` t tj d¡}| ¡ }t| d¡| }t ||¡ |d }t| d¡| }t ||¡ d S )NrA r: iPà ) r
|
r ré r rO rO r rX r )r< r Zser_copyr] r\ r r r! Ú*test_rolling_skew_kurt_numerical_stability¥ s rT )r< rò gä+Xë?gJ_9ïÿÿ?rq g¿BæÊ ô¿g333333ó¿gÍ >°ãÿ@c C sL t dddddddg}t| d¡| }t tjgd | }t ||¡ d S )NiÀÆ- r r r: r iç r6 )r< rò rt r\ r] r r r! Ú(test_rolling_skew_kurt_large_value_range± s rU c C s4 t jtdd ttdjddd W 5 Q R X d S )Nz!method must be 'table' or 'singler) r r6 )r< )r- r. r/ r
|
r, r r r r r! Útest_invalid_methodÀ s rV Ú1dc C s¢ t dddd}|tdd|d}|jddd ¡ }|tjdd
|
g|d}t ||¡ |jd d d
jddd ¡ }t ddd d}|tjd d
|
g|d}t ||¡ d S )Nr r rW r r r rF ro r r r+ z-1dr: ) r r, r r r r rX r~ rr )r% r0 r r0 r\ r] r r r! Ú.test_rolling_descending_date_order_with_offsetÅ s rX c C s\ t ddddg} | d¡ ¡ }t tjtjddg}tj||ddd t |dk|dk¡ d S )Nr« r# r: çUUUUUUõ?r çVç¯Ò<©ZatolZrtol)r
|
r r4 r r rX r r r r r! Ú,test_rolling_var_floating_artifact_precisionÔ s
|
r\ c C sB t dddg} | d¡ ¡ }t tjddg}tj||ddd d S )NgQöØ ¢>g 4PSªÈ¡>r gÌ[³¸^>>rZ r[ )r
|
r r5 r r rX r r r r r! Ú test_rolling_std_small_valuesß s ýÿr] zstart, exp_valuesç¸
ëQ¸?gX9´Èv¾?gºI +?gË¡E¶óý?çü©ñÒMbP?gú~j¼tX?ç<# n?c C s t dddtjddtjtjtjtjtjtjddg}|ddddtjtjddg }t |tt| t|| d }|j| d
jd
|
d d ¡ }t
|
||¡ d S ) Nr^ r_ çü©ñÒMb`?çü©ñÒMb?ç{®Gázt?çÉ?r` g=
|
×£p=º?rF r# r r ) r r r` r r, r rr r r rX rY )r Z
|
exp_valuesr rò r] r\ r r r! Ú3test_rolling_mean_all_nan_window_floating_artifactsí sB òÿø
|
þ re c C sP t dddtjtjtjg} | jddd ¡ }t dddd dd
|
g}t ||¡ d S ) Nra rb rc r: r r g{®Gáz?g¸
ëQ¸?g9´Èv¾?rS )r r r` r r rX rY rï r r r! Ú2test_rolling_sum_all_nan_window_floating_artifacts s rf c C s6 t td} | d¡ ¡ }t tjg}t ||¡ d S )Nr r )r
|
r, r r r r rX r r r r r! Útest_rolling_zero_window# s rg c C sd t tdtddd| d}t tjgd tdddd| d}|jddd ¡ }tj||d
|
d d S ) Nr# rA é r£ r rÈ r r rÍ Fry )r r, r r r r rX rY )Zfloat_numpy_dtyper r] r\ r r r! Útest_rolling_float_dtype+ s þri c C sÐ t t d¡ dd¡tdd dddd d
|
d d d ddd
|
¡} | jdddd ¡ }t tdddtdddtdddtdddtdddtdddtdddtdddtdddtdddd
|
dd}t
|
||¡ d S )Né( r rA Z
|
abcdefghijrà Úfloat16Úfloat32r" Zint8Úint16Úint32Zuint8Zuint16Zuint32Zuint64)
|
rí rÿ r1 ÚdÚeÚfÚgÚhrÊ Újr r r% r r: r# r r« r r ) r r rI Úreshaper r r r r, rX rY rï r r r! Útest_rolling_numeric_dtypes6 s: öÿ
|
|
|
|
|
|
|
|
|
|
ö órv ZaverageÚpctÚ ascendingÚ test_dataÚdefaultÚ
|
duplicatesÚnansc s° d}|dkr t tj |¡d}nN|dkr>t tj d|¡d}n0|dkrnt tj ddd tjtjtj g|¡d}| | ¡ fd
|
d ¡}| | ¡j d }t
|
||¡ d S ) NrÈ rz )r r{ r: r| rT ç Ð?ç è?c s | j djd S )N©r< rw rx r+ )Úrankrr )r» ©rx r< rw r r! Ú<lambda>j ó ztest_rank.<locals>.<lambda>r ) r
|
r ré r Úchoicer Úinfr ra r rX r ) r% r< rw rx ry Úlengthr r] r\ r r r! Ú test_rankY s ÿ
|
ÿr c C sz d} d}t d| dd}ttj | | ¡ | df¡|d}|jdd d
|
gd d }tj|d ddgd d }t |j
|
t |¡¡ d S )NrA r# Z20100101r rD r+ rF r} r r~ r rÍ é é2 éK ) r r r ré r ru r³ Z
|
percentilerX rP rò rØ )ÚrowÚcolr r Z df_quantileZ np_percentiler r r! Ú#test_rolling_quantile_np_percentileq s "r r³ r¸ gÍÌÌÌÌÌÜ?Ú interpolationZlinearÚlowerZhigherZnearestZmidpointr r! g @rd r g333333Ó?gffffffæ?g333333ã?c C sX t |}| | |¡}|jdd | |¡jd }t |¡rHt |¡sTtn ||ksTtd S )Nr r r+ )r
|
r³ Z expandingrr r ÚisnanrÅ )r³ r r rt Zq1Zq2r r r! Ú+test_rolling_quantile_interpolation_options s
|
r c C sP t d¡} t| }d}tjt|d" |jt| ddjddd W 5 Q R X d S ) Nr# z(Interpolation 'invalid' is not supportedr) r r r Úinvalid)r ) r rI r
|
r- r. r/ r r r³ )r rt r2 r r r! Útest_invalid_quantile_value s
|
|
r c C s t dddddg} d}tjt|d | d¡ d ¡ W 5 Q R X d
|
}tjt|d | d¡ d ¡ W 5 Q R X d }tjt|d | d¡ d ¡ W 5 Q R X d S )NrS r¸ r gÍÌÌÌÌÌì?rT z#quantile value -0.1 not in \[0, 1\]r) r: g¹¿z#quantile value 10.0 not in \[0, 1\]g $@z must be real number, not strr6 )r
|
r- r. r/ r r³ Ú TypeError)r r2 r r r! Útest_rolling_quantile_param¦ s r c C sª t dddddg} | jddd ¡ }t tjgd }t ||¡ | jdddjd d
|
}t d gd }t ||¡ t tjtjd d dgjd dd ¡ }t |d ¡s¦td S )NrT r5 rU rq r r r r# r )rx rS r: r r ) r
|
r r5 r r rX r r rÅ )Úvalsr\ r] r r r! Útest_rolling_std_1obsµ s $r c C sj t dddddg} | jdd ¡ }t |dd
¡ ¡ s<t| jdd ¡ }t |dd
¡ ¡ sftd S )Nge>¸¸ÁR?g¨R¨z1Ò2?r: r$ r )Úspan)r
|
r r5 r ÚisfiniteÚallrÅ Zewmrþ r r r! Útest_rolling_std_neg_sqrtÄ s ûÿ r c C s4 t jtdd ttdjddd W 5 Q R X d S )Nzstep must be an integerr) r r r6 rG ©r- r. r/ r r, r r r r r! Ú test_step_not_integer_raisesÙ s r c C s4 t jtdd ttdjddd W 5 Q R X d S )Nzstep must be >= 0r) r r r+ rG r r r r r! Ú test_step_not_positive_raisesÞ s r rò r r] r g«ªªªªª@@g @D@çUUUUUUÕ?gUUUUUU8@rY gUUUUUU@c C s~ t |}t | }|j||d ¡ }t ||¡ t |dk|dk¡ |j||d ¡ }t |t |¡¡ t |dk|dk¡ d S )Nr r )r
|
r r4 rX r r5 r Úsqrt)rò r% r r] ÚsrZ
|
result_varZ
|
result_stdr r r! Ú'test_rolling_var_same_value_count_logicã s 2 r¢ c C sd t dddddddg} | d¡}| ¡ }|dd
dk ¡ s@t| ¡ }|dd
dk ¡ s`td S )Nr r r r: r° )r
|
r r r rÅ r ©r¡ Úrr\ r r r! Ú(test_rolling_mean_sum_floating_artifacts- s
|
r¥ c C sd t dddddddg} | d¡}| ¡ }|dd
dk ¡ s@t| ¡ }|dd
dk ¡ s`td S )Nr r r éþÿÿÿr° )r
|
r r= r rÅ r> r£ r r r! Ú)test_rolling_skew_kurt_floating_artifacts8 s
|
r§ c C s | }t dgddd}|d t¡|d< |jddd}t||}||d}|rVdd gndd dg}|| |g¡jd
|
d t¡}t|j |kst
|
t ||¡ d S ) Nr r r: r& r1 r ©Ú numeric_onlyrí rÿ Trh ) r r Úobjectr rO rL rj r r r rÅ rX rY ) rk r© Úkernelr r Úopr\ r r] r r r! Útest_numeric_only_frameC s
|
|
r r« Úuse_argc C sÄ t dddgddd}|d t¡|d< |r2|fnd}|jddd}t|| }||d|i}|rhd d
|
gnd d
|
dg}|| t¡} |r| fnd}
|
| jddd} t| | } | |
|
d|i} t || ¡ d S ) Nr r r: r& r1 r r r© rí rÿ )r r rª r rO r rX rY )r« r© r® r Úargr r¬ r\ r Zdf2Úarg2Úrolling2Úop2r] r r r! Ú test_numeric_only_corr_cov_frameS s
|
|
r³ r c
|
C s | }t dg|d}|jddd}t||}|rh|tkrhd| d}tjt|d ||d W 5 Q R X n0||d}| |g¡jd d
|
t
|
¡} t || ¡ d S ) Nr r r r úRolling.ú does not implement numeric_onlyr) r¨ Trh ) r
|
r rO rª r- r. rJ rL rj r r rX r )
|
rk r© r r« r r r¬ r2 r\ r] r r r! Útest_numeric_only_seriesi s
|
|
r¶ c C sÖ t dddg|d}|r |fnd}|jddd}t|| }|rz|tkrzd| d}tjt|d ||d
|
|i W 5 Q R X nX||d
|
|i} | t¡}
|
|r|
|
fnd} |
|
jddd} t| | } | | d
|
|i}t
|
| |¡ d S ) Nr r r: r r r r´ rµ r) r© ) r
|
r rO rª r- r. rJ r r rX r )r« r® r© r r r¯ r r¬ r2 r\ Zser2r° r± r² r] r r r! Ú!test_numeric_only_corr_cov_seriesz s
|
|
|
r· )|r r Únumpyr r- Z pandas.compatr r r Zpandasr r r r
|
r r r r r r Zpandas._testingZ_testingrX Zpandas.api.indexersr Z pandas.core.indexers.objectsr Zpandas.tseries.offsetsr r" r3 ÚmarkZ parametrizerØ r9 rK rQ r^ rb rn r{ r r r r r r¢ r¨ r r® r¯ r± r´ r· r¹ r¼ r½ rÇ rÒ rÖ rÙ rÜ rì rð rñ ró rø rý r r r r8 r r r r r r$ r' r( r* r, r. r1 Zxfailr7 r; rA rB rD r rN rQ rS rT rU rV rX r\ r] re rf rg ri rv r r r r r r r r r r
r¢ r¥ r§ r r³ rª rª r¶ r· r r r r! Ú<module> sj 0
|
|
ÿüþ
|
|
|
|
ûþ ûþ ûþ ûþáþ-
|
øþ
|
üþ
|
|
|
|
øþ " $"" $"øþ
|
üþ
|
|
|
ÿþýüûúùø ÷
|
ö õ ô óòñðïîíìëêéèçæå ä ã â á à!ß"Þ#Ý$Üý+
|
ÿþýüûúùø ÷
|
ö õ ô óòñðïîíìëêéèçæå ä ã â á à!ß"Þ#Ý$ÜýÕþY
|
ýø ýø ýø ýø ýø
|
"ýøÊþD
|
ýú
|
ýú
|
ýúïþ
|
,ü ,ü (ü (ü22*äþ!
|
û÷ û÷ $ û÷éþ&
|
ÿþ ÿþùþ
|
|
|
þ
|
þ
|
þ þûþ ýþ
|
þüþ = øþ þþ
|
øþ(þ&þøþøþÝþ4!þþ
|
þþ þþ' # ÿ ùþ ü üü*ü*ü
|
ü üÛþ/
|
