o
    թZh
                     @   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g dg dfg dg dfg dg dfgdd ZdS )    N)_make_unique_kwarg_listmaybe_mangle_lambdasc                   C   sF   t ddksJ t dd jdksJ t dd gd jdks!J d S )Nmeanc                 S      | S N xr   r   T/var/www/html/lang_env/lib/python3.10/site-packages/pandas/tests/test_aggregation.py<lambda>       z7test_maybe_mangle_lambdas_passthrough.<locals>.<lambda><lambda>c                 S   r   r   r   r   r   r   r
   r      r   r   r   __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J |d jdksJ | d d |d d ks,J | d d |d d ks:J d S )Nc                 S      dS N   r   r   r   r   r
   r      r   z4test_maybe_mangle_lambdas_listlike.<locals>.<lambda>c                 S   r   )N   r   r   r   r   r
   r      r   r   
<lambda_0>r   
<lambda_1>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sJ |d d jdks$J d S )	NAc                 S   r   Nr   r   r   r   r   r
   r      r   z+test_maybe_mangle_lambdas.<locals>.<lambda>c                 S   r   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sJ |d d jdks%J | d d ddd	ks2J | d d ddd
dks@J | d d dd
dddksOJ d S )Nr   r   c                 S   s
   d||fS r   r   )r	   abr   r   r
   r   "   s   
 z0test_maybe_mangle_lambdas_args.<locals>.<lambda>c                 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
   test_maybe_mangle_lambdas_args!   s   "r!   c                  C   s.   t jt jt jdd} t| }|| ksJ d S )N)Zfoobar)CD)npr   r   r   r   r   r
   test_maybe_mangle_lambdas_named,   s   r&   zorder, expected_reorder)heightr   r(   maxweightr*   r'   r,   r   ))r(   
<lambda>_0r)   r+   )r(   
<lambda>_1r-   )Zcol2mincol1r   r2   r2   )r0   )r3   r.   )r3   r/   )r3   
<lambda>_2)colr   r5   r5   ))r6   r.   )r6   r/   )r6   r4   c                 C   s   t | }||ks
J d S r   )r   )orderZexpected_reorderr   r   r   r
   test_make_unique2   s   )r8   )numpyr%   ZpytestZpandas.core.applyr   r   r   r   r   r!   r&   markZparametrizer8   r   r   r   r
   <module>   s,    	'