3
pd
              	   @   s   d dl Zd dlZd dlmZmZ dd Zdd Zdd Zd	d
 Z	dd Z
ejjddddddgd d!d"d#d$gfd%d&d'd(gd)d*d+d,gfd-d.d/gd0d1d2gfgdd ZdS )3    N)_make_unique_kwarg_listmaybe_mangle_lambdasc               C   sF   t ddkstt dd jdks&tt dd gd jdksBtd S )Nmeanc             S   s   | S )N )xr   r   ?/tmp/pip-build-7vycvbft/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   dd dd g} t | }|d jdks*t|d jdks<t| d d |d d ksXt| d d |d d ksttd S )Nc             S   s   dS )N   r   )r   r   r   r   r      s    z4test_maybe_mangle_lambdas_listlike.<locals>.<lambda>c             S   s   dS )N   r   )r   r   r   r   r      s    r   z
<lambda_0>r   z
<lambda_1>)r   r
   r	   )Zaggfuncsresultr   r   r   "test_maybe_mangle_lambdas_listlike   s    r   c              C   sL   ddd dd gi} t | }|d d jdks2t|d d jdksHtd S )	NAc             S   s   dS )Nr   r   )r   r   r   r   r      s    z+test_maybe_mangle_lambdas.<locals>.<lambda>c             S   s   dS )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   dddddd gi} t | }|d d jdks4t|d d jdksJt| d d dddksdt| d d ddd	dkst| d d dd	d
ddkstd S )Nr   r   c             S   s
   d||fS )Nr   r   )r   abr   r   r   r      s    z0test_maybe_mangle_lambdas_args.<locals>.<lambda>c             S   s   dS )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 jt jt jdd} t| }|| ks*td S )N)Zfoobar)CD)npr   r   r	   )r   r   r   r   r   test_maybe_mangle_lambdas_named)   s    r   zorder, expected_reorderheight<lambda>maxweight
<lambda>_0
<lambda>_1col2mincol1
<lambda>_2colc             C   s   t | }||kstd S )N)r   r	   )orderZexpected_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%   )numpyr   ZpytestZpandas.core.aggregationr   r   r   r   r   r   r   markZparametrizer(   r   r   r   r   <module>   s:   	