o
    $jd	                  
   @   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   i/var/www/html/Deteccion_Ine/venv/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tmZ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   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 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   r/   r0   r   )r   r   r   r   r   r   foo26   r2   z4test_transform_listlike_func_with_args.<locals>.foo2z/foo1\(\) got an unexpected keyword argument 'b')matchr
      r   r   )r   )   	   r   )r1   r3   )r   r   )r	   r   )r   pytestZraises	TypeErrorr   r   r   r+   )r   r1   r3   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)r>   r?   )	r(   r)   r   r   absr*   r   r   r+   )r%   r=   r   r   r   r   r   test_transform_dictlikeB   s   
rA   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   r5   r   r@   r6   )      ?r   rB   )       @r5   rC   )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   rD   )numpyr(   r9   Zpandasr   r   r   r   Zpandas._testingZ_testingr   markZparametrizer   r   r@   arrayr.   r<   dictrA   rD   r   r   r   r   <module>   s,    *

		
	