o d @sddlmZmZddlZddlZddlmmZddl m Z m Z m Z ej ddgdddZej gd dd d Zej ddgdd d Zej ddgdddZej ddgdddZej dgdddZej dgdddZej ddgdddZej ddgdddZej ddgdddZej ejdedddgdd d!Zej ejd"eddd#d$gdd%d&Zej d'edd(edd)gdd*d+Zej d,d-Zej d.d/Zej gd0dd1d2Z dS)3)datetime timedeltaN) DataFrameSeries bdate_rangeTF)paramscC|jS)z&raw keyword argument for rolling.applyparamrequestr P/app/.heroku/python/lib/python3.10/site-packages/pandas/tests/window/conftest.pyrawr) summeanZmedianmaxminvarZstdZkurtZskewcountZsemcCrNr r r r rarithmetic_win_operatorssrcCrrr r r r rcenter+rcCrrr r r r r min_periods0rrcCr)z'parallel keyword argument for numba.jitr r r r rparallel5rrcCr)z$nogil keyword argument for numba.jitr r r r rnogil?rrcCr)z'nopython keyword argument for numba.jitr r r r rnopythonErrcCr)zadjust keyword argument for ewmr r r r radjustKrr cCr)z"ignore_na keyword argument for ewmr r r r r ignore_naQrr!cCr)znumeric_only keyword argumentr r r r r numeric_onlyWrr"numba)ZmarkscythoncCr)z)engine keyword argument for rolling.applyr r r r rengine]rr%)r#T)r$T)r$FcCr)z2engine and raw keyword arguments for rolling.applyr r r r rengine_and_rawcs r&z1 day)daysDcCr)z2Halflife argument for EWM when times is specified.r r r r rhalflife_with_timesorr)cCsBtjd}tdd}tj||<t|ttdddddd}|S)zMake mocked series as fixture.d(rZperiodsindex)nprandomrandnZarangeNaNrrr)ZarrZlocsseriesr r rr5us   r5cCs&ttjddttddddddS)zMake mocked frame as fixture.r* r-rr.r/)rr1r2r3rrr r r rframes r7)Nrr6cCr)z4step keyword argument for rolling window operations.r r r r rsteprr:)!rrnumpyr1ZpytestZpandas.util._test_decoratorsutilZ_test_decoratorstdZpandasrrrZfixturerrrrrrrr r!r"r Z skip_if_nor%r&Z timedelta64r)r5r7r:r r r rsX