U sVc @sddlmZmZddlZddlZddlmmZddl m Z m Z m Z ej ddgdddZej d d d d d ddddddg dddZej ddgdddZej ddgdddZej ddgdddZej dgdddZej dgddd Zej ddgdd!d"Zej ddgdd#d$Zej ddgdd%d&Zej ejd'ed'd(d)gdd*d+Zej ejd,ed'd(d-d.gdd/d0Zej d1edd2edd3gdd4d5Zej d6d7Zej d8d9Zej ddd:d;dZ dS)?)datetime timedeltaN) DataFrameSeries bdate_rangeTF)paramscCs|jS)z&raw keyword argument for rolling.applyparamrequestr @/tmp/pip-unpacked-wheel-xj8nt62q/pandas/tests/window/conftest.pyrawsrsumZmeanZmedianmaxminvarZstdZkurtZskewcountZsemcCs|jSNrr r r r arithmetic_win_operatorssrcCs|jSrrr r r r center+srcCs|jSrrr r r r min_periods0srcCs|jS)z'parallel keyword argument for numba.jitrr r r r parallel5srcCs|jS)z$nogil keyword argument for numba.jitrr r r r nogil?srcCs|jS)z'nopython keyword argument for numba.jitrr r r r nopythonEsrcCs|jS)zadjust keyword argument for ewmrr r r r adjustKsrcCs|jS)z"ignore_na keyword argument for ewmrr r r r ignore_naQsrcCs|jS)znumeric_only keyword argumentrr r r r numeric_onlyWsrnumba)ZmarkscythoncCs|jS)z)engine keyword argument for rolling.applyrr r r r engine]sr!)rT)r T)r FcCs|jS)z2engine and raw keyword arguments for rolling.applyrr r r r engine_and_rawcs r"z1 day)daysDcCs|jS)z2Halflife argument for EWM when times is specified.rr r r r halflife_with_timesosr%cCsBtjd}tdd}tj||<t|ttdddddd}|S)zMake mocked series as fixture.d(rZperiods)index)nprandomrandnarangeNaNrrr)ZarrZlocsseriesr r r r1us    r1cCs.ttjddttdddddtddS)zMake mocked frame as fixture.r& r)rr*)r+columns)rr,r-r.rrr/r r r r frames  r4r2cCs|jS)z4step keyword argument for rolling window operations.rr r r r stepsr7)!rrZnumpyr,ZpytestZpandas.util._test_decoratorsutilZ_test_decoratorstdZpandasrrrZfixturerrrrrrrrrrr Z skip_if_nor!r"Z timedelta64r%r1r4r7r r r r sl