
    I j                     H    S SK rS SKrS SKJrJrJr  S SKJr	   " S S5      r
g)    N)	DataFrameIndex
MultiIndexc                       \ rS rSrS rS rS rS r\R                  R                  SSSS	.S
S4SSS	.SS
4SS0S
S4SS0SS
4SSS.S
S
40 SS4/5      S 5       rSrg)TestDataFrameRenameAxis   c                 L   UR                  S5      nUR                  5       nUR                  SSS9=pEUb   eUb   e[        R                  " X25        UR                  SSS9nUR                  5       nUR                  SSSS9=pEUb   eUb   e[        R                  " X25        g )NfooT)inplacebar   axis)r   r   )rename_axiscopytmassert_frame_equal)selffloat_frameexpectedresultreturn_value	no_returns         |/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/pandas/tests/frame/methods/test_rename_axis.pytest_rename_axis_inplace0TestDataFrameRenameAxis.test_rename_axis_inplace   s    **51!!##)#5#5eT#5#JJ###   
f/**5q*9!!##)#5#5e!T#5#RR###   
f/    c                     [        SS/SS//SS/S9R                  SS	9nUR                  S
SS9n[        SS/SS//[        SS/S
S9SS/S9n[        R
                  " X#SS9  g )Nr            abcolumnsF)allows_duplicate_labelsidxr   r   nameindexr%   )check_flags)r   	set_flagsr   r   r   r   )r   dfr   r   s       r   3test_rename_axis_with_allows_duplicate_labels_falseKTestDataFrameRenameAxis.test_rename_axis_with_allows_duplicate_labels_false   s    AA'#s<FF$) G 
 A.VaVE1a&u$=Sz
 	fEBr   c                 f   [        SS/SS/S.5      n[        R                  " [        SS9   UR	                  [
        SS9  S S S 5        [        R                  " [        SS9   UR	                  SS	S
.SS9  S S S 5        [        R                  " [        SS9   UR	                  [
        SS9  S S S 5        [        R                  " [        SS9   US   R	                  [
        5        S S S 5        g ! , (       d  f       N= f! , (       d  f       N= f! , (       d  f       Nk= f! , (       d  f       g = f)Nr   r   )ABzUse `.rename`matchr   r   
      )r   r   r2   )r   pytestraises
ValueErrorr   id)r   r.   s     r   test_rename_axis_raises/TestDataFrameRenameAxis.test_rename_axis_raises+   s    aV1a&12]]:_=NN2AN& > ]]:_=NNrb>N2 > ]]:_=NN2AN& > ]]:_=sG# >= >= >= >= >=s/   C/"D DD"/
C= 
D
D"
D0c           	         [         R                  " / SQSS//SS/S9n[        [        [	        [        U5      5      5      [	        [        U5      5       Vs/ s H  o"S-  PM	     snS.US	9nUR                  S
SS9n[        R                  " UR                  [        SS/S
S95        UR                  S
S0SS9n[        R                  " UR                  [        SS/SS95        UR                  SS0S	9nUR                  R                  SS/:X  d   eUR                  [        R                  SS9nUR                  R                  SS/:X  d   eUR                  SS/S	9nUR                  R                  SS/:X  d   eUR                  5       R!                  S5      R#                  SS/S9nUR                  SSS9nUR                  R$                  S:X  d   eUR                  R$                  S:X  d   e[&        R(                  " [*        SS9   UR                  SS	9  S S S 5        [&        R(                  " [,        SS9   UR                  S/S	9  S S S 5        [&        R(                  " [*        SS9   UR                  S S9  S S S 5        g s  snf ! , (       d  f       Nw= f! , (       d  f       NV= f! , (       d  f       g = f) N)r"   r#   cr   r   llnn)namesr6   )xy)r+   colsr   rC   rD   r(   new)r%   r   r
   r   )r+   r   LLNNgoor$   mehr*   z	Must passr4   wrongzLength of namesbogus)rL   )r   from_productr   listrangelenr   r   assert_index_equalr%   r   r+   rB   strupperreset_index	set_indexdropr)   r8   r9   	TypeErrorr:   )r   miir.   r   sdfs         r   test_rename_axis_mapper/TestDataFrameRenameAxis.test_rename_axis_mapper:   sa   $$o1v%>tTlSuSW~&eCGn-Mn"fn-MNVX

 Q/
fnneS#JV.LM ##VUO!#D
fnneS#JU.KL tUm4||!!eT]222 ciia8||!!dD\111 uen5||!!eU^333 nn((.33T3K3Hue<||  E)))~~""e+++ ]]9K8NNN) 9 ]]:->?NN'N+ @ ]]9G4NNN& 54I .N< 98 @? 54s*   J
J=J/J/
J
J,/
J=z$kwargs, rename_index, rename_columnsNr   )mapperr   TFr   r+   r%   r*   c                 v   [        [        S5      SS9n[        SS/SS9n[        R                  " S5      R	                  SS	5      n[        XdU5      nUR                  " S
0 UD6nU(       a  UR                  S 5      OUn	U(       a  UR                  S 5      OUn
[        XiU
5      n[        R                  " X5        g )Nabcr
   r(   col1col2r      r    r    )
r   rN   nparangereshaper   r   renamer   r   )r   kwargsrename_indexrename_columnsr+   r%   datar.   r   expected_indexexpected_columnsr   s               r   test_rename_axis_none-TestDataFrameRenameAxis.test_rename_axis_nonee   s     d5k.(u5yy|##Aq)tG,)&)/;d+3A7>>$/wT3CD
f/r   rc   )__name__
__module____qualname____firstlineno__r   r/   r<   r[   r8   markparametrizern   __static_attributes__rc   r   r   r   r      s    0$
C$)'V [[.a($6a(%6t_dE*t,-tT:	

0
0r   r   )numpyrd   r8   pandasr   r   r   pandas._testing_testingr   r   rc   r   r   <module>r{      s&      
 o0 o0r   