
    Pi              
          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fddg dfd	d
g dfddg dfg          d             Z	e j                            dddg          d             Z
d Zd Zd ZdS )    N)Categoricalzto_replace,value,expected      r         r   )r   r   r   )      r   r   r   )r   r   r   c                     t          j        g dd          }|                    | |          }t          j        t          |g d                    }t	          j        ||           d S )Nr   categorydtype
categories)pdSeriesreplacer   tmassert_series_equal)
to_replacevalueexpectedserresults        /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/arrays/categorical/test_replace.pytest_replace_categorical_seriesr      sh     )IIIZ
0
0
0C[[U++FyX)))DDDEEH68,,,,,    zto_replace,value)r   r	   )r   r	   c                     t          j        g dd          }t          j        t          d          5  |                    | |           d d d            d S # 1 swxY w Y   d S )Nr   r   r   3Cannot setitem on a Categorical with a new categorymatch)r   r   pytestraises	TypeErrorr   )r   r   r   s      r   3test_replace_categorical_series_new_category_raisesr&      s     )IIIZ
0
0
0C	N
 
 
 ' ' 	J&&&' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 's   AAAc                      t          j        g dd          } t          j        g d|           }|                    dd          }t          j        g d|           }t	          j        ||d           d S )	N)r   r   r   T)orderedr   r   r   )r   r   r   )check_category_order)r   CategoricalDtyper   r   r   r   )r   r   r   r   s       r   test_replace_maintain_orderingr+   -   s|    			4888E
)IIIU
+
+
+C[[AFy%000H8V$GGGGGGr   c                  Z   t          t          j        g dd                    } t          j        |                               ddgddg          j        }t          t          j        dgdz  d          t          j        g dd                    }t          j        ||           d S )	N)abcstringr   r-   r.   r/   r   r   )r   r   arrayr   r   _valuesr   assert_categorical_equal)catr   r   s      r   !test_replace_categorical_ea_dtyper5   6   s    
bhh???
@
@CYs^^##S#Jc
;;CF
#(+++8OOO8<<<  H 11111r   c                  $   t          t          j        ddgd                    } t          j        t
          d          5  t          j        |                               ddgdt          j        g           d d d            d S # 1 swxY w Y   d S )Nr-   r.   r0   r   r    r!   r/   )	r   r   r1   r#   r$   r%   r   r   NA)r4   s    r   7test_replace_categorical_ea_dtype_different_cats_raisesr8   A   s    
bhSz:::
;
;C	N
 
 
 9 9 		#SzC<8889 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9s   7BB	B	)r#   pandasr   r   pandas._testing_testingr   markparametrizer   r&   r+   r5   r8    r   r   <module>r?      s*                     
Ayyy	
Ayyy	III	III
 
- -
 
-  		 ' ' 'H H H2 2 29 9 9 9 9r   