U
    *if                  
   @   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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ejdeegd	d
 Zdd ZdS )    N)	DataFrame
MultiIndexSeriesconcatz
ops, namessqrtabsolutec              	      sR   t jdd< t fdd|D dd}||_ |}t|| W 5 Q R X d S )Nignoreallc                    s   g | ]}| qS  r   ).0opstring_seriesr   [/home/mars/bis/venv/lib/python3.8/site-packages/pandas/tests/apply/test_series_transform.py
<listcomp>   s     z+test_transform_listlike.<locals>.<listcomp>   Zaxis)nperrstater   columns	transformtmassert_frame_equal)r   Zopsnamesexpectedresultr   r   r   test_transform_listlike   s
    
r   boxc              	   C   sf   t jdd" tt | t | gdd}W 5 Q R X 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_dictlike   s
    &
r"   c                  C   sZ   t ddg} | ddgdd}tdddgdddggtddgd	d
gd}t|| d S )Nr      r   r!   )bcg      ?g       @)r   r!   )r   r   r   )r   r   r   )r   )r   r   r   r   r   r   )Zdfr   r   r   r   r   test_transform_dictlike_mixed)   s    r&   )numpyr   ZpytestZpandasr   r   r   r   Zpandas._testingZ_testingr   markZparametrizer   r!   arrayr   dictr"   r&   r   r   r   r   <module>   s   	
	
	