a
    PfI                     @   sL   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d Z	d	d
 Z
dS )    N)Seriesc                  C   sT   t g d} d}tjt|d" | jddgdd W d    n1 sF0    Y  d S )N            z(No axis named foo for object type Seriesmatch   r   Zfoo)Zaxis)r   pytestraises
ValueErrortakesermsg r   _/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/series/indexing/test_take.pytest_take_validate_axis   s    r   c                  C   s   t g d} | g d}t g dg dd}t|| | g d}t g dg dd}t|| d}tjt|d	 | d
dg W d    n1 s0    Y  tjt|d	 | ddg W d    n1 s0    Y  d S )Nr   )r      r   )r   r   r   index)r   r   r   )r   r   r   )r   r   r   zindices are out-of-boundsr	   r   
   r   r   )r   r   tmassert_series_equalr   r   
IndexError)r   actualexpectedr   r   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'   c                  C   sT   t g d} d}tjt|d" | tddd W d    n1 sF0    Y  d S )Nr   z6Series.take requires a sequence of integers, not slicer	   r   r   r   )r   r   r   	TypeErrorr   slicer   r   r   r   test_take_slice_raises-   s    r*   )r   Zpandasr%   r   Zpandas._testingZ_testingr   r   r   r'   r*   r   r   r   r   <module>   s   	
