U
    >vfÃ  ã                   @   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-vdrwu74i/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Úcéþÿÿÿr   )Ú
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
   