o
    d                     @   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 g d} | g d}t g dg dd}t|| | g d}t g dg dd}t|| dd	 }tjt|d
d | dd
g W d    n1 sSw   Y  tjt|dd | ddg W d    d S 1 stw   Y  d S )N)            )      r   )r   r   r   index)r   r	   r   )r   r   r   )r   r	   r   c                 S   s   d|  dS )Nzindex z+ is out of bounds for( axis 0 with)? size 5 )xr   r   Z/app/.heroku/python/lib/python3.10/site-packages/pandas/tests/series/indexing/test_take.py<lambda>   s    ztest_take.<locals>.<lambda>
   )matchr   r   r   )r   taketmassert_series_equalpytestZraises
IndexError)seractualexpectedmsgr   r   r   	test_take   s   "r   c                  C   sR   t tg d} | g d}t tjg dg ddg dd}t|| d S )N)abc)r   r   )r   r   r   )
categories)r   r   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    