3
Ud*                 @   s.   d dl Zd dljZdd Zdd Zdd ZdS )    Nc              C   sz   t jdtdd} | dd  }|| jd ks0t| dd  }tj|| jddg  | d	d  }tj|| jd
ddg  d S )N   ABCD)indexc             S   s   dS )NA )xr   r   ^/var/www/html/virt/lib64/python3.6/site-packages/pandas/tests/series/indexing/test_callable.py<lambda>   s    z'test_getitem_callable.<locals>.<lambda>r   c             S   s   ddgS )Nr   Br   )r   r   r   r   r	      s    r
   c             S   s   ddddgS )NTFr   )r   r   r   r   r	      s    r         )pdSerieslistlocAssertionErrortmassert_series_equalZiloc)sresultr   r   r   test_getitem_callable   s    r   c              C   sL   t jddddgtdd} d	| dd < tj| t jd
dddgtdd d S )N   r   r   r   r   )r   c             S   s   dS )Nr   r   )r   r   r   r   r	      s    z'test_setitem_callable.<locals>.<lambda>r   )r   r   r   r   r   )r   r   r   r   test_setitem_callable   s    r   c              C   sH   dd } t jddddg}| ||dk < t jdd| dg}tj|| d S )Nc             S   s   | d S )Nr   r   )r   r   r   r   r	      s    z-test_setitem_other_callable.<locals>.<lambda>r   r   r   r   r   )r   r   r   r   )incr   expectedr   r   r   test_setitem_other_callable   s
    r   )Zpandasr   Zpandas._testingZ_testingr   r   r   r   r   r   r   r   <module>   s   
