U
    sVc                     @   s<   d dl Z d dlZd dlmZ d dlmZ dd Zdd ZdS )    N)Seriesc               	   C   s   t dddddg} | dddg}t dddgdddgd}t|| | dddg}t dddgdddgd}t|| d	d
 }tjt|dd | ddg W 5 Q R X tjt|dd | ddg W 5 Q R X d S )N                  indexc                 S   s   d|  dS )Nzindex z+ is out of bounds for( axis 0 with)? size 5 )xr   r   J/tmp/pip-unpacked-wheel-xj8nt62q/pandas/tests/series/indexing/test_take.py<lambda>       ztest_take.<locals>.<lambda>
   )match)r   taketmassert_series_equalpytestZraises
IndexError)seractualexpectedmsgr   r   r   	test_take   s    r   c                  C   s\   t tdddg} | dddg}t tjdddgdddgddddgd}t|| d S )	Nabcr   )
categoriesr   r
   )r   pdZCategoricalr   r   r   )r   resultr   r   r   r   test_take_categorical   s     r$   )	r   Zpandasr"   r   Zpandas._testingZ_testingr   r   r$   r   r   r   r   <module>   s
   