o
    g	                     @   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
g d	dd
d Zej
g dddd Zej
g dddd Zej
g d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
ejd'ed'd(d)d*gdd+d, Zej
ejd-ed'd(d)d.d/gdd0d1 Zej
d2d3 Zej
d4edd5gdd6d7 Zej
d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFejdGejdHd)gddIdJ Z ej
edKdLgddKgdMdKgdNdggdd gdOedKdLgddKgdMdKgdNdggddgdOedKdLgddKgdMdKgdNdggdPdPgdOedKdLgddKgdMdKgdNdggdQd gdOedKdLgddKgdMdKgdNdggdRdgdOedKdLgddKgdMdKgdNdggdPdgdOedSdTgdQdSgdUdSgdVdQggddRgdOedKdTgddSgdMdSgdNdQggd dQgdOedKdLgddKgdMdKgdNdQggdQdWgdOg	ddXdY Z!ej
dZd[ Z"ej
d\d] Z#dS )^    )	timedeltaN	DataFrameto_datetimeTF)paramsc                 C      | j S )z&raw keyword argument for rolling.applyparamrequest r   Z/home/ubuntu/cloudmapper/venv/lib/python3.10/site-packages/pandas/tests/window/conftest.pyraw      r   )triangblackmanhammingbartlettbohmanblackmanharrisnuttallbarthannc                 C   r   Nr   r
   r   r   r   	win_types   s   r   )kaisergaussiangeneral_gaussianexponentialc                 C   r   r   r   r
   r   r   r   win_types_special#      r   )summeanmedianmaxminvarstdkurtskewcountsemc                 C   r   r   r   r
   r   r   r   arithmetic_win_operators(   s   r+   )r    r!   r"   r#   r$   c                 C   r   r   r   r
   r   r   r   $arithmetic_numba_supported_operators;   s   
r,   )rightleftbothneitherc                 C   r   r   r   r
   r   r   r   closedH   r   r1   c                 C   r   r   r   r
   r   r   r   centerM   r   r2      c                 C   r   r   r   r
   r   r   r   min_periodsR   r   r4   c                 C   r   )z'parallel keyword argument for numba.jitr   r
   r   r   r   parallelW   r   r5   c                 C   r   )z$nogil keyword argument for numba.jitr   r
   r   r   r   nogila   r   r6   c                 C   r   )z'nopython keyword argument for numba.jitr   r
   r   r   r   nopythong   r   r7   c                 C   r   )zadjust keyword argument for ewmr   r
   r   r   r   adjustm   r   r8   c                 C   r   )z"ignore_na keyword argument for ewmr   r
   r   r   r   	ignore_nas   r   r9   numbaz0.46.0)markscythonc                 C   r   )z)engine keyword argument for rolling.applyr   r
   r   r   r   enginey   s   r=   )r:   T)r<   T)r<   Fc                 C   r   )z2engine and raw keyword arguments for rolling.applyr   r
   r   r   r   engine_and_raw   s   	r>   c                   C   s    t g dg dtg ddS )z0Frame for testing times argument in EWM groupby.)
abcr?   r@   rA   r?   r@   rA   r?   )
r   r   r   r3   r3   r3      rB   rB      )

2020-01-01rD   rD   z
2020-01-02z
2020-01-10z
2020-01-22z
2020-01-03
2020-01-23rE   z
2020-01-04)ABCr   r   r   r   r   times_frame   s   rI   z1 day)daysc                 C   r   )z2Halflife argument for EWM when times is specified.r   r
   r   r   r   halflife_with_times   r   rK   objectcategoryint8int16int32int64uint8uint16uint32uint64float16float32float64zm8[ns]zM8[ns]zdatetime64[ns, UTC]zKdirect creation of extension dtype datetime64[ns, UTC] is not supported ATMc                 C   r   )zDtypes for window testsr   r
   r   r   r   dtypes   s   rY   rB            columnsrH   g      ?g        g       @g      @g      @g       @Xc                 C   r   )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_pairwiserB   rZ   r3   r[   r\   r   r]   r   r   r   r   r   pairwise_target_frame   s   (rb   c                   C   s(   t g dg dg dg dgg ddS )z&Pairwise other frame for test_pairwise)Nr3   r3   )Nr3   rB   )NrC   rB   )Nr\   r3   )YZr_   r]   ra   r   r   r   r   pairwise_other_frame   s   re   )$datetimer   pytestpandas.util._test_decoratorsutil_test_decoratorstdpandasr   r   fixturer   r   r   r+   r,   r1   r2   r4   r5   r6   r7   r8   r9   r	   
skip_if_nor=   r>   rI   rK   markskiprY   r`   rb   re   r   r   r   r   <module>   s    




	



	








&&&&&&&&&

