a
    Pfd	                  
   @   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   a/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/apply/test_series_transform.pyf   s    ztest_agg_args.<locals>.fr   r	   )r   r   r   )r   	transformtmZassert_series_equal)argskwargs	incrementr   sresultexpectedr   r   r   test_agg_args   s
    
r   z
ops, namessqrtabsolutec                    sf   t jddF t fdd|D dd}||_ |}t|| W d    n1 sX0    Y  d S )Nignoreallc                    s   g | ]}| qS r   r   ).0opstring_seriesr   r   
<listcomp>(       z+test_transform_listlike.<locals>.<listcomp>r   Zaxis)nperrstater   columnsr   r   assert_frame_equal)r%   opsnamesr   r   r   r$   r   test_transform_listlike   s
    
r/   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 s`0    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   | | | S Nr   )r   r   r   r   r   r   foo13   s    z4test_transform_listlike_func_with_args.<locals>.foo1r	   c                 S   s   | | | S r0   r   )r   r   r   r   r   r   foo26   s    z4test_transform_listlike_func_with_args.<locals>.foo2z/foo1\(\) got an unexpected keyword argument 'b')matchr
      r   r   )r   )   	   r   )r1   r2   )r   r   )r	   r   )r   pytestZraises	TypeErrorr   r   r   r,   )r   r1   r2   msgr   r   r   r   r   &test_transform_listlike_func_with_args.   s    

6r;   boxc                 C   sz   t jdd, tt | t | gdd}W d    n1 s>0    Y  ddg|_| |t jt jd}t|| d S )Nr   r    r   r(   foobar)r=   r>   )	r)   r*   r   r   absr+   r   r   r,   )r%   r<   r   r   r   r   r   test_transform_dictlikeB   s
    :
r@   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   r4   r   r?   r5   )      ?r   rA   )       @r4   rB   )r   r?   )r   r   r   )r   r   r   )r+   )r   r   r   r   r   r,   )Zdfr   r   r   r   r   test_transform_dictlike_mixedL   s    rC   )numpyr)   r8   Zpandasr   r   r   r   Zpandas._testingZ_testingr   markZparametrizer   r   r?   arrayr/   r;   dictr@   rC   r   r   r   r   <module>   s*   *
	
	
	