
    Mh                     `    S SK J r   S SKrS SKrS SKrS SKJrJrJrJ	r	  S SK
Jr   " S S5      rg)    )datetimeN)Index
MultiIndexSeriesarrayc                   t    \ rS rSrS rS rS rS rS rS r	S r
S	 rS
 rS rS rS rS rS rS rS rSrg)
TestRename   c                     UnS nUR                  U5      nUR                  S   U" UR                  S   5      :X  d   e[        [        UR                  UR                  5      5      nUR                  U5      n[        R
                  " XF5        g )Nc                 $    U R                  S5      $ Nz%Y%m%dstrftimexs    Y/var/www/html/env/lib/python3.13/site-packages/pandas/tests/series/methods/test_rename.py<lambda>(TestRename.test_rename.<locals>.<lambda>       AJJx0    r   )renameindexdictziptmassert_series_equal)selfdatetime_seriestsrenamerrenamedrename_dictrenamed2s          r   test_renameTestRename.test_rename   su    0))G$}}Q7288A;#7777 3rxx7899[)
w1r   c                     [        [        R                  " S5      / SQSS9nUR                  SSS.5      n[        R
                  " UR                  [        / SQ5      5        g )	N   abcdint64r   dtypefoobar)r*   r,   )r)   r0   r+   r1   )r   nparanger   r   assert_index_equalr   r   )r   serr!   s      r   test_rename_partial_dict#TestRename.test_rename_partial_dict   sF    RYYq\)=WM**5u56
gmmU3K-LMr   c                     [        [        R                  " S5      [        / SQSS9SS9nUR	                  0 5      nUR
                  R                  UR
                  R                  :X  d   eg )Nr'   r(   namer9   r-   r.   )r   r2   r3   r   r   r   r9   )r   r    r!   s      r   test_rename_retain_index_name(TestRename.test_rename_retain_index_name"   sV    IIaL&: HPW
 ..$}}!!W]]%7%7777r   c                     [        [        S5      SS9n[        SSS.5      nUR                  U5      n[        [        S5      / SQSS9n[        R                  " X45        g )	N   r0   r:   
      )      )r   r?   r@      r'   )r   r9   )r   ranger   r   r   )r   r5   r    resultexpecteds        r   test_rename_by_series TestRename.test_rename_by_series*   sO    U1XE*RB(G$%(*;%H
v0r   c                    [        [        S5      [        S5      S9nSSS[        SSS5      S	4 H  nUR	                  U5      nUR
                  U:X  d   eU(       a@  [        R                  " UR                  R                  UR                  R                  5        O?[        R                  " UR                  R                  UR                  R                  5        UR
                  c  M   e   g )
Nr'   abcdr   r0   {        ^@     r0   )r   rD   listr   r   r9   r   assert_extension_array_equalr   valuesassert_numpy_array_equal)r   using_infer_stringr5   r9   rE   s        r   test_rename_set_nameTestRename.test_rename_set_name1   s    U1XT&\2Cr2(>IDZZ%F;;$&&&!//0C0CSYYEUEUV++FLL,?,?AQAQR88### Jr   c                    [        [        S5      [        S5      S9nSSS[        SSS5      S	4 H  nUR	                  US
S9  UR
                  U:X  d   e[        R                  " / SQ[        R                  S9nU(       a7  [        USS9n[        R                  " UR                  R                  U5        M  [        R                  " UR                  R                  U5        M     g )NrC   abcrK   r0   rL   rM   rN   rO   rP   Tinplace)r)   r*   r+   r/   zstring[pyarrow_numpy])r   rD   rQ   r   r   r9   r2   r   object_r   rR   r   rS   rT   )r   rU   r5   r9   exps        r   test_rename_set_name_inplace'TestRename.test_rename_set_name_inplace<   s    U1XT%[1Cr2(>IDJJtTJ*88t###((?"**=C!C'>?//		0@0@#F++CII,<,<cB Jr   c                     [        [        S5      5      nUR                  0 SS9  UR                  0 SS9  [        R                  " [
        SS9   UR                  0 SS9  S S S 5        g ! , (       d  f       g = f)Nr>   r   )axisr   zNo axis named 5match)r   rD   r   pytestraises
ValueError)r   r5   s     r   test_rename_axis_supported%TestRename.test_rename_axis_supportedH   s\    U1X

2A


2G
$]]:->?JJrJ" @??s   A((
A6c                 ~    S nU" UR                   S   5      nUR                  USS9  UR                   S   U:X  d   eg )Nc                 $    U R                  S5      $ r   r   r   s    r   r   0TestRename.test_rename_inplace.<locals>.<lambda>R   r   r   r   TrZ   )r   r   )r   r   r    rF   s       r   test_rename_inplaceTestRename.test_rename_inplaceQ   sH    0?0034w5$$Q'8333r   c                      " S S5      nU" 5       n[        / SQ5      R                  U5      nUR                  UL d   eg )Nc                       \ rS rSrSrg)=TestRename.test_rename_with_custom_indexer.<locals>.MyIndexerZ    N__name__
__module____qualname____firstlineno____static_attributes__rs   r   r   	MyIndexerrq   Z       r   rz   rA   rB   rC   r   r   r9   r   rz   ixr5   s       r   test_rename_with_custom_indexer*TestRename.test_rename_with_custom_indexerX   s:    	 	 [Y&&r*xx2~~r   c                      " S S5      nU" 5       n[        / SQ5      nUR                  USS9  UR                  UL d   eg )Nc                       \ rS rSrSrg)ETestRename.test_rename_with_custom_indexer_inplace.<locals>.MyIndexerc   rs   Nrt   rs   r   r   rz   r   c   r{   r   rz   r|   TrZ   r}   r~   s       r   'test_rename_with_custom_indexer_inplace2TestRename.test_rename_with_custom_indexer_inplacea   s?    	 	 [Y

2t
$xx2~~r   c           	         [        [        SS5      [        [        SS5      SS9S9nUR                  [        5      nUR                  S 5      n[
        R                  " X#5        UR                  UR                  :X  d   eg )	NrA      rB      IntIndexr:   rK   c                     [        U 5      $ )N)str)is    r   r   1TestRename.test_rename_callable.<locals>.<lambda>o   s    Ar   )r   rD   r   r   r   r   r   r9   r   r5   rE   rF   s       r   test_rename_callableTestRename.test_rename_callablek   sa    U1a[eAqk
(KLC::./
v0{{hmm+++r   c                     [        SS/SS9nUR                  S 5      n[        SS/5      n[        R                  " X#5        g )NrA   rB   r0   r:   )r   r   r   r   r   s       r   test_rename_noneTestRename.test_rename_nonet   s;    aV%(D!1a&>
v0r   c                 J   / SQ/ SQ/n[         R                  " USS/S9n[        [        R                  " S5      US9nUR                  SS	0SS
S9n/ SQ/ SQ/n[         R                  " USS/S9n[        [        R                  " S5      US9n[        R                  " XG5        g )N)r1   bazr   r0   qux)oner   twor   r   firstsecondnamesr>   rK   r   yesraise)r   levelerrors)r   r   r   r   r   )r   from_arraysr   r2   onesr   r   r   )r   arraysr   r5   rE   arrays_expectedindex_expectedseries_expecteds           r   "test_rename_series_with_multiindex-TestRename.test_rename_series_with_multiindex{   s     0/

 &&vgx5HIRWWQZu-5%.Q 0/

 $//GX#6
 !>B
v7r   c                 t   [        / SQSS9R                  S5      [        / SQSS9/n[        R                  " USS/S9n[	        SUS	9nUR                  SS
0SS9n[        / SQSS9R                  S5      [        / SQSS9/n[        R                  " USS/S9n[	        SUS	9n[        R                  " XG5        g )Nr|   Int64r\   categoryABr   rA   rK   r'   )r   )r'   rB   rC   )r   astyper   r   r   r   r   r   )r   r   mir5   rE   r   mi_expectedrF   s           r   2test_rename_series_with_multiindex_keeps_ea_dtypes=TestRename.test_rename_series_with_multiindex_keeps_ea_dtypes   s     )7+22:>)7+
 ##F3*=Qb!QF!, )7+22:>)7+
 !,,_S#JO!;/
v0r   c                     [        SS/5      n[        R                  " S5      n[        R                  " [
        US9   UR                  SS0SS9  S S S 5        g ! , (       d  f       g = f)	Nr0   r1   z[2] not found in axisrc   rB   	   r   )r   )r   reescapere   rf   KeyErrorr   )r   r5   rd   s      r   test_rename_error_arg TestRename.test_rename_error_arg   sM    eU^$		12]]851JJ1vgJ. 211s   A
A'c                 L   [        SS/5      nUR                  5       nUR                  SS0SS9n[        R                  " U5         SUS'   S S S 5        U(       a  US   US   :X  d   eUS   US   :X  d   eg US   US   :X  d   eUS   US   :X  d   eg ! , (       d  f       NO= f)	Nr0   r1   rA   r   F)copyfoobarr   )r   r   r   r   assert_cow_warning)r   using_copy_on_writewarn_copy_on_writer5   ser_origshallow_copys         r   test_rename_copy_false!TestRename.test_rename_copy_false   s    eU^$88:zz1a&uz5""#56CF 7A;,q/111A;,q/111q6\!_,,,q6\!_,,, 76s   B
B#rs   N)ru   rv   rw   rx   r$   r6   r;   rG   rV   r_   rh   rm   r   r   r   r   r   r   r   r   ry   rs   r   r   r	   r	      sT    	2N81	$
C#4,18.1&/-r   r	   )r   r   numpyr2   re   pandasr   r   r   r   pandas._testing_testingr   r	   rs   r   r   <module>r      s,     	    h- h-r   