
    bMh                     h    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dS )    )datetimeN)Index
MultiIndexSeriesarrayc                   n    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dS )
TestRenamec                 *   |}d }|                     |          }|j        d          ||j        d                   k    sJ t          t          |j        |j                            }|                     |          }t	          j        ||           d S )Nc                 ,    |                      d          S Nz%Y%m%dstrftimexs    g/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/tests/series/methods/test_rename.py<lambda>z(TestRename.test_rename.<locals>.<lambda>       AJJx0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   s    00))G$$}Q7728A;#7#77777 3rx778899[))
w11111r   c                     t          t          j        d          g dd          }|                    ddd          }t	          j        |j        t          g d                     d S )	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   sh    RYq\\)=)=)=WMMM**5u5566
gmU3K3K3K-L-LMMMMMr   c                     t          t          j        d          t          g dd          d          }|                    i           }|j        j        |j        j        k    sJ d S )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"   sn    IaLL&:&:&: H H HPW
 
 
 ..$$}!W]%7777777r   c                     t          t          d          d          }t          ddd          }|                    |          }t          t          d          g dd          }t          j        ||           d S )	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*   sx    U1XXE***RB((G$$%((*;*;*;%HHH
vx00000r   c                    t          t          d          t          d                    }dddt          ddd          d	fD ]}|                    |          }|j        |k    sJ |r*t          j        |j        j	        |j        j	                   n)t          j
        |j        j	        |j        j	                   |j        J d S )
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    U1XXT&\\222Cr2(>(>I 	$ 	$DZZ%%F;$&&&&! S/0CSYEUVVVV+FL,?AQRRR8####	$ 	$r   c                    t          t          d          t          d                    }dddt          ddd          d	fD ]}|                    |d
           |j        |k    sJ t          j        g dt          j                  }|r1t          |d          }t          j
        |j        j        |           {t          j        |j        j        |           d S )Nr>   abcrE   r-   rF   rG   rH   rI   rJ   Tinplace)r&   r'   r(   r,   str)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    U1XXT%[[111Cr2(>(>I 	C 	CDJJtTJ***8t####(???"*===C! CCu---/	0@#FFFF+CI,<cBBBB	C 	Cr   c                 0   t          t          d                    }|                    i d           |                    i d           t          j        t
          d          5  |                    i d           d d d            d S # 1 swxY w Y   d S )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   s    U1XX

2A


2G
$$$]:->??? 	# 	#JJrJ"""	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	#s   &BBBc                     d } ||j         d                   }|                    |d           |j         d         |k    sJ d S )Nc                 ,    |                      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   sY    007?0344w555$Q'8333333r   c                      G d d          } |            }t          g d                              |          }|j        |u sJ d S )Nc                       e Zd ZdS )=TestRename.test_rename_with_custom_indexer.<locals>.MyIndexerN__name__
__module____qualname__ r   r   	MyIndexerrg   Z           Dr   rm   r<   r=   r>   r   r   r5   r   rm   ixr2   s       r   test_rename_with_custom_indexerz*TestRename.test_rename_with_custom_indexerX   si    	 	 	 	 	 	 	 	 Y[[YYY&&r**x2~~~~~~r   c                      G d d          } |            }t          g d          }|                    |d           |j        |u sJ d S )Nc                       e Zd ZdS )ETestRename.test_rename_with_custom_indexer_inplace.<locals>.MyIndexerNrh   rl   r   r   rm   rv   c   rn   r   rm   ro   TrS   rp   rq   s       r   'test_rename_with_custom_indexer_inplacez2TestRename.test_rename_with_custom_indexer_inplacea   sq    	 	 	 	 	 	 	 	 Y[[YYY

2t
$$$x2~~~~~~r   c           	      ,   t          t          dd          t          t          dd          d                    }|                    t                    }|                    d           }t          j        ||           |j        |j        k    sJ d S )	Nr<      r=      IntIndexr6   rE   c                      t          |           S )N)rV   )is    r   r   z1TestRename.test_rename_callable.<locals>.<lambda>o   s    A r   )r   r?   r   r   rV   r   r   r5   r   r2   r@   rA   s       r   test_rename_callablezTestRename.test_rename_callablek   s    U1a[[eAqkk
(K(K(KLLLC::..//
vx000{hm++++++r   c                     t          ddgd          }|                    d           }t          ddg          }t          j        ||           d S )Nr<   r=   r-   r6   )r   r   r   r   r~   s       r   test_rename_nonezTestRename.test_rename_nonet   sQ    aV%(((D!!1a&>>
vx00000r   c                 p   g dg dg}t          j        |ddg          }t          t          j        d          |          }|                    dd	idd
          }g dg dg}t          j        |ddg          }t          t          j        d          |          }t          j        ||           d S )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IIRWQZZu---5%.QQ 0/////

 $/GX#6
 
 
 !>BBB
v77777r   c                    t          g dd                              d          t          g dd          g}t          j        |ddg          }t	          d|	          }|                    dd
id          }t          g dd                              d          t          g dd          g}t          j        |ddg          }t	          d|	          }t          j        ||           d S )Nro   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OO!;///
vx00000r   c                     t          ddg          }t          j        d          }t          j        t
          |          5  |                    ddid           d d d            d S # 1 swxY w Y   d S )	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   s    eU^$$	122]85111 	/ 	/JJ1vgJ...	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/s   A((A,/A,c                    t          ddg          }|                                }|                    ddid          }t          j        |          5  d|d<   d d d            n# 1 swxY w Y   |r*|d         |d         k    sJ |d         |d         k    sJ d S |d         |d         k    sJ |d         |d         k    sJ d S )	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"#566 	 	CF	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	-A;,q/1111A;,q/111111q6\!_,,,,q6\!_,,,,,,s   A%%A),A)N)ri   rj   rk   r"   r3   r7   rB   rP   rY   ra   rd   rs   rw   r   r   r   r   r   r   rl   r   r   r	   r	      s
       	2 	2 	2N N N8 8 81 1 1	$ 	$ 	$
C 
C 
C# # #4 4 4    , , ,1 1 18 8 8.1 1 1&/ / /- - - - -r   r	   )r   r   numpyr/   r^   pandasr   r   r   r   pandas._testing_testingr   r	   rl   r   r   <module>r      s          				                      h- h- h- h- h- h- h- h- h- h-r   