
    Pi                     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        d                    }|                     |          }t	          j        ||           d S )Nc                 ,    |                      d          S Nz%Y%m%dstrftimexs    {/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/series/methods/test_rename.py<lambda>z(TestRename.test_rename.<locals>.<lambda>       AJJx00     r   T)strict)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tDDDEE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r6   r+   r,   )r   r0   r1   r   r   r   r6   )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.   r7   
      )      )r   r;   r<      r%   )r   r6   )r   ranger   r   r   )r   r3   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   r6   r   assert_extension_array_equalr   valuesassert_numpy_array_equal)r   using_infer_stringr3   r6   rA   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?   abcrF   r.   rG   rH   rI   rJ   rK   Tinplace)r'   r(   r)   r-   str)r   r@   rL   r   r   r6   r0   r   object_r   rM   r   rN   rO   )r   rP   r3   r6   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 )Nr:   r   )axisr   zNo axis named 5match)r   r@   r   pytestraises
ValueError)r   r3   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   TrT   )r   r   )r   r   r   rB   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   	MyIndexerrh   Z           Dr   rn   r=   r>   r?   r   r   r6   r   rn   ixr3   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>.MyIndexerNri   rm   r   r   rn   rw   c   ro   r   rn   rp   TrT   rq   rr   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>      IntIndexr7   rF   c                      t          |           S )N)rW   )is    r   r   z1TestRename.test_rename_callable.<locals>.<lambda>o   s    A r   )r   r@   r   r   rW   r   r   r6   r   r3   rA   rB   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.   r7   )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namesr:   rF   r   yesraise)r   levelerrors)r   r   r   r   r   )r   from_arraysr   r0   onesr   r   r   )r   arraysr   r3   rA   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 )Nrp   Int64rV   categoryABr   r=   rF   r%   )r   )r%   r>   r?   )r   astyper   r   r   r   r   r   )r   r   mir3   rA   r   mi_expectedrB   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   r3   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|d<   |d         |d         k    sJ |d         |d         k    sJ d S )Nr.   r/   r=   r   foobarr   )r   copyr   )r   r3   ser_origshallow_copys       r   test_rename_copy_falsez!TestRename.test_rename_copy_false   sv    eU^$$88::zz1a&))A{l1o----{l1o------r   N)rj   rk   rl   r#   r4   r8   rC   rQ   rZ   rb   re   rt   rx   r   r   r   r   r   r   rm   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   numpyr0   r_   pandasr   r   r   r   pandas._testing_testingr   r	   rm   r   r   <module>r      s          				                      c. c. c. c. c. c. c. c. c. c.r   