3
���h � @ s> d dl Zd dlZd dlmZmZ d dljZdd� Z dd� Z
dS )� N)� DataFrame�Indexc C s� t jjd�} tddddddddg| jd�| jd�d��}dd� }dd � }|jd
�j|�j|�}tddgd d
d �}tj d
dgd|d�}t
j ||� d S )Ni��IZfoo�bar� )�A�B�Cc S s | j j� | jj� j� S )N)r �maxr �min)�dfgb� r �@/tmp/pip-build-5_djhm0z/pandas/pandas/tests/groupby/test_pipe.py�f s ztest_pipe.<locals>.fc S s | d S )N� r )Zsrsr r r
�square s ztest_pipe.<locals>.squarer �object)Zdtype�nameg�֮vq�!@g� ֯Y @r )r �index) �np�randomZ RandomStater Zrandn�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 jdddddgdddddgdd d
d#d$gd ��} d d
� }dd� }dd� }| jd�j|d�j|d�j|d�}t jdddgdd�}t jd%d&d'g|d�}tj||� t jddd d d d g�}|j|�jd d � �}t jd d!d"gt jdd d g�d�}tj||� d S )(Nr r r g �?g @g @g @g $@g Y@g @�@)�group�x�yc s | j � fdd�dd�j| j�S )Nc s | j j� � kS )N)r" Zmean)�grp)�arg1r r
�<lambda>6 s z+test_pipe_args.<locals>.f.<locals>.<lambda>F)Zdropna)�filterr Zgrouper)r r$ r )r$ r
r 5 s ztest_pipe_args.<locals>.fc S s | j � | j � j � | S )N)�sum)r Zarg2r r r
�g: s ztest_pipe_args.<locals>.gc S s | j | j | S )N)r! r" )r Zarg3r r r
�h= s ztest_pipe_args.<locals>.hr r �
�d )r g#���S@g��be��S@�P )r � r � c S s | j � | j� S )N)r' �count)r# r r r
r% J s z test_pipe_args.<locals>.<lambda>� r � g Y�g @��g#���S�g��be��S�i����) r r r r r r r r Z
Int64Index)r r r( r) r r r Zserr r r
�test_pipe_args) s " r2 ) Znumpyr Zpandasr r r Zpandas._testingZ_testingr r r2 r r r r
�<module> s
! |