gL i^` ddlZddlZddlmZmZmZmZmZm Z m Z ddl m Z dZdZej j#dddej$dggd Zej j#d d ej(d ej j+d gdZdZdZdZdZdZej j#degdgddd gdgddgfd dgddgddd gfgdgddgdfgdfegdgddd gdgddgfd dgddgddd gfgdgddgdfgdfegdgddd gdgddgfd dgddgddd gfgdgddgdfgd fed gdgdgdfeigfgd fed ej8dgej8ddgddgej8gddgfd ej8gej8dgddd gfd ej8dgej8ddgdgdfgdfed ej8dgej8ddgddgej8gddgfd ej8gej8dgddd gfd ej8dgej8ddgdgdfgdfed ej8dgej8ddgddgej8gddgfd ej8gej8dgddd gfd ej8dgej8ddgdgdfgd fgd Zej j#d!egdd gdgfd dgdd gfgdgdfgdfegdd gdgfd dgdd gfgdgdfgdfegdd gdgfd dgdd gfgdgdfgd fed dgd gdgfd dgdd gfgdfeej8dgej8gdgfej8dgdd gfgdfegd"#gdfgd$Zd%Zd&Z ej j#d'd(d)gd*Z!ej j#d+gd,ej j#d'gd-ej j#d.d/d0gej j#d1d/d0gej j#d2gd3d4Z"d5Z#d6Z$d7Z%d8Z&d9Z'd:Z(ej j#d;dfd?d@fdAdBfggdCDdEZ*ej j#d;dej j#dedfdggej j#dhd/d0gej j#djer  z7Support for axis=1 in DataFrame.expanding is deprecatedr=r$axis) rrones_get_axis_numberrrfloatr3assert_produces_warning FutureWarningrrr4) axis_framerrCr)ijr5r6s rtest_expanding_axisrLRs 2778$ %B  z *D qyGINr SAQ1 %2,?Qa?? ? S  Hrvvhla -M1eAh-MMNQSST # #M =8aj15578&(+ @ S .N88s* 'D; 1D6D; ;E 5"E6D; Ec|tdjdj}|tjtjdddg}t j ||y)Nrr?r@@@rrcountrrr3 assert_equalrr6r5s r%test_expanding_count_with_min_periodsrUfsR U1X & 0 0Q 0 ? E E GFS#>?HOOFH%rcdddtjdddg}gd}||jj}||}t j ||y)Nr r r?rr)?r!rNrNrOrPg@)rrrrRr3rS)rvaluesexpected_countsr6r5s r9test_expanding_count_default_min_periods_with_null_valuesr[msSArvvq!Q 'F9O V $ . . 0 6 6 8F/HOOFH%rc4|tdjdj}|tjtjtjtjtjg}t j ||y)NrrWrrQrTs r=test_expanding_count_with_min_periods_exceeding_series_lengthr]ws^ U1X & 0 0Q 0 ? E E GFGHHOOFH%rzdf,expected,min_periodsr r r?)rrrW)Ar rrrr r r?rWrXc|Dcgc]\}}t||}}}t||j|D]\}}tj||ycc}}wr0)rziprr3r4)rr5rrYr2actuals rtest_iter_expanding_dataframerd~s_~GOO?FE &.OHO",,{*CD0& fh/0PAzser,expected,min_periodsint64dtypec|Dcgc]\}}t||}}}t||j|D]\}}tj||ycc}}wr0)rrbrr3r9)serr5rrYr2rcs rtest_iter_expanding_seriesrks^DLLvU+LHL#-- *DE1& vx01Mrect}tjtd5|j ddddy#1swYyxYw)Nz.* got an unexpected keywordr$T)center)rr&r' TypeErrorrrs rtest_center_invalidros; B y(F G" D !"""s AA c|gd}|jj}t|trt |dj }t t jgdgdzz}tj||y)Nr`rg۞?r ) rsem isinstancerrrYrrr3r9)robjr6r5s rtest_expanding_semrtsi ) $C ]]_ "F&)$q (()rvvh(a/0H68,rmethodskewkurtc(ttjjdjd}t |j d|}|dz}t |j d|}t j||y)Nr r=r?i)rrrandom default_rnggetattrrr3r9)rusr5r6s r,test_expanding_skew_kurt_numerical_stabilityr}ss ryy$$Q'..r23A.wq{{1~v.0H DA ,WQ[[^V , .F68,rwindow)r r?r=r>)minmaxaveragepctTF ascending test_data)default duplicatesnansc d}|dk(r9ttjjdj|}n|dk(r:ttjjdj d|}no|dk(rjttjjdj dd d tj tj tj g|}j|jfd }|j|j }tj||y) Nr>rr )datarr?rrXg?g?cF|jjdS)Nrurr)rankiloc)r:rrurs rztest_rank..s!!&&C9&EJJ2Nrr) rrryrzchoicerinfrapplyrr3r9) r~rurrrlengthrjr5r6s ``` r test_rankrs FI"))//299&AB l ""))//299!VDE f &&q)00dD"&&"&&266':F  }}V$**NH]]6 " ' 'v3) ' TF68,rc^|j}|tjjdj t |zdd}|j j|}|jt |dj|}tj||yNr r r~r) dropnarryrzstandard_normallenrcorrrollingr3assert_almost_equalseriesr_r r6rolling_results rtest_expanding_corrr s A RYY " "1 % 5 5c!f = =sCA [[]   "FYYc!f!Y<AA!DN>62rc|jdj}tj||j t |djy)Nrrr)rrRr3rrr)rr6s rtest_expanding_countrsK   !  , 2 2 4Fc&kqAGGIrc|jjd}|jt|djd}t j ||y)N?r r)rquantilerrr3r)rr6rs rtest_expanding_quantilersM     ( ( -F^^3v;A^FOOPSTN6>2rcB|}|tjjdjt |zdd}|j j |}|jt |dj |}tj||yr) rryrzrrrcovrr3rrs rtest_expanding_covr%s}A RYY " "1 % 5 5c!f = =sCA [[]  q !FYYc!f!Y<@@CN>62rc|jj}|jt|dj}t j ||yNr r)rrrrr3r4framer6rs rtest_expanding_cov_pairwiser0sF __  " " $F]]#e*!]DHHJN&.1rc|jj}|jt|dj}t j ||yr)rrrrr3r4rs rtest_expanding_corr_pairwiser8sF __  # # %F]]#e*!]DIIKN&.1rzfunc,static_comprmeanc0tj|dSNrrBrrr:s rrrCs27711-rrc0tj|dSr)rrrs rrrD"&&+rrc0tj|dSr)rrrs rrrErr)rrrr)idsc |tjttdtjgdzz}d}t j t|5|jdd}dddt|}t||sJd}d}|tur|tjurt}t j ||d 5||dd }ddd|turt j|dyt j|j dd y#1swYxYw#1swY]xYw) Nr=z@The 'axis' keyword in (Series|DataFrame).expanding is deprecatedr$r rrrCz:The behavior of DataFrame.sum with axis=None is deprecatedF)r%check_stacklevelr@ check_names)rarraylistrrr3rGrHrr{rrrrrrr9r) func static_comprrr)rsr6warnr5s rtest_expanding_funcr?s 288DrOrvvhm$CD ED LC # #M =4nnn34 WS$  !F fo .. . FC D)# rvv(= # #De L*tCRy)*&  vbz84 v{{2eL44**s D3 D?3D<?Ecttjjdj d}d}t j t|5t|jdd|}dddddjjsJt j|jd ||ddt j t|5t|jd d|}dddt|jd sJt|jd sJttjjdj d }t j t|5t|jdd|}dddt|dsJt|dsJt j t|5t|jdd|}dddt j t|5t|jdd|}dddt jt j t|5t|jdd|}dddt j|jd ||ddy#1swYxYw#1swYxYw#1swYYxYw#1swYxYw#1swYxYw#1swYuxYw)Nr 2z4The 'axis' keyword in Series.expanding is deprecatedr$rrr r>rr?rr )rrryrzrr3rGrHr{rrallrrr )rrrjr)r6ser2result0result1s rtest_expanding_min_periodsr^s &&q)99"= >C @C # #M =HE2A>EGH #2;    ! ! ## #6;;r?KCR,AB # #M =HE2A>EGH  B  R !! ! "))''*::2> ?D # #M =HEAA>EGH q ??    # #M =HE'#--AA->EGH # #M =HE'#--AA->EGH7G, # #M =GD11=tDFG6;;r?KCR,AB5HH HH HH HHHHGGsH#K;*#L#L#L"#L.##L:;LLL"L+.L7:Mc |\}}|tjttdtjgdzz}|j dj d||}t||sJ|tur2tj|dtj|ddd ytj|jdtj|ddd d y) Nr=r rc"|jSNrrs rrz&test_expanding_apply.. !&&(rrawengine r@rrBFr)rrrrrrrrrrr3rrr9r)engine_and_rawrrrrr6s rtest_expanding_applyrs KFC 288DrOrvvhm$CD ED ^^^ * 0 0F1F fo .. .&  vay"''$s)!*DE  KKNBGGD"IA6E rc|\}}ttjjdj d}|j dj d||}|ddjjsJtj|jdtj|dd|j d j d ||}t|jd sJt|jd sJttjjdj d }|j dj d||}t|dsJt|dsJ|j dj d||}|j dj d||}tj|||j dj d||}tj|jdtj|ddy)Nr rrrc"|jSrrrs rrz2test_expanding_min_periods_apply..rrrrrrc"|jSrrrs rrz2test_expanding_min_periods_apply..rrrrr>rc"|jSrrrs rrz2test_expanding_min_periods_apply..rrr?rrc"|jSrrrs rrz2test_expanding_min_periods_apply..rrr c"|jSrrrs rrz2test_expanding_min_periods_apply..rrc"|jSrrrs rrz2test_expanding_min_periods_apply..rr)rrryrzrrrrrr3rrrr )rrrrjr6rrrs r test_expanding_min_periods_applyrs KFC &&q)99"= >C ]]r] * 0 0F1F #2;    ! ! ## #6;;r?BGGCH,=>]]r] * 0 0F1F  B  R !! ! "))''*::2> ?D ^^^ * 0 0F1F q ??   mmm*00F1Gmmm*00F1G7G, ]]q] ) / /F0F6;;r?BGGCH,=>rfcH|jdj|dSNrrTpairwiserrrs rrrs!1;;1;-11!d1CrcH|jdj|dSrrrrs rrrs!1;;1;-221t2Drct}ttdgdtgd}|djd|d<ttj|j |j g}ttj|j |j gddgtdgdd }||}tj||||}tj||y) Nar"namebarcolumnsr2float64r1)names)r2rrh) rrastyper from_productr2rr3r4)rdf1df2 df1_expected df2_expected df1_result df2_results r*test_moment_functions_zero_length_pairwisers +C E3%e4E"5uenUse%(L 3J*l33J*l3rc>|jjSr)rrRrs rrrs!++-%%'rcH|jdj|dSNrrFrrrs rrrs!!++!+,00U0CrcH|jdj|dSrrrs rrrs!!++!+,11!e1DrcB|jdjSNrr)rrrs rrr!++!+,002rcB|jdjSr)rrrs rrrrrcB|jdjSr)rrrs rrrrrcB|jdjSr)rrrs rrr!++!+,113rcB|jdjSr)rstdrs rrrrrcB|jdjSr)rvarrs rrrrrcB|jdjSr)rrvrs rrrr rcB|jdjSr)rrwrs rrrr rcD|jdjdS)Nrrr)rrrs rrrs!++!+,55c:rcB|jdjSr)rmedianrs rrrs!++!+,335rcP|jdjtdS)NrrFrrrrrs rrrs!!++!+,223E2BrcP|jdjtdS)NrrTrrrs rrrs!!++!+,223D2ArcTttj}|}t}|}tdg}|dj d|d<|}||}t j ||||}t j||||} t j| |y)Nrgrrr)rrrrrr3r9r4) rr| s_expectedrrrrs_resultrrs r!test_moment_functions_zero_lengthrs, RZZ AJ +CL SE "C3xy)CHLtH8Z03J*l33J*l3rc|\}}tgtj}tj||j j d||y)Nrgc"|jSrrrs rrz3test_expanding_apply_empty_series..s QVVXrr)rrrr3r9rr)rrrrjs r!test_expanding_apply_empty_seriesrsG KFC 2:: &C S]]_ " "#53v " Nrc|\}}tgd}|jdjd||}tgd}tj||y)N)NNNrrct|Sr)rrs rrz4test_expanding_apply_min_periods_0.. s Arr)rXr!rN)rrrr3r9)rrrr|r6r5s r"test_expanding_apply_min_periods_0r"sT KFC!"A [[Q[ ' - -.>CPV - WFo&H68,rc tgdgd}tddgddg}|jj|}tgd}tj||tgd gd}|jj|}tj||tgd gd }tgd gd }|jj|}tgd}tj||y)Nr^r`r1r r?rr )NNr!r Nr?r=rr r?r&rr=rr r?)NNNg@)rrrr3r9s1s2r6r5s2as rtest_expanding_cov_diff_indexr/s   +B Aq!f %B \\^   #F'(H68, Y /C \\^   $F68,  ) ,B  ) ,B \\^   #F-.H68,rc tgdgd}tddgddg}|jj|}tgd}tj||tgd gd}|jj|}tj||tgd gd }tgd gd }|jj|}tgd}tj||y)Nr^r`r1r r?rr )NNrXr$r%r(r)r*)NNNrX)rrrr3r9r+s rtest_expanding_corr_diff_indexr1!s   +B Aq!f %B \\^  $F'(H68, Y /C \\^  %F68,  ) ,B  ) ,B \\^  $F-.H68,rctddgddgddggtddgd  }tddgddggd dgtddgd  }tdd gddgddggtddgd  }tdd gddggd dgtddgd  }|jj|djd}|jj|djd}|jj|djd}|jj|djd}tddgddggtddgd tddgd }t j ||t j ||t j ||t j ||y)Nr rr?r rr_r r"rrrr2rrWXYTrgggg$r)rrrrlocr3r4 rdf1ardf2arresult2result3result4r5s r'test_expanding_cov_pairwise_diff_lengthr=4s aVaVaV,eS#JU6S TC  Q!QAsCju0M D  Q$1v&sCju0M C  Q!QAsCju0M D mmo!!#!599!G e}%sCju-S#JU+H '8,'8,'8,'8,rc tddgddgddggddgttdd }tddgddggtd dgdddg }td d gddgddggddgttdd }td d gddggtd dgdddg }|jj |dj d}|jj |dj d}|jj |dj d}|jj |dj d}tddgddggddgtddg }t j||t j||t j||t j||y)Nr r r?rr_r rrrrr3rrWr4r5Trg)rrrrrr6r3r4r7s r(test_expanding_corr_pairwise_diff_lengthr?Qs  Q!Q!Q 3*E%(QVGnn##C$#7;;A>Gnn##D4#8<.mean_w_argoswwqzE!!rr )r>r?)rrg4@)r>)rrargsrBr>)rkwargs) rrryrzrrrr3r4)rrCrrrr5r6s r test_expanding_apply_args_kwargsrFns"!KFC 299((+227; ros:  sE8288QC=9:;  ++##E$   ,  , -,(&&& II6 7s!%s+1vQF+aV4 y19=    II6 7s!%s+1vQF+aV4 y19=    II6 7s!%s+1vQF+aV4 y19=    !A3' ("a0 Bxj!$ QN"&&!Q@ AuBFF8,qc2266{"&&!51v>2661~RVVQN;YG    QN"&&!Q@ AuBFF8,qc2266{"&&!51v>2661~RVVQN;YG    QN"&&!Q@ AuBFF8,qc2266{"&&!51v>2661~RVVQN;YG   a9<z0{<z0  qcA3Z1a&1a&)9Iy;QRTUV  qcA3Z1a&1a&)9Iy;QRTUV  qcA3Z1a&1a&)9Iy;QRTUV AA3*1v1v&67;  1#"&&!q!f0EFJ ' "B*  1 1"-FF#34-5->2#<=u .tUm4&GH-I5/>3 -*33322  -. +, +,  & M M* RVV_vrww'%5"&&/J% C  CB  $?NDE44&'CD22232233:5BA(4)(4(--&-&-:-:,"0"FE?3T5M2,34,(3-0 11 1 FE?3T5M23-011341*/r