fume-manage-python.git

U  
®ý°d8Gã@sddlmZddlZddlZddlmmZddlm    Z    m
Z
mZmZm Z mZddlmZddlmZej dejdigejdigdd    d
igejdigejdigejd igdd    digdd    dddigdd    digdd    dddigg
¡ddZej dejdigejdigdd    d
igejdigejdigejd igdd    digdd    dddigdd    digdd    dddigg
¡ddZej dejdidgejdidgd d    d
idgejdidgejdidgejd idgd!d    didgd"d    dddidgd#d    didgd$d    dddidgg
¡d%d&Zej dejdidgejdidgd'd    d
idgejdidgejdidgejd idgd(d    didgd)d    dddidgd*d    didgd+d    dddidgg
¡d,d-Zej dejdigejdigejdigejdigejd igd.d    digd/d    dddigd0d    digd1d    dddigg    ¡d2d3Zd4d5Z ej d6digdigdigdigd igdigdddigdigdddigg    ¡ej d7dd8d9g¡d:d;Z!d<d=Z"ej d>did?gdid?gd
idgdid?gdid?gd id?gdid?gdddid?gdid?gdddid?gg
¡d@dAZ#ej dBdiddgdiddgd
iddgdiddgdiddgd iddgdiddgdddiddgdiddgdddiddgg
¡dCdDZ$ej dBdiddgdiddgd
iddgdiddgdiddgd iddgdiddgdddiddgdiddgdddiddgg
¡dEdFZ%ej dGdHd    dId    dJd    dKd    dLd    dMd    dNd    dOd    dPd    dQd    dRd    dSd    dTd    dUd    ej&dVd    ej'dWg¡dXdYZ(dZd[Z)d\d]Z*d^d_Z+d`daZ,dbdcZ-ej dddedfdgDdhdfdiD¡djdkZ.ej dGdld    dmd    dnd    dod    dpd    dqd    drd    dsd    dtd    dud    dvd    dwd    dxd    dyd    dzd    ej&d{d    ej'dWg¡d|d}Z/dS)~é©ÚdatetimeN)Ú    DataFrameÚ DatetimeIndexÚSeriesÚconcatÚisnaÚnotna)Úoffsetszcompare_func, roll_func, kwargsÚmeanÚsumcCst |¡ t¡ ¡S©N©ÚnpÚisfiniteÚastypeÚfloatr©Úx©rúad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/window/test_rolling_functions.pyÚ<lambda>órÚcountÚmedianÚminÚmaxcCstj|ddS©Né©Úddof©rÚstdrrrrr"rr"cCstj|ddS©Nrrr!rrrrr#rr cCstj|ddSr©rÚvarrrrrr$rr%cCstj|ddSr#r$rrrrr%rcCsjt|jd|d|f|}t|ts(ttdt||p8ddd}t |j    d|||d|¡dS)Né2©Ústeprréÿÿÿÿ)
ÚgetattrÚrollingÚ
isinstancerÚAssertionErrorÚrangeÚlenÚtmÚassert_almost_equalÚiloc)ÚseriesÚcompare_funcÚ    roll_funcÚkwargsr(ÚresultÚendrrrÚtest_seriessr9cCst |¡ t¡ ¡Sr rrrrrr5rcCstj|ddSrr!rrrrr<rcCstj|ddSr#r!rrrrr=rcCstj|ddSrr$rrrrr>rcCstj|ddSr#r$rrrrr?rcCst|jd|d|f|}t|ts(ttdt||p8ddd}tj|j    dddf|j    |d|ddfj
|d|ddddS)    Nr&r'rrr))ZaxisÚrawF©Zcheck_names)r*r+r,rr-r.r/r0Úassert_series_equalr2Úapply)r:Úframer4r5r6r(r7r8rrrÚ
test_frame/s$ýr?z%compare_func, roll_func, kwargs, minpé
cCst |¡ t¡ ¡Sr rrrrrrRrcCstj|ddSrr!rrrrrVrcCstj|ddSr#r!rrrrrWrcCstj|ddSrr$rrrrrXrcCstj|ddSr#r$rrrrrYrcCs~d}|ddd d¡ ¡}t|j||d|f|}|jd}|dt ¡}    |ddd |    |¡}
t     |d||
¡dS)NééÚB©ÚwindowÚmin_periodsr)é)
Úresamplerr*r+Úindexr
ÚBDayÚtruncater0r1)r3r4r5r6ÚminpÚwinZserZ series_resultÚ    last_dateÚ    prev_dateZtrunc_seriesrrrÚtest_time_rule_seriesMsÿ
rPcCst |¡ t¡ ¡Sr rrrrrrnrcCstj|ddSrr!rrrrrrrcCstj|ddSr#r!rrrrrsrcCstj|ddSrr$rrrrrtrcCstj|ddSr#r$rrrrrurcCsd}|ddd d¡ ¡}t|j||d|f|}|jd}    |    dt ¡}
|ddd |
|    ¡}tj    | 
|    ¡|j||ddd    dS)
NrArBrCrDr)rG©r:Fr;)rHrr*r+rIr
rJrKr0r<Úxsr=)r:r>r4r5r6rLrMÚfrmZframe_resultrNrOZtrunc_framerrrÚtest_time_rule_frameisÿ
ýrTcCstj|ddSrr!rrrrrrcCstj|ddSr#r!rrrrrrcCstj|ddSrr$rrrrrrcCstj|ddSr#r$rrrrrrcCsfttj d¡}tj|dd<tj|dd<t|jddd|f|}t |j    d||dd¡t|jddd|f|}t
|j    d    stt
|j    d
r¢tt
|j    dr´tt
|j    dsÆtttj d¡}t|jdd d|f|}t
|j    dstt|j    dst|dkrbt|jddd|f|}t|jddd|f|}t ||¡dS)Nr&r@éöÿÿÿé©rFr)ééérGiúÿÿÿéûÿÿÿééérrr) rrÚrandomÚrandnÚNaNr*r+r0r1r2rr-r    )r4r5r6Úobjr7Úobj2Zresult0Zresult1rrrÚ    test_nanss$
rdcCslttj d¡}tj|dd<tj|dd<|jddd ¡}t |j    dt 
|dd¡ t¡  ¡¡dS)Nr&r@rUrVrWr))rrr_r`rar+rr0r1r2rrrr)rbr7rrrÚtest_nans_count²sÿrezroll_func, kwargsrLécédcCszt|jt|d||d|f|}t|jt|||d|f|}t|}t |t|¡|}t ||||¡dS)Nr©rFr()r*r+r/rr0r<r1)r3rLr5r6r(r7ÚexpectedÚnan_maskrrrÚtest_min_periods¼s ÿþÿþrkcCsj|jt|dd|d ¡}|jt|d|d ¡}t|}t |t|¡|}t ||||¡dS)Nrrrh)r+r/rrr0r<r1)r3r(r7rirjrrrÚtest_min_periods_countÙsrlzroll_func, kwargs, minprYcCsttj d¡}tj|dd<tj|dd<t|jd|dd|f|}tt|ttjgdgjd|d|f|jddj    dd    }t
 ||¡dS)
Nr&r@rUrXT)rFÚcenteré    rW)Zdrop)rrr_r`rar*r+rr2Zreset_indexr0r<)r5r6rLrbr7rirrrÚtest_centerãs$ÿ ÿþýüÿroz#roll_func, kwargs, minp, fill_valuecCsddtdD}t| t|j|¡jd|d|f| d¡ |j¡}t|jd|dd|f|}|dk    rx| |¡}t     ||¡dS)    NcSsg|]}d|dqS©rÚdr©Ú.0rrrrÚ
<listcomp>sz.test_center_reindex_series.<locals>.<listcomp>érArDéôÿÿÿT©rErFrm)
r.r*ÚreindexÚlistrIr+ÚshiftÚfillnar0r<)r3r5r6rLÚ
fill_valueÚsZ    series_xpZ    series_rsrrrÚtest_center_reindex_seriess*þýüûÿÿþ
r~cCsddtdD}t| t|j|¡jd|d|f| d¡ |j¡}t|jd|dd|f|}|dk    rx| |¡}t     ||¡dS)    NcSsg|]}d|dqSrprrrrrrrt8sz-test_center_reindex_frame.<locals>.<listcomp>rurArDrvTrw)
r.r*rxryrIr+rzr{r0Úassert_frame_equal)r>r5r6rLr|r}Zframe_xpZframe_rsrrrÚtest_center_reindex_frame's*þýüûÿÿþ
rÚfcCs|jdddj|ddS©Nr@r\rDF)Úpairwise©r+ZcovrrrrrMrcCs|jdddj|ddSr©r+ZcorrrrrrrNrcCs|jddd ¡S©Nr@r\rD©r+rrrrrrOrcCs|jddd ¡Sr©r+rrrrrrPrcCs|jddd ¡Sr©r+rrrrrrQrcCs|jddd ¡Sr©r+rrrrrrRrcCs|jddd ¡Sr©r+r"rrrrrSrcCs|jddd ¡Sr©r+r%rrrrrTrcCs|jddd ¡Sr©r+ZskewrrrrrUrcCs|jddd ¡Sr©r+ZkurtrrrrrVrcCs|jdddjddS)Nr@r\rDçà?)Úquantile©r+rrrrrrWrcCs|jddd ¡Sr©r+rrrrrrXrcCs|jdddjtddS©Nr@r\rDFrQ©r+r=rrrrrrYrcCs|jdddjtddS©Nr@r\rDTrQrrrrrrZrcCs|jdddd ¡S©NZboxcarr@r\)Zwin_typerErFrrrrrr\r)ZmarkscCsttd}ttj|jd}tddgddgddgdd    ggd
dgd}ttj|j|jd }||}t ||¡||}t     ||¡dS)Nr^©rIrr\r]rBrnr)rÚArC©Úcolumns)rIr)
rr.rÚnanrIrrr0r<r)rr}Ú
s_expectedZdfZdf_expectedÚs_resultZ    df_resultrrrÚ+test_rolling_functions_window_non_shrinkageJs(rcCs¶ddtddD}| tddddd¡ttdd|d    }| d
d¡}| ¡}tdd dddgtddtddDddd    dd|}| d¡ ¡j    d|d ¡}t
 ||¡dS)z%Replicate result expected in GH #6297cSsg|]}tdd|qS©é·rr©rsÚirrrrtqsz+test_rolling_max_gh6297.<locals>.<listcomp>rér r]rércSst|Sr ©rrrrrrvrz)test_rolling_max_gh6297.<locals>.<lambda>çð?ç@g@ç@g@cSsg|]}tdd|dqS©r rrrr¡rrrrt|sÚD©ÚfreqN©rEr()r.ÚappendrrÚmapÚ
sort_indexrrHrr+r0r<)r(Úindicesr3rirrrrÚtest_rolling_max_gh6297osþýr²cCsddtddD}| tdddd¡| tdddd¡tttddd    d
g|d}| dd ¡}| ¡}tdddddgtddtddDddddd|}| d¡     ¡j
d|d     ¡}t ||¡tdddddgtddtddDddddd|}| d¡  ¡j
d|d     ¡}t ||¡d}tdddd|gtddtddDddddd|}| d¡ ¡j
d|d     ¡}t ||¡dS)NcSsg|]}tdd|qSrrr¡rrrrtsz-test_rolling_max_resample.<locals>.<listcomp>rr£r r\rBrr@rXrcSst|Sr r¥rrrrrrz+test_rolling_max_resample.<locals>.<lambda>çr¦r§ç@g4@cSsg|]}tdd|dqSr©rr¡rrrrtsrªr«rg$@cSsg|]}tdd|dqSr©rr¡rrrrtsg«ªªªªª&@cSsg|]}tdd|dqSr©rr¡rrrrt¡s)r.r®rrryr¯r°rrHrr+r0r<rr)r(r±r3rirÚvrrrÚtest_rolling_max_resamples>þýþýþýr¶cCsÔddtddD}| tdddd¡| tdddd¡tttddd    d
g|d}| dd ¡}| ¡}tdddddgtddtddDddddd|}| d¡     ¡j
d|d}t ||     ¡¡dS)NcSsg|]}tdd|qSrrr¡rrrrt¨sz-test_rolling_min_resample.<locals>.<listcomp>rr£r r\rBrr@rXrcSst|Sr r¥rrrrr®rz+test_rolling_min_resample.<locals>.<lambda>r³r¦r§r´r¨cSsg|]}tdd|dqSr©rr¡rrrrtµsrªr«r) r.r®rrryr¯r°rrHrr+r0r<)r(r±r3riÚrrrrÚtest_rolling_min_resample§sþýr¸cCsÈddtddD}| tdddd¡| tdddd¡tttddd    d
g|d}| dd ¡}| ¡}tddddd    gtddtddDddd}| d¡     ¡j
dd     ¡}t ||¡dS)NcSsg|]}tdd|qSrrr¡rrrrt¼sz0test_rolling_median_resample.<locals>.<listcomp>rr£r r\rBrr@rXrcSst|Sr r¥rrrrrÂrz.test_rolling_median_resample.<locals>.<lambda>r³r¦r§r´cSsg|]}tdd|dqSr©rr¡rrrrtÉsrªr«©rE) r.r®rrryr¯r°rrHrr+r0r<)r±r3rirrrrÚtest_rolling_median_resample»sþrºcCsDd}ttj |¡jddd ¡ttj |¡jddd ¡dS)Ni NrBF)rErm)rrr_r`r+r)ÚnrrrÚ test_rolling_median_memory_errorÏsr¼Ú    data_typecCsg|]}t d|¡qS)r©rÚdtype)rsÚwidthrrrrtØsrt)r^écCs(g|] }dD]}t ||¡qqS)Zuir¾)rsrÀÚsignrrrrtÙs)rrBr^rÁcCspttjd|djdd ¡}|jdt d¡ks6tttjd|djdd ¡}|jdt d¡ksltdS)NrX©r¿r\r¹rZf8)    rrZaranger+rZdtypesr¿r-r)r½r7rrrÚ"test_rolling_min_max_numeric_typesÖsrÄcCs|jddd ¡S)Nr@rrD)r+rrrrrrêrcCs|jdddj|ddSrrrrrrrërcCs|jdddj|ddSrrrrrrrìrcCs|jddd ¡SrrrrrrrírcCs|jddd ¡SrrrrrrrîrcCs|jddd ¡SrrrrrrrïrcCs|jddd ¡SrrrrrrrðrcCs|jddd ¡SrrrrrrrñrcCs|jddd ¡SrrrrrrròrcCs|jddd ¡SrrrrrrrórcCs|jddd ¡SrrrrrrrôrcCs|jddd d¡S)Nr@r\rDrrrrrrrõrcCs|jddd ¡SrrrrrrrörcCs|jdddjtddSrrrrrrr÷rcCs|jdddjtddSrrrrrrrørcCs|jdddd ¡Srrrrrrrúrc
Cs|ttjd}|}t}|}tdgd}|d d¡|d<|}||}t ||¡||}t ||¡||}    t |    |¡dS)NrÃÚarÚfloat64)rrrÆrrr0r<r)
rr}rZdf1Zdf1_expectedZdf2Zdf2_expectedrZ
df1_resultZ
df2_resultrrrÚ!test_moment_functions_zero_lengthçsrÇ)0rÚnumpyrZpytestZpandas.util._test_decoratorsÚutilZ_test_decoratorsÚtdZpandasrrrrrr    Zpandas._testingZ_testingr0Zpandas.tseriesr
ÚmarkZparametrizerZnansumrrrr9r?rPrTrdrerkrlror~rÚparamZskip_if_no_scipyrr²r¶r¸rºr¼rÄrÇrrrrÚ<module>s 
 
ý
 
 
òþ
 
 
ý
 
 
òþ
öþ
 öþ
 
 
 
 
 
÷þ
 
 
 
÷þ
öþ
 
 
 
 
 
 
 
 
öþ
 
 
 
 
 
 
 
 
öþ
þñþ
%ÿþ
þðþ