
    g2                     >    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
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    j/var/www/html/nodeJS/PythonScripts/venv3.11/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   object)dtype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   )r0   mean)grparg1s    r   <lambda>z+test_pipe_args.<locals>.f.<locals>.<lambda>9   s    35::<<$+> r   F)dropnar.   )filterr!   )r   r5   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/   r0   )r(   arg3s     r   hztest_pipe_args.<locals>.h?   s    tbd{T!!r   r.   r   
   d   r
   r   )r   g#SgbeS)r   )   rC   r   r      rD   c                 T    |                                  |                                 z  S r   )r;   count)r4   s    r   r6   z test_pipe_args.<locals>.<lambda>L   s    swwyy399;;/F r   )   r	      )rC   r   rD   )r   )
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   