a
    <b                     @   s  d dl Zd dlZd dl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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 dd Ze 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ejd)i fd*d+d,ifd*d+d ifd-d+d,ifd-d+d ifd.i fgdd/d0 ZdS )1    N	DataFrame)reduction_kernelstransformation_kernelsTF)paramsc                 C   s   | j S Nparamrequest r   m/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/pandas/tests/groupby/conftest.pysort   s    r   c                 C   s   | j S r   r   r
   r   r   r   as_index   s    r   c                 C   s   | S r   r   )Z multiindex_dataframe_random_datar   r   r   mframe   s    r   c                   C   s*   t g dg dtjdtjddS )Nfoobarr   r   r   r   r   r   oner   twothreer   r   r   r      ABCDr   nprandomrandnr   r   r   r   df   s    

r"   c                   C   s   t  S r   )tmZmakeTimeSeriesr   r   r   r   ts'   s    r$   c                   C   s   t  S r   )r#   ZgetTimeSeriesDatar   r   r   r   tsd,   s    r%   c                 C   s   t | S r   r   )r%   r   r   r   tsframe1   s    r&   c                   C   s4   t g dg dtjdtjtjddddS )Nr   r   r   Zfloat32)Zdtyper   )r   r   r    r!   arrayr   r   r   r   df_mixed_floats6   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_groupB   s    


r.   c               	   C   sN   g dg dg dg dg dg dg dg dg} t | g d	d
}|dS )N)r   aZa0_at_0)   bZb0_at_1)   r/   Za1_at_2)   r1   Zb1_at_3)   cZc0_at_4)   r/   Za2_at_5)   r/   Za3_at_6)   r/   Za4_at_7)IndexGroupValue)columnsr9   )r   Z	set_index)datar"   r   r   r   slice_test_dft   s    
r>   c                 C   s   | j dddS )Nr:   F)r   )groupby)r>   r   r   r   slice_test_grouped   s    r@   c                 C   s   | j S )zT
    yields the string names of all groupby reduction functions, one at a time.
    r   r
   r   r   r   reduction_func   s    rA   c                 C   s   | j S )z@yields the string names of all groupby transformation functions.r   r
   r   r   r   transformation_func   s    rB   c                 C   s   | j S )z5yields both aggregation and transformation functions.r   r
   r   r   r   groupby_func   s    rC   c                 C   s   | j S )z'parallel keyword argument for numba.jitr   r
   r   r   r   parallel   s    rD   c                 C   s   | j S )z$nogil keyword argument for numba.jitr   r
   r   r   r   nogil   s    rE   c                 C   s   | j S )z'nopython keyword argument for numba.jitr   r
   r   r   r   nopython   s    rF   meanvarZddofr0   Zstdsumc                 C   s   | j S )z(reductions supported with engine='numba'r   r
   r   r   r   numba_supported_reductions   s    rJ   )Znumpyr   ZpytestZpandasr   Zpandas._testingZ_testingr#   Zpandas.core.groupby.baser   r   Zfixturer   r   r   r"   r$   r%   r&   r(   r.   r>   r@   sortedrA   rB   rC   rD   rE   rF   rJ   r   r   r   r   <module>   s`   








1





	






