
    Egd	              
          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                            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                            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                            deeg          d             Zd ZdS )    N)	DataFrame
MultiIndexSeriesconcatzargs, kwargs, increment a   )          )r	   cr
      c                     dd}t          ddg          } |j        |dg| R i |}||z   }t          j        ||           d S )Nr   c                 $    | |z   d|z  z   d|z  z   S )N
   d   r   )xr   br   s       ]/var/www/sysmax/venv/lib/python3.11/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!      sl    ( ( ( ( 	1vAQ[A//////F9}H68,,,,,r   z
ops, namessqrtabsolutec                      t          j        d          5  t           fd|D             d          }||_                             |          }t          j        ||           d d d            d S # 1 swxY w Y   d S )Nignoreallc                 &    g | ]} |          S r   r   ).0opstring_seriess     r   
<listcomp>z+test_transform_listlike.<locals>.<listcomp>(   s#    ;;;22m,,;;;r   r	   axis)nperrstater   columnsr   r   assert_frame_equal)r+   opsnamesr    r   s   `    r   test_transform_listliker5      s     
	"	"	" 0 0;;;;s;;;!DDD ((--
fh///	0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0s   AA33A7:A7c                  f   t          g d          } dd}dd}d}t          j        t          |          5  |                     ||gdd	d	d
           d d d            n# 1 swxY w Y   |                     ||gdd	d
          }t          g dg dd          }t          j        ||           d S )N)r	   r
   r   r	   r   c                     | |z   |z   S Nr   )r   r   r   s      r   foo1z4test_transform_listlike_func_with_args.<locals>.foo13       1uqyr   r
   c                     | |z   |z   S r8   r   )r   r   r   s      r   foo2z4test_transform_listlike_func_with_args.<locals>.foo26   r:   r   z/foo1\(\) got an unexpected keyword argument 'b')matchr      r   r   )r   )   	   r   )r9   r<   )r	   r   )r
   r   )r   pytestraises	TypeErrorr   r   r   r2   )r   r9   r<   msgr   r    s         r   &test_transform_listlike_func_with_argsrF   .   s#    	yyyA       =C	y	,	,	, 2 2	T4L!Q!q1112 2 2 2 2 2 2 2 2 2 2 2 2 2 2 [[$q!q[11F***jjjAABBH&(+++++s   A  A$'A$boxc                 t   t          j        d          5  t          t          j        |           t          j        |           gd          }d d d            n# 1 swxY w Y   ddg|_        |                      |t           j        t           j        d                    }t          j        ||           d S )Nr%   r&   r	   r-   foobar)rI   rJ   )	r/   r0   r   r"   absr1   r   r   r2   )r+   rG   r    r   s       r   test_transform_dictlikerL   B   s     
	"	"	" S S27=1126-3H3HIPQRRRS S S S S S S S S S S S S S Su~H$$SS)H)H%I%IJJF&(+++++s   8AA!Ac                      t          ddg          } |                     ddgdd          }t          g dg dgt          ddgd	d
g                    }t	          j        ||           d S )Nr	   r>   r"   rK   r?   )      ?r	   rN   )       @r>   rO   )r"   rK   )r   r   r	   )r   r	   r   )r1   )r   r   r   r   r   r2   )dfr   r    s      r   test_transform_dictlike_mixedrQ   L   s    	AB\\f==>>F	&J89i:PQQ  H &(+++++r   )numpyr/   rB   pandasr   r   r   r   pandas._testing_testingr   markparametrizer!   r"   rK   arrayr5   rF   dictrL   rQ   r   r   r   <module>rZ      s                         "a[2Qx#fb"%5sAh7LM - -	 - 
'VH
&"'	Z01	27)		vh'	2627#	$	$z6&:;	 0 0 0, , ,( v//, , 0/,, , , , ,r   