o
    Û­j"  ã                   @   s@   d dl Zd dlZd dlmZmZ d dlmZ dd„ Z	dd„ Z
dS )é    N)Ú	DataFrameÚIndexc                  C   s†   t j d¡} tg d¢|  d¡|  d¡dœƒ}dd„ }dd„ }| d	¡ |¡ |¡}td
dgd	d�}tj	ddgd|d�}t
 ||¡ d S )Né   )ÚfooÚbarr   r   r   r   r   r   é   )ÚAÚBÚCc                 S   s   | j  ¡ | j ¡  ¡  S ©N)r	   Úmaxr
   Úmin)Údfgb© r   ú[/var/www/html/CropPilot/venv/lib/python3.10/site-packages/pandas/tests/groupby/test_pipe.pyÚf   ó   ztest_pipe.<locals>.fc                 S   s   | d S )Nr   r   )Úsrsr   r   r   Úsquare   s   ztest_pipe.<locals>.squarer   r   r   ©Únameg¦8t”þ@gtÝÐîÞ@r	   )r   Úindex)ÚnpÚrandomÚdefault_rngr   Ústandard_normalÚgroupbyÚpiper   ÚpdÚSeriesÚtmÚassert_series_equal)Úrandom_stateÚdfr   r   Úresultr   Úexpectedr   r   r   Ú	test_pipe   s   ýÿr&   c                  C   sÖ   t g d¢g d¢g d¢dœƒ} dd„ }dd„ }d	d
„ }|  d¡ |d¡ |d¡ |d¡}tddgdd�}tjddg|d�}t ||¡ t g d¢¡}| |¡ dd„ ¡}tjg d¢tg d¢tj	d�d�}t ||¡ d S )N)r   r   r	   r	   r
   )g      ð?ç       @g      @r'   g      @)g      $@g      Y@g     @�@g      YÀg     @�À)ÚgroupÚxÚyc                    s    | j ‡ fdd„dd�}| d¡S )Nc                    s   | j  ¡ ˆ kS r   )r*   Úmean©Úgrp©Úarg1r   r   Ú<lambda>9   s    z+test_pipe_args.<locals>.f.<locals>.<lambda>F)Údropnar(   )Úfilterr   )r   r/   Úfilteredr   r.   r   r   8   s   
ztest_pipe_args.<locals>.fc                 S   s   |   ¡ |   ¡   ¡  | S r   )Úsum)r   Úarg2r   r   r   Úg<   r   ztest_pipe_args.<locals>.gc                 S   s   | j | j | S r   )r)   r*   )r#   Úarg3r   r   r   Úh?   s   ztest_pipe_args.<locals>.hr(   r   é
   éd   r   r	   r   g#�šáSÀgùÜbeøžSÀ)r   )é   r;   r   r   é   r<   c                 S   s   |   ¡ |  ¡  S r   )r4   Úcountr,   r   r   r   r0   L   s    z test_pipe_args.<locals>.<lambda>)é   r   é   )r;   r   r<   )Údtype)
r   r   r   r   r   r   r    r!   r   Úint64)r#   r   r6   r8   r$   r   r%   Úserr   r   r   Útest_pipe_args,   s"   ýÿ" rC   )Únumpyr   Úpandasr   r   r   Úpandas._testingÚ_testingr    r&   rC   r   r   r   r   Ú<module>   s    !