
    Owgd	              
       $   d dl Zd dlZd dlmZmZmZmZ d dlm	Z
 ej                  j                  ddi d fdddidfdi dfd	d
didfg      d        Zej                  j                  dej                  gdgfej                  ej                  gddgf ej                   ej                  g      dgf ej                   ej                  ej                  g      ddgfg      d        Zd Zej                  j                  deeg      d        Zd Zy)    N)	DataFrame
MultiIndexSeriesconcatzargs, kwargs, increment a   )          )r	   cr
      c                     dd}t        ddg      } |j                  |dg| i |}||z   }t        j                  ||       y )Nr   c                 $    | |z   d|z  z   d|z  z   S )N
   d   r   )xr   br   s       _/var/www/horilla/myenv/lib/python3.12/site-packages/pandas/tests/apply/test_series_transform.pyfztest_agg_args.<locals>.f   s    1urAv~a''    r	   r
   )r   r   r   )r   	transformtmassert_series_equal)argskwargs	incrementr   sresultexpecteds          r   test_agg_argsr!      sK    ( 	1vAQ[[A///F9}H68,r   z
ops, namessqrtabsolutec           	         t        j                  d      5  t        |D cg c]
  } ||        c}d      }||_        | j	                  |      }t        j                  ||       d d d        y c c}w # 1 sw Y   y xY w)Nignoreallr	   axis)nperrstater   columnsr   r   assert_frame_equal)string_seriesopsnamesopr    r   s         r   test_transform_listliker2      sq     
	" 0s;2m,;!D ((-
fh/	0 0;0 0s   
A5A0
7A50A55A>c                  @   t        g d      } dd}dd}d}t        j                  t        |      5  | j	                  ||gdddd	       d d d        | j	                  ||gddd
      }t        g dg dd      }t        j                  ||       y # 1 sw Y   IxY w)N)r	   r
   r   r   c                     | |z   |z   S Nr   )r   r   r   s      r   foo1z4test_transform_listlike_func_with_args.<locals>.foo13       1uqyr   c                     | |z   |z   S r5   r   )r   r   r   s      r   foo2z4test_transform_listlike_func_with_args.<locals>.foo26   r7   r   z/foo1\(\) got an unexpected keyword argument 'b')matchr      r   r   )r   )   	   r   )r6   r9   )r	   r   )r
   r   )r   pytestraises	TypeErrorr   r   r   r-   )r   r6   r9   msgr   r    s         r   &test_transform_listlike_func_with_argsrC   .   s     	yA =C	y	, 2	T4L!Q!q12 [[$q!q[1F*jABH&(+2 2s   BBboxc                 n   t        j                  d      5  t        t        j                  |       t        j                  |       gd      }d d d        ddg_        | j                   |t         j                  t         j                  d            }t        j                  ||       y # 1 sw Y   _xY w)Nr%   r&   r	   r(   foobar)rF   rG   )	r*   r+   r   r"   absr,   r   r   r-   )r.   rD   r    r   s       r   test_transform_dictlikerI   B   s     
	" S277=1266-3HIPQRSu~H$$S)H%IJF&(+	S Ss   6B++B4c                      t        ddg      } | j                  ddgdd      }t        g dg dgt        ddgd	d
g            }t	        j
                  ||       y )Nr	   r;   r"   rH   r<   )      ?r	   rK   )       @r;   rL   )r"   rH   )r   r   r	   )r   r	   r   )r,   )r   r   r   r   r   r-   )dfr   r    s      r   test_transform_dictlike_mixedrN   L   s^    	AB\\f=>F	&J89i:PQH &(+r   )numpyr*   r?   pandasr   r   r   r   pandas._testing_testingr   markparametrizer!   r"   rH   arrayr2   rC   dictrI   rN   r   r   r   <module>rW      s7       "a[2Qx#fb"%5sAh7LM-	- 
''VH
&&"''	Z01	277)	vh'	266277#	$z6&:;	00,( v/, 0,,r   