
    Pi                     P    d dl Zd dlZd dlmZmZmZ d dlmZ	  G d d          Z
dS )    N)	DataFrameIndex
MultiIndexc                       e Zd Zd Zd Zd Zd Zej        	                    ddddd	d
fdddd
d	fddid	d
fddid
d	fdddd	d	fi d
d
fg          d             Z
dS )TestDataFrameRenameAxisc                    |                     d          }|                                }|                     dd          x}}|J |J t          j        ||           |                     dd          }|                                }|                     ddd          x}}|J |J t          j        ||           d S )NfooT)inplacebar   axis)r   r
   )rename_axiscopytmassert_frame_equal)selffloat_frameexpectedresultreturn_value	no_returns         /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/frame/methods/test_rename_axis.pytest_rename_axis_inplacez0TestDataFrameRenameAxis.test_rename_axis_inplace   s    **511!!###)#5#5eT#5#J#JJy###   
fh///**5q*99!!###)#5#5e!T#5#R#RRy###   
fh/////    c                    t          ddgddggddg                              d	          }|                    d
d          }t          ddgddggt          ddgd
          ddg          }t	          j        ||d           d S )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_falsezKTestDataFrameRenameAxis.test_rename_axis_with_allows_duplicate_labels_false   s    AA'#s<<<FF$) G 
 
 A..VaVE1a&u$=$=$=Sz
 
 
 	fhEBBBBBBr   c                    t          ddgddgd          }t          j        t          d          5  |                    t
          d           d d d            n# 1 swxY w Y   t          j        t          d          5  |                    dd	d
d           d d d            n# 1 swxY w Y   t          j        t          d          5  |                    t
          d           d d d            n# 1 swxY w Y   t          j        t          d          5  |d                             t
                     d d d            d S # 1 swxY w Y   d S )Nr   r   )ABzUse `.rename`matchr   r   
      )r   r   r/   )r   pytestraises
ValueErrorr   id)r   r,   s     r   test_rename_axis_raisesz/TestDataFrameRenameAxis.test_rename_axis_raises+   s<   aV1a&1122]:_=== 	' 	'NN2AN&&&	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' ]:_=== 	3 	3NNrb>>N222	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 ]:_=== 	' 	'NN2AN&&&	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' ]:_=== 	$ 	$sG###	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$sG   AA"AB((B,/B,C77C;>C;!EEEc           	         t          j        g dddggddg          }t          t          t	          t          |                              d t	          t          |                    D             d|	          }|                    d
d          }t          j        |j	        t          ddgd
                     |                    d
did          }t          j        |j	        t          ddgd                     |                    ddi	          }|j        j        ddgk    sJ |                    t          j        d          }|j        j        ddgk    sJ |                    ddg	          }|j        j        ddgk    sJ |                                                    d                              ddg          }|                    dd          }|j        j        dk    sJ |j	        j        dk    sJ t'          j        t*          d          5  |                    d	           d d d            n# 1 swxY w Y   t'          j        t,          d          5  |                    dg	           d d d            n# 1 swxY w Y   t'          j        t*          d          5  |                    d            d d d            d S # 1 swxY w Y   d S ) N)r    r!   cr   r   llnn)namesc                     g | ]}|d z  S )r3    ).0is     r   
<listcomp>zCTestDataFrameRenameAxis.test_rename_axis_mapper.<locals>.<listcomp>>   s    -M-M-Ma"f-M-M-Mr   )xy)r)   colsr   rD   rE   r&   new)r#   r   r	   r   )r)   r   LLNNgoor"   mehr(   z	Must passr1   wrongzLength of namesbogus)rM   )r   from_productr   listrangelenr   r   assert_index_equalr#   r   r)   r>   strupperreset_index	set_indexdropr'   r5   r6   	TypeErrorr7   )r   mir,   r   sdfs        r   test_rename_axis_mapperz/TestDataFrameRenameAxis.test_rename_axis_mapper:   s   $ooo1v%>tTlSSSuSWW~~&&-M-MeCGGnn-M-M-MNNVX
 
 

 Q//
fneS#JV.L.L.LMMM ##VUO!#DD
fneS#JU.K.K.KLLL tUm44|!eT]2222 cia88|!dD\1111 uen55|!eU^3333 nn((..33T3K3HHue<<| E))))~"e++++ ]9K888 	* 	*NNN)))	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* ]:->??? 	, 	,NN'N+++	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, ]9G444 	' 	'NNN&&&	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	's6   $III-JJJ7KK"Kz$kwargs, rename_index, rename_columnsNr   )mapperr   TFr   r)   r#   r(   c                    t          t          d          d          }t          ddgd          }t          j        d                              dd	          }t          |||          } |j        d
i |}|r|                    d           n|}	|r|                    d           n|}
t          ||	|
          }t          j	        ||           d S )Nabcr	   r&   col1col2r      r   r   r@   )
r   rO   nparangereshaper   r   renamer   r   )r   kwargsrename_indexrename_columnsr)   r#   datar,   r   expected_indexexpected_columnsr   s               r   test_rename_axis_nonez-TestDataFrameRenameAxis.test_rename_axis_nonee   s     d5kk...(u555y||##Aq))tUG,,))&))/;Fd+++3AN7>>$///wT>3CDD
fh/////r   )__name__
__module____qualname__r   r-   r9   r[   r5   markparametrizerl   r@   r   r   r   r      s        0 0 0$
C 
C 
C$ $ $)' )' )'V [.a(($6a((%6t_dE*t,--tT:	

 
0 0
 
0 0 0r   r   )numpyrb   r5   pandasr   r   r   pandas._testing_testingr   r   r@   r   r   <module>rv      s                 
      o0 o0 o0 o0 o0 o0 o0 o0 o0 o0r   