
    bMh"                     >    d dl Zd dlZd dlmZmZ d dlmZ d Z	d Z
dS )    N)	DataFrameIndexc                     t           j                            d          } t          g d|                     d          |                     d          d          }d }d }|                    d                              |                              |          }t          dd	gd
          }t          j	        ddgd|          }t          j        ||           d S )N   )foobarr   r   r   r   r   r      )ABCc                     | j                                         | j                                                                        z
  S N)r   maxr   min)dfgbs    ^/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/tests/groupby/test_pipe.pyfztest_pipe.<locals>.f   s-    vzz||dfjjll..0000    c                     | dz  S )Nr    )srss    r   squareztest_pipe.<locals>.square   s    Avr   r
   r   r   nameg8t@gt@r   )r   index)nprandomdefault_rngr   standard_normalgroupbypiper   pdSeriestmassert_series_equal)random_statedfr   r   resultr   expecteds          r   	test_piper*      s     9((++L	III--a00--a00	
 	

 
B1 1 1   ZZ__!!!$$))&11F5%.s+++Ey+->?cQVWWWH8V,,,,,r   c                  z   t          g dg dg dd          } d }d }d }|                     d                              |d	                              |d
                              |d          }t          ddgd          }t	          j        ddg|          }t          j        ||           t	          j        g d          }|                    |                              d           }t	          j        g dt          g dt          j	                            }t          j        ||           d S )N)r
   r
   r   r   r   )g      ?       @g      @r,   g      @)g      $@g      Y@g     @@g      Yg     @)groupxyc                 b    |                      fdd          }|                    d          S )Nc                 >    | j                                         k    S r   )r/   mean)grparg1s    r   <lambda>z+test_pipe_args.<locals>.f.<locals>.<lambda>9   s    35::<<$+> r   F)dropnar-   )filterr    )r   r4   filtereds    ` r   r   ztest_pipe_args.<locals>.f8   s6    ;;>>>>u;MM(((r   c                 ~    |                                  |                                                                   z  |z   S r   )sum)r   arg2s     r   gztest_pipe_args.<locals>.g<   s,    xxzzDHHJJNN,,,t33r   c                 &    | j         | j        z   |z
  S r   )r.   r/   )r'   arg3s     r   hztest_pipe_args.<locals>.h?   s    tbd{T!!r   r-   r   
   d   r
   r   r   g#SgbeS)r   )   rB   r   r      rC   c                 T    |                                  |                                 z  S r   )r:   count)r3   s    r   r5   z test_pipe_args.<locals>.<lambda>L   s    swwyy399;;/F r   )   r	      )rB   r   rC   )dtype)
r   r    r!   r   r"   r#   r$   r%   r   int64)r'   r   r<   r?   r(   r   r)   sers           r   test_pipe_argsrK   ,   sk    
...***777	
 	

 
B) ) )4 4 4" " " ZZ  %%a++00B77<<QDDF 3*7+++Ey..9GGGH68,,, )&&&
'
'C[[""#F#FGGFy5"(+K+K+KLLLH68,,,,,r   )numpyr   pandasr"   r   r   pandas._testing_testingr$   r*   rK   r   r   r   <module>rP      s                        - - -B$- $- $- $- $-r   