
    Pi6                         d dl mZ d dlmZ 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 G d d	          ZdS )
    )deepcopy)methodcaller)LiteralN)	DataFrame
MultiIndexSeries
date_rangec                       e Zd Zej                            dddg          d             Zej                            dddg          d             Zd Zd Z	d Z
d	 Zd
 Zd ZdS )TestDataFramefunc_set_axis_namerename_axisc                    t          ddgddgg          } t          |d          |          }|j        j        J |j        j        dk    sJ  t          |dd          |          }|j        j        J |j        j        dk    sJ d S )N            foocolsaxis)r   r   indexnamecolumns)selfr   dfresults       s/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/generic/test_frame.pytest_set_axis_namez TestDataFrame.test_set_axis_name   s    AA'((*dE**2..x}$$$| E))))3dF333B77z&&&~"f,,,,,,    c           
         t          t          j        d          t          j        d t          d          D                       t          j        d t          d          D                                 }ddg} t          ||          |          }|j        j        |k    sJ |j	        j        d d gk    sJ  t          ||d	
          |          }|j	        j        ddgk    sJ |j        j        d d gk    sJ d S )N)r   r   c                     g | ]}d |fS )A .0xs     r   
<listcomp>z7TestDataFrame.test_set_axis_name_mi.<locals>.<listcomp>#   s    )H)H)Hq3()H)H)Hr    aBcc                     g | ]}d |fS )Cr$   r%   s     r   r(   z7TestDataFrame.test_set_axis_name_mi.<locals>.<listcomp>$   s    +J+J+JS!H+J+J+Jr    xyz)r   r   L1L2r   r   )
r   npemptyr   from_tupleslistr   r   namesr   )r   r   r   level_namesr   s        r   test_set_axis_name_miz#TestDataFrame.test_set_axis_name_mi   s   HV()H)HDKK)H)H)HII*+J+Jd5kk+J+J+JKK
 
 
 Tl0dK0044|![0000~#d|33338dKa888<<~#d|3333|!dD\111111r    c                     t          ddgg          }d}t          j        t          |          5  t	          |           d d d            d S # 1 swxY w Y   d S )NFz+The truth value of a DataFrame is ambiguousmatch)r   pytestraises
ValueErrorbool)r   r   msg_errs      r   test_nonzero_single_elementz)TestDataFrame.test_nonzero_single_element1   s    '((?]:W555 	 	HHH	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	s   AAAc                 n   t          g dg dt          j                            d                              d          t          j                            d                              d          d          }|                    d                                          }t          j        ||           d S )N)r   barr   r@   r   r@   r   r   )onerA   twothreerB   rB   rA   rC   r      )r#   Br+   Dr#   )	r   r/   randomdefault_rngstandard_normalgroupbysumtmassert_metadata_equivalentr   r   r   s      r   'test_metadata_propagation_indiv_groupbyz5TestDataFrame.test_metadata_propagation_indiv_groupby7   s    MMMQQQY**1--==a@@Y**1--==a@@	 
 
 C$$&&
%b&11111r    c                     t          t          j                            d                              d          t          ddd                    }|                    d          }t          j        ||           d S )	Nr   )  r   20130101rQ   s)periodsfreq)r   1min)	r   r/   rG   rH   rI   r	   resamplerL   rM   rN   s      r   (test_metadata_propagation_indiv_resamplez6TestDataFrame.test_metadata_propagation_indiv_resampleD   su    I!!!$$44Y??ZC@@@
 
 
 V$$
%b&11111r    c                    	 ddt           dt           dt          d         d z  fd}|                                5 }|                    t           ddg           |                    t           d|           t          t          j                            d	                              d
dd          ddg          }t          t          j                            d	                              d
dd          ddg          }dgt           _        d|_	        d|_	        |
                    |dgdgd          }|j	        dk    sJ t          t          j                            d	                              d
dd          t          d                    }d|_	        t          j        ||g          }|j	        dk    sJ 	 d d d            d S # 1 swxY w Y   d S )Nr   othermethod)mergeconcatc           
         | j         D ]Ɋ|dk    rM|j        \  }}t          |d          dz   t          |d          z   }t                              | |           U|dk    rCd                    fd|j        D                       }t                              | |           t                              | t          |d                     | S )Nr\    |r]   +c                 R    g | ]#}t          |d           t          |          $S N)getattr)r&   or   s     r   r(   zSTestDataFrame.test_metadata_propagation_indiv.<locals>.finalize.<locals>.<listcomp>^   sE        !&q$55#At,,  r    )	_metadata
input_objsrd   object__setattr__join)r   rZ   r[   kwargsleftrightvaluer   s          @r   finalizez?TestDataFrame.test_metadata_propagation_indiv.<locals>.finalizeQ   s     M MW$$"'"2KD%#D$33c9GE4QS<T<TTE&&tT59999x''HH   %*%5   E &&tT59999&&tT75$3K3KLLLLKr    rf   filename__finalize__r   r   r   )r   r   abr   cdz
fname1.csvz
fname2.csvinner)left_onright_onhowzfname1.csv|fname2.csvabr   zfoo+foorc   )r   r   contextsetattrr/   rG   rH   integersrf   rp   r\   r2   pdr]   )r   monkeypatchro   mdf1df2r   s          r   test_metadata_propagation_indivz-TestDataFrame.test_metadata_propagation_indivM   s(    9=	 			 -.5	 	 	 	2   "" 	0aIIizl;;;IIi:::	%%a((11!Q??#s  C 	%%a((11!Q??#s  C $.,I'CL'CLYYsSESEwYOOF?&===== 	%%a((11!Q??d  C !CLYSz**F?i/////3	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0s   FGGGc                     t          dg di          }d|_        g d|d<   d|_        |j        dk    sJ t          j        |d         t	          g dd                     d S )Nr'   r   r   r   r   )r   r      y   )r   )r   r   rL   assert_series_equalr   )r   r   s     r   test_set_attributez TestDataFrame.test_set_attribute   sr     YYY'(())3tqyyyy
r#wyyys(C(C(CDDDDDr    c                 t    t          g g dg          }t          |          }t          j        ||           d S )Nr#   )datar   r   )r   r   rL   assert_frame_equal)r   empty_frameempty_frame_copys      r   test_deepcopy_emptyz!TestDataFrame.test_deepcopy_empty   sB      RrC5AAA#K00
.<<<<<r    N)__name__
__module____qualname__r9   markparametrizer   r5   r>   rO   rX   r   r   r   r$   r    r   r   r      s        [V&6%FGG	- 	- HG	- [V&6%FGG2 2 HG2"  2 2 22 2 260 60 60p
E 
E 
E= = = = =r    r   c                   h    e Zd Zej                            dddg ddg          d             Zd ZdS )	TestDataFrame2rn   r   Truer   g      @c                 n   t          g dg dd          }d}t          j        t          |          5  |                                                    dddd|	           d d d            n# 1 swxY w Y   t          j        t          |          5  |                                                    d
d|           d d d            n# 1 swxY w Y   t          j        t          |          5  |                                                    d|           d d d            n# 1 swxY w Y   t          j        t          |          5  |                                                    dd|           d d d            n# 1 swxY w Y   t          j        t          |          5  |                                	                    |           d d d            n# 1 swxY w Y   t          j        t          |          5  |                                
                    |j        dk    |           d d d            n# 1 swxY w Y   t          j        t          |          5  |                                                    |j        dk    |           d d d            d S # 1 swxY w Y   d S )Nr   )r   r   r   )rr   rs   z8For argument "inplace" expected type bool, received typer7   r'   r   r   )mapperr   inplacerr   )r   r   r   )rn   r      )
to_replacern   r   r   r   )condr   )r   r9   r:   r;   copyr   dropfillnareplaceinterpolate_whererr   mask)r   rn   r   msgs       r   test_validate_bool_argsz&TestDataFrame2.test_validate_bool_args   s@   YYYYYY7788H]:S111 	V 	VGGII!!s)=)=Au!UUU	V 	V 	V 	V 	V 	V 	V 	V 	V 	V 	V 	V 	V 	V 	V ]:S111 	7 	7GGIINN3QN666	7 	7 	7 	7 	7 	7 	7 	7 	7 	7 	7 	7 	7 	7 	7 ]:S111 	5 	5GGII1e444	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 ]:S111 	D 	DGGII!UCCC	D 	D 	D 	D 	D 	D 	D 	D 	D 	D 	D 	D 	D 	D 	D ]:S111 	1 	1GGII!!%!000	1 	1 	1 	1 	1 	1 	1 	1 	1 	1 	1 	1 	1 	1 	1 ]:S111 	; 	;GGII"$(E:::	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; ]:S111 	9 	9GGIINNq%N888	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9s}   .A..A25A2+CCC1*D''D+.D++FFF*)GG#&G#3III*3J**J.1J.c                    t          t          j                            d                              d          ddg          }t          j        dddddt          j        g          }|d                                         }t          j        |d<   d}t          j
        t          |	          5  |                    dd
d           d d d            n# 1 swxY w Y   t          j
        t          |	          5  |                    d
dgd           d d d            n# 1 swxY w Y   t          j
        t          |	          5  |                    dd           d d d            n# 1 swxY w Y   t          j
        t          |	          5  |                    dd           d d d            d S # 1 swxY w Y   d S )Nr   )r   r   jimjoert   r   r   zunexpected keywordr7   r   T)r   in_placer   )r   )r   r/   rG   rH   rI   r   Categoricalnanr   r9   r:   	TypeErrorr   reindexr   )r   r   catsr   s        r   test_unexpected_keywordz&TestDataFrame2.test_unexpected_keyword   s   I!!!$$44V<<uen
 
 
 ^Q1aBF344Y^^1"]9C000 	2 	2GGEDG111	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 ]9C000 	- 	-JJ1vtJ,,,	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- ]9C000 	' 	'IIaI&&&	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' ]9C000 	( 	(IIa$I'''	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	(sH   .CCC9DD#&D#E))E-0E-F44F8;F8N)r   r   r   r9   r   r   r   r   r$   r    r   r   r      s]        [Wq&)))S&ABB9 9 CB92( ( ( ( (r    r   )r   r   operatorr   typingr   numpyr/   r9   pandasr   r   r   r   r	   pandas._testing_testingrL   r   r   r$   r    r   <module>r      s         ! ! ! ! ! !                                E= E= E= E= E= E= E= E=R/( /( /( /( /( /( /( /( /( /(r    