a
    é´«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   s   | j S ©N©Úparam©Úrequest© r   ú\/var/www/html/laksyashrimp/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   | j S r	   r
   r   r   r   r   Údropna   s    r   c                 C   s   | j S r	   r
   r   r   r   r   Úobserved   s    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   ©Z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)Zdtyper&   r%   r   r'   )Úcolumnsr)   )	r   r    r!   r"   r#   r   ÚlistÚobjectr   r   r   r   r   Útsframe8   s
    ýr0   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   )Údullr1   Úshinyr1   r1   r2   r2   r1   r2   r2   r2   r   é   )r   r   r   r   ÚEÚFr   r   r   r   r   Úthree_groupA   s    Öÿr6   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   r7   Za1_at_2)é   r9   Zb1_at_3)r+   ÚcZc0_at_4)é   r7   Za2_at_5)é   r7   Za3_at_6)é   r7   Za4_at_7)r   ÚGroupÚValue)r-   r   )r   Z	set_index)Údatar$   r   r   r   Úslice_test_dfs   s    ø
rB   c                 C   s   | j ddd�S )Nr?   F)r   )Úgroupby)rB   r   r   r   Úslice_test_groupedƒ   s    rD   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    rE   c                 C   s   | j S )z@yields the string names of all groupby transformation functions.r
   r   r   r   r   Útransformation_func�   s    rF   c                 C   s   | j S )z5yields both aggregation and transformation functions.r
   r   r   r   r   Úgroupby_func–   s    rG   c                 C   s   | j S )z'parallel keyword argument for numba.jitr
   r   r   r   r   Úparallelœ   s    rH   c                 C   s   | j S )z$nogil keyword argument for numba.jitr
   r   r   r   r   Únogil¦   s    rI   c                 C   s   | j S )z'nopython keyword argument for numba.jitr
   r   r   r   r   Únopython¬   s    rJ   ÚmeanÚvarZddofr8   ZstdÚsumÚminÚmaxZ	min_countr   )rK   Zvar_1Zvar_0Zstd_1Zstd_0rM   rN   rO   zsum-min_countzmin-min_countzmax-min_count)r   Úidsc                 C   s   | j S )z(reductions supported with engine='numba'r
   r   r   r   r   Únumba_supported_reductions²   s    rQ   )Únumpyr    ZpytestZpandasr   r   r   r   Zpandas.core.groupby.baser   r   Zfixturer   r   r   r   r$   r*   r0   r6   rB   rD   ÚsortedrE   rF   rG   rH   rI   rJ   rQ   r   r   r   r   Ú<module>   sf   







1





	








õò