U
|
®ý°dw> ã @ s d dl Zd dlZd dlmZmZmZmZmZ d dl m
|
Z d dl m Z mZ d dlmZmZmZ d dlmZ dd Zdd Zd
|
d Zd d Zej ddedddddddddej g
|
i fdej dddddddddej g
|
i fdej dddddddddej g
|
i fd ej dddd d d ddd ej g
|
d!d"ifd#ej dddd$d%d&ddd'ej g
|
d!d"ifd(ej!ddddddddd)ej g
|
i fg¡d*d+ Z"d,d- Z#ej d.d/dddd0dd1ddej ej g
|
fd2dddd3d4d5ddej ej g
|
fg¡d6d7 Z$ej d8d9dddddd:ddddg
|
gd;ddddd<dd<dddg
|
gg¡d=d> Z%d?d@ Z&ej dAd"dd"d"dBdCgfdDdd"d dBd"gfg¡ej dEd(g fdFd'gfg¡dGdH Z'ej dIeeeg¡ej dJd"dCdKg¡ej dLd"d"gd d"gdMd"dCgd d"gdMd"gdN ej d"dCej ge(e)dOdN dMg¡dPdQ Z*ej dRd"d"d gd"gfd"dCd d"gd"dCgfdCd"d gd"gfdCdCd d"gdCdCgfdSdKe)dKe(e)dSdKdKgdS fdKdSe)dSdSgdS fd d e +g ¡e +g ¡fd"d e +g ¡e +g ¡fd d"d gd gfg ¡dTdU Z,ej dVed d"dCgd"dCdCgdWdCed dXdgej-d"dCdCge)dBgdYdgdZd[ej.d\feej d"dCej ge(e)dOd] d"gd^ dCgd_ e)d]d`dKedadadbd<d<dcdd:dddedfdgdhdidjdkdldmgej-d"gd^ dCgd_ e)d]gdYdgdZd[ej.d\fg¡dndo Z/ej dpdqdrdsdtdudvdwdxdyg ¡ej dJd"dCdBdOdSdzd{g¡d|d} Z0d~d Z1dd Z2dS )é N)Ú DataFrameÚ
|
MultiIndexÚSeriesÚconcatÚ
|
date_range)Ú BaseIndexerÚFixedForwardWindowIndexer)ÚExpandingIndexerÚFixedWindowIndexerÚVariableOffsetWindowIndexer)Ú BusinessDayc C sF G dd dt } | }tjtdd ttd |¡ W 5 Q R X d S )Nc @ s e Zd Zdd ZdS )z8test_bad_get_window_bounds_signature.<locals>.BadIndexerc S s d S ©N© )Úselfr r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/window/test_base_indexer.pyÚget_window_bounds s zJtest_bad_get_window_bounds_signature.<locals>.BadIndexer.get_window_boundsN©Ú__name__Ú
|
__module__Ú __qualname__r r r r r Ú
|
BadIndexer s r z BadIndexer does not implement©Úmatché )r ÚpytestÚraisesÚ
|
ValueErrorr ÚrangeÚrolling)r Úindexerr r r Ú$test_bad_get_window_bounds_signature s r c C s< t td} t }| |¡ ¡ }| ¡ ¡ }t ||¡ d S )Né
|
)r r r r ÚmeanZ expandingÚtmÚassert_series_equal)Úsr ÚresultÚexpectedr r r Útest_expanding_indexer# s
|
r( c C sn dddddg} t dtdi}G dd dt}|d| d}| |¡ ¡ }t dd d
|
d d d gi}t ||¡ d S ) NTFÚvaluesr c @ s e Zd Zdd ZdS )z3test_indexer_constructor_arg.<locals>.CustomIndexerc S sl t j|t jd}t j|t jd}t|D ]:}| j| rLd||< |d ||< q(|||< || j ||< q(||fS ©N©Údtyper é )ÚnpÚemptyÚint64r Ú use_expandingÚ window_size© r Ú
|
num_valuesÚ min_periodsÚcenterÚclosedÚstepÚstartÚendÚir r r r 1 s
|
zEtest_indexer_constructor_arg.<locals>.CustomIndexer.get_window_boundsNr r r r r Ú CustomIndexer0 s r< r- ©r2 r1 ç ç ð?ç @ç $@©r r r r Úsumr# Úassert_frame_equal)r1 Údfr< r r& r' r r r Ú test_indexer_constructor_arg+ s rF c C sh t dtdi} G dd dt}|dd}| j|ddddd ¡ }t dd
|
d d d dgi}t ||¡ d S )Nr) r c @ s e Zd Zdd ZdS )z8test_indexer_accepts_rolling_args.<locals>.CustomIndexerc S s t j|t jd}t j|t jd}t|D ]P}|rb|dkrb|dkrb|dkrb|dkrbd||< |||< q(|||< || j ||< q(||fS )Nr+ r- Úbothé r )r. r/ r0 r r2 r3 r r r r G s$ ÿþýüû
|
zJtest_indexer_accepts_rolling_args.<locals>.CustomIndexer.get_window_boundsNr r r r r r< F s r< r- ©r2 TrG )r6 r5 r7 r8 r> r? rA r@ ç @rB )rE r< r r& r' r r r Ú!test_indexer_accepts_rolling_argsC s
|
ÿ
|
rK z func,np_func,expected,np_kwargsÚcountr@ g @Úminr> r? rJ g @g @g @Úmaxç Y@g "@Ústdgû·ÛK@gþDéû¾mK@gµ!µ?þJ@gß'Üe æ?Zddofr- Úvarg[{ªª@¨@g[{ªª§@g[{ªªÄ¦@g à?Úmediang !@c sN t d¡}d|d< tdd}d}tjt|d$ | |j|dd } t| | W 5 Q R X d
|
}tjt|d$ | |j|d d } t| | W 5 Q R X | |j|d |d} t| | }
|
| |d d |
}t |
|
|¡ | |
|
fdd¡} t |
|
| ¡ |dkrdnd } | |j|| d} t| | }| |
|
fdd¡}t ||¡ d S )NrA rO r é rI z.Forward-looking windows can't have center=Truer T)Úwindowr6 zAForward-looking windows don't support setting the closed argumentÚright)rT r7 rH ©rT r5 r8 c s | fS r r ©Úx©Únp_funcÚ np_kwargsr r Ú<lambda>² ó z-test_rolling_forward_window.<locals>.<lambda>rL r ©rT r5 c s | fS r r rW rY r r r\ ¼ r] ) r. Úaranger r r r r Úgetattrr# Ú assert_equalÚapply)Úframe_or_seriesÚfuncrZ r' r[ r8 r) r r r r& Z expected2r5 Zrolling3Zresult3Z expected3r rY r Útest_rolling_forward_windowa s, 9
|
|
re c C sr t d¡}d|d< tdd}| |j|d|d}| ¡ }| ddd d
|
d d ddt jt jg
|
d d |
}t ||¡ d S ) NrA rO r rI rS rV r> g Å oòÛ@gê\QJÖ@gúòì£Ó@g§v©-Õ@gA Û@)r. r_ r r ZskewÚnanr# ra )rc r8 r) r r r& r' r r r Ú test_rolling_forward_skewnessÀ s*
|
|
öÿ órg z func,expectedÚcovg @X@g @WÀÚcorrgîá°óÚë?g¥hÆ"¥ª?g¡´1=ë¿c C s¢ t d¡ dd¡}|d }d|d< t j||gdd}tdd }t|j|dd
|
}t|| jt d dfd f }|j
|
d d }t |j
|
d d }|j |_ t ||¡ d S )Nr! éÿÿÿÿr- rH éd )r r )ZaxisrS rI r^ r T)Zdrop)r. r_ ZreshapeZ concatenater r r r` ÚlocÚsliceZ reset_indexr Únamer# ra )rd r' Zvalues1Zvalues2r) r r r& r r r Ú test_rolling_forward_cov_corrÙ s
|
ro zclosed,expected_datarU g (@Úleftg @c C s^ t ddd}ttd|d}td}t||d}|j|| d ¡ }t||d}t ||¡ d S )NZ2020r! )Zperiods)Úindexr- )rq Úoffset)r7 ) r r r r r r rC r# rD )r7 Z expected_datarq rE rr r r& r' r r r Ú&test_non_fixed_variable_window_indexerþ s rs c C s` t dd d d dgi}tdd}|j|d| d ¡ }t dddddgid d |
}t ||¡ d S ) NÚbé rH rI r rV r> r? )r r r rL r# rD )r8 rE r r& r' r r r Ú test_fixed_forward_indexer_count s
|
|
rv )Ú end_valuer) rS rH rj )rd ÚargsZquantilec
|
sn G fdddt }dddddg}tdtdi}|d|d}t| |¡|| }td|i} t || ¡ d S ) Nc s e Zd Z fddZdS )z0test_indexer_quantile_sum.<locals>.CustomIndexerc sr t j|t jd}t j|t jd}t|D ]@}| j| rRd||< t| d||< q(|||< || j ||< q(||fS r* )r. r/ r0 r r1 rN r2 r3 ©rw r r r s
|
zBtest_indexer_quantile_sum.<locals>.CustomIndexer.get_window_boundsNr r ry r r r< s r< TFr) r r- r= )r r r r` r r# rD )
|
rw r) rd rx r< r1 rE r r& r' r ry r Útest_indexer_quantile_sum s rz Ú indexer_classr2 é Údf_data)Úart é é c C sX t |}d}| |d}|j}t|D ].}| d¡d j|dd ¡ |j|ks$tq$d S )NrS rI r~ rt r- r^ )r r2 r Úgroupbyr r" ÚAssertionError)r{ r2 r} rE Z
|
num_trialsr Zoriginal_window_sizer; r r r Ú0test_indexers_are_reusable_after_groupby_rolling4 s
|
r z5window_size, num_values, expected_start, expected_endr c C st t | d}|j||d\}}tj|t |d d |
¡dd tj|t |d d |
¡dd t|t|ksptd S )NrI )r4 r8 F)Z check_dtype)r r r# Zassert_numpy_array_equalr. ÚarrayÚlenr )r2 r4 Zexpected_startZ expected_endr8 r r9 r: r r r Ú!test_fixed_forward_indexer_boundsM s
|
ÿ r zdf, window_size, expected)rt r~ g ø?r~ )Únamesrt )rq rn r, é ru é )rt r~ ÚcgÍÌÌÌÌÌ @g @g @g )@g *@g +@g ,@g -@g .@g /@g 0@g 0@g 1@c C s6 t |d}| d¡d j|dd ¡ }t ||¡ d S )NrI r~ rt r- r^ )r r r r" r# r$ )rE r2 r' r r& r r r Ú0test_rolling_groupby_with_fixed_forward_specifici s 6
|
r Ú
|
group_keys)r- )r- rH )rH r- )r- r- rH )r- rH r- )r- r- rH rH )r- rH rS rH rS ) r- r- rH r- r- rH r- r- rH r- r- rH )r- rH rS r- rH rS r- rH rS r- rH rS r- rH rS é é c s¼ t t t| ¡tjt| tjdd tjt| tjdd}t d}| d¡d j
|
|dd ¡ }dd g|j _ | d¡ddd g }t fd
|
d |D }| dd g¡d }t ||¡ d S ) Nr+ é )r~ rt r rI r~ rt r- r^ r c s4 g | ],\} j fd dtt D dqS )c s* g | ]"} d j ||
jddqS )rt r- )Z min_count)ZilocrC )Ú.0r; )Úgr2 r r Ú
|
<listcomp>Å s ÿzKtest_rolling_groupby_with_fixed_forward_many.<locals>.<listcomp>.<listcomp>)rt )Zassignr r
)r Ú_rI )r r r à s ú
|
þÿz@test_rolling_groupby_with_fixed_forward_many.<locals>.<listcomp>)r r. r Úlistr_ r
Úfloat64r0 r r r rC rq r r Z set_indexr# r$ )r r2 rE r r& ÚgroupsZmanualr rI r Ú,test_rolling_groupby_with_fixed_forward_many¤ s" ýÿ
|
|
ùÿ r c C sÀ G dd dt } | }td |¡}d}tjt|d | ¡ W 5 Q R X tjt|d tt| W 5 Q R X tjt|d |j dd W 5 Q R X tjt|d |j
|
dd W 5 Q R X d S )Nc @ s e Zd Zdd ZdS )z4test_unequal_start_end_bounds.<locals>.CustomIndexerc S s t dg¡t ddg¡fS ©Nr- rH ©r. r ©r r4 r5 r6 r7 r8 r r r r Ô s zFtest_unequal_start_end_bounds.<locals>.CustomIndexer.get_window_boundsNr r r r r r< Ó s r< r- r9 r T©Úpairwise© r r r r r r r" ÚnextÚiterri rh ©r< r Zrollr r r r Ú test_unequal_start_end_boundsÒ s r¡ c C sÄ G dd dt } | }tddg |¡}d}tjt|d | ¡ W 5 Q R X tjt|d tt| W 5 Q R X tjt|d |j dd W 5 Q R X tjt|d |j
|
dd W 5 Q R X d S )Nc @ s e Zd Zdd ZdS )z4test_unequal_bounds_to_object.<locals>.CustomIndexerc S s t dg¡t dg¡fS r r r r r r r ê s zFtest_unequal_bounds_to_object.<locals>.CustomIndexer.get_window_boundsNr r r r r r< é s r< r- z start and endr Tr r r r r r Ú test_unequal_bounds_to_objectç s r¢ )3Únumpyr. r Zpandasr r r r r Zpandas._testingZ_testingr# Zpandas.api.indexersr r Z pandas.core.indexers.objectsr r
|
r Zpandas.tseries.offsetsr r r( rF rK ÚmarkZ parametrizer
rf rM rN rP rQ rR re rg ro rs rv rz r r r r r Z from_arraysr r r r¡ r¢ r r r r Ú<module> sp
|
"üö ñö ñüÕþ5
|
* öþþþ
|
þþ
|
|
"ÿ ÿ(ýþ "÷þ
|
üý ýÿî ÿç÷õþ4
|
÷þ
|
