
    Owg                     `    d dl m Z  d dlZd dlZd dlZd dlmZmZmZm	Z	 d dl
mZ  G d d      Zy)    )datetimeN)Index
MultiIndexSeriesarrayc                   l    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
 Zd Zd Zd Zd Zd Zd Zy)
TestRenamec                    |}d }|j                  |      }|j                  d    ||j                  d         k(  sJ t        t        |j                  |j                              }|j                  |      }t	        j
                  ||       y )Nc                 $    | j                  d      S Nz%Y%m%dstrftimexs    ^/var/www/horilla/myenv/lib/python3.12/site-packages/pandas/tests/series/methods/test_rename.py<lambda>z(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_renamezTestRename.test_rename   sw    0))G$}}Q7288A;#7777 3rxx7899[)
w1r   c                     t        t        j                  d      g dd      }|j                  ddd      }t	        j
                  |j                  t        g d             y )	N   abcdint64r   dtypefoobar)r'   r)   )r&   r-   r(   r.   )r   nparanger   r   assert_index_equalr   r   )r   serr   s      r   test_rename_partial_dictz#TestRename.test_rename_partial_dict   sF    RYYq\)=WM**5u56
gmmU3K-LMr   c                     t        t        j                  d      t        g dd      d      }|j	                  i       }|j
                  j                  |j
                  j                  k(  sJ y )Nr$   r%   namer5   r*   r+   )r   r/   r0   r   r   r   r5   )r   r   r   s      r   test_rename_retain_index_namez(TestRename.test_rename_retain_index_name"   sV    IIaL&: HPW
 ..$}}!!W]]%7%7777r   c                     t        t        d      d      }t        ddd      }|j                  |      }t        t        d      g dd      }t        j                  ||       y )	N   r-   r6   
      )      )r   r:   r;      r$   )r   r5   )r   ranger   r   r   )r   r2   r   resultexpecteds        r   test_rename_by_seriesz TestRename.test_rename_by_series*   sQ    U1XE*RB(G$%(*;%H
vx0r   c                    t        t        d      t        d            }dddt        ddd      d	fD ]  }|j	                  |      }|j
                  |k(  sJ |r?t        j                  |j                  j                  |j                  j                         n>t        j                  |j                  j                  |j                  j                         |j
                  J  y )
Nr$   abcdr   r-   {        ^@     r-   )r   r?   listr   r   r5   r   assert_extension_array_equalr   valuesassert_numpy_array_equal)r   using_infer_stringr2   r5   r@   s        r   test_rename_set_namezTestRename.test_rename_set_name1   s    U1XT&\2Cr2(>I 	$DZZ%F;;$&&&!//0C0CSYYEUEUV++FLL,?,?AQAQR88###	$r   c                    t        t        d      t        d            }dddt        ddd      d	fD ]  }|j	                  |d
       |j
                  |k(  sJ t        j                  g dt        j                        }|r8t        |d      }t        j                  |j                  j                  |       t        j                  |j                  j                  |        y )Nr>   abcrE   r-   rF   rG   rH   rI   rJ   Tinplace)r&   r'   r(   r,   zstring[pyarrow_numpy])r   r?   rK   r   r   r5   r/   r   object_r   rL   r   rM   rN   )r   rO   r2   r5   exps        r   test_rename_set_name_inplacez'TestRename.test_rename_set_name_inplace<   s    U1XT%[1Cr2(>I 	CDJJtTJ*88t###((?"**=C!C'>?//		0@0@#F++CII,<,<cB	Cr   c                     t        t        d            }|j                  i d       |j                  i d       t        j                  t
        d      5  |j                  i d       d d d        y # 1 sw Y   y xY w)Nr9   r   )axisr   zNo axis named 5match)r   r?   r   pytestraises
ValueError)r   r2   s     r   test_rename_axis_supportedz%TestRename.test_rename_axis_supportedH   sd    U1X

2A


2G
$]]:->? 	#JJrJ"	# 	# 	#s   A33A<c                     d } ||j                   d         }|j                  |d       |j                   d   |k(  sJ y )Nc                 $    | j                  d      S r   r   r   s    r   r   z0TestRename.test_rename_inplace.<locals>.<lambda>R   r   r   r   TrS   )r   r   )r   r   r   rA   s       r   test_rename_inplacezTestRename.test_rename_inplaceQ   sH    0?0034w5$$Q'8333r   c                 ~     G d d      } |       }t        g d      j                  |      }|j                  |u sJ y )Nc                       e Zd Zy)=TestRename.test_rename_with_custom_indexer.<locals>.MyIndexerN__name__
__module____qualname__ r   r   	MyIndexerrf   Z       r   rl   r<   r=   r>   r   r   r5   r   rl   ixr2   s       r   test_rename_with_custom_indexerz*TestRename.test_rename_with_custom_indexerX   s:    	 	 [Y&&r*xx2~~r   c                      G d d      } |       }t        g d      }|j                  |d       |j                  |u sJ y )Nc                       e Zd Zy)ETestRename.test_rename_with_custom_indexer_inplace.<locals>.MyIndexerNrg   rk   r   r   rl   ru   c   rm   r   rl   rn   TrS   ro   rp   s       r   'test_rename_with_custom_indexer_inplacez2TestRename.test_rename_with_custom_indexer_inplacea   s?    	 	 [Y

2t
$xx2~~r   c           	         t        t        dd      t        t        dd      d            }|j                  t              }|j                  d       }t        j                  ||       |j                  |j                  k(  sJ y )	Nr<      r=      IntIndexr6   rE   c                     t        |       S )N)str)is    r   r   z1TestRename.test_rename_callable.<locals>.<lambda>o   s
    A r   )r   r?   r   r   r|   r   r   r5   r   r2   r@   rA   s       r   test_rename_callablezTestRename.test_rename_callablek   sc    U1a[eAqk
(KLC::./
vx0{{hmm+++r   c                     t        ddgd      }|j                  d       }t        ddg      }t        j                  ||       y )Nr<   r=   r-   r6   )r   r   r   r   r~   s       r   test_rename_nonezTestRename.test_rename_nonet   s=    aV%(D!1a&>
vx0r   c                 `   g dg dg}t        j                  |ddg      }t        t        j                  d      |      }|j                  dd	idd
      }g dg dg}t        j                  |ddg      }t        t        j                  d      |      }t        j                  ||       y )N)r.   bazr   r-   qux)oner   twor   r   firstsecondnamesr9   rE   r   yesraise)r   levelerrors)r   r   r   r   r   )r   from_arraysr   r/   onesr   r   r   )r   arraysr   r2   r@   arrays_expectedindex_expectedseries_expecteds           r   "test_rename_series_with_multiindexz-TestRename.test_rename_series_with_multiindex{   s     0/

 &&vgx5HIRWWQZu-5%.Q 0/

 $//GX#6
 !>B
v7r   c                    t        g dd      j                  d      t        g dd      g}t        j                  |ddg      }t	        d|	      }|j                  dd
id      }t        g dd      j                  d      t        g dd      g}t        j                  |ddg      }t	        d|	      }t        j                  ||       y )Nrn   Int64rU   categoryABr   r<   rE   r$   )r   )r$   r=   r>   )r   astyper   r   r   r   r   r   )r   r   mir2   r@   r   mi_expectedrA   s           r   2test_rename_series_with_multiindex_keeps_ea_dtypesz=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!;/
vx0r   c                     t        ddg      }t        j                  d      }t        j                  t
        |      5  |j                  ddid       d d d        y # 1 sw Y   y xY w)	Nr-   r.   z[2] not found in axisr[   r=   	   r   )r   )r   reescaper]   r^   KeyErrorr   )r   r2   r\   s      r   test_rename_error_argz TestRename.test_rename_error_arg   sV    eU^$		12]]851 	/JJ1vgJ.	/ 	/ 	/s   AA&c                 4   t        ddg      }|j                         }|j                  ddid      }t        j                  |      5  d|d<   d d d        |r|d   |d   k(  sJ |d   |d   k(  sJ y |d   |d   k(  sJ |d   |d   k(  sJ y # 1 sw Y   AxY w)	Nr-   r.   r<   r   F)copyfoobarr   )r   r   r   r   assert_cow_warning)r   using_copy_on_writewarn_copy_on_writer2   ser_origshallow_copys         r   test_rename_copy_falsez!TestRename.test_rename_copy_false   s    eU^$88:zz1a&uz5""#56 	CF	A;,q/111A;,q/111q6\!_,,,q6\!_,,,	 	s   BBN)rh   ri   rj   r"   r3   r7   rB   rP   rX   r`   rc   rr   rv   r   r   r   r   r   r   rk   r   r   r	   r	      sT    	2N81	$
C#4,18.1&/-r   r	   )r   r   numpyr/   r]   pandasr   r   r   r   pandas._testing_testingr   r	   rk   r   r   <module>r      s,     	    h- h-r   