a
    é´«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/laksyashrimp/venv/lib/python3.9/site-packages/pandas/tests/groupby/test_pipe.pyÚf   s    ztest_pipe.<locals>.fc                 S   s   | d S )Nr   r   )Zsrsr   r   r   Úsquare   s    ztest_pipe.<locals>.squarer   r   r   ©Únameg¦8t”þ@gtÝÐîÞ@r	   )r   Úindex)ÚnpÚrandomZdefault_rngr   Zstandard_normalÚgroupbyÚpiper   ÚpdÚSeriesÚtmÚassert_series_equal)Z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   ó    z+test_pipe_args.<locals>.f.<locals>.<lambda>F)Zdropnar#   )Úfilterr   )r   r*   Úfilteredr   r)   r   r   8   s    ztest_pipe_args.<locals>.fc                 S   s   |   ¡ |   ¡   ¡  | S r   )Úsum)r   Zarg2r   r   r   Úg<   s    ztest_pipe_args.<locals>.gc                 S   s   | j | j | S r   )r$   r%   )r   Zarg3r   r   r   Úh?   s    ztest_pipe_args.<locals>.hr#   r   é
   éd   r   r	   r   g#�šáSÀgùÜbeøžSÀ)r   )é   r4   r   r   é   r5   c                 S   s   |   ¡ |  ¡  S r   )r/   Úcountr'   r   r   r   r+   L   r,   z test_pipe_args.<locals>.<lambda>)é   r   é   )r4   r   r5   )Zdtype)
r   r   r   r   r   r   r   r   r   Zint64)r   r   r0   r1   r   r   r    Zserr   r   r   Útest_pipe_args,   s"    ýÿ" r9   )Únumpyr   Zpandasr   r   r   Zpandas._testingZ_testingr   r!   r9   r   r   r   r   Ú<module>   s
   !