U
    sVc
                  	   @   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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dddgfgdd ZdS )    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   A/tmp/pip-unpacked-wheel-xj8nt62q/pandas/tests/test_aggregation.py<lambda>       z7test_maybe_mangle_lambdas_passthrough.<locals>.<lambda><lambda>c                 S   s   | S r   r   r   r   r   r	   r
      r   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
      r   z4test_maybe_mangle_lambdas_listlike.<locals>.<lambda>c                 S   s   dS )N   r   r   r   r   r	   r
      r   r   
<lambda_0>r   
<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
      r   z+test_maybe_mangle_lambdas.<locals>.<lambda>c                 S   s   dS r   r   r   r   r   r	   r
      r   r   r   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 r   r   )r   abr   r   r	   r
   "   r   z0test_maybe_mangle_lambdas_args.<locals>.<lambda>c                 S   s   d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	   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	   test_maybe_mangle_lambdas_named,   s    r%   zorder, expected_reorder)heightr   )r&   max)weightr'   )r(   r   )r&   
<lambda>_0)r&   
<lambda>_1)Zcol2min)col1r   )r,   r)   )r,   r*   )r,   
<lambda>_2)colr   )r.   r)   )r.   r*   )r.   r-   c                 C   s   t | }||kstd S r   )r   r   )orderZexpected_reorderr   r   r   r	   test_make_unique2   s    )r0   )Znumpyr$   ZpytestZpandas.core.applyr   r   r   r   r   r    r%   markZparametrizer0   r   r   r   r	   <module>   sN   	'