U
|
®ý°dY[ ã @ s| d dl Zd dlZd dlmZmZmZmZmZm Z m
|
Z
|
d dl m Z dd Zdd Zej ddd e d
|
g¡g¡d d Zej d dejdejjdddg¡dd Zdd Zdd Zdd Zdd Zd d Zej d edd
|
d gd!d"d#gd$dgd!gd$d gfdd
|
gd!d"gd$d dgfdd
|
d gd!d"d#gd$d dd
|
gfgd fedd
|
d gd!d"d#gd$dgd!gd$d gfdd
|
gd!d"gd$d dgfdd
|
d gd!d"d#gd$d dd
|
gfgd
|
fedd
|
d gd!d"d#gd$dgd!gd$d gfdd
|
gd!d"gd$d dgfdd
|
d gd!d"d#gd$d dd
|
gfgdfedgd!gd$g d
|
fe i g fgdfedej d gej d"d#gd$d%gej gd$d gfdej gej d"gd$d dgfdej d gej d"d#gd$d dd
|
gfgd fedej d gej d"d#gd$d%gej gd$d gfdej gej d"gd$d dgfdej d gej d"d#gd$d dd
|
gfgd
|
fedej d gej d"d#gd$d%gej gd$d gfdej gej d"gd$d dgfdej d gej d"d#gd$d dd
|
gfgdfg¡d&d' Z ej d(edd
|
d gdgd gfdd
|
gd dgfdd
|
d gd dd
|
gfgd fedd
|
d gdgd gfdd
|
gd dgfdd
|
d gd dd
|
gfgd
|
fedd
|
d gdgd gfdd
|
gd dgfdd
|
d gd dd
|
gfgdfedd
|
gdgd gfdd
|
gd dgfgd
|
feej d
|
gej gd gfej d
|
gd dgfgd
|
feg d)d*g d
|
fg¡d+d, Z d-d. Z d/d0 Z ej d1d2d3g¡d4d5 Z!ej d6dd d7d8g¡ej d1d9d:d;g¡ej d<d=d>g¡ej d?d=d>g¡ej d@dAdBdCg¡dDdE Z"dFdG Z#dHdI Z$dJdK Z%dLdM Z&dNdO Z'dPdQ Z(ejjdRdSej)fdTdUdV fd:dWdV fd9dXdV fgdSdTd:d9gdYdZd[ Z*ejjdRdSej)fdTej+fd:ej,fd9ej-fgdSdTd:d9gdYd\d] Z.d^d_ Z/d`da Z0ej dbdcdV dddV g¡dedf Z1ej dbdgdV dhdV didV djdV dkdV dldV dmdV dndV dodV dpdV dqdV drdV dsdV dtdV dudV g¡dvdw Z2dxdy Z3dzd{ Z4d|d} Z5d~d Z6dd Z7dd Z8dd
Z9dd Z:ej dddg¡ej dd=d>g¡dd Z;ej de<e=g¡dd Z>ej dddg¡ej dd=d>g¡ej de<e=g¡dd Z?dS )é N)Ú DataFrameÚ DatetimeIndexÚIndexÚ
|
MultiIndexÚSeriesÚisnaÚnotnac C s. t ddddtjdgi} | | d¡ ¡ d S )NÚBr é é é )r ÚnpÚnanÚ expandingÚsum©Údf© r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/window/test_expanding.pyÚtest_doc_string s r c C s | t dj}|dd d S )Né r
|
©Ú min_periods)Úranger )Úframe_or_seriesÚcr r r Útest_constructor s r Úwç @Úfoor c C s: | t dj}d}tjt|d ||d W 5 Q R X d S )Nr z min_periods must be an integer©Úmatchr )r r ÚpytestÚraisesÚ
|
ValueError)r r r Úmsgr r r Útest_constructor_invalid s r& Úexpanderr
|
Zlsz,GH#16425 expanding with offset not supported)Úreason)Zmarksc C sX t }t | ¡ ¡ }t ||¡ t tg d}t tg d | ¡ ¡ }t ||¡ d S )N©Úindex)r r r ÚtmÚassert_frame_equalr )r' ÚexpectedÚresultr r r Útest_empty_df_expanding) s r/ c C s^ t tjg} | jdd ¡ }t dg}t ||¡ | jdd ¡ }t tjg}t ||¡ d S )Nr r g r
|
)r r r r r r+ Úassert_series_equal)Úxr. r- r r r Útest_missing_minp_zeroD s
|
r2 c C s t t d¡}| | ¡}|dkr8t dd tdD }n*t tjgd dd td d
|
D gd }|jd | d ¡ }t ||¡ d S ) N)é
|
é r c S s. i | ]&}|t jgd dd tddD qS )r c S s g | ] }t |qS r ©Úfloat)Ú.0Újr r r Ú
|
<listcomp>Y s z2test_expanding_axis.<locals>.<dictcomp>.<listcomp>é é )r r r ©r7 Úir r r Ú
|
<dictcomp>Y s z'test_expanding_axis.<locals>.<dictcomp>r4 r c S s g | ] }t |qS r r5 r< r r r r9 ] s z'test_expanding_axis.<locals>.<listcomp>r: é r3 ©Úaxis)
|
r r ZonesZ_get_axis_numberr r r r r+ r, )Z
|
axis_framer rA r- r. r r r Útest_expanding_axisR s
|
ÿ*rB c C s> | t djdd ¡ }| tjtjdddg}t ||¡ d S )Nr r: r ç @ç @ç @©r r Úcountr r r+ Ú assert_equal©r r. r- r r r Ú%test_expanding_count_with_min_periodsc s rJ c C sN dddt jdddg}ddd d d
|
d d g}| | ¡ ¡ }| |}t ||¡ d S ) Nr
|
r r: r r é ç ð?r rC rD rE g @)r r r rG r+ rH )r ÚvaluesZexpected_countsr. r- r r r Ú9test_expanding_count_default_min_periods_with_null_valuesj s
|
rN c C sD | t djdd ¡ }| tjtjtjtjtjg}t ||¡ d S )Nr rK r rF rI r r r Ú=test_expanding_count_with_min_periods_exceeding_series_lengtht s rO zdf,expected,min_periodsr: r r rK )ÚAr rL c C s8 dd |D }t || |¡D ]\}}t ||¡ q d S )Nc S s g | ]\}}t ||d qS ©r) )r ©r7 rM r* r r r r9 º s z1test_iter_expanding_dataframe.<locals>.<listcomp>)Úzipr r+ r, )r r- r Úactualr r r Ú test_iter_expanding_dataframe{ s ?rU zser,expected,min_periodsÚint64©Údtypec C s8 dd |D }t || |¡D ]\}}t ||¡ q d S )Nc S s g | ]\}}t ||d qS rQ )r rR r r r r9 Í s z.test_iter_expanding_series.<locals>.<listcomp>)rS r r+ r0 )Úserr- r rT r r r Útest_iter_expanding_seriesÀ s rZ c C s0 t } tjtdd | jdd W 5 Q R X d S )Nz .* got an unexpected keywordr T)Úcenter)r r" r# Ú TypeErrorr r r r r Útest_center_invalidÓ s r] c C sX | dddg}| ¡ ¡ }t|tr2t|d j}ttjgdgd }t ||¡ d S )Nr r
|
r g»¹øÛ æ?)
|
r ZsemÚ
|
isinstancer r rM r r r+ r0 )r Úobjr. r- r r r Útest_expanding_semÚ s
|
r` ÚmethodÚskewÚkurtc C sL t tj d¡}t| d¡| }|d }t| d¡| }t ||¡ d S )Nr3 r: i )r r ÚrandomÚrandÚgetattrr r+ r0 )ra Úsr- r. r r r Ú,test_expanding_skew_kurt_numerical_stabilityä s
|
rh Úwindowr3 r4 ÚminÚmaxZaverageÚpctTFÚ 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 ) Nr4 ro )Údatarp r: rq rL g Ð?g è?c s | j djd S )N©ra rl rm éÿÿÿÿ)ÚrankÚiloc©r1 ©rm ra rl r r Ú<lambda>ÿ ó ztest_rank.<locals>.<lambda>rs ) r r rd re Úchoicer Úinfr Úapplyru r+ r0 ) ri ra rl rm rn ÚlengthrY r- r. r rx r Ú test_rankî s ÿ
|
ÿr c C sZ | ¡ }|tj t|¡ d d
}| ¡ |¡}|jt|dd |¡}t ||¡ d S ©Néûÿÿÿr
|
©ri r )
|
Zdropnar rd ÚrandnÚlenr ÚcorrÚrollingr+ Úassert_almost_equal©ÚseriesrP r r. Úrolling_resultr r r Útest_expanding_corr s
|
r c C s2 | j dd ¡ }t || jt| dd ¡ ¡ d S )Nr r r )r rG r+ r r r )r r. r r r Útest_expanding_count s
|
ÿr c C s6 | ¡ d¡}| jt| dd d¡}t ||¡ d S )Nç à?r
|
r )r Úquantiler r r+ r )r r. r r r r Útest_expanding_quantile s r c C sV | }|t j t|¡ d d
}| ¡ |¡}|jt|dd |¡}t ||¡ d S r ) r rd r r r Úcovr r+ r r r r r Útest_expanding_cov s
|
r c C s2 | ¡ ¡ }| jt| dd ¡ }t ||¡ d S ©Nr
|
r )r r r r r+ r, ©Úframer. r r r r Útest_expanding_cov_pairwise+ s r c C s2 | ¡ ¡ }| jt| dd ¡ }t ||¡ d S r )r r
r r r+ r, r r r r Ú test_expanding_corr_pairwise3 s r zfunc,static_compr Úmeanc C s t j| ddS ©Nr r@ ©r r rw r r r ry > rz ry c C s t j| ddS r )r rk rw r r r ry ? rz c C s t j| ddS r )r rj rw r r r ry @ rz )Úidsc C s |t ttdt jgd ¡}t|jddd| }t||sFt||d d
}|t krpt
|
|d |¡ nt
|
j |j d |dd d S )Nr3 r
|
r ©r rA r; F©Z check_names)r ÚarrayÚlistr r rf r r^ ÚAssertionErrorr r+ r r0 rv )ÚfuncÚ static_compr rr r. r- r r r Útest_expanding_func: s "r¢ c C sL t tj d¡}t|jddd| }|d d
¡ ¡ s>tt
|
|j d ||d d
¡ t|jddd| }t|j d stt |j d stt tj d
|
¡}t|jd dd| }t|d sÌtt |d sÜtt|jddd| }t|jddd| }t
|
||¡ t|jddd| }t
|
|j d ||d d
¡ d S )Né2 é r r é rt é é é r4 r r: r r
|
) r r rd r rf r r Úallr r+ r rv r )r r¡ rY r. Úser2Úresult0Úresult1r r r Útest_expanding_min_periodsP s r c C s¬ | \}}|t ttdt jgd ¡}|jddjdd ||d}t||sTt|t krt
|
|d t j |d d
d d
|
¡ n(t
|
j |jd t j |d d
d d
|
d d d S ) Nr3 r
|
r c S s | ¡ S ©N©r rw r r r ry s rz z&test_expanding_apply.<locals>.<lambda>©ÚrawÚengineé r; r r@ Fr )r r r r r r r} r^ r r r+ r r r0 rv )Úengine_and_rawr r² r± rr r. r r r Útest_expanding_applyo s " ÿ$ ÿrµ c C s| | \}}t tj d¡}|jddjdd ||d}|d d
¡ ¡ sLtt
|
|j d t |d d
¡¡ |jd djd
|
d ||d}t|j d stt |j d s¬tt tj d ¡}|jddjdd ||d}t|d sètt |d søt|jddjdd ||d}|jddjdd ||d}t
|
||¡ |jddjdd ||d}t
|
|j d t |d d
¡¡ d S )Nr£ r¤ r c S s | ¡ S r® r¯ rw r r r ry rz z2test_expanding_min_periods_apply.<locals>.<lambda>r° r¥ rt r¦ c S s | ¡ S r® r¯ rw r r r ry rz r§ r¨ r4 r c S s | ¡ S r® r¯ rw r r r ry rz r: r r c S s | ¡ S r® r¯ rw r r r ry rz r
|
c S s | ¡ S r® r¯ rw r r r ry rz c S s | ¡ S r® r¯ rw r r r ry ¡ rz )r r rd r r r} r r© r r+ r rv r r )r´ r² r± rY r. rª r« r¬ r r r Ú test_expanding_min_periods_apply sR ÿ ÿ ÿ ÿ ÿ ÿr¶ Úfc C s | j ddj| ddS ©Nr r T©Úpairwise©r r rw r r r ry © rz c C s | j ddj| ddS r¸ ©r r
rw r r r ry ª rz c C s¨ t }t tdgddtg ddd}|d d¡|d< t t |j|jg¡d}t tj|j|jgddgdtdgdddd }| |}t ||¡ | |}t ||¡ d S )
|
NÚar ©ÚnameÚbar©Úcolumnsr* Úfloat64r) )Únames)r* r rX ) r r Úastyper Z from_productr* r r+ r, )r· Údf1Údf2Ú df1_expectedÚ df2_expectedÚ
|
df1_resultÚ
|
df2_resultr r r Ú*test_moment_functions_zero_length_pairwise¦ s ý rÌ c C s | ¡ ¡ S r® )r rG rw r r r ry à rz c C s | j ddj| ddS ©Nr r Fr¹ r» rw r r r ry Ä rz c C s | j ddj| ddS rÍ r¼ rw r r r ry Å rz c C s | j dd ¡ S ©Nr r )r rk rw r r r ry Æ rz c C s | j dd ¡ S rÎ )r rj rw r r r ry Ç rz c C s | j dd ¡ S rÎ )r r rw r r r ry È rz c C s | j dd ¡ S rÎ )r r rw r r r ry É rz c C s | j dd ¡ S rÎ )r Zstdrw r r r ry Ê rz c C s | j dd ¡ S rÎ )r Úvarrw r r r ry Ë rz c C s | j dd ¡ S rÎ )r rb rw r r r ry Ì rz c C s | j dd ¡ S rÎ )r rc rw r r r ry Í rz c C s | j dd d¡S )Nr r r )r r rw r r r ry Î rz c C s | j dd ¡ S rÎ )r Zmedianrw r r r ry Ï rz c C s | j ddjtddS )Nr r F©r± ©r r} r rw r r r ry Ð rz c C s | j ddjtddS )Nr r TrÐ rÑ rw r r r ry Ñ rz c
|
C s| t tjd}|}t }|}tdgd}|d d¡|d< |}| |}t ||¡ | |}t ||¡ | |} t | |¡ d S )NrW r½ ©r rà )r r rà r rÅ r+ r0 r, )
|
r· rg Z
|
s_expectedrÆ rÈ rÇ rÉ Zs_resultrÊ rË r r r Ú!test_moment_functions_zero_lengthÀ s rÓ c C s: | \}}t g tjd}t || ¡ jdd ||d¡ d S )NrW c S s | ¡ S r® r¯ rw r r r ry ì rz z3test_expanding_apply_empty_series.<locals>.<lambda>r° )r r rà r+ r0 r r} )r´ r² r± rY r r r Ú!test_expanding_apply_empty_seriesè s ÿrÔ c C sP | \}}t d d d g}|jddjdd ||d}t dddg}t ||¡ d S ) Nr r c S s t | S r® )r rw r r r ry ô rz z4test_expanding_apply_min_periods_0.<locals>.<lambda>r° rL r rC )r r r} r+ r0 )r´ r² r± rg r. r- r r r Ú"test_expanding_apply_min_periods_0ð s
|
rÕ c C sä t dddgdddgd} t ddgddgd}| ¡ |¡}t d d dg}t ||¡ t dd dgdddgd}| ¡ |¡}t ||¡ t ddd gdddgd} t dd
|
d gdddgd}| ¡ |¡}t d d d d g}t ||¡ d S ) Nr
|
r r: r r) r é é r3 r³ g @)r r r r+ r0 ©Ús1Ús2r. r- Zs2ar r r Ú test_expanding_cov_diff_indexù s rÛ c C sä t dddgdddgd} t ddgddgd}| ¡ |¡}t d d dg}t ||¡ t dd dgdddgd}| ¡ |¡}t ||¡ t ddd gdddgd} t dd
|
d gdddgd}| ¡ |¡}t d d d dg}t ||¡ d S ) Nr
|
r r: r r) rL rÖ r× r3 r³ )r r r
r+ r0 rØ r r r Ú test_expanding_corr_diff_index s rÜ c C sn t ddgddgddggtddgdd d
|
} t ddgddggd dgtddgdd d }t dd gd d gddggtddgdd d
|
}t dd gddggd dgtddgdd d }| ¡ j|ddjd }| ¡ j|ddjd }| ¡ j|ddjd }| ¡ j|ddjd }t ddgddggtddgdd tddgdd d}t ||¡ t ||¡ t ||¡ t ||¡ d S )Nr
|
r r: r r³ rP r r r¾ rÒ r ©r* r rK ÚXÚYTr¹ g Àg Àg Àg $ÀrÁ )r r r r Úlocr+ r, © rÆ Zdf1arÇ Zdf2ar¬ Zresult2Zresult3Zresult4r- r r r Ú'test_expanding_cov_pairwise_diff_length s8 * ÿ ÿ ÿý râ c C sn t ddgddgddggddgttdddd } t ddgddggtd
|
dgddddgd }t d d gd d gddggddgttdddd }t d d gddggtd
|
dgddddgd }| ¡ j|ddjd }| ¡ j|ddjd }| ¡ j|ddjd }| ¡ j|ddjd }t ddgddggddgtddgd }t ||¡ t ||¡ t ||¡ t ||¡ d S )Nr
|
r r: r rP r rÀ r¾ rÁ r rÝ r rK rÞ rß Tr¹ g ð¿)r r r r r
rà r+ r, rá r r r Ú(test_expanding_corr_pairwise_diff_length< sB ÿ ÿý ÿ
|
ÿ rã c C s dd }| \}}t tj dd¡}| ¡ jtj||dd }| ¡ j|||dd}t ||¡ | ¡ j||d did
|
}t ||¡ d S ) Nc S s t | ¡| S r® r )r1 Úconstr r r Ú
|
mean_w_argZ s z4test_expanding_apply_args_kwargs.<locals>.mean_w_argr4 r: )r² r± g 4@)r4 )r² r± Úargsrä )r± Úkwargs) r r rd re r r} r r+ r, )r´ rå r² r± r r- r. r r r Ú test_expanding_apply_args_kwargsY s rè c C s¤ | }t dgddd}|d t¡|d< | ¡ }t||d }|d k r ||d}|rZddgndddg}|| |g¡jd d
|
t¡}t|j |kst
|
t ||¡ d S ) Nr
|
r r: ©r½ Úbr r ©Ú numeric_onlyr½ rê T©Zdrop) r rÅ Úobjectr rf ÚaggÚ reset_indexr6 r r r r+ r, ) Úarithmetic_win_operatorsrì Úkernelr r Úopr. r r- r r r Útest_numeric_only_framej s
|
rô rò r
r Úuse_argc C s¸ t dddgddd}|d t¡|d< |r2|fnd}| ¡ }t|| }||d|i}|rbdd gndd dg}|| t¡} |r| fnd}
|
| ¡ } t| | } | |
|
d|i} t || ¡ d S )
|
Nr
|
r r: ré r r rì r½ rê )r rÅ rî r rf r6 r+ r, )rò rì rõ r Úargr ró r. r rÇ Úarg2Ú
|
expanding2Úop2r- r r r Ú test_numeric_only_corr_cov_frame{ s
|
|
rú rX c
|
C s | }t dg|d}| ¡ }t||}|rb|tkrbd| d}tjt|d ||d W 5 Q R X n0||d}| |g¡jdd t
|
¡} t || ¡ d S ) Nr
|
rW ú
|
Expanding.ú does not implement numeric_onlyr rë Trí ) r r rf rî r" r# ÚNotImplementedErrorrï rð rÅ r6 r+ r0 )
|
rñ rì rX rò rY r ró r% r. r- r r r Útest_numeric_only_series s
|
|
rþ c C sÊ t dddg|d}|r |fnd}| ¡ }t|| }|rt|tkrtd| d}tjt|d ||d |i W 5 Q R X nR||d |i} | t¡}
|
|r|
|
fnd} |
|
¡ } t| | } | | d |i}t
|
| |¡ d S )
|
Nr
|
r r: rW r rû rü r rì ) r r rf rî r" r# rý rÅ r6 r+ r0 )rò rõ rì rX rY rö r ró r% r. rª r÷ rø rù r- r r r Ú!test_numeric_only_corr_cov_series¢ s
|
|
|
rÿ )@Únumpyr r" Zpandasr r r r r r r Zpandas._testingZ_testingr+ r r ÚmarkZ parametrizer r& ÚparamZxfailr/ r2 rB rJ rN rO r rU rZ r] r` rh r r r r r r r r r¢ r rk rj r rµ r¶ rÌ rÓ rÔ rÕ rÛ rÜ râ rã rè rô rú Úintrî rþ rÿ r r r r Ú<module> sD $
|
ÿþþþ
|
|
ýù
|
ýù
|
ýù "ýù
|
"ýù
|
"ýùÐþ=
|
<<<(.úþ
|
|
|
|
|
|
ü
|
ø
|
|
"
|
ý
|
'þþ
|
ñþ
|
|
|
