o
    gY                     @   sN  d dl Zd dlZd dlmZmZ d dlmZ d dl	m
Z
mZ ejddgddd Zejd	d
 Zejdd Zejdd Zejdd Zejdd Zejdd Zejdd Zejee
ddd Zejeeddd Zejee
ee 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dS )#    N)	DataFrame
MultiIndex)reduction_kernelstransformation_kernelsTF)paramsc                 C      | j S Nparamrequest r   [/home/ubuntu/cloudmapper/venv/lib/python3.10/site-packages/pandas/tests/groupby/conftest.pyas_index   s   r   c                  C   sF   t g dg dgg dg dgddgd} ttjdd	| g d
dS )N)foobarbazqux)onetwothree)
r   r   r      r      r      r   r   )
r   r   r   r   r   r   r   r   r   r   firstsecond)levelscodesnames
   r   )ABC)indexcolumns)r   r   nprandomrandn)r#   r   r   r   mframe   s   r(   c                   C   s*   t g dg dtjdtjddS )Nr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r      r    r!   r"   Dr   r%   r&   r'   r   r   r   r   df   s   

r/   c                   C      t  S r   )tmmakeTimeSeriesr   r   r   r   ts*      r3   c                   C   r0   r   )r1   getTimeSeriesDatar   r   r   r   tsd/   r4   r6   c                 C   s   t | S r   )r   )r6   r   r   r   tsframe4   r4   r7   c                   C   s4   t g dg dtjdtjtjddddS )Nr)   r*   r+   float32)dtyper,   )r   r%   r&   r'   arrayr   r   r   r   df_mixed_floats9   s   
r;   c                	   C   s:   t g dg dg dtjdtjdtjddS )N)r   r   r   r   r   r   r   r   r   r   r   )r   r   r   r   r   r   r   r   r   r   r   )dullr<   shinyr<   r<   r=   r=   r<   r=   r=   r=      )r    r!   r"   r-   EFr.   r   r   r   r   three_groupE   s   


rA   c                 C   r   )zT
    yields the string names of all groupby reduction functions, one at a time.
    r	   r   r   r   r   reduction_funcw   s   rB   c                 C   r   )z@yields the string names of all groupby transformation functions.r	   r   r   r   r   transformation_func      rC   c                 C   r   )z5yields both aggregation and transformation functions.r	   r   r   r   r   groupby_func   rD   rE   c                 C   r   )z'parallel keyword argument for numba.jitr	   r   r   r   r   parallel   rD   rF   c                 C   r   )z$nogil keyword argument for numba.jitr	   r   r   r   r   nogil   rD   rG   c                 C   r   )z'nopython keyword argument for numba.jitr	   r   r   r   r   nopython   rD   rH   )numpyr%   pytestpandasr   r   pandas._testing_testingr1   pandas.core.groupby.baser   r   fixturer   r(   r/   r3   r6   r7   r;   rA   sortedrB   rC   rE   rF   rG   rH   r   r   r   r   <module>   sB    

	





1



	
