o
    5c
                     @   s`  d dl Z d dlZd dlmZ d dlmZ e jdddg ddfddg d	dfddg d
dfddg d
dfdgdg ddfddgdg ddfddgdg ddfddg ddfddg d
dfdgdgg ddfddgddgg ddfddg ddfg ddg ddfgdd Z	e jdddddgd fdd!ddgdfddddgdfddddgd"fgd#d$ Z
dS )%    N)Categoricalz)to_replace,value,expected,flip_categories      )r   r      F   )r   r   r   r   r   r         r   )r   r   r   )r   r   r   )r   r   r   )r   r   r   )r   r	   )r   r   r   4)r   r   r
   )r   r   35)r   r   r   Tc                 C   s~   t jg ddd}|| |}t j|dd}|j| |dd |r-|j|jjd d d }tj||dd tj||dd d S )	Nr   category)dtypeTinplaceF)check_category_order)pdSeriesreplacecatset_categories
categoriestmassert_series_equal)
to_replacevalueexpectedflip_categoriesserresult r!   b/var/www/html/gps/gps/lib/python3.10/site-packages/pandas/tests/arrays/categorical/test_replace.pytest_replace_categorical_series   s   r#   z-to_replace, value, result, expected_error_msgbcaz$Categorical.categories are differentdz+Categorical.categories length are differentc                 C   s   t ddg}t |}tjtdd || |}W d    n1 s"w   Y  t|| | dkrNtjt|d t|| W d    n1 sIw   Y  tjtdd |j| |dd W d    n1 shw   Y  t|| d S )Nr&   r$   zSeries.replace)matchTr   )	r   r   assert_produces_warningFutureWarningr   assert_categorical_equalpytestraisesAssertionError)r   r   r    expected_error_msgr   r   r!   r!   r"   test_replace_categorical,   s   r0   )r,   pandasr   r   pandas._testing_testingr   markparametrizer#   r0   r!   r!   r!   r"   <module>   s>    

