o
    ŀgd	                  
   @   s   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d Zejdejgdgfejejgddgfeejgdgfeejejgddgfgdd Zdd Zejdeegdd Zdd ZdS )    N)	DataFrame
MultiIndexSeriesconcatzargs, kwargs, increment a   )          )r   cr	      c                 C   sH   ddd}t ddg}|j|dg| R i |}|| }t|| d S )Nr   c                 S   s   | | d|  d|  S )N
   d   r   )xr   br   r   r   \/var/www/html/myenv/lib/python3.10/site-packages/pandas/tests/apply/test_series_transform.pyf   s   ztest_agg_args.<locals>.fr   r	   )r   r   r   )r   	transformtmassert_series_equal)argskwargs	incrementr   sresultexpectedr   r   r   test_agg_args   s
   
r   z
ops, namessqrtabsolutec                    sh   t jdd$ t fdd|D dd}||_ |}t|| W d    d S 1 s-w   Y  d S )Nignoreallc                    s   g | ]}| qS r   r   ).0opstring_seriesr   r   
<listcomp>(   s    z+test_transform_listlike.<locals>.<listcomp>r   axis)nperrstater   columnsr   r   assert_frame_equal)r&   opsnamesr   r   r   r%   r   test_transform_listlike   s   
"r0   c                  C   s   t g d} ddd}ddd}d	}tjt|d
 | j||gddddd W d    n1 s0w   Y  | j||gdddd}tg dg dd}t|| d S )N)r   r	   r
   r   r   c                 S      | | | S Nr   )r   r   r   r   r   r   foo13      z4test_transform_listlike_func_with_args.<locals>.foo1r	   c                 S   r1   r2   r   )r   r   r   r   r   r   foo26   r4   z4test_transform_listlike_func_with_args.<locals>.foo2z/foo1\(\) got an unexpected keyword argument 'b')matchr
      r   r   )r   )   	   r   )r3   r5   )r   r   )r	   r   )r   pytestraises	TypeErrorr   r   r   r-   )r   r3   r5   msgr   r   r   r   r   &test_transform_listlike_func_with_args.   s   

r?   boxc                 C   sz   t jdd tt | t | gdd}W d    n1 sw   Y  ddg|_| |t jt jd}t|| d S )Nr    r!   r   r(   foobar)rA   rB   )	r*   r+   r   r   absr,   r   r   r-   )r&   r@   r   r   r   r   r   test_transform_dictlikeB   s   
rD   c                  C   sV   t ddg} | ddgdd}tg dg dgtddgd	d
gd}t|| d S )Nr   r7   r   rC   r8   )      ?r   rE   )       @r7   rF   )r   rC   )r   r   r   )r   r   r   )r,   )r   r   r   r   r   r-   )dfr   r   r   r   r   test_transform_dictlike_mixedL   s   rH   )numpyr*   r;   pandasr   r   r   r   pandas._testing_testingr   markparametrizer   r   rC   arrayr0   r?   dictrD   rH   r   r   r   r   <module>   s,    *

		
	