o
    �¨Êhd	  ã                
   @   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   úZ/var/www/html/env/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,    *þ
üþ
		
	