a
    Pf                     @   s:   d dl Z d dlmZmZmZ d dlmZ G dd dZdS )    N)Index
MultiIndexSeriesc                   @   s@   e Zd Zdd Zdd Zejdddiddii gd	d
 ZdS )TestSeriesRenameAxisc                 C   s   t jg dddggddgd}tttt||d}|jddid}|jjddgksZJ |jt	j
d	d
}|jjddgks~J |jddgd}|jjddgksJ tjtdd |jdd W d    n1 s0    Y  d S )N)abc      Zllnn)namesindexfoor   )r   ZaxisZLLZNNZgooZ
unexpected)matchZwrong)columns)r   Zfrom_productr   listrangelenrename_axisr   r   strupperpytestZraises	TypeError)selfmiserresult r   e/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/series/methods/test_rename_axis.pytest_rename_axis_mapper   s    z,TestSeriesRenameAxis.test_rename_axis_mapperc                 C   s8   | d}|}|j ddd}|d u s(J t|| d S )Nr   T)Zinplace)r   tmassert_series_equal)r   Zdatetime_seriesexpectedr   Z	no_returnr   r   r   test_rename_axis_inplace   s
    
z-TestSeriesRenameAxis.test_rename_axis_inplacekwargsZmapperNr   c                 C   sb   t tddd}tg d|d}|jf i |}|r>|d n|}tg d|d}t|| d S )Nabcr   )name)r	   r
      r   )r   r   r   r   renamer!   r"   )r   r%   r   r   r   Zexpected_indexr#   r   r   r   test_rename_axis_none&   s    z*TestSeriesRenameAxis.test_rename_axis_none)	__name__
__module____qualname__r    r$   r   markZparametrizer*   r   r   r   r   r      s   	r   )	r   Zpandasr   r   r   Zpandas._testingZ_testingr!   r   r   r   r   r   <module>   s   