3
���h�
� @ s� d dl Zd dlZd dlmZmZ dd� Zdd� Zdd� Zd d
� Z d d � Z
ej j d
dd d d d gd d!d"d#d$gfd%d&d'd(gd)d*d+d,gfd-d.d/gd0d1d2gfg�dd� �Z
dS )3� N)�_make_unique_kwarg_list�maybe_mangle_lambdasc C sF t d�dkst�t dd� �jdks&t�t dd� g�d jdksBt�d S )N�meanc S s | S )N� )�xr r �?/tmp/pip-build-5_djhm0z/pandas/pandas/tests/test_aggregation.py�<lambda> s z7test_maybe_mangle_lambdas_passthrough.<locals>.<lambda>z<lambda>c S s | S )Nr )r r r r r s r )r �AssertionError�__name__r r r r �%test_maybe_mangle_lambdas_passthrough s r c C sx dd� dd� g} t | �}|d jdks*t�|d jdks<t�| d d �|d d �ksXt�| d d �|d d �kstt�d S )Nc S s dS )N� r )r r r r r s z4test_maybe_mangle_lambdas_listlike.<locals>.<lambda>c S s dS )N� r )r r r r r s r z
<lambda_0>r z
<lambda_1>)r r
r )Zaggfuncs�resultr r r �"test_maybe_mangle_lambdas_listlike s r c C sL ddd� dd� gi} t | �}|d d jdks2t�|d d jdksHt�d S ) N�Ac S s dS )Nr r )r r r r r s z+test_maybe_mangle_lambdas.<locals>.<lambda>c S s dS )Nr r )r r r r r s r z
<lambda_0>r z
<lambda_1>)r r
r )�funcr r r r �test_maybe_mangle_lambdas s r c C s� dd dd�dd� gi} t | �}|d d jdks4t�|d d jdksJt�| d d dd�d
ksdt�| d d ddd �dks�t�| d d dd d
d �dks�t�d S )Nr r c S s
d||fS )Nr r )r �a�br r r r s z0test_maybe_mangle_lambdas_args.<locals>.<lambda>c S s dS )Nr r )r r r r r s r z
<lambda_0>z
<lambda_1>r
� )r )r )r r r )r r r
)r r
r )r r
r )r r r r r � test_maybe_mangle_lambdas_args s r c C s. t jt jt jd�d�} t| �}|| ks*t�d S )N)Zfoo�bar)�C�D)�npr r r )r r r r r � test_maybe_mangle_lambdas_named) s r zorder, expected_reorder�height�<lambda>�max�weight�
<lambda>_0�
<lambda>_1�col2�min�col1�
<lambda>_2�colc C s t | �}||kst�d S )N)r r )�orderZexpected_reorderr r r r �test_make_unique/ s )r( )r r )r r )r r )r r )r r )r r )r r )r r )r r! )r r )r" r# )r$ r )r$ r )r$ r )r" r# )r$ r )r$ r! )r$ r% )r& r )r& r )r& r )r& r )r& r! )r& r% )Znumpyr ZpytestZpandas.core.aggregationr r r r r r r �markZ parametrizer( r r r r �<module> s: |