
    Eg                     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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         `/var/www/sysmax/venv/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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      )ABzUse `.rename`matchr   r   
      )r   r   r   )r   pytestraises
ValueErrorr   id)r   dfs     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)abcr   r   llnn)namesc                     g | ]}|d z  S )r"    ).0is     r   
<listcomp>zCTestDataFrameRenameAxis.test_rename_axis_mapper.<locals>.<listcomp>2   s    -M-M-Ma"f-M-M-Mr   )xy)indexcolsr   r6   r7   namenew)columnsr   r	   r   )r8   r   LLNNgoo)r=   mehr8   r=   z	Must passr    wrongzLength of namesbogus)rD   )r   from_productr   listrangelenr   r   assert_index_equalr=   r   r8   r0   strupperreset_index	set_indexdropr;   r$   r%   	TypeErrorr&   )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   r8   r=   rB   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   r2   )
r   rF   nparangereshaper   r   renamer   r   )r   kwargsrename_indexrename_columnsr8   r=   datar(   r   expected_indexexpected_columnsr   s               r   test_rename_axis_nonez-TestDataFrameRenameAxis.test_rename_axis_noneY   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)   rR   r$   markparametrizerd   r2   r   r   r   r      s        0 0 0$$ $ $)' )' )'V [.a(($6a((%6t_dE*t,--tT:	

 
0 0
 
0 0 0r   r   )numpyrZ   r$   pandasr   r   r   pandas._testing_testingr   r   r2   r   r   <module>rn      s                 
      c0 c0 c0 c0 c0 c0 c0 c0 c0 c0r   