o
    �¨Êh2  ã                   @   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	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   úP/var/www/html/env/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   Úobject)ÚdtypeÚ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   r0   Ú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   )r5   Úcountr-   r   r   r   r1   L   s    z test_pipe_args.<locals>.<lambda>)é   r   é   )r<   r   r=   )r   )
r   r   r   r   r   r    r!   r"   r   Úint64)r$   r   r7   r9   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    !