
    Owg2                     <    d dl Zd dlZd dlmZmZ d dlmZ d Z	d Z
y)    N)	DataFrameIndexc                     t         j                  j                  d      } t        g d| j	                  d      | j	                  d      d      }d }d }|j                  d      j                  |      j                  |      }t        dd	gd
d      }t        j                  ddgd|      }t        j                  ||       y )N   )foobarr   r   r   r   r   r      )ABCc                     | j                   j                         | j                  j                         j                         z
  S N)r   maxr   min)dfgbs    U/var/www/horilla/myenv/lib/python3.12/site-packages/pandas/tests/groupby/test_pipe.pyfztest_pipe.<locals>.f   s*    vvzz|dffjjl..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     99((+L	I--a0--a0	

B1 ZZ_!!!$))&1F5%.s;Eyy+->?cQVWH8V,r   c                  B   t        g dg dg dd      } d }d }d }| j                  d      j                  |d	      j                  |d
      j                  |d      }t        ddgd      }t	        j
                  ddg|      }t        j                  ||       t	        j
                  g d      }|j                  |      j                  d       }t	        j
                  g dt        g dt        j                              }t        j                  ||       y )N)r
   r
   r   r   r   )g      ?       @g      @r-   g      @)g      $@g      Y@g     @@g      Yg     @)groupxyc                 R    | j                  fdd      }|j                  d      S )Nc                 >    | j                   j                         kD  S r   )r0   mean)grparg1s    r   <lambda>z+test_pipe_args.<locals>.f.<locals>.<lambda>9   s    355::<$+> r   F)dropnar.   )filterr!   )r   r5   filtereds    ` r   r   ztest_pipe_args.<locals>.f8   s'    ;;>u;M((r   c                 f    | j                         | j                         j                         z  |z   S r   )sum)r   arg2s     r   gztest_pipe_args.<locals>.g<   s%    xxzDHHJNN,,t33r   c                 :    | j                   | j                  z   |z
  S r   )r/   r0   )r(   arg3s     r   hztest_pipe_args.<locals>.h?   s    ttbdd{T!!r   r.   r   
   d   r
   r   )r   g#SgbeS)r   )   rC   r   r      rD   c                 D    | j                         | j                         z  S r   )r;   count)r4   s    r   r6   z test_pipe_args.<locals>.<lambda>L   s    swwy399;/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   ,   s     
.*7	

B)4" ZZ %%a+00B7<<QDF 3*7+Eyy..9GH68, ))&
'C[[""#FGFyy5"((+KLH68,r   )numpyr   pandasr#   r   r   pandas._testing_testingr%   r+   rK   r   r   r   <module>rP      s       -B$-r   