
    g
                         d dl Zd dlZd dlmZmZ d Zd Zd Zd Z	d Z
ej                  j                  dg d	g d
fg dg dfg dg dfg      d        Zy)    N)_make_unique_kwarg_listmaybe_mangle_lambdasc                      t        d      dk(  sJ t        d       j                  dk(  sJ t        d g      d   j                  dk(  sJ y )Nmeanc                     | S N xs    R/var/www/openai/venv/lib/python3.12/site-packages/pandas/tests/test_aggregation.py<lambda>z7test_maybe_mangle_lambdas_passthrough.<locals>.<lambda>   s    !    <lambda>c                     | S r   r	   r
   s    r   r   z7test_maybe_mangle_lambdas_passthrough.<locals>.<lambda>   s    1r   r   r   __name__r	   r   r   %test_maybe_mangle_lambdas_passthroughr   
   sN    '6111,55CCC.q1::jHHHr   c                      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 y )Nc                      yN   r	   r
   s    r   r   z4test_maybe_mangle_lambdas_listlike.<locals>.<lambda>   s    !r   c                      y)N   r	   r
   s    r   r   z4test_maybe_mangle_lambdas_listlike.<locals>.<lambda>   s    qr   r   
<lambda_0>r   
<lambda_1>r   )aggfuncsresults     r   "test_maybe_mangle_lambdas_listliker      s    [)H!(+F!9---!9---8A;t	q	$///8A;t	q	$///r   c                      dd d gi} t        |       }|d   d   j                  dk(  sJ |d   d   j                  dk(  sJ y )NAc                      yNr   r	   r
   s    r   r   z+test_maybe_mangle_lambdas.<locals>.<lambda>   s    Ar   c                      yr   r	   r
   s    r   r   z+test_maybe_mangle_lambdas.<locals>.<lambda>   s    r   r   r   r   r   r   funcr   s     r   test_maybe_mangle_lambdasr&      sS    +{+,D!$'F#;q>""l222#;q>""l222r   c                     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(  sJ  | d   d   ddd	      d
k(  sJ  | d   d   dd	d      dk(  sJ y )Nr    r   c                     d||fS r"   r	   )r   abs      r   r   z0test_maybe_mangle_lambdas_args.<locals>.<lambda>"   s
    Q1Ir   c                      yr   r	   r
   s    r   r   z0test_maybe_mangle_lambdas_args.<locals>.<lambda>"   s    r   r   r   r   )r   r   r   r   )r   r   r      )r*   )r   r   r,   )r   r   r$   s     r   test_maybe_mangle_lambdas_argsr-   !   s    -{;<D!$'F#;q>""l222#;q>""l22249Q<1***49Q<1a I---49Q<1"i///r   c                      t         j                  t         j                  t         j                  dd} t        |       }|| k(  sJ y )N)foobar)CD)npr   r   r$   s     r   test_maybe_mangle_lambdas_namedr4   ,   s3    rwwrww?@D!$'FT>>r   zorder, expected_reorder)heightr   r6   maxweightr8   r5   r:   r   ))r6   
<lambda>_0r7   r9   )r6   
<lambda>_1r;   )col2mincol1r   rA   rA   )r>   )rB   r<   )rB   r=   )rB   
<lambda>_2)colr   rD   rD   ))rE   r<   )rE   r=   )rE   rC   c                 (    t        |       }||k(  sJ y r   )r   )orderexpected_reorderr   s      r   test_make_uniquerI   2   s    R %U+F%%%%r   )numpyr3   pytestpandas.core.applyr   r   r   r   r&   r-   r4   markparametrizerI   r	   r   r   <module>rO      s     I030 	
"	
 LQ	
?#&N&O&N&r   