
    I j              
           S SK r S SKrS SKJr  S SKJr  \ R                  R                  SSS/ SQ4SS/ SQ4S	S
/ SQ4SS/ SQ4/5      S 5       r	\ R                  R                  SSS/5      S 5       r
S rS rS rg)    N)Categoricalzto_replace,value,expected      r         r   )r   r   r   )      r   r   r   )r   r   r   c                     [         R                  " / SQSS9nUR                  X5      n[         R                  " [        U/ SQS95      n[        R
                  " XB5        g )Nr   categorydtype
categories)pdSeriesreplacer   tmassert_series_equal)
to_replacevalueexpectedserresults        }/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/pandas/tests/arrays/categorical/test_replace.pytest_replace_categorical_seriesr      sE     ))IZ
0C[[+FyyX)DEH6,    zto_replace,value)r   r	   )r   r	   c                     [         R                  " / SQSS9n[        R                  " [        SS9   UR                  X5        S S S 5        g ! , (       d  f       g = f)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&      sD     ))IZ
0C	N
 	J&
 
 
s   A
Ac                      [         R                  " / SQSS9n [         R                  " / SQU S9nUR                  SS5      n[         R                  " / SQU S9n[        R
                  " X2SS9  g )	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+   -   sT    	48E
))IU
+C[[AFyy%0H8$Gr   c                  D   [        [        R                  " / SQSS95      n [        R                  " U 5      R	                  SS/SS/5      R
                  n[        [        R                  " S/S-  SS9[        R                  " / SQSS9S9n[        R                  " X5        g )	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{    
bhhh?
@CYYs^##S#Jc
;CCF
#(+88O8<H 1r   c                     [        [        R                  " SS/SS95      n [        R                  " [
        SS9   [        R                  " U 5      R                  SS/S[        R                  /5        S S S 5        g ! , (       d  f       g = f)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   sc    
bhhSz:
;C	N
 			#SzC<8
 
 
s   9A<<
B
)r#   pandasr   r   pandas._testing_testingr   markparametrizer   r&   r+   r5   r8    r   r   <module>r?      s         
Ay	
Ay	I	I
-
-  		''H29r   