o
    EhA                     @   s  d dl m Z mZ d dlZd dlZd dlm  mZ d dl	m
Z
mZmZ ejddgddd Zejg d	dd
d Zejddgddd Zejddgddd Zejddgddd Zejdgddd Zejdgddd Zejddgddd Zejddgddd Zejddgddd Zejejdedejjgddgdd d! Zejejd"edejjgdd#d$gdd%d& Zejd'edd(edd)gdd*d+ Zejd,d- Z ejd.d/ Z!ejg d0dd1d2 Z"dS )3    )datetime	timedeltaN)	DataFrameSeriesbdate_rangeTF)paramsc                 C      | j S )z&raw keyword argument for rolling.applyparamZrequest r   <lib/python3.10/site-packages/pandas/tests/window/conftest.pyraw      r   )sumZmeanZmedianmaxminvarZstdZkurtZskewcountZsemc                 C   r   Nr	   r   r   r   r   arithmetic_win_operators   s   r   c                 C   r   r   r	   r   r   r   r   center+      r      c                 C   r   r   r	   r   r   r   r   min_periods0   r   r   c                 C   r   )z'parallel keyword argument for numba.jitr	   r   r   r   r   parallel5   r   r   c                 C   r   )z$nogil keyword argument for numba.jitr	   r   r   r   r   nogil?   r   r   c                 C   r   )z'nopython keyword argument for numba.jitr	   r   r   r   r   nopythonE   r   r   c                 C   r   )zadjust keyword argument for ewmr	   r   r   r   r   adjustK   r   r   c                 C   r   )z"ignore_na keyword argument for ewmr	   r   r   r   r   	ignore_naQ   r   r   c                 C   r   )znumeric_only keyword argumentr	   r   r   r   r   numeric_onlyW   r   r    numba)Zmarkscythonc                 C   r   )z)engine keyword argument for rolling.applyr	   r   r   r   r   engine]   s   r#   )r!   T)r"   T)r"   Fc                 C   r   )z2engine and raw keyword arguments for rolling.applyr	   r   r   r   r   engine_and_rawh   s   r$   z1 day)ZdaysDc                 C   r   )z2Halflife argument for EWM when times is specified.r	   r   r   r   r   halflife_with_timesv   r   r&   c                  C   sH   t jdd} t dd}t j| |< t| ttdddddd}|S )	zMake mocked series as fixture.   d      (     r   Zperiodsindex)	nprandomdefault_rngstandard_normalZarangenanr   r   r   )ZarrZlocsseriesr   r   r   r4   |   s
   
r4   c                   C   s*   t tjddttddddddS )zMake mocked frame as fixture.r'   )r(   
   r+   r   r(   r,   r-   )r   r/   r0   r1   r2   r   r   r   r   r   r   frame   s   r6   )Nr   r'      r5   c                 C   r   )z4step keyword argument for rolling window operations.r	   r   r   r   r   step   r   r8   )#r   r   Znumpyr/   ZpytestZpandas.util._test_decoratorsutilZ_test_decoratorsZtdZpandasr   r   r   Zfixturer   r   r   r   r   r   r   r   r   r    r
   Z
skip_if_noZmarkZ
single_cpur#   r$   Ztimedelta64r&   r4   r6   r8   r   r   r   r   <module>   sd    




	






	

	
