o
    Û­j±  ã                   @   s   d dl Zd dlZd dlmZmZmZ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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 ¡ 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d-i fd.i fd,d/d0ifd-d/d0ifd.d/d0ifgg d1¢d2�d3d4„ ƒZdS )5é    N)Ú	DataFrameÚIndexÚSeriesÚ
date_range)Úreduction_kernelsÚtransformation_kernelsTF)Úparamsc                 C   ó   | j S ©N©Úparam©Úrequest© r   úZ/var/www/html/CropPilot/venv/lib/python3.10/site-packages/pandas/tests/groupby/conftest.pyÚsort   ó   r   c                 C   r	   r
   r   r   r   r   r   Úas_index   r   r   c                 C   r	   r
   r   r   r   r   r   Údropna   r   r   c                 C   r	   r
   r   r   r   r   r   Úobserved   r   r   c                   C   s6   t g d¢g d¢tj d¡ d¡tj d¡ d¡dœƒS )N)ÚfooÚbarr   r   r   r   r   r   )Úoner   ÚtwoÚthreer   r   r   r   é   é   )ÚAÚBÚCÚD©r   ÚnpÚrandomÚdefault_rngÚstandard_normalr   r   r   r   Údf$   s   üÿr&   c                   C   s$   t tj d¡ d¡tdddd�d�S )Nr   é   ú
2000-01-01r   ©ÚperiodsÚfreq)Úindex)r   r"   r#   r$   r%   r   r   r   r   r   Úts0   s   þr-   c                   C   s2   t tj d¡ d¡ttdƒtd�tdddd�d	�S )
Nr   )r'   é   ÚABCD)Údtyper(   r'   r   r)   )Úcolumnsr,   )	r   r"   r#   r$   r%   r   ÚlistÚobjectr   r   r   r   r   Útsframe8   s
   ýr4   c                	   C   sL   t g d¢g d¢g d¢tj d¡ d¡tj d¡ d¡tj d¡ 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   )Údullr5   Úshinyr5   r5   r6   r6   r5   r6   r6   r6   r   é   )r   r   r   r    ÚEÚFr!   r   r   r   r   Úthree_groupA   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   ÚaÚa0_at_0)é   ÚbÚb0_at_1)r   r;   Úa1_at_2)é   r>   Úb1_at_3)r.   ÚcÚc0_at_4)é   r;   Úa2_at_5)é   r;   Úa3_at_6)é   r;   Úa4_at_7)r   ÚGroupÚValue)r1   r   )r   Ú	set_index)Údatar&   r   r   r   Úslice_test_dfs   s   ø

rO   c                 C   s   | j ddd�S )NrK   F)r   )Úgroupby)rO   r   r   r   Úslice_test_groupedƒ   s   rQ   c                 C   r	   )zT
    yields the string names of all groupby reduction functions, one at a time.
    r   r   r   r   r   Úreduction_funcˆ   s   rR   c                 C   r	   )z@yields the string names of all groupby transformation functions.r   r   r   r   r   Útransformation_func�   ó   rS   c                 C   r	   )z5yields both aggregation and transformation functions.r   r   r   r   r   Úgroupby_func–   rT   rU   c                 C   r	   )z'parallel keyword argument for numba.jitr   r   r   r   r   Úparallelœ   rT   rV   c                 C   r	   )z$nogil keyword argument for numba.jitr   r   r   r   r   Únogil¦   rT   rW   c                 C   r	   )z'nopython keyword argument for numba.jitr   r   r   r   r   Únopython¬   rT   rX   ÚmeanÚvarÚddofr=   ÚstdÚsumÚminÚmaxÚ	min_countr   )rY   Úvar_1Úvar_0Ústd_1Ústd_0r]   r^   r_   zsum-min_countzmin-min_countzmax-min_count)r   Úidsc                 C   r	   )z(reductions supported with engine='numba'r   r   r   r   r   Únumba_supported_reductions²   s   rf   )Únumpyr"   ÚpytestÚpandasr   r   r   r   Úpandas.core.groupby.baser   r   Úfixturer   r   r   r   r&   r-   r4   r:   rO   rQ   ÚsortedrR   rS   rU   rV   rW   rX   rf   r   r   r   r   Ú<module>   sh    







1





	








õò