B •xŠ\Gd2d3„d3e>ƒZ?Gd4d5„d5e>ƒZ@Gd6d7„d7e@ƒZAGd8d9„d9e@ƒZBGd:d;„d;e@ƒZCGdd?„d?e>ƒZEGd@dA„dAeEƒZFGdBdC„dCeEƒZGGdDdE„dEeEƒZHGdFdG„dGeEƒZIGdHdI„dIe>ƒZJGdJdK„dKeJƒZKGdLdM„dMeJƒZLGdNdO„dOeJƒZMGdPdQ„dQe>ƒZNGdRdS„dSe>ƒZOGdTdU„dUeOƒZPGdVdW„dWeOƒZQGdXdY„dYeOƒZRej8 9d'¡GdZd[„d[e6ƒƒZSGd\d]„d]e5ƒZTd^d_„ZUeUƒZVd`da„ZWGdbdc„dce6ƒZXGddde„dee5ƒZYGdfdg„dge5ƒZZdS)hé)ÚOrderedDict)ÚdatetimeÚ timedelta)ÚproductN)Úcatch_warnings)Úrandn)ÚrangeÚzip)ÚUnsupportedFunctionCall)Ú DataFrameÚIndexÚSeriesÚ TimestampÚbdate_rangeÚconcatÚisnaÚnotna)ÚSpecificationError)Ú safe_sort)édé cCs(t|tƒrt ||¡nt ||¡dS)N)Ú isinstancer ÚtmÚassert_series_equalÚassert_frame_equal)ÚleftÚright©rú7lib/python3.7/site-packages/pandas/tests/test_window.pyÚassert_equals rTF)ZparamscCs|jS)N)Úparam)ÚrequestrrrÚraw#sr"ÚtriangÚblackmanÚhammingÚbartlettÚbohmanÚblackmanharrisÚnuttallÚbarthanncCs|jS)N)r )r!rrrÚ win_types(sr+ÚkaiserÚgaussianÚgeneral_gaussiancCs|jS)N)r )r!rrrÚwin_types_special.sr/c@s*eZdZe dd¡Ze g¡Zdd„ZdS)ÚBaseéé(cCsfttƒ}tj||j<||_ttdddƒtd|_t | ¡|jd|_tttt ƒ|jt t ¡d|_dS)NiÙé)Úperiods)Úindex)r5Úcolumns)rÚNÚnpÚNaNÚ _nan_locsÚarrrrÚrngr ÚcopyÚseriesrÚKÚarangeÚframe)Úselfr;rrrÚ_create_data8szBase._create_dataN) Ú__name__Ú __module__Ú__qualname__r8r@r:ÚarrayZ _inf_locsrCrrrrr03s r0c@sHeZdZdd„Zdd„Zdd„Zdd„Zd d „Zdd„Zd d„Z dd„Z dd„Zdd„Zdd„Z ejej d¡dd„ƒƒZdd„Zej dddejejejejgdd d!d gd!d"d#d"gejejejejgd d"d#d"gd#d$d%d$gd%d&d'd(ggfd)d*d+d+d d gdd d!d gd d"d"d gd+d+d"d"gd d"d#d"gd,d$d-d"gd$d&d%d"ggfg¡d.d/„ƒZd*S)0ÚTestApicCs| ¡dS)N)rC)rBÚmethodrrrÚsetup_methodEszTestApi.setup_methodcCs°|jjdd}t |jj|jj¡|jjddd}|jj|jjdksLt‚|jjddd}t |jj|jjddg¡|jjddddg}t |jj|jjddg¡dS)Né)Úwindowr3)r3érM)rAÚrollingrÚassert_index_equalZ _selected_objr6ÚnameÚAssertionError)rBÚrrrrÚtest_getitemHs zTestApi.test_getitemc Csrtddggddgd}|jdd}t t|jdg¡t t|jddg¡tjtd d |ddgWdQRXdS)Nr3éÚAÚB)r6rK)rLÚCz^[^A]+$)Úmatch)rrNÚpytestÚraisesÚKeyErrorÚ__getitem__)rBÚdfÚgrrrÚtest_select_bad_colsYszTestApi.test_select_bad_colscsTtddggddgd}|jdd‰t ˆj ¡ˆd ¡¡t t‡fdd „¡dS) Nr3rTrUrV)r6rK)rLcsˆjS)N)ÚFr)rRrrÚisz/TestApi.test_attribute_access..) rrNrrrUÚsumrYrZÚAttributeError)rBr]r)rRrÚtest_attribute_accessdszTestApi.test_attribute_accesscCsxttdƒtddƒddœƒ}|jdd}|ddg ¡}ttjtjdd d gtjtjddd gdœtdƒd}t ||¡dS)NrKrÚfoo)rUrVrWrM)rLrUrVéé ééé)rUrVÚAB)r6) rrrNrbr8ÚnanÚlistrr)rBr]rRÚresultÚexpectedrrrÚtests_skip_nuisancekszTestApi.tests_skip_nuisancec CsLttdƒtddƒddœƒ}|jdd}tjtdd| ¡WdQRXdS) NrKrre)rUrVrWrM)rLzcannot handle this type)rX)rrrNrYrZÚ TypeErrorrb)rBr]rRrrrÚtest_skip_sum_object_raisesusz#TestApi.test_skip_sum_object_raisesc Cs¤ttdƒtdddƒdœƒ}|jdd}|d ¡}|d ¡}|d ¡}|d ¡}|d ¡}|d ¡}| tjtjg¡} t||||gd d} t j dd gdd gg¡| _t | | ¡| tjtjdœ¡} t||gd d} t j| | dd| ddd gi¡} t||gd d} t j ddg¡| _t | | ¡|d ddg¡} t||gd d} ddg| _t | | ¡tdd&t dt¡| ddddœi¡} WdQRXt||gd d} t j ddg¡| _t j| | ddtdd.t dt¡| dddœdddœdœ¡} WdQRXt||||gd d} ddddg}t j |¡| _t j| | dd| dd gdd gdœ¡} t||||gd d} ddddg}t j |¡| _t j| | dddS)NrKrrrT)rUrVrM)rLrUrVr3)ÚaxisÚmeanÚstdT)Ú check_like)rUrt)rUrurb)ÚrecordÚignore)rtrb)rUrb)Úmean2Úsum2)rVry)rVrz)rVrt)rVru)rrrNrtrurbÚ aggregater8rÚpdÚ MultiIndexÚfrom_productr6rrÚfrom_tuplesrÚwarningsÚsimplefilterÚ FutureWarning)rBr]rRZa_meanZa_stdÚa_sumZb_meanZb_stdZb_sumrnroZexp_colsrrrÚtest_agg|s\ zTestApi.test_aggcCs‚ttdƒtdddƒdœƒ}|jdd}|d ¡}| tjd d „dœ¡}|djdd „|d }t||gdd}tj ||dddS)NrKrrrT)rUrVrM)rLrUcSstj|ddS)Nr3)Úddof)r8ru)ÚxrrrraÀsz(TestApi.test_agg_apply..rVcSstj|ddS)Nr3)r…)r8ru)r†rrrraÁs)r"r3)rsT)rv) rrrNrbÚaggr8Úapplyrrr)rBr"r]rRrƒrnZrcustomrorrrÚtest_agg_apply¸szTestApi.test_agg_applycCsÆttdƒtdddƒdœƒ}|jdd}| tjtjg¡j}tj tdƒd d gg¡}t ||¡|d tjtjg¡j}td d gƒ}t ||¡| dtjtjgi¡j}tj dd g¡}t ||¡dS)NrKrrrT)rUrVrM)rLrkrbrtrU)rUrb)rUrt)rrrNr‡r8rbrtr6r|r}r~rmrrOrr)rBr]rRrnrorrrÚtest_agg_consistencyÅszTestApi.test_agg_consistencyc sNttdƒtdddƒdœƒ}|jdd‰‡fdd „}t t|¡tˆd ¡ˆd ¡ˆd ¡ˆd ¡gdd }t j ddddg¡|_t dd:t dt¡ˆd dg dddgidddgidœ¡}WdQRXtj||ddt dd2t dt¡ˆ dddgidddgidœ¡}WdQRXt j ddddg¡|_tj||dddS)NrKrrrT)rUrVrM)rLcs$ˆ dddgidddgidœ¡dS)NrUrtrbrV)Zr1Zr2)r{r)rRrrÚfÜsz(TestApi.test_agg_nested_dicts..frUrVr3)rs)Úrart)rŒru)Úrbrt)rruT)rwrxrŒrtrur)rv)rUrŒrt)rUrŒru)rVrrt)rVrru)rrrNrYrZrrrtrur|r}rr6rr€rr‚r‡rr)rBr]r‹rornr)rRrÚtest_agg_nested_dictsÖs(zTestApi.test_agg_nested_dictscCslddddddddd d dg}tdd dgdddgtdƒtjdddtjddddt d¡t d¡t d¡gddtjgddtjgddtjgt dƒt dƒt dƒgt d¡t d¡t d¡gd œ|d!}tdddgdddgdddgdddgdddgdddgdddgdddgdddgdddgdddgd œ|d!}|j d d" ¡}t ||¡| d¡ ¡}| ¡ t¡}t ||¡dS)#NÚintÚfloatÚstringrrr4Úfl_infÚfl_nanÚstr_nanÚdt_natÚperiods_natr3rTrMg@g@g@ÚabcZ20170101)r4z1 sÚs)r4Úfreqz2012-01z2012-02z2012-03gð?g@ZaaZbbZ20170203)rrr‘rrr4r’r“r”r•r–)r6)rL)rrmr|Ú date_rangeZtimedelta_rangeZPeriodr8ZInfr9rrNÚcountrrrÚastyper)rBÚcolsr]rornrrrÚtest_count_nonnumeric_typesôsH z#TestApi.test_count_nonnumeric_typesz"ignore:can't resolve:ImportWarningcCsÀttj d¡ƒjdddd}t|jdd|jddgdd}d d g|_| d d„dd„g¡}t ||¡d d„}dd„}t|jdd|jddgdd}ddg|_| ||g¡}t ||¡dS)Nrrr3r-)rLÚmin_periodsÚwin_type)rug{®Gáz„?)rszcSs|jddS)Nr)ru)rt)r†rrrra'sz/TestApi.test_window_with_args..cSs|jddS)Ng{®Gáz„?)ru)rt)r†rrrra(scSs|jddS)Nr)ru)rt)r†rrrÚa+sz(TestApi.test_window_with_args..acSs|jddS)Ng{®Gáz„?)ru)rt)r†rrrÚb.sz(TestApi.test_window_with_args..br¡r¢)r r8ÚrandomrrNrrtr6r{rr)rBrRrornr¡r¢rrrÚtest_window_with_argss zTestApi.test_window_with_argscCsNtt d¡dd}| d¡ ¡}| d¡ ¡}|jdks@g€A@gD@gN@gT@g@U@éPÚ expandingNg$@g‹RB°ªª:@gI@c Csètjdddgdddgdddgdddgdddgdddgdddggd d dgd}t| d ¡|ƒ}|rj||ƒ}n|ƒ}tjjd ddddddgd dgd}tj ddddg¡}tj|||d} | td ddgfdddgffƒ¡} t | | ¡dS)NrUrr1r¥r2rVr©éZZstockÚlowÚhigh)r6)rUr)rUr3)rUrT)rVrM)rVé)rVrK)rVrf)Únames)r¬rt)r¬Úmax)rrt)rÚmin)r5r6rtr°r±) r|rÚgetattrÚgroupbyr}rr‡rrr)rBÚfuncZwindow_sizeZ expected_valsr]r‹rLr5r6rornrrrÚtest_multiple_agg_funcs?s. zTestApi.test_multiple_agg_funcs)rDrErFrJrSr_rdrprrr„r‰rŠrŽržÚtdÚskip_if_no_scipyrYÚmarkÚfilterwarningsr¤r¨Úparametrizer8rlrµrrrrrHCs: < + rHz*ignore:can't resolve package:ImportWarningc@sneZdZdd„Zejej dddg¡dd„ƒƒZ ejej dddg¡dd „ƒƒZ ej d ddg¡d d„ƒZdS)Ú TestWindowcCs| ¡dS)N)rC)rBrIrrrrJrszTestWindow.setup_methodÚwhichr>rAc CsØt||ƒ}|j}|dddd|ddddd|dddddxddd t dg¡gD]N}t t¡|dd|dWdQRXt t¡|ddd|dWdQRXqRWx.d D]&}t t¡||ddWdQRXqªWdS)NÚboxcarrTr3)r rLrŸT)r rLrŸÚcenterFg@re)Úfoobarr3)r rL)r²rNr8rGrYrZÚ ValueError)rBr¼ÚoÚcÚwZwtrrrÚtest_constructorus zTestWindow.test_constructorcCs t||ƒ}|j}||dddS)NrT)r rL)r²rN)rBr¼r+rÁrÂrrrÚtest_constructor_with_win_typeŽs z)TestWindow.test_constructor_with_win_typerIrbrtc Cs|tjtdddgƒddgd}d}tjt|dt||ƒddd ƒWdQRXtjt|dt||ƒtjd WdQRXdS)NrTr®rfr)rLz2numpy operations are not valid with window objects)rXr3rM)Údtype) ÚrwindowZWindowr rYrZr r²r8Úfloat64)rBrIrÃÚmsgrrrÚtest_numpy_compat—szTestWindow.test_numpy_compatN)rDrErFrJr¶r·rYr¸rºrÄrÅrÊrrrrr»osr»c@sœeZdZdd„Zdd„Zej dddg¡dd „ƒZe j ej dddg¡d d„ƒƒZej ded de jd dg¡dd„ƒZej ded de jd ddg¡dd„ƒZej dddddddg¡dd„ƒZdd„Zej dddg¡d d!„ƒZej dddg¡d"d#„ƒZej d$d%d&g¡ej d'dd(d)d*d*d+d,d d-d.d/d0g fdd1d)d*d*d*d+d,d d-d.d/g fdd2ejd*d*d+d,d d-d.d/d0g fdd3ejd*d*d*d+d,d d-d.d/g fdd(d)d+d,d d-d.d/d0d4d5g fdd1d)d+d,d d-d.d/d0d4d5g fdd2ejd*d+d,d d-d.d/d0d4g fdd3ejd*d+d,d d-d.d/d0d4g fg¡d6d7„ƒƒZd8d9„Zej d'dd(ejd*d*d+d,d d-d.ejejg fdd1ejd*d*d*d+d,d d-d.ejg fdd2ejejd*d+d,d d-d.ejejg fdd3ejejd*d*d+d,d d-d.ejg fdd(ejd+d,d d-d.d/d/ejejg fdd1ejd+d,d d-d.d/d/d/ejg fdd2ejejd+d,d d-d.d/ejejg fdd3ejejd+d,d d-d.d/d/ejg fg¡d:d;„ƒZej dd?„ƒZd@dA„ZdBdC„ZdDdE„Zej dFe je jg¡dGdH„ƒZ dIdJ„Z!dKS)LÚTestRollingcCs| ¡dS)N)rC)rBrIrrrrJ§szTestRolling.setup_methodcCs@tddddtjdgiƒ}|| d¡ ¡|jddd ¡dS)NrVrr3rTr®)rŸ)rr8rlrNrb)rBr]rrrÚtest_doc_stringªszTestRolling.test_doc_stringr¼r>rAc Csît||ƒ}|j}|dd|ddd|dddd|ddddt t¡|dƒ|d ƒWdQRXx€d dt dg¡gD]j}t t¡||dWdQRXt t¡|d|dWdQRXt t¡|dd|dWdQRXq|WdS)NrT)rLr3)rLrŸT)rLrŸr¾Fréÿÿÿÿg@re)r²rNrYrZrÀr8rG)rBr¼rÁrÂrÃrrrrÄ±s zTestRolling.test_constructorc Cs6t||ƒ}|j}t t¡|dddWdQRXdS)NrÍr½)r )r²rNrYrZrÀ)rBr¼rÁrÂrrrrÅÍs z*TestRolling.test_constructor_with_win_typerLrM)ZdayscCs–d}tdt |¡itjd|ddd}t ddgt d d d¡¡}|j|d ¡}td|itjd|ddd}t ||¡| d ¡ ¡}t ||¡dS)NrÚvaluez 2015-12-24ÚD)r4r™)r5ggð?g@g;@rM)rLÚ3D) rr8r@r|ršÚappendrNrbrr)rBrLÚnr]Z expected_datarnrorrrÚ&test_constructor_with_timedelta_window×sz2TestRolling.test_constructor_with_timedelta_windowrÐc Cs d}tdt |¡itjd|ddd}tdt tjdgt dd d ¡¡itjd|ddd}|j|dd ¡}|j|ddj t|d }t ||¡t ||¡dS)NrrÎz 2017-08-08rÏ)r4r™)r5gð?g@g;@rMrT)rLrŸ)r")rr8r@r|ršrÑr9rNrbrˆrr)rBrLr"rÒr]roZresult_roll_sumZresult_roll_genericrrrÚ0test_constructor_timedelta_window_and_minperiodsèsztddddddgiƒ}t t¡|jdddWdQRXdS) NrUrr3rTrMr®Úneither)rLÚclosed)rrYrZrÀrN)rBr]rrrÚtest_closedszTestRolling.test_closedr´cCsPtjdgtjdddd}t|jddd|ƒƒ}t |tjtjg|j d ¡dS) NrTÚ2000r3)r4)Údatar5Ú10Dr)r×)r5) r|r ršr²rNrrr8rlr5)rBr´ÚserrnrrrÚtest_closed_one_entrysz!TestRolling.test_closed_one_entrycCs”tjdddgdddgdœtjdddd}t|jdd d djdd d|ƒƒ}tjjdddg|jgdd}tj t jdt jg|dd}t ||¡dS)Nr3rTrM)rUrVrÙ)r4)rÚr5rUF)ÚsortrVrÛr)r×)rUN)Zarraysr¯)rÚr5rP)r|rršr²r³rNr}Zfrom_arraysr5r r8rlrr)rBr´rÜrnZexp_idxrorrrÚtest_closed_one_entry_groupbysz)TestRolling.test_closed_one_entry_groupbyÚinput_dtyperrzfunc,closed,expectedrgrr3rTr®rKrféÚbothrÖrérgcCsZtjt d¡ |¡tjdddd}t|jd|d|ƒƒ}tj||jd}t ||¡dS)NrrÙ)r4)rÚr5rÐ)r×)r5)r|r r8r@rœršr²rNr5rr)rBràr´r×rorÜrnrrrÚtest_closed_min_max_datetimes z(TestRolling.test_closed_min_max_datetimec Csxtjt d¡tjdddd}|j|jddgd}|jdd d ¡}tjtj dddd dddg|jd}t ||¡dS)NrrÙ)r4)rÚr5r3rK)r5rÐr)r×rrTrMr®rf)r|r r8r@ršÚdropr5rNr±rlrr)rBrÜrnrorrrÚtest_closed_uneven5s zTestRolling.test_closed_unevencCsjtjt d¡tjdddd}tj||jdd…<t|jdd|d|ƒƒ}tj||jd }t ||¡dS) NrrÙ)r4)rÚr5éýÿÿÿrÐrT)rŸr×)r5)r|r r8r@ršrlr5r²rNrr)rBr´r×rorÜrnrrrÚtest_closed_min_max_minpAsz$TestRolling.test_closed_min_max_minpÚrollerÚ1scCs\tƒ}tƒ |¡ ¡}t ||¡tt g¡d}tt g¡d |¡ ¡}t ||¡dS)N)r5)rrNrbrrr|Ú DatetimeIndex)rBrérornrrrÚtests_empty_df_rollingUsz"TestRolling.tests_empty_df_rollingcCsht tjg¡}|jddd ¡}t dg¡}t ||¡|jddd ¡}t tjg¡}t ||¡dS)Nr3r)rŸg)r|r r8rlrNrbrr)rBr†rnrorrrÚtest_missing_minp_zerocsz"TestRolling.test_missing_minp_zerocCs\tjtjgdt ddddg¡d}|jt d¡dd ¡}tjd |jd}t ||¡dS)Nr®z 2017-01-01z 2017-01-04z 2017-01-06z 2017-01-07)r5Ú2dr)rŸg)r|r r8rlrërNÚ Timedeltarbr5rr)rBr†rnrorrrÚtest_missing_minp_zero_variablepsz+TestRolling.test_missing_minp_zero_variablecCsltjjddgdddggddgd}tt d ¡|d }| d¡ ¡}t |j |j ¡|jjdddgksht ‚dS)NrUrVrWrÏÚEÚ1Ú2)r¯)rrf)r6rM)r|r}r~rr8ÚonesrNÚcovrrOr6r5r¯rQ)rBrr]rnrrrÚtest_multi_index_namesysz"TestRolling.test_multi_index_namesÚklassc Cs8|ddddgƒ}t t¡t| d¡ƒWdQRXdS)Nr3rTrMr®)rYrZÚNotImplementedErrorÚiterrN)rBr÷ÚobjrrrÚtest_iter_raises„szTestRolling.test_iter_raisescCsztt d¡ƒ}| |¡}|dkr8tdd„tdƒDƒƒ}n ttjgddgdgd ƒ}|jd |d ¡}t ||¡dS)N)rr1rcSs$i|]}tjgddgd|“qS)rTg@rã)r8rl)Ú.0Úirrrú ’sz1TestRolling.test_rolling_axis..r1rTg@rhrrM)rs) rr8rôÚ_get_axis_numberrrlrNrbrr)rBÚ axis_framer]rsrornrrrÚtest_rolling_axisŒs zTestRolling.test_rolling_axisN)"rDrErFrJrÌrYr¸rºrÄr¶r·rÅrr|rïrÓrÔrÊrØrÝrßr8rlrärærèrìrírðrör rrûrrrrrrË¥sT"$ . " $"" $0 rËc @s´eZdZdd„Zdd„Zej dddg¡dd „ƒZej d ddd dddg¡dd„ƒZ ej ddej dejjdddg¡dd„ƒZdd„Z ej dejejg¡dd„ƒZd d!„Zd"S)#Ú TestExpandingcCs| ¡dS)N)rC)rBrIrrrrJ¢szTestExpanding.setup_methodcCs.tddddtjdgiƒ}|| d¡ ¡dS)NrVrr3rTr®)rr8rlrªrb)rBr]rrrrÌ¥szTestExpanding.test_doc_stringr¼r>rAc Cs”t||ƒ}|j}|dd|ddd|dddx\ddt dg¡gD]F}t t¡||dWdQRXt t¡|d|dWdQRXqFWdS) Nr3)rŸT)rŸr¾Fg@rerT)r²rªr8rGrYrZrÀ)rBr¼rÁrÂrÃrrrrÄ«s zTestExpanding.test_constructorrIrurtrbr°r±rÕc Csxtjtdddgƒdd}d}tjt|dt||ƒdddƒWdQRXtjt|dt||ƒtjd WdQRXdS) NrTr®rf)rLz2numpy operations are not valid with window objects)rXr3rM)rÆ) rÇZ Expandingr rYrZr r²r8rÈ)rBrIÚerÉrrrrÊ¿szTestExpanding.test_numpy_compatÚexpanderr3Zlsz,GH#16425 expanding with offset not supported)Úreason)ZmarkscCs\tƒ}tƒ |¡ ¡}t ||¡tt g¡d}tt g¡d |¡ ¡}t ||¡dS)N)r5)rrªrbrrr|rë)rBrrornrrrÚtest_empty_df_expandingÌs z%TestExpanding.test_empty_df_expandingcCsdt tjg¡}|jdd ¡}t dg¡}t ||¡|jdd ¡}t tjg¡}t ||¡dS)Nr)rŸgr3)r|r r8rlrªrbrr)rBr†rnrorrrríàsz$TestExpanding.test_missing_minp_zeror÷c Cs8|ddddgƒ}t t¡t| d¡ƒWdQRXdS)Nr3rTrMr®)rYrZrørùrª)rBr÷rúrrrrûíszTestExpanding.test_iter_raisescCs„tt d¡ƒ}| |¡}|dkr8tdd„tdƒDƒƒ}n*ttjgddd„td d ƒDƒgdƒ}|jd |d ¡}t ||¡dS) N)rr1rcSs.i|]&}tjgddd„tddƒDƒ|“qS)rTcSsg|]}t|ƒ‘qSr)r)rüÚjrrrú üsz@TestExpanding.test_expanding_axis...rMé)r8rlr)rürýrrrrþûsz5TestExpanding.test_expanding_axis..r1rTcSsg|]}t|ƒ‘qSr)r)rürýrrrrsz5TestExpanding.test_expanding_axis..rMrir)rs) rr8rôrÿrrlrªrbrr)rBrr]rsrornrrrÚtest_expanding_axisõs z!TestExpanding.test_expanding_axisN)rDrErFrJrÌrYr¸rºrÄrÊr Zxfailrrír|r rrûr rrrrr s rc@sReZdZdd„Zdd„Zej dddg¡dd „ƒZej d ddd g¡dd„ƒZ dS)ÚTestEWMcCs| ¡dS)N)rC)rBrIrrrrJszTestEWM.setup_methodcCs0tddddtjdgiƒ}||jdd ¡dS)NrVrr3rTr®gà?)Úcom)rr8rlÚewmrt)rBr]rrrrÌszTestEWM.test_doc_stringr¼r>rAc CsVt||ƒ}|j}|dd|dd|dd|dd|ddd|ddd |ddd t t¡|dddWdQRXt t¡|dddWdQRXt t¡|ddd WdQRXt t¡|ddWdQRXt t¡|ddWdQRXt t¡|ddWdQRXx.dD]&}t t¡||dWdQRXq(WdS)Ngà?)rgø?)Úspan)Úalphagè?)Úhalflife)rr)rr)rr)rr)rr)rrgà¿r)gà¿gø?)r²r rYrZrÀ)rBr¼rÁrÂrrrrrÄs0 zTestEWM.test_constructorrIrurtrÕc Csxtjtdddgƒdd}d}tjt|dt||ƒddd ƒWdQRXtjt|dt||ƒtjd WdQRXdS)NrTr®rfgà?)rz2numpy operations are not valid with window objects)rXr3rM)rÆ) rÇZEWMr rYrZr r²r8rÈ)rBrIrrÉrrrrÊ<szTestEWM.test_numpy_compatN) rDrErFrJrÌrYr¸rºrÄrÊrrrrr s'rc @sneZdZdZdd„dd„dd„dd„dd„dd„d d„d d„dœZdd „Zdd„Zdd„Zdd„Zdd„Z dS)ÚDtyperTcCs| ¡S)N)r›)ÚvrrrraVszDtype.cCs| ¡S)N)r°)rrrrraWscCs| ¡S)N)r±)rrrrraXscCs| ¡S)N)rb)rrrrraYscCs| ¡S)N)rt)rrrrraZscCs| ¡S)N)ru)rrrrra[scCs| ¡S)N)rÕ)rrrrra\scCs| ¡S)N)Úmedian)rrrrra]s)r›r°r±rbrtrurÕrcCstdddddgddttjddddgddttjddddgddttjdddd gddttjd ddd gddttjgt d ¡gdddttjd d d d gddttjd ddd gdddœtdddddgddttjddddgddttjddddgddttjddddgddttjdd ddgddttjgt d¡gdddttjddddgddttjdd ddgdddœttdddddgƒtdddddgƒdœddtttjddddgƒttjddd dgƒdœddtttjddddgƒttjdddd gƒdœddtttjddddgƒttjddddgƒdœddtttjdddd gƒttjddddgƒdœddtttjgt d¡gdƒttjgt d¡gdƒdœddtttjddddgƒttjddddgƒdœddtttjdddd gƒttjddddgƒdœdddœdœ}|S)Nr3rTrÈ)rÆrMr®rrKrágà?gø?g@g@)r›r°r±rbrtrurÕrrrãrfrhérg)rr3éé)Úsr1Úsr2r])r r8rlZsqrtr)rBÚexpectsrrrÚget_expects`sRzDtype.get_expectscCsPtt d¡|d}tt ddd¡|d}tt d¡ d¡|d}|||dœ}|S)NrK)rÆrréþÿÿÿ)rKrT)rrr])r r8r@rÚreshape)rBrÆrrr]rÚrrrÚ_create_dtype_data“szDtype._create_dtype_datacCs| |j¡|_| ¡|_dS)N)rrÆrÚrr)rBrrrrC szDtype._create_datacCsd| ¡xVt|j ¡|j ¡ƒD]<\}}|j|}|j|}|j||}| |||||¡q WdS)N)rCrÚfuncsÚkeysrÚrÚcheck_dtypes)rBÚf_nameÚd_namer‹ÚdÚexprrrÚtest_dtypes¤s zDtype.test_dtypescCs&|j|jd}||ƒ}t ||¡dS)N)rL)rNrLrÚassert_almost_equal)rBr‹r!r#r"r$Úrollrnrrrr szDtype.check_dtypesN) rDrErFrLrrrrCr%r rrrrrRs3 rc@seZdZeZdS)ÚTestDtype_objectN)rDrErFÚobjectrÆrrrrr(³sr(c@seZdZdS)Ú Dtype_integerN)rDrErFrrrrr*·sr*c@seZdZejZdS)ÚTestDtype_int8N)rDrErFr8Zint8rÆrrrrr+»sr+c@seZdZejZdS)ÚTestDtype_int16N)rDrErFr8Zint16rÆrrrrr,¿sr,c@seZdZejZdS)ÚTestDtype_int32N)rDrErFr8Zint32rÆrrrrr-Ãsr-c@seZdZejZdS)ÚTestDtype_int64N)rDrErFr8Zint64rÆrrrrr.Çsr.c@seZdZdS)ÚDtype_uintegerN)rDrErFrrrrr/Ësr/c@seZdZejZdS)ÚTestDtype_uint8N)rDrErFr8Zuint8rÆrrrrr0Ïsr0c@seZdZejZdS)ÚTestDtype_uint16N)rDrErFr8Zuint16rÆrrrrr1Ósr1c@seZdZejZdS)ÚTestDtype_uint32N)rDrErFr8Zuint32rÆrrrrr2×sr2c@seZdZejZdS)ÚTestDtype_uint64N)rDrErFr8Zuint64rÆrrrrr3Ûsr3c@seZdZdS)ÚDtype_floatN)rDrErFrrrrr4ßsr4c@seZdZejZdS)ÚTestDtype_float16N)rDrErFr8Zfloat16rÆrrrrr5ãsr5c@seZdZejZdS)ÚTestDtype_float32N)rDrErFr8Zfloat32rÆrrrrr6çsr6c@seZdZejZdS)ÚTestDtype_float64N)rDrErFr8rÈrÆrrrrr7ësr7c@seZdZdZdZdd„ZdS)ÚTestDtype_categoryÚcategoryFcCs2ttdƒ|d}ttdddƒ|d}||dœ}|S)NrK)rÆrrr)rr)r r)rBrÆrrrÚrrrrós z%TestDtype_category._create_dtype_dataN)rDrErFrÆZ include_dfrrrrrr8ïsr8c@seZdZdd„ZdS)ÚDatetimeLikec CsN|j|jd}|dkr,||ƒ}t ||¡nt t¡||ƒWdQRXdS)N)rLr›)rNrLrr&rYrZrø)rBr‹r!r#r"r$r'rnrrrr szDatetimeLike.check_dtypesN)rDrErFr rrrrr:ÿsr:c@seZdZe d¡ZdS)ÚTestDtype_timedeltazm8[ns]N)rDrErFr8rÆrrrrr;sr;c@seZdZe d¡ZdS)ÚTestDtype_datetimezM8[ns]N)rDrErFr8rÆrrrrr<sr<c@seZdZdZdd„ZdS)ÚTestDtype_datetime64UTCzdatetime64[ns, UTC]cCst d¡dS)NzKdirect creation of extension dtype datetime64[ns, UTC] is not supported ATM)rYÚskip)rBrrrrCsz$TestDtype_datetime64UTC._create_dataN)rDrErFrÆrCrrrrr=sr=c@søeZdZdd„Zdd„Zdd„Zdd„Zd d „Zej dd„ƒZ ej d d„ƒZej dd„ƒZej dd„ƒZ ej dd„ƒZej dd„ƒZej dd„ƒZej dd„ƒZej dd„ƒZej dd„ƒZdd „Zd!d"„Zd#d$„Zej d%d&d'd(d)d*g¡d+d,„ƒZd-d.„Zej d/d&d'd0d(d1g¡ej d2d3d4d5d6d7g¡ej d8d*d9d:d;dgd?d*d:d;dgd&ejd@ejdAgejejejejgejd'ejdBdAd(gd(gejdCdDgg¡dEdF„ƒƒƒZdGdH„ZdIdJ„Z dKdL„Z!ej dMe"e#g¡ej dNdOdP„dQdP„g¡dRdS„ƒƒZ$dTdU„Z%ej dVdWdXg¡dYdZ„ƒZ&d[d\„Z'd]d^„Z(d_d`„Z)dadb„Z*ej dcdd„ƒZ+ej dedf„ƒZ,d‰didj„Z-dkdl„Z.ej dmdgdng¡ej dodgdng¡dpdq„ƒƒZ/drds„Z0dtdu„Z1dvdw„Z2dxdy„Z3dzd{„Z4d|d}„Z5d~d„Z6d€d„Z7ej dNd‚dƒd„g¡d…d†„ƒZ8dŠd‡dˆ„Z9dhS)‹ÚTestMomentscCs| ¡dS)N)rC)rBrIrrrrJ#szTestMoments.setup_methodc CsÆtt d¡ƒjdddd ¡t t¡$tt d¡ƒjdddd ¡WdQRXtt d¡ƒjdddd ¡tt d¡ƒjdddd ¡t t¡$tt d¡ƒjdddd ¡WdQRXdS) NrrMTr)rLr¾rsr3)rrrT) r r8rôrNrtrYrZrÀr)rBrrrÚtest_centered_axis_validation&s(z)TestMoments.test_centered_axis_validationcCs|jtjddddS)NrbF)rPÚzero_min_periods_equal)Ú_check_moment_funcr8Únansum)rBrrrÚtest_rolling_sum:s zTestMoments.test_rolling_sumcCsdd„}|j|dddddS)NcSst |¡ t¡ ¡S)N)r8Úisfiniterœrrb)r†rrrra?sz0TestMoments.test_rolling_count..r›Fr)rPÚhas_min_periodsÚ fill_value)rB)rBZcounterrrrÚtest_rolling_count>s zTestMoments.test_rolling_countcCs|jtjdddS)Nrt)rP)rBr8rt)rBrrrÚtest_rolling_meanCszTestMoments.test_rolling_meancCsht ddddddddd d g ¡}t|ƒjddd ¡}ttjtjddddddtjtjg ƒ}t ||¡dS)NgÍÌÌÌÌÌ@gìQ¸…k.@gáz®Gá@g= ×£p="@g…ëQ¸ž+@g{®Gáú*@g®Gáz®0@gö(\Âõ"@gÃõ(\B%@gö(\Âõ,@rKT)r¾g•C‹ì#@g ×£p=Š&@gTã¥›Ä '@gÕxé&1)@g‰A`åÐ¢)@g•C‹lç)@)r8rGr rNrtrlrr)rBÚvalsrnrorrrÚtest_cmov_meanFs zTestMoments.test_cmov_meancCsjt ddddddddd d g ¡}t|ƒjddd d ¡}ttjtjddddddtjtjg ƒ}t ||¡dS)NgÍÌÌÌÌÌ@gìQ¸…k.@gáz®Gá@g= ×£p="@g…ëQ¸ž+@g{®Gáú*@g®Gáz®0@gö(\Âõ"@gÃõ(\B%@gö(\Âõ,@rKr½T)r r¾g•C‹ì#@g ×£p=Š&@gTã¥›Ä '@gÕxé&1)@g‰A`åÐ¢)@g•C‹lç)@)r8rGr rNrtrlrr)rBrJrnrorrrÚtest_cmov_windowPs zTestMoments.test_cmov_windowcCs°t tjgd¡}|jdddd ¡}t |¡ ¡s8t‚t g¡}|jdddd ¡}t |ƒdksft‚t tj d¡¡}|jddd ¡}t |¡ ¡sœt‚t |ƒdks¬t‚dS)NrrKTr½)r¾r r)r )r|r r8rlrNrtÚisnanÚallrQÚlenr£r)rBrJrnrrrÚtest_cmov_window_cornerZs z#TestMoments.test_cmov_window_cornerc Cstt ddgddgddgddgd d gddgd dgddgddgddgg ¡}t tjtjgtjtjgddgddgddgddgddgddgtjtjgtjtjgg ¡}t|ƒjd d!d"d# ¡}t t|ƒ|¡t t ¡t|ƒjd d!d"d# ¡WdQRXt tjtjgtjtjgd$d%gd&d'gd(d)gd*d+gd,d&gd-d.gtjtjgtjtjgg ¡}t|ƒjd d!d"d# ¡}t t|ƒ|¡dS)/Ng\Âõ(\(@g…ëQ¸ @g\Âõ(\$@gR¸…ëQ"@g{®Gáz*@g¸…ëQ8-@g ×£p= @g¸…ëQ8 @gš™™™™™@gáz®Gá&@gHáz®G"@g×£p= ×@g…ëQ¸ž&@g×£p= W%@g…ëQ¸…@g ×£p= @g×£p= ×"@gHáz®Ç @gáz®Gá(@g\Âõ(\@g/Ý$"@gbX9´È"@g}?5^ºI!@gZd;ßÏ#@g= ×£p½!@gÑ"Ûù~j$@g= ×£p=@g“V-!@g^ºI+@gš™™™™"@g²ï§ÆK@rKr½T)r r¾gáz®G!G@g{®GázG@g\Âõ(œE@g¤p= ×ÃH@gÍÌÌÌÌ,F@g…ëQ¸…I@gfffffA@g¸…ëQxE@g{®GázC@g F@gÂõ(\C@) r8rGrlrrNrtrrrYrZrcrurb)rBrJÚxpÚrsrrrÚtest_cmov_window_framems$ "z"TestMoments.test_cmov_window_framecCs^ttj d¡ƒ}tj|d<tj|d<|jdddd ¡}|jddddd ¡}t ||¡dS) Nrr®rãrKT)rŸr¾r½)r rŸr¾) r r8r£rrlrNrtrr)rBrJrQrRrrrÚtest_cmov_window_na_min_periodsŒs z+TestMoments.test_cmov_window_na_min_periodscCsHt ddddddddd d g ¡}tjtjddd dddtjtjg tjtjddddddtjtjg tjtjddddddtjtjg tjtjdddd d!d"tjtjg tjtjd#d$d%d&d'd(tjtjg tjtjd)d*d+d,d-d.tjtjg tjtjd/d0d1d2d3d4tjtjg tjtjddddddtjtjg d5œ}t||ƒ}t|ƒjd6|d7d8 ¡}t ||¡dS)9NgÍÌÌÌÌÌ@gìQ¸…k.@gáz®Gá@g= ×£p="@g…ëQ¸ž+@g{®Gáú*@g®Gáz®0@gö(\Âõ"@gÃõ(\B%@gö(\Âõ,@g='½o|m!@gqæWs€ #@gÂ(›Â(@g4hèŸà,@gf÷äa¡¶+@gâ#bJ$¡'@g}iÆ’"@g6ñº~±$@gÞT¤ÂØ(@gâuý‚*@gÃõ(\Â*@g±Ÿ`¬(@gÃõ(\â @gÃõ(\b"@g= ×£p)@g\Âõ(¼,@g ×£p=*,@gÃõ(\"'@gô¦"Æv@g_ÎQZ"@g²ºÕsÒ«)@g*Æ3hX,@gŒ5˜†Q-@g÷éxÌ@5&@gœŠT[è@g£;ˆ)T"@g}–çÁÝ*@g³Íé ,@g OäIÒ5.@gY£¢Ñ}%@gÈêVÏI/@gõœô¾ñU"@gËóàî¬ *@g!<Ú8,@g”Þ7¾ö.@g½Þý‘%@g;M„ ï@g–>tA}["@gH§®|–—)@gfÚþ••f,@gó“jŸŽ'-@g,}è‚úV&@)r%r#r*r'r(r)r$r&rKT)r r¾)r8rGrlr rNrtrr)rBr+rJÚxpsrQrRrrrÚtest_cmov_window_regular˜s* z$TestMoments.test_cmov_window_regularcCshtjtdƒtjd}| ¡}tj|dd…<tj|dd…<t|ƒ}t|ƒjd|dd ¡}t ||¡dS)Nr)rÆrTrrKT)r r¾)r8rGrrr=rlr rNrtrr)rBr+rJrQrRrrrÚ%test_cmov_window_regular_linear_range´sz1TestMoments.test_cmov_window_regular_linear_rangecCs*t dddddddtjdd g ¡}tjtjd ddd ddddg tjtjddddddddg tjtjd ddd ddddg tjtjddddddddg tjtjd d!d"d#d$d%d&d'g tjtjd(d)d*d+d,d-d.d/g tjtjd0d1d2d3d4d5d6d7g tjtjd8d9d:d;dd?g d@œ}t||ƒ}t|ƒjdA|dBdC ¡}t ||¡dS)DNgÍÌÌÌÌÌ@gìQ¸…k.@gáz®Gá@g= ×£p="@g…ëQ¸ž+@g{®Gáú*@g®Gáz®0@gÃõ(\B%@gö(\Âõ,@gc—¨Þh#@gìQ¸…%@gÃõ(\â @gÃõ(\b"@g= ×£p)@g\Âõ(¼,@g@Ù”+¼;/@gÂõ(\O+@ge¸u"@g"àªÔ&@g;M„ ï@g–>tA}["@gH§®|–—)@gfÚþ••f,@g’ËH¿½/@gO¯”eˆã!@gÆ¢éìd '@gô¦"Æv@g_ÎQZ"@g²ºÕsÒ«)@g*Æ3hX,@gýjÌÑ/@gR›8¹/#@gœ¿ …˜$@g='½o|m!@gqæWs€ #@gÂ(›Â(@gEØðôJi,@g&pën~.@g÷¯¬4)e+@gjö@+0ô @g8gDio(@gÈêVÏI/@gõœô¾ñU"@gËóàî¬ *@gmIFÎ,@gætYLl0@gÏ ¡O+@gï v¦Ð©"@gö(\Â…#@g}iÆ’"@g6ñº~±$@gÞT¤ÂØ(@g®Gáz¤+@gÌz1”ý,@gHáz®‡+@g˜À»Ù @gx( ô‰¼(@gœŠT[è@g£;ˆ)T"@g}–çÁÝ*@gB`åÐ",@gÎQÚ0@gßÛôgO+@)r&r$r*r'r%r)r#r(rKrM)r rŸ)r8rGrlr rNrtrr)rBr+rJrUrQrRrrrÚ%test_cmov_window_regular_missing_dataÀs* z1TestMoments.test_cmov_window_regular_missing_datacCsÔddiddidddœdœ}t ddd d ddd dddg ¡}tjtjddddddtjtjg tjtjddddddtjtjg tjtjdddd d!d"tjtjg d#œ}t||ƒ}t|ƒjd$|d%d&jf||Ž}t ||¡dS)'NÚbetagð?rug@)ÚpowerÚwidth)r,r-r.gÍÌÌÌÌÌ@gìQ¸…k.@gáz®Gá@g= ×£p="@g…ëQ¸ž+@g{®Gáú*@g®Gáz®0@gö(\Âõ"@gÃõ(\B%@gö(\Âõ,@gnú³)ò!@gLŽ;¥ƒ…#@gWÿ[É~(@g¹6TŒóÇ+@gŽé K.scoreatpercentilecs ˆ|ˆƒS)Nr)r†)rbrfrrÚ quantile_func6sz8TestMoments.test_rolling_quantile..quantile_funcÚquantile)rPrh)rB)rBrbrgr)rbrfrÚtest_rolling_quantile"sz!TestMoments.test_rolling_quantilecCs|d}d}tjd|dd}ttj ||¡ |df¡|d}|jdd d gdd}tj|d ddgdd}t |jt |¡¡dS)NrrKZ20100101rV)r4r™rÍ)r5gÐ?gà?gè?r)rséé2éK) r|ršrr8r£ÚrandrrhZ percentilerr&rdrG)rBÚrowÚcolrer]Zdf_quantileZ np_percentilerrrÚ#test_rolling_quantile_np_percentile<s"z/TestMoments.test_rolling_quantile_np_percentilerhgÍÌÌÌÌÌÜ?r3Ú interpolationZlinearÚlowerZhigherZnearestZmidpointrÚg@g@g@g@g@g@g @gš™™™™™É?gš™™™™™Ù?g333333Ó?gffffffæ?g333333ã?cCsXt|ƒ}| ||¡}|jdd ||¡jd}t |¡rHt |¡sTt‚n||ksTt‚dS)Nr3)rŸrÍ)r rhrªÚilocr8rMrQ)rBrhrqrÚr˜Zq1Zq2rrrÚ+test_rolling_quantile_interpolation_optionsIs z7TestMoments.test_rolling_quantile_interpolation_optionsc CsLt d¡}t|ƒ}tjtdd"|jt|ƒddjdddWdQRXdS) NrKz(Interpolation 'invalid' is not supported)rXr3)rŸgà?Zinvalid)rq) r8r@r rYrZrÀrNrOrh)rBrÚr˜rrrÚtest_invalid_quantile_valueas z'TestMoments.test_invalid_quantile_valuec Csˆtdddddgƒ}t t¡| d¡ d¡WdQRXt t¡| d¡ d¡WdQRXt t¡| d¡ d ¡WdQRXdS) Nggš™™™™™¹?gà?gÍÌÌÌÌÌì?gð?rMgš™™™™™¹¿g$@re)r rYrZrÀrNrhrq)rBrÜrrrÚtest_rolling_quantile_paramjsz'TestMoments.test_rolling_quantile_paramc CsØt ¡^tjddtddd„}|jtjd||dtgƒ}| d¡j d d „|d}t ||¡WdQRXtdddgƒ}|jdd dj dd „|d}tdddgƒ}t ||¡|jdd dj t|d}t ||¡dS)Nrxz.*(empty slice|0 for slice).*)Úmessager9cSs|t |¡ ¡S)N)r8rErt)r†rrrr‹sz)TestMoments.test_rolling_apply..frˆ)rPr´r"rcSs| ¡S)N)rt)r†rrrra…sz0TestMoments.test_rolling_apply..)r"rTr)rŸcSst|ƒS)N)rO)r†rrrraŠsgð?g@) r€rr¹ÚRuntimeWarningrBr8rtr rNrˆrrrO)rBr"r‹rornr˜rrrÚtest_rolling_applyvs zTestMoments.test_rolling_applyr÷rIcCs|jddS)NrT)rL)rN)r†rrrra“szTestMoments.cCs| ¡S)N)rª)r†rrrra“sc Cs:|t d¡ƒ}t t¡||ƒ dd„¡WdQRXdS)NrMcSst|ƒS)N)rO)r†rrrrašsz7TestMoments.test_apply_future_warning..)r8r@rZassert_produces_warningr‚rˆ)rBr÷rIr˜rrrÚtest_apply_future_warning‘sz%TestMoments.test_apply_future_warningcCsxt ddddg¡}| d¡jtj|d}| ¡ ¡s8t‚|jdddjtj|d}tjddddgt d }t ||¡dS) Nr3rTrMr®r)r")rŸrf)rÆ)r|r rNrˆr8rbrrNrQrrr&)rBr"rJrnrorrrÚ test_rolling_apply_out_of_boundsœsz,TestMoments.test_rolling_apply_out_of_boundsrLrTÚ2sc s¤tjtj d¡tjjdddddœtjddddd ‰‡fd d„}ˆ |¡j|dd }ˆj dd… ˆ¡}t ||¡t t¡ˆ |¡j|dd WdQRXdS)NrKrr)Úsize)rUrVÚ20130101r˜)r4r™)r5cs$|jdˆjdkrtjS|jdS)NrrÍ)r5r8rlrs)r†)r]rrr‹°sz=TestMoments.test_rolling_apply_with_pandas_objects..fF)r"rTT)r|rr8r£rZrandintršrNrˆrsZreindex_likerrrYrZrc)rBrLr‹rnror)r]rÚ&test_rolling_apply_with_pandas_objects§sz2TestMoments.test_rolling_apply_with_pandas_objectscCs*|jdd„dd|jdd„ddddS)NcSstj|ddS)Nr3)r…)r8ru)r†rrrra½sz.TestMoments.test_rolling_std..ru)rPcSstj|ddS)Nr)r…)r8ru)r†rrrra¿sr)rPr…)rB)rBrrrÚtest_rolling_std¼s zTestMoments.test_rolling_stdcCs²t dddddg¡}|jddd ¡}t tjgd¡}t ||¡|jdddjd d }t dgd¡}t ||¡t tjtjdd dg¡jddd ¡}t |d¡s®t ‚dS)Ngð?g@g@g@g@r3)rŸrKr)r…grMr®rT) r|r rNrur8rlrrrMrQ)rBrJrnrorrrÚtest_rolling_std_1obsÂsz!TestMoments.test_rolling_std_1obscCslt dddddg¡}|jdd ¡}t |dd…¡ ¡s>t‚|jdd ¡}t |dd…¡ ¡sht‚dS)Nge>¸¸ÁR?g¨R¨z1Ò2?rM)rLrT)r) r|r rNrur8rErNrQr )rBr¡r¢rrrÚtest_rolling_std_neg_sqrtÑsz%TestMoments.test_rolling_std_neg_sqrtcCs*|jdd„dd|jdd„ddddS)NcSstj|ddS)Nr3)r…)r8rÕ)r†rrrraàsz.TestMoments.test_rolling_var..rÕ)rPcSstj|ddS)Nr)r…)r8rÕ)r†rrrraâsr)rPr…)rB)rBrrrÚtest_rolling_varßs zTestMoments.test_rolling_varcs&ddlm‰|j‡fdd„dddS)Nr)Úskewcsˆ|ddS)NF)Úbiasr)r†)r„rrraèsz/TestMoments.test_rolling_skew..r„)rP)Úscipy.statsr„rB)rBr)r„rÚtest_rolling_skewåszTestMoments.test_rolling_skewcs&ddlm‰|j‡fdd„dddS)Nr)Úkurtosiscsˆ|ddS)NF)r…r)r†)rˆrrraísz/TestMoments.test_rolling_kurt..Úkurt)rP)r†rˆrB)rBr)rˆrÚtest_rolling_kurtêszTestMoments.test_rolling_kurtTNc sÚd)‡‡fdd„ } | |jdd} t| tƒs,t‚t | jd||jdd…ƒ¡| |jdd}t|tƒsht‚tj |jddd…f|jjdd…dd…fj |dtd dd |r¢d}d} |jddd … d¡ ¡}|jddd … d¡ ¡}|r| ||| d} | ||| d}n| ||d} | ||d}| jd}|dt ¡}|jddd … ||¡}|jddd … ||¡}t | d||ƒ¡tj | |¡|j |tddd ttdƒƒ}tj|dd…<tj|dd…<|rÌ| |ddd}t |jd||dd…ƒ¡| |ddd}t|jdƒst‚t|jdƒr2t‚t|jdƒrFt‚t|jdƒsZt‚ttdƒƒ}| |ddd}t|jdƒsˆt‚t|jdƒsœt‚|rô| |ddd}| |ddd}t ||¡n(| |dƒ}t |jd||dd…ƒ¡|r„xâdt|jƒdt|jƒfD]h} | |jt|jƒd| d}| |jt|jƒ| d}t|ƒ}t |t|ƒ¡|}t ||||¡qWnZ| |jt|jƒdƒ}| |jt|jƒƒ}t|ƒ}t |t|ƒ¡|}t ||||¡|rÖ|r0| |dddd}| t |ttjgd ƒg¡dddd d…jdd!}n>| |ddd"}| t |ttjgd ƒg¡dƒd d…jdd!}t ||¡d#d$„td%ƒDƒ}|r d} | |j t|jjƒ|¡d| d d&¡ |jj¡}| |j t|jjƒ|¡d| d d&¡ |jj¡}| |jd| dd'}| |jd| dd'}n€| |j t|jjƒ|¡dd d&¡ |jj¡}| |j t|jjƒ|¡dd d&¡ |jj¡}| |jddd(}| |jddd(}|dk r¾| |¡}| |¡}t ||¡t !||¡dS)*NFcs |j|||d}t|ˆƒfˆŽS)N)rLrŸr¾)rNr²)rúrLrŸr¾rR)ÚkwargsrPrrÚ get_resultõsz2TestMoments._check_moment_func..get_resultrk)rLrÍiÎÿÿÿr)rsr")Úcheck_namesrjrrTrV)rLrŸrj)r"iöÿÿÿr¥)rŸr1ééiúÿÿÿéûÿÿÿrKrMr®r3T)rŸr¾rg)rå)r¾cSsg|]}d|‘qS)zx%dr)rür†rrrrgsz2TestMoments._check_moment_func..riôÿÿÿ)rLrŸr¾)rLr¾)NF)"r>rr rQrr&rsrArrrˆr"Úresamplertr5ÚoffsetsZBDayÚtruncateZxsrr8r9rrrOr|rÚreset_indexrZreindexrmÚshiftÚfillnar) rBÚstatic_comprPrFZ has_centerÚ has_time_rulerGrAr‹rŒÚ series_resultÚframe_resultÚwinÚminpr>rAZ last_dateZ prev_dateZtrunc_seriesZtrunc_framerúrnÚobj2Úresult0Úresult1roZnan_maskr˜Z series_xpZframe_xpZ series_rsZframe_rsr)r‹rPrrBðsÚ" zTestMoments._check_moment_funccCsT|jddt t d¡¡}d|d<|jddd ¡ ¡}t |d¡d ksPt ‚dS) Nrt)rPièr3rKrF)rÚadjustg{®Gáz„?) Ú _check_ewr|r r8Úzerosr rtrbÚabsrQ)rBrJrnrrrÚ test_ewma‹s zTestMoments.test_ewmar FÚ ignore_nacCsZtddddgƒ}|r&tddddgƒ}ntddd d gƒ}|jd||d ¡}t ||¡dS)Ngð?g@g@g @gš™™™™™ù?gÊ"k å@gYÂÚ;±@gR×ÚûTUõ?góSÇ@g ßû´—@)rr r¥)r r rtrr)rBr r¥r˜rornrrrÚtest_ewma_cases“szTestMoments.test_ewma_casescCsZtdgtjgddgƒ}|jdd ¡}t |tdgt|ƒƒ¡ttjgddgtjgddgƒ}|jdd ¡}t |ttjgddgdƒ¡ttjddgƒ}tdtjdgƒ}ttjdtjtjdtjgƒ}tdtjddgƒ}d}dd|}d d „} xL|ddtjd|dgf|ddtjd|dgf|ddtjd||gf|ddtjd||gf|ddd|dtjdgf|ddd|tjdgf|ddd|dtj|gf|ddd|tj|gf|ddtjd|d tjtjdtjgf|ddtjd|tjtjdtjgf|ddtjd|d tjtj|tjgf|ddtjd|tjtj|tjgf|ddd|d tjd|dgf|ddd|dtjd|dgf|ddd|d tjd|||d|d|gf|ddd|dtjd|||gfgD]d\}} }}| |t|ƒƒ} |j|| |d ¡}t || ¡|dkrî|j|| d ¡}t || ¡qîWdS)Ngð?rK)rrTr®g@Y@gI@g@cSs| |¡ ¡| ¡jddS)NÚffill)rI)ÚmultiplyÚcumsumr–)r˜rÃrrrÚ simple_wma³sz6TestMoments.test_ewma_nan_handling..simple_wmaTFrM)rr r¥)rr )r r8rlr rtrrrO)rBr˜rnZs0Ús1r¦r§rrrªr r¥rÃrorrrÚtest_ewma_nan_handling¢sZ(" $ z"TestMoments.test_ewma_nan_handlingcCs|jdddS)NrÕ)rP)r¡)rBrrrÚtest_ewmvarÙszTestMoments.test_ewmvarcCs|jdddS)NÚvol)rP)r¡)rBrrrÚtest_ewmvolÜszTestMoments.test_ewmvolc Cs~|jjdd ¡}|jjdd ¡}t ||¡t t¡|jjdddWdQRXt t¡|j ¡ ¡WdQRXdS)Ng#@)rr1)r)rr)r>r rtrr&rYrZrÀ)rBrUrVrrrÚtest_ewma_span_com_argsßsz#TestMoments.test_ewma_span_com_argsc CsÈ|jjdd ¡}|jjdd ¡}t ||¡t t¡|jjdddWdQRXt t¡|jjddd WdQRXt t¡|jjdddd WdQRXt t¡|j ¡WdQRXdS)Ng ‘ñHŽÝ+@)rg$@)rr1rk)rrg#@)rr)rrr)r>r rtrr&rYrZrÀ)rBrUrVrrrÚtest_ewma_halflife_argész"TestMoments.test_ewma_halflife_argcCsrt|jƒ}|jdd ¡}|jdd ¡}|jdd ¡}|jdd ¡}t ||¡t ||¡t ||¡dS) NgŽVÎÕRÀã?)rg¿Æ·CØã?)rg‚_ãÛ!ì@)rgXlWYíç?)r)r r;r rtrr)rBr˜r¡r¢rÂr#rrrÚtest_ewm_alpha÷s zTestMoments.test_ewm_alphac Cs”|j}t t¡| ¡WdQRXt t¡|jdddWdQRXt t¡|jdddWdQRXt t¡|jdddWdQRXdS)Ng$@gà?)rr)rr)rr)r>rYrZrÀr )rBr˜rrrÚtest_ewm_alpha_argszTestMoments.test_ewm_alpha_argcCst|jƒ}tjt|jdd|jdd|jddtjt|jddtjt|jddtjt|jdd|jdd|jddtjt|jdd tjt|jdd |jdd tjt|jdd tjt|jdd |jdd |jdd tjt|jdd dS)Ngš™™™™™¹¿)rggš™™™™™¹?)rgÍÌÌÌÌÌì?gð?gš™™™™™ñ?)r)r)r r;rYrZrÀr )rBr˜rrrÚtest_ewm_domain_checkss" z"TestMoments.test_ewm_domain_checksrtr®rÕcCs6tjgtjd}| d¡}t||ƒƒ}t ||¡dS)N)rÆrM)r|r r8rÈr r²rr&)rBrIrJr rnrrrÚtest_ew_empty_series&s z TestMoments.test_ew_empty_seriesc Cs0t|jjdd|ƒƒ}t|tƒs$t‚t|jjdd|ƒƒ}t|ƒtksJt‚t|jjdd|ƒƒ}|rz||j ¡ ¡szt‚tdƒ}t j|dd…<t j|dd…<t|ƒ}t|jddd|ƒƒ}|dd… ¡ ¡sÔt‚|dd… ¡ ¡rìt‚xdD]}t|jd|d|ƒƒ}|d krP|dd… ¡ ¡s4t‚|dd… ¡ ¡r„t‚n4|dd… ¡ ¡sjt‚|dd… ¡ ¡r„t‚ttƒjd|d|ƒƒ}t |tƒ¡ttd gƒjd|d|ƒƒ}|d krät |td gƒ¡qôt |tt jgƒ¡qôWttt d¡ƒjdd|ƒƒ} | jt jks,t‚dS) Nr)rrkiöÿÿÿrT)rrŸr )rr3rtgð?)rŸ)r)r²r>r rr rQrAÚtyperr:rrNrr8r9Úanyrrr@rÆZfloat_) rBrPÚpreserve_nanr™ršrnr;r˜rŸÚresult2rrrr¡.s> zTestMoments._check_ew)TTTNT)NF):rDrErFrJr@rDrHrIr¶r·rKrLrPrSrTrVrWrXr]r^r_r`rarYr¸rºrirpr8rlrtrurvryr rrzr{rr€rr‚rƒr‡rŠrBr¤r¦r¬rr¯r°r±r²r³r´rµr¡rrrrr? sx 7 r?c@seZdZeddgddgddgddggddgdeddgddgddgddggddgdeddgddgddgddggddgdeddgddgddgddggddgdeddgddgddgddggd dgdeddgddgddgddggd dgdeddgddgddgddggddgdeddgd dgd dgdd ggdd gdeddgddgddgdd ggdd gdeddgddgddgdd ggd dgdg ZedddgdddgdddgdddggdddgdZeddddgƒZdd„Ze j ddd„dd„g¡dd„ƒZe j ddd„dd„dd„dd„d d„d!d„g¡d"d#„ƒZ e j dd$d„d%d„d&d„d'd„d(d„d)d„g¡d*d+„ƒZe j dd,d„d-d„d.d„d/d„d0d„d1d„g¡d2d3„ƒZe j dd4d„d5d„d6d„d7d„d8d„d9d„g¡d:d;„ƒZe j ddd„d?d„d@d„dAd„g¡dBdC„ƒZdS)DÚTestPairwiserTr®r3rKrãr)r6rWgð?gg@g@g@g @ÚXNrMÚYÚZcCs(| ¡j}| ¡j}tj||dddS)NF)Zcheck_dtype)ÚdropnardrÚassert_numpy_array_equal)rBrnrorrrÚcompareys zTestPairwise.comparer‹cCs| ¡S)N)rõ)r†rrrra‚szTestPairwise.cCs| ¡S)N)Úcorr)r†rrrra‚scs€‡fdd„|jDƒ}x8t|j|ƒD](\}}t |j|j¡t |j|j¡q"Wx,t|ƒD] \}}|dkrX| ||d¡qXWdS)Ncsg|]}ˆ|ƒ‘qSrr)rür])r‹rrr‡sz-TestPairwise.test_no_flex..r)Údf1sr rrOr5r6Ú enumeraterÀ)rBr‹Úresultsr]rnrýr)r‹rÚtest_no_flex‚szTestPairwise.test_no_flexcCs| ¡jddS)NT)Úpairwise)rªrõ)r†rrrrascCs| ¡jddS)NT)rÆ)rªrÁ)r†rrrra‘scCs|jddjddS)NrM)rLT)rÆ)rNrõ)r†rrrra’scCs|jddjddS)NrM)rLT)rÆ)rNrÁ)r†rrrra“scCs|jddjddS)NrM)rT)rÆ)r rõ)r†rrrra”scCs|jddjddS)NrM)rT)rÆ)r rÁ)r†rrrra•scCs¬g}xtt|jƒD]f\}}||ƒ}tj|jjd|jddt t|jjdƒt|j ¡ƒ¡t |j|j¡| |¡qWx,t|ƒD] \}}|dkr„| ||d¡q„WdS)NrF)rr3)rÃrÂrrOr5Úlevelsr¿rr6ÚuniquerÑrÀ)rBr‹rÄrýr]rnrrrÚtest_pairwise_with_selfsz$TestPairwise.test_pairwise_with_selfcCs| ¡jddS)NF)rÆ)rªrõ)r†rrrra«scCs| ¡jddS)NF)rÆ)rªrÁ)r†rrrra¬scCs|jddjddS)NrM)rLF)rÆ)rNrõ)r†rrrrascCs|jddjddS)NrM)rLF)rÆ)rNrÁ)r†rrrra®scCs|jddjddS)NrM)rF)rÆ)r rõ)r†rrrra¯scCs|jddjddS)NrM)rF)rÆ)r rÁ)r†rrrra°scs€‡fdd„|jDƒ}x8t|j|ƒD](\}}t |j|j¡t |j|j¡q"Wx,t|ƒD] \}}|dkrX| ||d¡qXWdS)Ncsg|]}ˆ|ƒ‘qSrr)rür])r‹rrr´sz;TestPairwise.test_no_pairwise_with_self..r)rÂr rrOr5r6rÃrÀ)rBr‹rÄr]rnrýr)r‹rÚtest_no_pairwise_with_selfªs z'TestPairwise.test_no_pairwise_with_selfcCs| ¡j|ddS)NT)rÆ)rªrõ)r†Úyrrrra½scCs| ¡j|ddS)NT)rÆ)rªrÁ)r†rËrrrra¾scCs|jddj|ddS)NrM)rLT)rÆ)rNrõ)r†rËrrrra¿scCs|jddj|ddS)NrM)rLT)rÆ)rNrÁ)r†rËrrrraÀscCs|jddj|ddS)NrM)rT)rÆ)r rõ)r†rËrrrraÁscCs|jddj|ddS)NrM)rT)rÆ)r rÁ)r†rËrrrraÂscs ‡‡fdd„ˆjDƒ}xVtˆj|ƒD]F\}}tj|jjd|jddt t|jjdƒtˆjj ¡ƒ¡q$Wx,t|ƒD] \}}|dkrxˆ ||d¡qxWdS)Ncsg|]}ˆ|ˆjƒ‘qSr)Údf2)rür])r‹rBrrrÆsz9TestPairwise.test_pairwise_with_other..rF)rr3) rÂr rrOr5rÇr¿rrÌr6rÈrÃrÀ)rBr‹rÄr]rnrýr)r‹rBrÚtest_pairwise_with_other¼s z%TestPairwise.test_pairwise_with_othercCs| ¡j|ddS)NF)rÆ)rªrõ)r†rËrrrraÒscCs| ¡j|ddS)NF)rÆ)rªrÁ)r†rËrrrraÓscCs|jddj|ddS)NrM)rLF)rÆ)rNrõ)r†rËrrrraÔscCs|jddj|ddS)NrM)rLF)rÆ)rNrÁ)r†rËrrrraÕscCs|jddj|ddS)NrM)rF)rÆ)r rõ)r†rËrrrraÖscCs|jddj|ddS)NrM)rF)rÆ)r rÁ)r†rËrrrra×sc sè‡‡fdd„ˆjDƒ}xÌtˆj|ƒD]¼\}}|dk r”tdd2t dt¡|j ˆjj¡}|j ˆjj ¡}WdQRXt |j|¡t |j |¡q$tj tddˆ|ˆjƒWdQRXtj tddˆˆj|ƒWdQRXq$WdS) Ncs$g|]}|jjrˆ|ˆjƒnd‘qS)N)r6Z is_uniquerÌ)rür])r‹rBrrrÛsz.T)rwrxz'arg1' columns are not unique)rXz'arg2' columns are not unique)rÂr rr€rrxr5ÚunionrÌr6rrOrYrZrÀ)rBr‹rÄr]rnZexpected_indexZexpected_columnsr)r‹rBrÚtest_no_pairwise_with_otherÑs z(TestPairwise.test_no_pairwise_with_othercCs| ¡ |¡S)N)rªrõ)r†rËrrrraïscCs| ¡ |¡S)N)rªrÁ)r†rËrrrraðscCs|jdd |¡S)NrM)rL)rNrõ)r†rËrrrrañscCs|jdd |¡S)NrM)rL)rNrÁ)r†rËrrrraòscCs|jdd |¡S)NrM)r)r rõ)r†rËrrrraóscCs|jdd |¡S)NrM)r)r rÁ)r†rËrrrraôscs˜‡‡fdd„ˆjDƒ‡‡fdd„ˆjDƒ}x8tˆj|ƒD](\}}t |j|j¡t |j|j¡q:Wx,t|ƒD] \}}|dkrpˆ ||d¡qpWdS)Ncsg|]}ˆ|ˆjƒ‘qSr)r˜)rür])r‹rBrrrøsz:TestPairwise.test_pairwise_with_series..csg|]}ˆˆj|ƒ‘qSr)r˜)rür])r‹rBrrrùsr)rÂr rrOr5r6rÃrÀ)rBr‹rÄr]rnrýr)r‹rBrÚtest_pairwise_with_seriesîs z&TestPairwise.test_pairwise_with_series)rDrErFrrÂrÌr r˜rÀrYr¸rºrÅrÉrÊrÍrÏrÐrrrrrºdsl&&& &&& " rºcsDdd„‰‡fdd„}dd„‰dd„‰ˆƒ|ƒ}‡‡fd d „|DƒS)NcSs’tƒttjgƒttjtjgƒtdgƒttjdgƒtdtjgƒtddgƒtddgƒtddgƒtddddtjtjtjddtjtjgƒttjdddtjtjtjddtjtjgƒttjtjddtjtjtjddtjtjgƒttjdtjddddtjtjddd d dgƒttjdtjdddd tjtjddd d dgƒtddtjddddtjtjddd d dgƒtddtjdddd tjtjddd d dgƒttdƒƒttdddƒƒgS)Ng@gð?g@g@g@g@g@g(@g*@g,@g.@gg"@rr1rr)r r8rlrrrrrÚ create_seriess2 z/_create_consistency_data..create_seriesc s~tƒtdgdtddgdtddgdtt d¡ d¡ƒtt d¡ d¡ƒtt d¡ d¡dddd d gdgd d„ˆƒDƒS)Nr¡)r6r¢r)rKrTrj)rKrKécr#cSsg|]}t|ƒ‘qSr)r)rür˜rrrr(szG_create_consistency_data..create_dataframes..)rr8r@rr)rÑrrÚcreate_dataframess z3_create_consistency_data..create_dataframescSs"|j ¡}tt|t|ƒƒƒdkS)Nr3)rdÚravelrOÚsetr)r†rdrrrÚis_constant*s z-_create_consistency_data..is_constantcSs| ¡ ¡ ¡S)N)rrN)r†rrrÚno_nans.sz)_create_consistency_data..no_nanscsg|]}|ˆ|ƒˆ|ƒf‘qSrr)rür†)rÖr×rrr4sz,_create_consistency_data..r)rÓrÚr)rÑrÖr×rÚ_create_consistency_datasrØccsTxNdD]F}x@ddddd|hD],}|r.||kr.qxdD]}|||fVq4WqWqWdS)N)r3rTrMrr1rr3rTrMr®)FTr)rLrŸr¾rrrÚ_rolling_consistency_cases:s rÙcseZdZdd„ddfdd„ddfdd„ddfd d„dd fdd„ddfd d„ddfdd„ddfdd„ddfdd„ddfdd„ddfejddfejddfejdd fejddfdd„ddfdd„ddfejddfgZ ej ddfejddfejdd fej ddfdd„ddfdd„ddfejddfgZ‡fdd„Zdd„ZdŸd d!„Zejjej d"d#dd$d%d&g¡ej d'd(d)g¡ej d*d(d)g¡d+d,„ƒƒƒƒZejjej d"d#dd$d%d&g¡d-d.„ƒƒZejjej d/eeƒƒ¡d0d1„ƒƒZd2d3„Zd4d5„Zd6d7„Zd8d9„Zej d:e d;ƒ¡dd?„Z"d@dA„Z#dBdC„Z$ej dDddg¡dEdF„ƒZ%dGdH„Z&dIdJ„Z'dKdL„Z(dMdN„Z)dOdP„Z*dQdR„Z+dSdT„Z,dUdV„Z-dWdX„Z.dYdZ„Z/d[d\„Z0d]d^„Z1d_d`„Z2dadb„Z3dcdd„Z4dedf„Z5ej dgdhd„did„djd„dkd„dld„dmd„dnd„dod„dpd„dqd„drd„dsd„dtd„dud„dvd„g¡dwdx„ƒZ6dydz„Z7d{d|„Z8d}d~„Z9dd€„Z:dd‚„Z;dƒd„„Zd‰dŠ„Z?ejjd‹d ejfdej fdej fdejfgd dddgdŒddŽ„ƒZ@dd„ZAd d‘d’„ZBd“d”„ZCd•d–„ZDd—d˜„ZEd™dš„ZFd›dœ„ZGddž„ZH‡ZIS)¡ÚTestMomentsConsistencycCst|ƒ ¡S)N)r r›)rrrrraEszTestMomentsConsistency.Nr›cCst|ƒ ¡S)N)r r°)rrrrraFsr°cCst|ƒ ¡S)N)r r±)rrrrraGsr±cCst|ƒ ¡S)N)r rb)rrrrraHsrbcCst|ƒ ¡S)N)r rt)rrrrraIsrtcCst|ƒ ¡S)N)r ru)rrrrraJsr3rucCst|ƒ t|ƒ¡S)N)r rõ)rrrrraKsrõcCst|ƒ t|ƒ¡S)N)r rÁ)rrrrraLsrÁcCst|ƒ ¡S)N)r rÕ)rrrrraMsrÕcCst|ƒ ¡S)N)r r)rrrrraVsrcCstj|ddS)Nr3)r…)r8Znanstd)rrrrra[scCstj|ddS)Nr3)r…)r8Znanvar)rrrrra\scCstj|ddS)Nr3)r…)r8ru)rrrrradscCstj|ddS)Nr3)r…)r8rÕ)rrrrraescstt|ƒ ¡t|_dS)N)ÚsuperrÚrCÚ_consistency_datarÚ)rB)Ú __class__rrrCisz#TestMomentsConsistency._create_datacCs| ¡dS)N)rC)rBrIrrrrJmsz#TestMomentsConsistency.setup_methodc *Csdd„} xò|jD]æ\}}}||ƒ}||ƒ}|rJ||ƒ}t|| d¡ƒ|||ƒ}|r´t|tƒrj| ¡n | ¡ ¡}|tj}|||t|dƒk<t||ƒtj|dd…<t||ƒ|ræ| ræ|ræ||ƒ}| |ƒ}||ƒ}t|||ƒx| | |f|||fgD]ö\}}}||ƒ}||ƒ}|dk ¡ ¡r0t ‚|dk ¡ ¡rFt ‚|rv|||ƒ}|dk ¡ ¡rlt ‚t||ƒt|||ƒ|| kr¬|||ƒ}t||||ƒ|r|dk ¡ ¡rÈt ‚|tj}d||t|dƒk<||krütj||dk<t||ƒt|tƒrþxä|jD]Ú\} }}| ¡ | ¡¡s:q||| ƒ}!|| |ƒ}"t|!|"ƒ|r||| ƒ}#|| |ƒ}$t|#|$ƒ||| ƒ}%|| ƒ}&t|#d|%||&ƒ|| ƒ}'t|!|#||'ƒ||kr|| ƒ}(||| ƒ})t|#|)||(ƒqWqþWqWdS) NcSs|j ¡}t|t|ƒ ¡ƒS)N)rdrÔrÕrÚtolist)r†rdrrrÚ_non_null_valuesus zJTestMomentsConsistency._test_moments_consistency.._non_null_valuesrÈr3rgrTgà?)rÚrrœrr r°r8rlr·rQrZequals)*rBrŸr›rtÚ mock_meanrÁÚvar_unbiasedÚstd_unbiasedÚcov_unbiasedÚ var_biasedÚ std_biasedÚ cov_biasedÚvar_debiasing_factorsrßr†rÖr×Zcount_xZmean_xroZcorr_x_xr$Zvar_unbiased_xZvar_biased_xZvar_debiasing_factors_xrurÕrõZvar_xZstd_xZcov_x_xZmean_x2rËZcorr_x_yZcorr_y_xZcov_x_yZcov_y_xZvar_x_plus_yZvar_yZstd_yZmean_yZmean_x_times_yrrrÚ_test_moments_consistencyps€ z0TestMomentsConsistency._test_moments_consistencyrŸrrTrMr®r TFr¥csà‡fdd„‰‡fdd„‰‡fdd„‰d‰|jˆdd „‡‡‡‡fd d „‡‡‡‡‡fdd „‡‡‡‡fdd „‡‡‡‡fd d „‡‡‡‡fdd „‡‡‡‡fdd „‡‡‡‡fdd „‡‡‡‡fdd „‡‡‡‡fdd „‡‡‡‡fdd „ddS)Nc sltˆtƒr`tˆjƒs$tˆjˆjdSt‡‡‡‡‡fdd„tˆjƒDƒdd}ˆj|_ˆj|_|Sttj ˆjd}ddˆ}ˆr¢ˆˆˆ ¡ˆˆdd |ˆ ¡<nÆˆrêxÀttˆƒƒD]0}ˆj|ˆj|kr´t dd||ƒ|j|<q´Wn~d }d}xtttˆƒƒD]d}ˆj|ˆj|kr|dkr0d|j|<n ||t d|||ƒ|j|<||j|7}|}qW|S)N)r5r6cs.g|]&\}}ˆˆjdd…|fˆˆˆd‘qS)N)rr r¥)rs)rürýÚ_)Ú_weightsr rr¥r˜rrríszQTestMomentsConsistency.test_ewm_consistency.._weights..r3)rs)r5gð?F)rr r¥grÍ)rrrOr6r5rrÃr r8rlrrZiatÚpow) r˜rr r¥rÃrrýZsum_wtsZprev_i)rê)r rr¥r˜rrêès: z=TestMomentsConsistency.test_ewm_consistency.._weightsc sZˆ||||d}| ¡jdd}|| ¡jdd}||}||}tj||dk<||S)N)rr r¥r§)rIg)r©r–r8rl) r˜rr r¥ÚweightsZcum_sumZ cum_sum_sqÚ numeratorÚdenominator)rêrrÚ_variance_debiasing_factors szPTestMomentsConsistency.test_ewm_consistency.._variance_debiasing_factorscsXˆ||||d}| |¡ ¡ | ¡¡jdd}tj|| ¡ ¡|rNt|dƒndk<|S)N)rr r¥r§)rIr3) r¨r©Údivider–r8rlrªr›r°)r˜rrŸr r¥rìrn)rêrrÚ_ewma s $z:TestMomentsConsistency.test_ewm_consistency.._ewmag@cSs| ¡ ¡S)N)rªr›)r†rrrra sz=TestMomentsConsistency.test_ewm_consistency..cs|jˆˆˆˆd ¡S)N)rrŸr r¥)r rt)r†)r rr¥rŸrrra! scsˆ|ˆˆˆˆdS)N)rrŸr r¥r)r†)rñr rr¥rŸrrra$ scs|jˆˆˆˆd |¡S)N)rrŸr r¥)r rÁ)r†rË)r rr¥rŸrrra( scs|jˆˆˆˆdjddS)N)rrŸr r¥F)r…)r rÕ)r†)r rr¥rŸrrra+ scs|jˆˆˆˆdjddS)N)rrŸr r¥F)r…)r ru)r†)r rr¥rŸrrra/ s cs|jˆˆˆˆdj|ddS)N)rrŸr r¥F)r…)r rõ)r†rË)r rr¥rŸrrra3 s cs|jˆˆˆˆdjddS)N)rrŸr r¥T)r…)r rÕ)r†)r rr¥rŸrrra7 s cs|jˆˆˆˆdjddS)N)rrŸr r¥T)r…)r ru)r†)r rr¥rŸrrra; scs|jˆˆˆˆdj|ddS)N)rrŸr r¥T)r…)r rõ)r†rË)r rr¥rŸrrra> s csˆ|ˆˆˆdS)N)rr r¥r)r†)rïr rr¥rrraB s)rŸr›rtràrÁrárârãrärårærç)rè)rBrŸr r¥r)rñrïrêr rr¥rŸrÚtest_ewm_consistencyãs"# z+TestMomentsConsistency.test_ewm_consistencycspt ¡\tjddtd|jˆdd„‡fdd„‡fdd„‡fdd„‡fd d„‡fd d„‡fdd„‡fdd„‡fd d„‡fdd„dd„dxÒ|jD]È\}}}|j}|r¶|j|j}x¦|D]ž\}}}t|j ˆd|ƒ} |rîˆdk rîˆ|krîq¼|dkr| ƒ} |j ddj |dd}n2|dkr,| dd} n| ƒ} |j ˆdj |dd}|dkr¼t| |ƒq¼Wq–WWdQRXdS)Nrxz.*(empty slice|0 for slice).*)rwr9cSs| ¡ ¡S)N)rªr›)r†rrrraU szCTestMomentsConsistency.test_expanding_consistency..cs|jˆd ¡S)N)rŸ)rªrt)r†)rŸrrraV scs|jˆd ¡| ¡ ¡S)N)rŸ)rªrbr›)r†)rŸrrraX scs|jˆd |¡S)N)rŸ)rªrÁ)r†rË)rŸrrraZ scs|jˆd ¡S)N)rŸ)rªrÕ)r†)rŸrrra\ scs|jˆd ¡S)N)rŸ)rªru)r†)rŸrrra^ scs|jˆd |¡S)N)rŸ)rªrõ)r†rË)rŸrrra` scs|jˆdjddS)N)rŸr)r…)rªrÕ)r†)rŸrrrab scs|jˆdjddS)N)rŸr)r…)rªru)r†)rŸrrrad scs|jˆdj|ddS)N)rŸr)r…)rªrõ)r†rË)rŸrrraf scSs&| ¡ ¡| ¡ ¡d dtj¡S)Ngð?g)rªr›Úreplacer8rl)r†rrrrah s )rŸr›rtràrÁrárârãrärårærç)rŸr›rT)r´r")rõrÁF)rÆ)rbÚprod)r€rr¹rxrèrÚÚbase_functionsÚno_nan_functionsr²rªrˆr)rBrŸr†rÖr×Ú functionsr‹Úrequire_min_periodsrPZexpanding_fZexpanding_f_resultZexpanding_apply_f_resultr)rŸrÚtest_expanding_consistencyF sR z1TestMomentsConsistency.test_expanding_consistencyzwindow,min_periods,centercs²t ¡žtjddtd|jˆ‡‡fdd„‡‡‡fdd„‡‡‡fdd„‡‡‡fdd„‡‡‡fd d„‡‡‡fd d„‡‡‡fdd„‡‡‡fdd„‡‡‡fd d„‡‡‡fdd„‡‡fdd„dxä|jD]Ú\}}}|j}|ræ|j|j}x¸|D]°\}} } t|j ˆˆˆd| ƒ}| r(ˆdk r(ˆ| kr(qì| dkrT|ƒ}|j ˆdˆdj |dd} n6| dkrj|dd}n|ƒ}|j ˆˆˆdj |dd} | dkrìt|| ƒqìWqÆWWdQRXdS)Nrxz.*(empty slice|0 for slice).*)rwr9cs|jˆˆd ¡S)N)rLr¾)rNr›)r†)r¾rLrrra szATestMomentsConsistency.test_rolling_consistency..cs|jˆˆˆd ¡S)N)rLrŸr¾)rNrt)r†)r¾rŸrLrrra£ scs*|jˆˆˆd ¡ |jˆˆˆd ¡¡S)N)rLrŸr¾)rNrbrðr›)r†)r¾rŸrLrrra¦ scs|jˆˆˆd |¡S)N)rLrŸr¾)rNrÁ)r†rË)r¾rŸrLrrra scs|jˆˆˆd ¡S)N)rLrŸr¾)rNrÕ)r†)r¾rŸrLrrra± scs|jˆˆˆd ¡S)N)rLrŸr¾)rNru)r†)r¾rŸrLrrraµ scs|jˆˆˆd |¡S)N)rLrŸr¾)rNrõ)r†rË)r¾rŸrLrrra¹ scs|jˆˆˆdjddS)N)rLrŸr¾r)r…)rNrÕ)r†)r¾rŸrLrrra½ scs|jˆˆˆdjddS)N)rLrŸr¾r)r…)rNru)r†)r¾rŸrLrrraÁ scs|jˆˆˆdj|ddS)N)rLrŸr¾r)r…)rNrõ)r†rË)r¾rŸrLrrraÅ scs4|jˆˆd ¡ |jˆˆd ¡d dtj¡¡S)N)rLr¾gð?g)rNr›rðrór8rl)r†)r¾rLrrraÈ s)rŸr›rtràrÁrárârãrärårærç)rLr¾rŸr›r)rLrŸr¾T)r´r")rõrÁF)rÆ)rbrô)r€rr¹rxrèrÚrõrör²rNrˆr)rBrLrŸr¾r†rÖr×r÷r‹rørPZ rolling_fZrolling_f_resultZrolling_apply_f_resultr)r¾rŸrLrÚtest_rolling_consistency‘ sX z/TestMomentsConsistency.test_rolling_consistencyc CsZ|j}|tt|ƒƒ}|jddd |¡}t |dt |dd…|dd…¡d¡dS)Nrkrj)rLrŸrÍiÎÿÿÿ)rr3)r>rrOrNrõrr&r8)rBrUrVrnrrrÚtest_rolling_covö sz'TestMomentsConsistency.test_rolling_covcCs|jddddddS)NrNrõrrK)rLrŸ)Ú_check_pairwise_moment)rBrrrÚtest_rolling_cov_pairwiseý sz0TestMomentsConsistency.test_rolling_cov_pairwisec Cs´|j}|tt|ƒƒ}|jddd |¡}t |dt |dd…|dd…¡d¡t ¡}t ¡}tj |dd…<tj |dd…<|jt|ƒd d |¡}t |d| |¡¡dS) Nrkrj)rLrŸrÍiÎÿÿÿ)rr3rKrr3)r>rrOrNrÁrr&r8ZcorrcoefZmakeTimeSeriesrl)rBrUrVrnr¡r¢rrrÚtest_rolling_corr s,z(TestMomentsConsistency.test_rolling_corrcCs|jddddddS)NrNrÁrrK)rLrŸ)rü)rBrrrÚtest_rolling_corr_pairwise s z1TestMomentsConsistency.test_rolling_corr_pairwiserLrácCsDt t d¡¡}t t d¡¡}|j|dj|d ¡ ¡s@t ‚dS)Nr1)rL)Úother) r|r r8r¢r@rNrÁrrNrQ)rBrLr˜rrrrÚ$test_rolling_corr_with_zero_variance sz;TestMomentsConsistency.test_rolling_corr_with_zero_variancecsjd‡‡‡fdd„ }||jƒ}|jtdƒdfdf}|j d¡|_||jd|jdƒ}tj||dddS)Ncstt|ˆƒfˆŽˆƒ|ƒS)N)r²)rúr)Údispatchr‹rPrrrŒ szATestMomentsConsistency._check_pairwise_moment..get_resultr3rKF)r)N)rAÚlocÚslicer5Z droplevelrr)rBrrPr‹rŒrnror)rr‹rPrrü s z-TestMomentsConsistency._check_pairwise_momentcCst ttjddd¡dS)NrKrf)rYrZrqrÇZ_flex_binary_moment)rBrrrÚtest_flex_binary_moment& sz.TestMomentsConsistency.test_flex_binary_momentc Csàtt ddgddgddgddgd d gddgg¡ƒ}|d jddd |d¡}tdd„|Dƒƒsdt‚xvtdƒD]j}ttj dd¡ƒ}|d jddd |d¡}ytdd„|Dƒƒs¸t‚Wqntk rÖt |ƒYqnXqnWdS)Ng†F$ÈÙë?g)—Òé¯Ç?g‰ËT5ÿ—ä?g´?«'NÉÓ?gFr1í?g+ÿ” C-à?gR ²±`?gÕ1:b%§?g9T!X4!ë?gXÀÊÂS_Õ?gS®=é?g\–+“Ltä?rrKT)r¾r3css"|]}t t |¡¡dkVqdS)r3N)r8r£Ú nan_to_num)rür†rrrú 3 sz:TestMomentsConsistency.test_corr_sanity..rr¥rTcss"|]}t t |¡¡dkVqdS)r3N)r8r£r)rür†rrrr: s)rr8rGrNrÁrNrQrr£rmÚprint)rBr]ÚresrérrrÚtest_corr_sanity+ sz'TestMomentsConsistency.test_corr_sanityrIcsÎˆjd‰tˆjddˆƒˆjƒ}tˆjjddˆƒˆƒ}ˆj ‡‡fdd„¡}t ||¡t ||¡ˆj ¡‰tjj ˆj Žˆjdd…<tˆjjddˆƒˆƒ}t‡‡‡fdd„ˆjDƒƒ}t ||¡dS)Nr3r)rLcstˆjddˆƒ|ƒS)Nr)rL)r²rN)r†)rIr>rrraD sz?TestMomentsConsistency.test_flex_binary_frame..cs.i|]&}tˆj|jddˆƒˆ|ƒ|“qS)r)rL)r²rArN)rüÚk)Úframe2rIrBrrrþN szATestMomentsConsistency.test_flex_binary_frame..) rAr²rNrˆrrr=r8r£rrcrdr)rBrIr Zres2r$Zres3r)rrIrBr>rÚtest_flex_binary_frame> s z-TestMomentsConsistency.test_flex_binary_framecCs| d¡dS)Nrõ)Ú_check_binary_ew)rBrrrÚtest_ewmcovR sz"TestMomentsConsistency.test_ewmcovcCs|jddddddS)Nr rõrrK)rrŸ)rü)rBrrrÚtest_ewmcov_pairwiseU sz+TestMomentsConsistency.test_ewmcov_pairwisecCs| d¡dS)NrÁ)r)rBrrrÚtest_ewmcorrX sz#TestMomentsConsistency.test_ewmcorrcCs|jddddddS)Nr rÁrrK)rrŸ)rü)rBrrrÚtest_ewmcorr_pairwise[ sz,TestMomentsConsistency.test_ewmcorr_pairwisecsf‡fdd„}ttdƒt d¡d}|dd…tdƒ}tj|dd…<tj|dd…<|||d d d}t |jdd…¡ ¡s~t‚t |jdd…¡ ¡ršt‚x¬d D]¤}|||d |d}t |jdd…¡ ¡sÐt‚t |jdd…¡ ¡rìt‚|tgƒtgƒd|d}t |tgƒ¡|tdgƒtdgƒd|d}t |ttjgƒ¡q Wtj t||tdƒd d ddS)Ncst|j|f|Žˆƒ|ƒS)N)r²r )rUrVrr‹)rPrrr´_ sz5TestMomentsConsistency._check_binary_ew..funcrk)r5rTé0riöÿÿÿr1rK)rŸr)rr3rTr gð?)r rr8r@r9rMrdrNrQr·rrrYrZÚ Exception)rBrPr´rUrVrnrŸr)rPrr^ s$ z'TestMomentsConsistency._check_binary_ewcCszdd„}ttj dd¡ƒ}| ¡jtj|dd}| ¡j||dd}t ||¡| ¡j||d did }t ||¡dS)NcSst |¡|S)N)r8rt)r†ÚconstrrrÚ mean_w_arg szKTestMomentsConsistency.test_expanding_apply_args_kwargs..mean_w_argr1rM)r"g4@)r1)r"Úargsr)r"r‹) rr8r£rmrªrˆrtrr)rBr"rr]rornrrrÚ test_expanding_apply_args_kwargs s z7TestMomentsConsistency.test_expanding_apply_args_kwargscCsX|j ¡}|tt|ƒƒdd…}| ¡ |¡}|jt|ƒdd |¡}t ||¡dS)Nrr3)rLrŸ) r>r¾rrOrªrÁrNrr&)rBrUrVrnÚrolling_resultrrrÚtest_expanding_corr’ s z*TestMomentsConsistency.test_expanding_corrcCs2|j ¡ ¡}t ||jjt|jƒd ¡¡dS)N)rL)r>rªr›rr&rNrO)rBrnrrrÚtest_expanding_countœ sz+TestMomentsConsistency.test_expanding_countcCs<|j ¡ d¡}|jjt|jƒdd d¡}t ||¡dS)Ngà?r3)rLrŸ)r>rªrhrNrOrr&)rBrnrrrrÚtest_expanding_quantile¡ sz.TestMomentsConsistency.test_expanding_quantilecCsT|j}|tt|ƒƒdd…}| ¡ |¡}|jt|ƒdd |¡}t ||¡dS)Nrr3)rLrŸ)r>rrOrªrõrNrr&)rBrUrVrnrrrrÚtest_expanding_cov© s z)TestMomentsConsistency.test_expanding_covcCs8|j ¡ ¡}|jjt|jƒdd ¡}t ||¡dS)Nr3)rLrŸ)rArªrÁrNrOrr)rBrnrrrrÚtest_expanding_cov_pairwise³ sz2TestMomentsConsistency.test_expanding_cov_pairwisecCs8|j ¡ ¡}|jjt|jƒdd ¡}t ||¡dS)Nr3)rLrŸ)rArªrÁrNrOrr)rBrnrrrrÚtest_expanding_corr_pairwise» sz3TestMomentsConsistency.test_expanding_corr_pairwisecCsätdddgdddgd}tddgddgd}| ¡ |¡}tdddgƒ}t ||¡tdddgdddgd}| ¡ |¡}t ||¡tddd gdddgd}tdd d gdddgd}| ¡ |¡}tddddgƒ}t ||¡dS)Nr3rTrMr)r5g@rárãrrgg@)r rªrõrr)rBr«r¦rnroÚs2arrrÚtest_expanding_cov_diff_indexÂ sz4TestMomentsConsistency.test_expanding_cov_diff_indexcCsätdddgdddgd}tddgddgd}| ¡ |¡}tdddgƒ}t ||¡tdddgdddgd}| ¡ |¡}t ||¡tddd gdddgd}tdd d gdddgd}| ¡ |¡}tddddgƒ}t ||¡dS)Nr3rTrMr)r5gð?rárãrrg)r rªrÁrr)rBr«r¦rnror rrrÚtest_expanding_corr_diff_indexÔ sz5TestMomentsConsistency.test_expanding_corr_diff_indexcCs–tdddgdddgd}tddgddgd}|jddd |¡}tdddgƒ}t ||¡tdddgdddgd}|jddd |¡}t ||¡dS)Nr3rTrMr)r5)rLrŸg@)r rNrõrr)rBr«r¦rnror rrrÚtest_rolling_cov_diff_lengthæ sz3TestMomentsConsistency.test_rolling_cov_diff_lengthcCs–tdddgdddgd}tddgddgd}|jddd |¡}tdddgƒ}t ||¡tdddgdddgd}|jddd |¡}t ||¡dS)Nr3rTrMr)r5)rLrŸgð?)r rNrÁrr)rBr«r¦rnror rrrÚtest_rolling_corr_diff_lengthò sz4TestMomentsConsistency.test_rolling_corr_diff_lengthr‹cCs|jdddj|ddS)NrrK)rLrŸF)rÆ)rNrõ)r†rrrrascCs|jdddj|ddS)NrrK)rLrŸF)rÆ)rNrÁ)r†rrrrascCs|jddd ¡S)NrrK)rLrŸ)rNr°)r†rrrrascCs|jddd ¡S)NrrK)rLrŸ)rNr±)r†rrrrascCs|jddd ¡S)NrrK)rLrŸ)rNrb)r†rrrrascCs|jddd ¡S)NrrK)rLrŸ)rNrt)r†rrrrascCs|jddd ¡S)NrrK)rLrŸ)rNru)r†rrrra scCs|jddd ¡S)NrrK)rLrŸ)rNrÕ)r†rrrra scCs|jddd ¡S)NrrK)rLrŸ)rNr„)r†rrrrascCs|jddd ¡S)NrrK)rLrŸ)rNr‰)r†rrrrascCs|jdddjddS)NrrK)rLrŸgà?)rh)rNrh)r†rrrra scCs|jddd ¡S)NrrK)rLrŸ)rNr)r†rrrrascCs|jdddjtddS)NrrK)rLrŸF)r")rNrˆrb)r†rrrrascCs|jdddjtddS)NrrK)rLrŸT)r")rNrˆrb)r†rrrrascCs|jdddd ¡S)Nr½rrK)r rLrŸ)rNrt)r†rrrrascCs¨ttdƒƒ}ttj|jd}tddgddgddgdd ggd dgd}ttj|j|jd }y,||ƒ}t ||¡||ƒ}t ||¡Wnt k r¢t d¡YnXdS)Nr®)r5r3rKrMrTrgrÍrrUrV)r6)r5r6zscipy not available) r rr8rlr5rr6rrrÚImportErrorrYr>)rBr‹r˜Ú s_expectedr]Údf_expectedÚs_resultÚ df_resultrrrÚ+test_rolling_functions_window_non_shrinkageþ s(zBTestMomentsConsistency.test_rolling_functions_window_non_shrinkagecCs¦tddgddgddgddggtdd gd dttdƒd dd}ttdd gd dtjj|j|jgd d gddd}dd„dd„g}x |D]}||ƒ}t ||¡q†WdS)Nr3rKrMrTrgrÍrrUrVre)rPr®Úbar)r6r5)r¯rÈ)r6r5rÆcSs|jdddj|ddS)NrrK)rLrŸT)rÆ)rNrõ)r†rrrra3sz[TestMomentsConsistency.test_rolling_functions_window_non_shrinkage_binary..cSs|jdddj|ddS)NrrK)rLrŸT)rÆ)rNrÁ)r†rrrra5s) rrrr|r}r~r5r6rr)rBr]r'r÷r‹r)rrrÚ2test_rolling_functions_window_non_shrinkage_binary(s zITestMomentsConsistency.test_rolling_functions_window_non_shrinkage_binaryc Cs`tƒ}|}tƒ}|}tdgd}|d d¡|d<|}dd„dd„dd„dd„d d„d d„dd„dd„d d„dd„dd„dd„dd„dd„dd„dd„dd„dd„dd„dd„dd„dd„dd„dd„dd„dd„dd„d d„d!d„d"d„d#d„g}xf|D]^}y@||ƒ} t | |¡||ƒ} t | |¡||ƒ}t ||¡Wqútk rVwúYqúXqúWdS)$Nr¡)r6rÈcSs| ¡ ¡S)N)rªr›)r†rrrraEszJTestMomentsConsistency.test_moment_functions_zero_length..cSs|jddj|ddS)NrK)rŸF)rÆ)rªrõ)r†rrrraFscSs|jddj|ddS)NrK)rŸF)rÆ)rªrÁ)r†rrrraHscSs|jdd ¡S)NrK)rŸ)rªr°)r†rrrraJscSs|jdd ¡S)NrK)rŸ)rªr±)r†rrrraKscSs|jdd ¡S)NrK)rŸ)rªrb)r†rrrraLscSs|jdd ¡S)NrK)rŸ)rªrt)r†rrrraMscSs|jdd ¡S)NrK)rŸ)rªru)r†rrrraNscSs|jdd ¡S)NrK)rŸ)rªrÕ)r†rrrraOscSs|jdd ¡S)NrK)rŸ)rªr„)r†rrrraPscSs|jdd ¡S)NrK)rŸ)rªr‰)r†rrrraQscSs|jdd d¡S)NrK)rŸgà?)rªrh)r†rrrraRscSs|jdd ¡S)NrK)rŸ)rªr)r†rrrraSscSs|jddjtddS)NrK)rŸF)r")rªrˆrb)r†rrrraTscSs|jddjtddS)NrK)rŸT)r")rªrˆrb)r†rrrraVscSs|jdd ¡S)Nr)rL)rNr›)r†rrrraXscSs|jdddj|ddS)NrrK)rLrŸF)rÆ)rNrõ)r†rrrraYscSs|jdddj|ddS)NrrK)rLrŸF)rÆ)rNrÁ)r†rrrra[scSs|jddd ¡S)NrrK)rLrŸ)rNr°)r†rrrra]scSs|jddd ¡S)NrrK)rLrŸ)rNr±)r†rrrra^scSs|jddd ¡S)NrrK)rLrŸ)rNrb)r†rrrra_scSs|jddd ¡S)NrrK)rLrŸ)rNrt)r†rrrra`scSs|jddd ¡S)NrrK)rLrŸ)rNru)r†rrrraascSs|jddd ¡S)NrrK)rLrŸ)rNrÕ)r†rrrrabscSs|jddd ¡S)NrrK)rLrŸ)rNr„)r†rrrracscSs|jddd ¡S)NrrK)rLrŸ)rNr‰)r†rrrradscSs|jddd d¡S)NrrK)rLrŸgà?)rNrh)r†rrrraescSs|jddd ¡S)NrrK)rLrŸ)rNr)r†rrrragscSs|jdddjtddS)NrrK)rLrŸF)r")rNrˆrb)r†rrrrahscSs|jdddjtddS)NrrK)rLrŸT)r")rNrˆrb)r†rrrrajscSs|jdddd ¡S)Nr½rrK)r rLrŸ)rNrt)r†rrrrals)r rrœrrrr%)rBr˜r&Údf1Údf1_expectedrÌÚdf2_expectedr÷r‹r(Ú df1_resultÚ df2_resultrrrÚ!test_moment_functions_zero_length;s` z8TestMomentsConsistency.test_moment_functions_zero_lengthc Csàtƒ}|}ttdgddtgddd}|d d¡|d<ttj |j|jg¡tgƒd}ttjj|j|jgddgdtdgdddd }d d„dd„d d„dd„g}x4|D],}||ƒ}t ||¡||ƒ}t ||¡q¬WdS)Nr¡re)rPr+)r6r5rÈ)r5r6)r¯)r5r6rÆcSs|jddj|ddS)NrK)rŸT)rÆ)rªrõ)r†rrrraszSTestMomentsConsistency.test_moment_functions_zero_length_pairwise..cSs|jddj|ddS)NrK)rŸT)rÆ)rªrÁ)r†rrrra‘scSs|jdddj|ddS)NrrK)rLrŸT)rÆ)rNrõ)r†rrrra“scSs|jdddj|ddS)NrrK)rLrŸT)rÆ)rNrÁ)r†rrrra•s) rrrœr|r}r~r5r6rr) rBr-r.rÌr/r÷r‹r0r1rrrÚ*test_moment_functions_zero_length_pairwise~s, zATestMomentsConsistency.test_moment_functions_zero_length_pairwisec Csntddgddgddggtddgdd d }tddgddggddgtddgdd d}tdd gddgddggtddgdd d }tdd gddggddgtddgdd d}| ¡j|ddjd}| ¡j|ddjd}| ¡j|ddjd}| ¡j|ddjd}tddgddggtddgdd tddgdd d} t || ¡t || ¡t || ¡t || ¡dS)Nr3rKrMrTrgrUrVre)rP)r6r)r5r6rfr»r¼T)rÆgÀgÀgÀg$À)r6r5)rrrªrõrrr) rBr-Údf1arÌÚdf2arŸr¹Úresult3Úresult4rorrrÚ'test_expanding_cov_pairwise_diff_lengthŸs*z>TestMomentsConsistency.test_expanding_cov_pairwise_diff_lengthc Csntddgddgddggddgttdƒddd }tddgddggtd dgddddgd}tdd gddgddggddgttdƒddd }tdd gddggtd dgddddgd}| ¡j|ddjd}| ¡j|ddjd}| ¡j|ddjd}| ¡j|ddjd}tddgddggddgtddgƒd } t || ¡t || ¡t || ¡t || ¡dS)Nr3rTrMr®rUrVr+)rP)r6r5r)r5r6rKrfr»r¼T)rÆgð¿)rrrrªrÁrrr) rBr-r4rÌr5rŸr¹r6r7rorrrÚ(test_expanding_corr_pairwise_diff_length¹s.z?TestMomentsConsistency.test_expanding_corr_pairwise_diff_lengthcCs°ttjgdƒ}tdgdƒ}|jdd ¡}t ||¡ttj d¡ƒ}|jdd ¡}t ||¡tddddd gƒ}ttjtjtjd dgƒ}|jdd ¡}t ||¡dS) NrKr3)rLrTgá¸òH"ø¿g¡òT,#šÀ¿gsRNjöÈ?gm ;÷äÂû?g6ÊÜ×Ú?gA.qäÈÆ?g‘(´¬ûÇø?r®) r r8r9rNr„rrr£r)rBÚall_nanr#r†rorrrÚtest_rolling_skew_edge_casesÓsz3TestMomentsConsistency.test_rolling_skew_edge_casescCs°ttjgdƒ}tdgdƒ}|jdd ¡}t ||¡ttj d¡ƒ}|jdd ¡}t ||¡tddddd gƒ}ttjtjtjd dgƒ}|jdd ¡}t ||¡dS) NrKr3)rLrMgá¸òH"ø¿g¡òT,#šÀ¿gsRNjöÈ?gm ;÷äÂû?g6ÊÜ×Ú?gÒ8ÔïÂ–ó?g¦_"Þ:_@r®) r r8r9rNr‰rrr£r)rBr:r#r†rorrrÚtest_rolling_kurt_edge_casesèsz3TestMomentsConsistency.test_rolling_kurt_edge_casescCs0tdgdƒjdd ¡}t |¡ ¡s,t‚dS)Ngš™™™™™ñ?rŽr)rL)r rNr„r8rMrNrQ)rBr¡rrrÚ test_rolling_skew_eq_value_fperrýsz7TestMomentsConsistency.test_rolling_skew_eq_value_fperrcCs0tdgdƒjdd ¡}t |¡ ¡s,t‚dS)Ngš™™™™™ñ?rŽr)rL)r rNr‰r8rMrNrQ)rBr¡rrrÚ test_rolling_kurt_eq_value_fperrsz7TestMomentsConsistency.test_rolling_kurt_eq_value_fperrzfunc,static_comp)Zidscs"d‡fdd„ }|j||dddS)Nr3Frcs|j|||d}t|ˆƒƒS)N)rŸr¾rs)rªr²)r†rŸr¾rsr$)r´rrÚexpanding_func s zBTestMomentsConsistency.test_expanding_func..expanding_func)r¸)r3Fr)Ú_check_expanding)rBr´r—r?r)r´rÚtest_expanding_funcsz*TestMomentsConsistency.test_expanding_funccsŒd‡fdd„ }|j|tjddtgƒ}t || ¡jdd„ˆd¡tdddgƒ}|jd d jdd„ˆd}tdd dgƒ}t ||¡dS)Nr3cs"|j|d}|jdd„ˆd}|S)N)rŸcSs| ¡S)N)rt)r†rrrraszUTestMomentsConsistency.test_expanding_apply..expanding_mean..)r")rªrˆ)r†rŸr$rn)r"rrÚexpanding_meanszCTestMomentsConsistency.test_expanding_apply..expanding_meanF)r¸cSs| ¡S)N)rt)r†rrrra!sz=TestMomentsConsistency.test_expanding_apply..)r"r)rŸcSst|ƒS)N)rO)r†rrrra%sgð?g@g@)r3)r@r8rtr rrrªrˆ)rBr"rBrÜr˜rnror)r"rÚtest_expanding_applysz+TestMomentsConsistency.test_expanding_applyc Cs„||jƒ}t|tƒst‚||jƒ}t|tƒs0t‚||jƒ}t |d||jdd…ƒ¡|rt|j|j ¡ ¡stt‚ttdƒƒ} |rZ|| dd}|dd… ¡ ¡sªt‚t |jd|| dd…ƒ¡|| dd}t |jd ƒsæt‚t |jd ƒsøt‚ttdƒƒ} || dd}t |d ƒs"t‚t |dƒs4t‚|| dd}|| dd}t ||¡n&|| ƒ}t |jd|| dd…ƒ¡dS)Nrr rkr¥)rŸérÍrŽé rr1rKrMr®rr3)r>rr rQrArrr&rsr:rrNrr) rBr´r—rFr˜r¸r™ršrnrÜZser2ržrŸrrrr@)s2 z'TestMomentsConsistency._check_expandingcCs¢dd„tddƒDƒ}| tdddddƒ¡ttddƒ|d }| d d„¡}| ¡}tdd dddgdd„tddƒDƒd }| d¡ ¡jdd ¡}t ||¡dS)z%Replicate result expected in GH #6297cSsg|]}tdd|ƒ‘qS)i·r3)r)rürýrrrrSszBTestMomentsConsistency.test_rolling_max_gh6297..r3rfi·rMrrá)r5cSst|ƒS)N)r)r†rrrraXsz@TestMomentsConsistency.test_rolling_max_gh6297..gð?g@g@g@g@cSsg|]}tdd|dƒ‘qS)i·r3r)r)rürýrrrr]srÏ)rLN)rrÑrr ÚmapÚ sort_indexr‘r°rNrr)rBÚindicesr>ror†rrrÚtest_rolling_max_gh6297Psz.TestMomentsConsistency.test_rolling_max_gh6297cCs\dd„tddƒDƒ}| tddddƒ¡| tddddƒ¡tttddƒƒd d g|d}| dd „¡}| ¡}tdddddgdd„tddƒDƒd}| d¡ ¡j dd ¡}t ||¡tdddddgdd„tddƒDƒd}| d¡ ¡j dd ¡}t ||¡d}tdddd|gdd„tddƒDƒd}| d¡ ¡j dd ¡}t ||¡dS)NcSsg|]}tdd|ƒ‘qS)i·r3)r)rürýrrrrcszDTestMomentsConsistency.test_rolling_max_resample..r3rfi·rKrTrrr1)r5cSst|ƒS)N)r)r†rrrraiszBTestMomentsConsistency.test_rolling_max_resample..ggð?g@g@g4@cSsg|]}tdd|dƒ‘qS)i·r3r)r)rürýrrrrosrÏ)rLg$@cSsg|]}tdd|dƒ‘qS)i·r3r)r)rürýrrrrusg«ªªªªª&@cSsg|]}tdd|dƒ‘qS)i·r3r)r)rürýrrrr|s)rrÑrr rmrFrGr‘r°rNrrrrt)rBrHr>ror†rrrrÚtest_rolling_max_resampleas&z0TestMomentsConsistency.test_rolling_max_resamplecCsÀdd„tddƒDƒ}| tddddƒ¡| tddddƒ¡tttddƒƒd d g|d}| dd „¡}| ¡}tdddddgdd„tddƒDƒd}| d¡ ¡j dd}t || ¡¡dS)NcSsg|]}tdd|ƒ‘qS)i·r3)r)rürýrrrr‚szDTestMomentsConsistency.test_rolling_min_resample..r3rfi·rKrTrrr1)r5cSst|ƒS)N)r)r†rrrraˆszBTestMomentsConsistency.test_rolling_min_resample..ggð?g@g@g@cSsg|]}tdd|dƒ‘qS)i·r3r)r)rürýrrrrŽsrÏ)rL)rrÑrr rmrFrGr‘r±rNrr)rBrHr>rorRrrrÚtest_rolling_min_resample€sz0TestMomentsConsistency.test_rolling_min_resamplecCsÀdd„tddƒDƒ}| tddddƒ¡| tddddƒ¡tttddƒƒd d g|d}| dd „¡}| ¡}tddddd gdd„tddƒDƒd}| d¡ ¡j dd ¡}t ||¡dS)NcSsg|]}tdd|ƒ‘qS)i·r3)r)rürýrrrr”szGTestMomentsConsistency.test_rolling_median_resample..r3rfi·rKrTrrr1)r5cSst|ƒS)N)r)r†rrrrašszETestMomentsConsistency.test_rolling_median_resample..ggð?g@g@cSsg|]}tdd|dƒ‘qS)i·r3r)r)rürýrrrr srÏ)rL)rrÑrr rmrFrGr‘rrNrr)rBrHr>ror†rrrÚtest_rolling_median_resample’sz3TestMomentsConsistency.test_rolling_median_resamplecCsDd}ttj |¡ƒjddd ¡ttj |¡ƒjddd ¡dS)Ni NrTF)rLr¾)r r8r£rrNr)rBrÒrrrÚ test_rolling_median_memory_error¤sz7TestMomentsConsistency.test_rolling_median_memory_errorcCs dd„dDƒ}| dd„dDƒ¡xx|D]p}ttjd|dƒjdd ¡}|jd t d¡ksbt‚ttjd|dƒjdd ¡}|jd t d¡ks(t‚q(WdS)NcSsg|]}t d |¡¡‘qS)zf{})r8rÆÚformat)rür[rrrrszMTestMomentsConsistency.test_rolling_min_max_numeric_types..)r®rãc Ss(g|] }dD]}t d ||¡¡‘qqS)Zuiz{}{})r8rÆrN)rür[Zsignrrrr®s)r3rTr®rãr1)rÆrK)rLrZf8) Úextendrr8r@rNr°ZdtypesrÆrQr±)rBZ types_testZ data_typernrrrÚ"test_rolling_min_max_numeric_typesªs z9TestMomentsConsistency.test_rolling_min_max_numeric_types)NNNNNNN)TTT)JrDrErFr8ZnanmaxZnanminrCZnanmeanZ nanmedianrõr°r±rbrtrrörCrJrèrYr¸ZslowrºròrùrmrÙrúrûrýrþrÿrrrürr r rrrrrrrrrrrrr!r"r#r$r*r,r2r3r8r9r;r<r=r>rArCr@rIrJrKrLrMrPÚ __classcell__rr)rÝrrÚCsÒ o `Ic ! C! &rÚc@sdeZdZdd„Zdd„Zdd„Zdd„Zd d „Zdd„Zd d„Z dd„Z dd„Zdd„Zdd„Z dS)ÚTestGrouperGroupingcCsDtt d¡ƒ|_tdgddgddgdt d¡d œƒ|_dS) Nrr3r1rTrrMrãr2)rUrV)r r8r@r>rrA)rBrIrrrrJ¾sz TestGrouperGrouping.setup_methodcsL‡fdd„}t t|¡ˆj d¡}|jr.t‚ˆjjddd}|jsHt‚dS)NcsˆjjddddS)NrUr3)re)rAr³r)rBrrr‹Åsz+TestGrouperGrouping.test_mutated..frUT)Úmutated)rYrZrqrAr³rSrQ)rBr‹r^r)rBrÚtest_mutatedÃs z TestGrouperGrouping.test_mutatedcCs¬|j d¡}|jjddd}|j dd„¡}| d¡ ¡j}t ||¡| d¡j ¡}t ||¡|j d¡ ¡}t ||¡|jj |jj¡ d¡ ¡}t ||¡dS)NrUT)rScSs| d¡ ¡S)NrT)rNrt)r†rrrraÒsz2TestGrouperGrouping.test_getitem..rT) rAr³rVrˆrNrtrrrU)rBr^Ú g_mutatedrornrrrrSÎsz TestGrouperGrouping.test_getitemcCsf|j d¡}| d¡}|jjddd}|j dd„¡}|j ¡}t ||¡|j ¡}t ||¡dS)NrUrTT)rScSs| d¡ ¡S)NrT)rNr›)r†rrrraæsz;TestGrouperGrouping.test_getitem_multiple..)rAr³rNrVrˆr›rr)rBr^rRrUrornrrrÚtest_getitem_multipleàs z)TestGrouperGrouping.test_getitem_multiplecs´|j d¡}|jdd}x6dD].‰t|ˆƒƒ}| ‡fdd„¡}t ||¡qWx:dD]2‰t|ˆƒdd }| ‡fd d„¡}t ||¡qVW| d¡}| dd„¡}t ||¡dS) NrUr®)rL)rbrtr±r°r›r‰r„cst| d¡ˆƒƒS)Nr®)r²rN)r†)r‹rrraõsz2TestGrouperGrouping.test_rolling..)rurÕr3)r…cst| d¡ˆƒddS)Nr®r3)r…)r²rN)r†)r‹rrraúsgà?cSs| d¡ d¡S)Nr®gà?)rNrh)r†rrrraþs)rAr³rNr²rˆrrrh)rBr^rRrnror)r‹rÚtest_rollingîs z TestGrouperGrouping.test_rollingcs’ˆj d¡}|jdd}xtdD]l‰t|ˆƒˆjƒ}‡‡fdd„}| |¡}t ||¡t|jˆƒdd}‡fd d„}| |¡}t ||¡qWdS) NrUr®)rL)rÁrõcst| d¡ˆƒˆjƒS)Nr®)r²rNrA)r†)r‹rBrrr´ sz7TestGrouperGrouping.test_rolling_corr_cov..funcT)rÆcst|j d¡ˆƒddS)Nr®T)rÆ)r²rVrN)r†)r‹rrr´ s) rAr³rNr²rˆrrrVr)rBr^rRrnr´ror)r‹rBrÚtest_rolling_corr_cov s z)TestGrouperGrouping.test_rolling_corr_covcsL|j d¡}|jdd}|jdd„ˆd}| ‡fdd„¡}t ||¡dS)NrUr®)rLcSs| ¡S)N)rb)r†rrrra sz8TestGrouperGrouping.test_rolling_apply..)r"cs| d¡jdd„ˆdS)Nr®cSs| ¡S)N)rb)rËrrrra szJTestGrouperGrouping.test_rolling_apply....)r")rNrˆ)r†)r"rrra s)rAr³rNrˆrr)rBr"r^rRrnror)r"rry sz&TestGrouperGrouping.test_rolling_applycCs´t dgddgddgddœ¡}| d¡}tj dd d ddd g¡}ddg|_tjtjddgddg|d}|jdd ¡}t ||¡| ¡|jdd ¡}t ||¡dS)NrerMr+r3rf)rUrVrU)r+rM)r+r®)r+rK)rer)rer3)rerTg@rTrV)r6r5)rL)r|rr³r}rr¯r8rlrNrbrr)rBr]r^ZmirornrrrÚtest_rolling_apply_mutability s& z1TestGrouperGrouping.test_rolling_apply_mutabilitycs°|j d¡}| ¡}x6dD].‰t|ˆƒƒ}| ‡fdd„¡}t ||¡qWx:dD]2‰t|ˆƒdd}| ‡fdd„¡}t ||¡qRW| d ¡}| d d„¡}t ||¡dS)NrU)rbrtr±r°r›r‰r„cst| ¡ˆƒƒS)N)r²rª)r†)r‹rrra; sz4TestGrouperGrouping.test_expanding..)rurÕr)r…cst| ¡ˆƒddS)Nr)r…)r²rª)r†)r‹rrra@ sgà?cSs| ¡ d¡S)Ngà?)rªrh)r†rrrraD s)rAr³rªr²rˆrrrh)rBr^rRrnror)r‹rÚtest_expanding4 s z"TestGrouperGrouping.test_expandingcsŽˆj d¡}| ¡}xtdD]l‰t|ˆƒˆjƒ}‡‡fdd„}| |¡}t ||¡t|jˆƒdd}‡fdd„}| |¡}t ||¡qWdS)NrU)rÁrõcst| ¡ˆƒˆjƒS)N)r²rªrA)r†)r‹rBrrr´N sz9TestGrouperGrouping.test_expanding_corr_cov..funcT)rÆcst|j ¡ˆƒddS)NT)rÆ)r²rVrª)r†)r‹rrr´U s) rAr³rªr²rˆrrrVr)rBr^rRrnr´ror)r‹rBrÚtest_expanding_corr_covG s z+TestGrouperGrouping.test_expanding_corr_covcsH|j d¡}| ¡}|jdd„ˆd}| ‡fdd„¡}t ||¡dS)NrUcSs| ¡S)N)rb)r†rrrra_ sz:TestGrouperGrouping.test_expanding_apply..)r"cs| ¡jdd„ˆdS)NcSs| ¡S)N)rb)rËrrrraa szLTestGrouperGrouping.test_expanding_apply....)r")rªrˆ)r†)r"rrraa s)rAr³rªrˆrr)rBr"r^rRrnror)r"rrCZ sz(TestGrouperGrouping.test_expanding_applyN)rDrErFrJrTrSrVrWrXryrYrZr[rCrrrrrR¼s rRc@süeZdZdd„Zdd„Zdd„Zdd„Zd d „Zdd„Zd d„Z dd„Z dd„Zdd„Zdd„Z dd„Zdd„Zdd„Zdd„Zdd „Zd!d"„Zd#d$„Zd%d&„Zd'd(„Zd)d*„Zd+d,„Zd-d.„Zd/d0„Zd1d2„Zd3d4„Zd5d6„Zd7d8„Zd9d:„Zd;d<„Z d=S)>Ú TestRollingTScCsbttjddddtdƒdœƒ d¡|_tdtdƒiƒ|_tdƒtd ƒtd ƒtdƒtdƒg|j_dS) Nr~rKr˜)r4r™)rUrVrUrVz20130101 09:00:00z20130101 09:00:02z20130101 09:00:03z20130101 09:00:05z20130101 09:00:06) rr|ršrÚ set_indexÚregularÚraggedrr5)rBrIrrrrJj szTestRollingTS.setup_methodcCsPtddddtjdgitdƒtdƒtdƒtd ƒtd ƒgd}|| d¡ ¡dS) NrVrr3rTr®z20130101 09:00:00z20130101 09:00:02z20130101 09:00:03z20130101 09:00:05z20130101 09:00:06)r5r|)rr8rlrrNrb)rBr]rrrrÌx szTestRollingTS.test_doc_stringc Cs|j}t t¡|jddWdQRXt t¡| ¡jddWdQRXx:dtj d¡gD]&}t t¡|j|dWdQRXq`Wx&dtj d¡dgD]}|j|dqžWxBddt d dd g¡gD](}t t¡|jd|dWdQRXqÊWt t¡|jddd WdQRXdS)Nr¿)rLZ2MSrTÚ1DZ2msgð?rer3rM)rLrŸT)rLr¾) r^rYrZrÀrNr”r|r’Z MonthBeginZDayr8rGrø)rBr]r™rœrrrÚ test_validƒ szTestRollingTS.test_validc Cs–|j}t t¡|jdddWdQRX| ¡}tjdt|ƒd|d<|jddd ¡t t¡|jdddWdQRX|jdddj ¡dS) Nr|r¿)rLÚonr~)r4rWrîrV)r^rYrZrÀrNr=r|ršrOrbrV)rBr]rrrÚtest_on szTestRollingTS.test_onc CsÔttjddddtdƒdœƒ}|jjs*t‚|jddd ¡| d¡}|j jsRt‚| d¡ ¡t|j ¡ƒ|_ |j jr|t‚t t¡| d¡ ¡WdQRX| ¡}t t¡|jddd ¡WdQRXdS) Nr~rKr˜)r4r™)rUrVr|rU)rb)rr|ršrrUZis_monotonicrQrNrbr]r5ÚreversedrÞrYrZrÀr”)rBr]rrrÚtest_monotonic_on´ s zTestRollingTS.test_monotonic_oncCsÐttdƒtjdddddœƒ}tdƒtdƒtdƒtdƒtd ƒg|d <| d ¡ d¡j ¡j dd }|jdd dj ¡}t ||¡| d ¡ d¡dg ¡ ¡dd g}|jdd ddg ¡}t ||¡dS)NrKz20130101 09:00:00Ú3s)r4r™)rVrWz20130101 09:00:02z20130101 09:00:03z20130101 09:00:05z20130101 09:00:06rUr|T)rå)rbrV) rrr|ršrr]rNrVrbr”rrr)rBr]rornrrrÚ test_frame_onÎ s& zTestRollingTS.test_frame_onc Cstdddddgdddtjdgttdƒtdƒtdƒtd ƒtd ƒgƒdœdd dgd}tdddddgdddtjdg|d dœdd dgd}|jdd d ¡}|}t ||¡t dddtjdgdd}|jdd dj ¡}t ||¡|ddd g}|jdd dddd g ¡}t ||¡dS)Nrr3rTrMr®z20130101 09:00:00z20130101 09:00:02z20130101 09:00:03z20130101 09:00:05z20130101 09:00:06)rUrVrWrUrWrV)r6grár|)rb)rP)rr8rlrrrNrbrrr rVr)rBr]Z expected1rnrorrrÚtest_frame_on2ú s* zTestRollingTS.test_frame_on2cCsì|j ¡}tjdddd|_|jddd ¡}|jdd ¡}t ||¡tjddd d|_|jddd ¡}|jd dd ¡}t ||¡|jddd ¡}|jd dd ¡}t ||¡|jdd ¡}|jd d ¡}t ||¡dS) Nr~rKrÏ)r4r™r3)rLrŸr`)rLZ2D) r^r=r|ršr5rNrbrr)rBr]rornrrrÚtest_basic_regulars z TestRollingTS.test_basic_regularcCsf|j}|jddd ¡}| d¡ ¡}t ||¡|jddd ¡}|jddd ¡}t ||¡dS)NrTr3)rŸr|)r^rNrbrr)rBr]rornrrrÚtest_min_periods-szTestRollingTS.test_min_periodsc Cs^tddgditdƒtdƒtdƒtdƒtdƒgd }t t¡|jjd ddWdQRX| ¡}d ddddg|d<|jd dd ¡}t ||¡| d ¡ ¡}t ||¡| ¡}d ddddg|d<|jd dd ¡}t ||¡| ¡}tjd dddg|d<|jd dd ¡}t ||¡| ¡}tjd ddtjg|d<|jd dd ¡}t ||¡dS)NrUr3rKz20130101 09:00:01z20130101 09:00:02z20130101 09:00:03z20130101 09:00:04z20130101 09:00:06)r5r|Zblabla)rLr×gð?rTr)r×rMrârrÖ) rrrYrZrÀr^rNr=rbrrr8rl)rBr]rornrrrrØ;s4zTestRollingTS.test_closedcCsÒ|j}|jddd ¡}| ¡}dddddg|d<t ||¡|jd dd ¡}| ¡}ddddd g|d<t ||¡|jd dd ¡}| ¡}tjtjdtjd g|d<t ||¡|jddd ¡}| ¡}ddddd g|d<t ||¡|jdd ¡}| ¡}ddddd g|d<t ||¡|jddd ¡}| ¡}dddddg|d<t ||¡|jddd ¡}| ¡}tjtjdddg|d<t ||¡|jddd ¡}| ¡}dddddg|d<t ||¡dS)Nrêr3)rLrŸgrTrMr®rVr|rárfrK)rLÚ4srfrgÚ5sr)r_rNrbr=rrr8rl)rBr]rnrorrrÚtest_ragged_sumbsBzTestRollingTS.test_ragged_sumcCsz|j}|jddd ¡}| ¡}dddddg|d<t ||¡|jd dd ¡}| ¡}ddd ddg|d<t ||¡dS) Nrêr3)rLrŸgrTrMr®rVr|gø?g@g@)r_rNrtr=rr)rBr]rnrorrrÚtest_ragged_meanszTestRollingTS.test_ragged_meancCsz|j}|jddd ¡}| ¡}dddddg|d<t ||¡|jd dd ¡}| ¡}ddd ddg|d<t ||¡dS) Nrêr3)rLrŸgrTrMr®rVr|gø?g@g@)r_rNrr=rr)rBr]rnrorrrÚtest_ragged_medianšsz TestRollingTS.test_ragged_mediancCs~|j}|jddd d¡}| ¡}dddddg|d <t ||¡|jd dd d¡}| ¡}ddddd g|d <t ||¡dS)Nrêr3)rLrŸgà?grTrMr®rVr|gø?g@g@)r_rNrhr=rr)rBr]rnrorrrÚtest_ragged_quantile§sz"TestRollingTS.test_ragged_quantilecCsø|j}|jdddjdd}| ¡}dgd|d<t ||¡|jdddjdd}| ¡}tjgd|d<t ||¡|jd ddjdd}| ¡}dgd gd|d<t ||¡|jdddjdd}| ¡}tjd dddg|d<t ||¡dS)Nrêr3)rLrŸr)r…grKrVrfgà?r®rlg»¹øÛž æ?gð?gÊPSé§ô?)r_rNrur=rrr8rl)rBr]rnrorrrÚtest_ragged_std´s"zTestRollingTS.test_ragged_stdcCsø|j}|jdddjdd}| ¡}dgd|d<t ||¡|jdddjdd}| ¡}tjgd|d<t ||¡|jd ddjdd}| ¡}dgd gd|d<t ||¡|jdddjdd}| ¡}tjd dddg|d<t ||¡dS)Nrêr3)rLrŸr)r…grKrVrfgÐ?r®rlgà?gð?gªªªªªªú?)r_rNrÕr=rrr8rl)rBr]rnrorrrÚtest_ragged_varËs"zTestRollingTS.test_ragged_varcCs€|j}|jddd ¡}| ¡}tjgd|d<t ||¡|jddd ¡}| ¡}tjgddddg|d<t ||¡dS) Nrfr3)rLrŸrKrVrlrTg)r_rNr„r=r8rlrr)rBr]rnrorrrÚtest_ragged_skewâszTestRollingTS.test_ragged_skewcCs||j}|jddd ¡}| ¡}tjgd|d<t ||¡|jddd ¡}| ¡}tjgddg|d<t ||¡dS) Nrfr3)rLrŸrKrVrlr®g333333ó¿)r_rNr‰r=r8rlrr)rBr]rnrorrrÚtest_ragged_kurtïszTestRollingTS.test_ragged_kurtcCsÚ|j}|jddd ¡}| ¡}dddddg|d<t ||¡|j}|jdd ¡}t ||¡|jddd ¡}| ¡}dddddg|d<t ||¡|jddd ¡}| ¡}tjtjdtjdg|d<t ||¡dS) Nrêr3)rLrŸgð?rV)rLr|rT)r_rNr›r=rrr8rl)rBr]rnrorrrÚtest_ragged_countüs zTestRollingTS.test_ragged_countcCsttjddddddddd gd œƒ d¡}| d¡ ¡}| ¡}ddddd g|d <t ||¡ttjdddddd dd dgd œƒ d¡}t ||¡| d¡ ¡}| ¡}dd ddd g|d <t ||¡| d¡ ¡}| ¡}dd dddg|d <t ||¡dS)Nr~rKr˜)r4r™gr3rTrMr®)rUrVrUrêrVr|g@rl) rr|ršr]rNr±r=rr)rBr]rnrorrrÚtest_regular_mins*zTestRollingTS.test_regular_mincCs²|j}|jddd ¡}| ¡}dddddg|d<t ||¡|jd dd ¡}| ¡}dddddg|d<t ||¡|jd dd ¡}| ¡}dddddg|d<t ||¡dS)Nrêr3)rLrŸgrTrMr®rVr|rlr)r_rNr±r=rr)rBr]rnrorrrÚtest_ragged_min-szTestRollingTS.test_ragged_mincCsœd}tdtj |¡itjd|ddd}|jddd ¡}| d ¡ ¡}||dk ¡ ¡s`t ‚|jddd ¡}| d ¡ ¡}||dk ¡ ¡s˜t ‚dS)Ni'rVr~r˜)r4r™)r5rTr3)rŸr|g{®Gáz„?éÈZ200s)rr8r£rr|ršrNr±rNÚboolrQ)rBr7ZdfprornrrrÚ test_perf_min@szTestRollingTS.test_perf_mincCs²|j}|jddd ¡}| ¡}dddddg|d<t ||¡|jd dd ¡}| ¡}dddddg|d<t ||¡|jd dd ¡}| ¡}dddddg|d<t ||¡dS)Nrêr3)rLrŸgrTrMr®rVr|rl)r_rNr°r=rr)rBr]rnrorrrÚtest_ragged_maxPszTestRollingTS.test_ragged_maxcCs®|j}dd„}|jdddj||d}| ¡}d|d<t ||¡|jd ddj||d}| ¡}d|d<t ||¡|jd ddj||d}| ¡}d|d<t ||¡dS)NcSsdS)Nr3r)r†rrrragsz1TestRollingTS.test_ragged_apply..rêr3)rLrŸ)r"gð?rVr|rl)r_rNrˆr=rr)rBr"r]r‹rnrorrrÚtest_ragged_applycszTestRollingTS.test_ragged_applycCsx|jd}|jdd}|jdd}x0dD](}t||ƒƒ}t||ƒƒ}t ||¡q(W| d¡}| d¡}t ||¡dS)NrTr3)rLrê) rbrtr›rrurÕr‰r„r±r°gà?)r^rNr²rrrh)rBr]ÚerrRr‹rnrorrrÚtest_allws zTestRollingTS.test_allcCsV|jd}|jdd}|jdd}|jdd„|d}|jdd„|d}t ||¡dS) NrTr3)rLrêcSsdS)Nr3r)r†rrrrasz.TestRollingTS.test_all_apply..)r"cSsdS)Nr3r)r†rrrras)r^rNrˆrr)rBr"r]r}rRrnrorrrÚtest_all_apply‰s zTestRollingTS.test_all_applycsŽtdt d¡itjddddd}| dd¡}|jd d }xNdD]F‰t|ˆƒƒ}‡fdd „}| |j j ¡ |¡jddd}t ||¡q@WdS)NrVrkr~ÚH)r4r™)r5z09:00z16:00Z5H)rL) rbrtr›rrurÕr‰r„r±r°cs"| dd¡}t|jdddˆƒƒS)Nz09:00z16:00rKr3)rŸ)Úbetween_timer²rN)r†)r‹rrÚ agg_by_day©sz+TestRollingTS.test_all2..agg_by_dayrT)Úlevelrå)rr8r@r|ršrrNr²r³r5Zdayrˆr”rr)rBr]ZdftrRrnr‚ror)r‹rÚ test_all2“s zTestRollingTS.test_all2cCs¾dddgdddgdddgdddgd d dgd ddgd ddgd ddgd ddgd ddgg }t|dddgd}t |d¡|d<| d¡ d¡ dd„¡}| d¡jdddd ¡}t ||¡dS)NZDavidz1/1/2015rz1/5/2015iôz 5/30/2015rkz 7/25/2015ZRyanz1/4/2014z 1/19/2015z 3/31/2016ZJoez7/1/2015z9/9/2015z 10/15/2015rPZdateÚamount)rÚr6cSs| d¡d ¡S)NÚ180Dr…)rNrb)r†rrrraÁsz6TestRollingTS.test_groupby_monotonic..r†)rb) rr|Zto_datetimer]r³rˆrNrbrr)rBrÚr]rornrrrÚtest_groupby_monotonic±s z$TestRollingTS.test_groupby_monotoniccCs¾tjdddd}tdgddgddgd t ||f¡t d ¡dœƒ}| d¡jd ddj ¡}| d¡ d¡ dd„¡}t ||¡| d¡}| d¡jd ddj ¡}t ||¡dS)Nz2016-01-01 09:30:00r1r˜)Ústartr4r™r3rTrrMrãr2)rUrVrWrUrkrV)rbcSs| d¡d ¡S)NrkrW)rNrt)r†rrrraÐsz2TestRollingTS.test_non_monotonic..)r|ršrr8Zconcatenater@r³rNrWrtr]rˆrrZsort_values)rBZdatesr]rnrorÌrrrÚtest_non_monotonicÅs z TestRollingTS.test_non_monotoniccCsàtjdddd}tt t|ƒ¡|d}| d¡ ¡}ttjgdgt|ƒd|d}t ||¡|jd dd ¡}t ||¡| d¡ ¡}ttjdgdgt|ƒd |d}t ||¡|jd dd ¡}t ||¡dS)Nz 2017-01-01rjZ1h)r4r™)r5Z2hgà?r3rT)rŸZ3hgð?rM)r|ršr r8r@rOrNrõrlrr)rBreZssrnroZ expected2rrrÚtest_rolling_cov_offset×s"$z%TestRollingTS.test_rolling_cov_offsetN)!rDrErFrJrÌrarcrergrhrirjrØrmrnrorprqrrrsrtrurvrwrzr{r|r~rr„r‡r‰rŠrrrrr\e s<,'+ r\)[ÚcollectionsrrrÚ itertoolsrr€rZnumpyr8Znumpy.randomrrYZ pandas.compatrr Z pandas.errorsr Zpandas.util._test_decoratorsÚutilZ_test_decoratorsr¶Zpandasr|rrr rrrrrZpandas.core.baserZpandas.core.sortingrZpandas.core.windowZcorerLrÇZpandas.util.testingZtestingrZpandas.tseries.offsetsZtseriesr’r7r?rZfixturer"r+r/r)r0rHr¸r¹r»rËrrrr(r*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r:r;r<r=r?rºrØrÜrÙrÚrRr\rrrrÚs (.5|iIaI 4 *