
    d                     d    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 d dl	m
Z  G d d          ZdS )    )datetimeN)Index
MultiIndexSeriesc                   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    c/var/www/html/t/fyr/venv311/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r4   r)   r*   )r   r.   r/   r   r   r   r4   )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,   r5   
      )      )r   r9   r:      r#   )r   r4   )r   ranger   r   r   )r   r1   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                 6   t          t          d          t          d                    }dddt          ddd          d	fD ]V}|                    |          }|j        |k    sJ t          j        |j        j	        |j        j	                   |j        J Wd S )
Nr#   abcdr   r,   {        ^@     r,   )
r   r>   listr   r   r4   r   assert_numpy_array_equalr   values)r   r1   r4   r?   s       r   test_rename_set_namezTestRename.test_rename_set_name0   s    U1XXT&\\222Cr2(>(>I 	$ 	$DZZ%%F;$&&&&'(;SY=MNNN8####		$ 	$r   c                 X   t          t          d          t          d                    }dddt          ddd          d	fD ]g}|                    |d
           |j        |k    sJ t          j        g dt          j                  }t          j
        |j        j        |           hd S )Nr=   abcrD   r,   rE   rF   rG   rH   rI   Tinplace)r%   r&   r'   r+   )r   r>   rJ   r   r   r4   r.   arrayobject_r   rK   r   rL   )r   r1   r4   exps       r   test_rename_set_name_inplacez'TestRename.test_rename_set_name_inplace8   s    U1XXT%[[111Cr2(>(>I 	? 	?DJJtTJ***8t####(???"*===C'	(8#>>>>	? 	?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 )Nr8   r   )axisr   zNo axis named 5match)r   r>   r   pytestraises
ValueError)r   r1   s     r   test_rename_axis_supportedz%TestRename.test_rename_axis_supportedA   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>K   r   r   r   TrP   )r   r   )r   r   r   r@   s       r   test_rename_inplacezTestRename.test_rename_inplaceJ   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   	MyIndexerrd   S           Dr   rj   r;   r<   r=   r   r   r4   r   rj   ixr1   s       r   test_rename_with_custom_indexerz*TestRename.test_rename_with_custom_indexerQ   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>.MyIndexerNre   ri   r   r   rj   rs   \   rk   r   rj   rl   TrP   rm   rn   s       r   'test_rename_with_custom_indexer_inplacez2TestRename.test_rename_with_custom_indexer_inplaceZ   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<      IntIndexr5   rD   c                      t          |           S )N)str)is    r   r   z1TestRename.test_rename_callable.<locals>.<lambda>h   s    A r   )r   r>   r   r   rz   r   r   r4   r   r1   r?   r@   s       r   test_rename_callablezTestRename.test_rename_callabled   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,   r5   )r   r   r   r   r|   s       r   test_rename_nonezTestRename.test_rename_nonem   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namesr8   rD   r   yesraise)r   levelerrors)r   r   r   r   r   )r   from_arraysr   r.   onesr   r   r   )r   arraysr   r1   r?   arrays_expectedindex_expectedseries_expecteds           r   "test_rename_series_with_multiindexz-TestRename.test_rename_series_with_multiindext   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 )Nrl   Int64rR   categoryABr   r;   rD   r#   )r   )r#   r<   r=   )r   astyper   r   r   r   r   r   )r   r   mir1   r?   r   mi_expectedr@   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 axisrY   r<   	   r   )r   )r   reescaper[   r\   KeyErrorr   )r   r1   rZ   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                 4   t          ddg          }|                                }|                    ddid          }d|d<   |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   using_copy_on_writer1   ser_origshallow_copys        r   test_rename_copy_falsez!TestRename.test_rename_copy_false   s    eU^$$88::zz1a&uz55A 	-A;,q/1111A;,q/111111q6\!_,,,,q6\!_,,,,,,r   N)rf   rg   rh   r!   r2   r6   rA   rM   rV   r^   ra   rp   rt   r}   r   r   r   r   r   ri   r   r   r   r      s       	2 	2 	2N N N8 8 81 1 1$ $ $? ? ?# # #4 4 4    , , ,1 1 18 8 8.1 1 1&/ / /- - - - -r   r   )r   r   numpyr.   r[   pandasr   r   r   pandas._testing_testingr   r   ri   r   r   <module>r      s          				              
      a- a- a- a- a- a- a- a- a- a-r   