a j=ic @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)paramscCs|jS)z&raw keyword argument for rolling.applyparamrequestr m/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/pandas/tests/window/conftest.pyrawsr) summeanZmedianmaxminvarZstdZkurtZskewcountZsemcCs|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")r T)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)Zarrlocsseriesr r r r3us    r3cCs.ttjddttdddddtddS)zMake mocked frame as fixture.r' r*rr+)r,columns)rr-r.r/rrr0r r r r frames  r6)Nrr4cCs|jS)z4step keyword argument for rolling window operations.rr r r r stepsr9)!rrnumpyr-ZpytestZpandas.util._test_decoratorsutilZ_test_decoratorstdZpandasrrrZfixturerrrrrrrrrrr Z skip_if_nor"r#Z timedelta64r&r3r6r9r r r r sV