3
Ud                 @   s:   d dl Z d dlZd dlmZ d dljZdd Zdd ZdS )    N)Seriesc              C   s   t dddddg} | jdddg}t dddgdddgd}tj|| | jdddg}t dddgdddgd}tj|| dd	 }tjt|d
d | jdd
g W d Q R X tjt|dd | jddg W d 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
   Z/var/www/html/virt/lib64/python3.6/site-packages/pandas/tests/series/indexing/test_take.py<lambda>   s    ztest_take.<locals>.<lambda>
   )matchr   )r   taketmassert_series_equalpytestZraises
IndexError)seractualexpectedmsgr
   r
   r   	test_take   s    r   c              C   s\   t tjdddg} | jd	d
dg}t tjdddgdddgddddgd}tj|| d S )Nabcr   r   )
categoriesr   )r	   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
   
