U
    f/e	                     @   s  d dl mZ d dlZd dlm  mZ d dlmZm	Z	 ej
ddgddd Zej
d	d
ddddddgddd Zej
ddddgddd Zej
dddddddd d!d"d#gdd$d% Zej
dddddgdd&d' Zej
d(d)d*d+gdd,d- Zej
ddgdd.d/ Zej
dd0gdd1d2 Zej
ddgdd3d4 Zej
dgdd5d6 Zej
dgdd7d8 Zej
ddgdd9d: Zej
ddgdd;d< Zej
ejd=ed=d>d?d@gddAdB Zej
ejdCed=d>d?dDdEgddFdG Zej
dHdI Zej
dJed0dKgddLdM Zej
dNdOdPdQdRdSdTdUdVdWdXdYdZd[d\ejd]ejd^d?gdd_d` Z ej
edadbgd0dagdcdagddd0ggd0d gdeedadbgd0dagdcdagddd0ggd0d0gdeedadbgd0dagdcdagddd0ggdfdfgdeedadbgd0dagdcdagddd0ggdgd gdeedadbgd0dagdcdagddd0ggdhd0gdeedadbgd0dagdcdagddd0ggdfd0gdeedidjgdgdigdkdigdldgggd0dhgdeedadjgd0digdcdigdddgggd dggdeedadbgd0dagdcdagdddgggdgdmgdeg	ddndo Z!ej
dpdq Z"ej
drds Z#dS )t    )	timedeltaN	DataFrameto_datetimeTF)paramsc                 C   s   | j S )z&raw keyword argument for rolling.applyparamrequest r   @/tmp/pip-unpacked-wheel-tiezk1ph/pandas/tests/window/conftest.pyraw   s    r   ZtriangZblackmanZhammingZbartlettZbohmanZblackmanharrisZnuttallZbarthannc                 C   s   | j S Nr   r	   r   r   r   	win_types   s    r   ZkaiserZgaussianZgeneral_gaussianZexponentialc                 C   s   | j S r   r   r	   r   r   r   win_types_special#   s    r   sumZmeanZmedianmaxminvarZstdZkurtZskewcountZsemc                 C   s   | j S r   r   r	   r   r   r   arithmetic_win_operators(   s    r   c                 C   s   | j S r   r   r	   r   r   r   $arithmetic_numba_supported_operators;   s    
r   rightleftZbothZneitherc                 C   s   | j S r   r   r	   r   r   r   closedH   s    r   c                 C   s   | j S r   r   r	   r   r   r   centerM   s    r      c                 C   s   | j S r   r   r	   r   r   r   min_periodsR   s    r   c                 C   s   | j S )z'parallel keyword argument for numba.jitr   r	   r   r   r   parallelW   s    r   c                 C   s   | j S )z$nogil keyword argument for numba.jitr   r	   r   r   r   nogila   s    r   c                 C   s   | j S )z'nopython keyword argument for numba.jitr   r	   r   r   r   nopythong   s    r    c                 C   s   | j S )zadjust keyword argument for ewmr   r	   r   r   r   adjustm   s    r!   c                 C   s   | j S )z"ignore_na keyword argument for ewmr   r	   r   r   r   	ignore_nas   s    r"   numbaz0.46.0)Zmarkscythonc                 C   s   | j S )z)engine keyword argument for rolling.applyr   r	   r   r   r   enginey   s    r%   )r#   T)r$   T)r$   Fc                 C   s   | j S )z2engine and raw keyword arguments for rolling.applyr   r	   r   r   r   engine_and_raw   s    	r&   c                   C   sP   t ddddddddddg
ddddddddddg
tdddd	d
dddddg
dS )z0Frame for testing times argument in EWM groupby.abcr   r         z
2020-01-01z
2020-01-02z
2020-01-10z
2020-01-22z
2020-01-03z
2020-01-23z
2020-01-04)ABCr   r   r   r   r   times_frame   s$    r/   z1 day)daysc                 C   s   | j S )z2Halflife argument for EWM when times is specified.r   r	   r   r   r   halflife_with_times   s    r1   objectcategoryZint8Zint16Zint32Zint64Zuint8Zuint16Zuint32Zuint64Zfloat16Zfloat32Zfloat64zm8[ns]zM8[ns]zdatetime64[ns, UTC]zKdirect creation of extension dtype datetime64[ns, UTC] is not supported ATMc                 C   s   | j S )zDtypes for window testsr   r	   r   r   r   dtypes   s    r4   r*            columnsr.   g      ?g        g       @g      @g      @g       @Xc                 C   s   | j S )zPairwise frames test_pairwiser   r	   r   r   r   pairwise_frames   s    r;   c                   C   s(   t ddgddgddgddggddgdS )z'Pairwise target frame for test_pairwiser*   r5   r   r6   r7   r   r8   r   r   r   r   r   pairwise_target_frame   s    r=   c                   C   s2   t dddgdddgdddgdddggdddgd	S )
z&Pairwise other frame for test_pairwiseNr   r*   r+   r7   YZr:   r8   r<   r   r   r   r   pairwise_other_frame   s    "r@   )$datetimer   ZpytestZpandas.util._test_decoratorsutilZ_test_decoratorstdZpandasr   r   Zfixturer   r   r   r   r   r   r   r   r   r   r    r!   r"   r   Z
skip_if_nor%   r&   r/   r1   markskipr4   r;   r=   r@   r   r   r   r   <module>   s   



	




	








&&&&&&&&&

