U
    ÎZj½  ã                   @   sL   d dl Zd dlZd dlmZmZ d dlmZ d dl	m
Z
 dd„ Zdd„ ZdS )é    N)Ú	DataFrameÚIndex)Ú
Int64Indexc               	   C   s”   t j d¡} tddddddddg|  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 )NiÒ–IZfooÚbaré   )ÚAÚBÚCc                 S   s   | j  ¡ | j ¡  ¡  S ©N)r   Úmaxr	   Úmin)Údfgb© r   úa/var/www/html/TRUCKING_PROJECT/venv/lib/python3.8/site-packages/pandas/tests/groupby/test_pipe.pyÚf   s    ztest_pipe.<locals>.fc                 S   s   | d S )Né   r   )Zsrsr   r   r   Úsquare   s    ztest_pipe.<locals>.squarer   Úobject)ZdtypeÚnamegÂÖ®vqû!@g‰	Ö¯Y @r   )r   Úindex)ÚnpÚrandomZRandomStater   ZrandnÚ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 dddddgdddddgdd	d
ddgdœƒ} dd„ }dd„ }dd„ }|  d¡ |d¡ |d¡ |d¡}tdddgdd�}tjdddg|d�}t ||¡ t ddddddg¡}| |¡ d d!„ ¡}tjd"d#d$gtdddgƒd�}t ||¡ d S )%Nr   r   r	   g      ð?g       @g      @g      @g      $@g      Y@g     @�@g      YÀg     @�À)ÚgroupÚxÚyc                    s   | j ‡ fdd„dd� | j¡S )Nc                    s   | j  ¡ ˆ kS r
   )r$   Zmean©Úgrp©Úarg1r   r   Ú<lambda>:   ó    z+test_pipe_args.<locals>.f.<locals>.<lambda>F)Zdropna)Úfilterr   Zgrouper)r   r(   r   r'   r   r   9   s    ÿztest_pipe_args.<locals>.fc              
   S   sB   d}t jt|d��& |  ¡ |  ¡  ¡  | W  5 Q R £ S Q R X d S )Nz!The default value of numeric_only)Úmatch)r   Zassert_produces_warningÚFutureWarningÚsum)r   Zarg2Úmsgr   r   r   Úg>   s    ztest_pipe_args.<locals>.gc                 S   s   | j | j | S r
   )r#   r$   )r   Zarg3r   r   r   ÚhC   s    ztest_pipe_args.<locals>.hr"   r   é
   éd   )r   g#�šáSÀgùÜbeøžSÀi°ÿÿÿ)r   é   r   é   c                 S   s   |   ¡ |  ¡  S r
   )r.   Úcountr%   r   r   r   r)   P   r*   z test_pipe_args.<locals>.<lambda>é   r   é   )	r   r   r   r   r   r   r   r   r   )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   Zpandas.core.apir   r!   r9   r   r   r   r   Ú<module>   s   !