
    bMhI                     F    d dl Z d dlZd dlmZ d dlmZ d Zd Zd Z	d Z
dS )    N)Seriesc                      t          g d          } d}t          j        t          |          5  |                     ddgd           d d d            d S # 1 swxY w Y   d S )N            z(No axis named foo for object type Seriesmatch   r	   foo)axis)r   pytestraises
ValueErrortakesermsgs     f/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/tests/series/indexing/test_take.pytest_take_validate_axisr      s    
!!!
"
"C
4C	z	-	-	- % %!Qe$$$% % % % % % % % % % % % % % % % % %s   AAAc                  Z   t          g d          } |                     g d          }t          g dg d          }t          j        ||           |                     g d          }t          g dg d          }t          j        ||           d}t	          j        t          |	          5  |                     d
dg           d d d            n# 1 swxY w Y   t	          j        t          |	          5  |                     ddg           d d d            d S # 1 swxY w 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   s       r   	test_taker#      s   
!!!
"
"CXXiii  Fiiiyyy111H68,,,XXjjj!!Fiiiyyy111H68,,,
%C	z	-	-	-  !R              	z	-	-	-  !Q                 s$   1CCC;D  D$'D$c                      t          t          j        g d                    } |                     g d          }t          t          j        g dg d          g d          }t	          j        ||           d S )N)abc)r(   r   )r&   r&   r%   )
categories)r   r   r   r   )r   pdCategoricalr   r   r   )r   resultr"   s      r   test_take_categoricalr-   #   s    
00
1
1CXXkkk""F
???CCC999  H 68,,,,,    c                      t          g d          } d}t          j        t          |          5  |                     t          ddd                     d d d            d S # 1 swxY w Y   d S )Nr   z6Series.take requires a sequence of integers, not slicer   r   r   r   )r   r   r   	TypeErrorr   slicer   s     r   test_take_slice_raisesr2   -   s    
!!!
"
"C
BC	y	,	,	, ! !q!Q   ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !s   %A!!A%(A%)r   pandasr*   r   pandas._testing_testingr   r   r#   r-   r2    r.   r   <module>r7      s                    % % %  $- - -! ! ! ! !r.   