
    d5                         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)methodcallerN)	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       [/var/www/html/t/fyr/venv311/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gg          }|                                sJ t          dgg          }|                                rJ t          ddgg          }d}t          j        t          |          5  |                                 d d d            n# 1 swxY w Y   t          j        t          |          5  t          |           d d d            d S # 1 swxY w Y   d S )NTFz+The truth value of a DataFrame is ambiguousmatch)r   boolpytestraises
ValueError)r   r   msgs      r   test_nonzero_single_elementz)TestDataFrame.test_nonzero_single_element0   s`   x  wwyyy!!7799'((;]:S111 	 	GGIII	 	 	 	 	 	 	 	 	 	 	 	 	 	 	]:S111 	 	HHH	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	s$   >BB#&B#C""C&)C&c                 "   t          g dg dt          j                            d          t          j                            d          d          }|                    d                                          }t          j        ||           d S )N)r   barr   r?   r   r?   r   r   )oner@   twothreerA   rA   r@   rB      )r"   Br*   Dr"   )r   r.   randomrandngroupbysumtmassert_metadata_equivalentr   r   r   s      r   'test_metadata_propagation_indiv_groupbyz5TestDataFrame.test_metadata_propagation_indiv_groupby?   s    MMMQQQY__Q''Y__Q''	 
 
 C$$&&
%b&11111r   c                     t          t          j                            dd          t	          ddd                    }|                    d          }t          j        ||           d S )Ni  r   20130101s)periodsfreq)r   1T)r   r.   rF   rG   r   resamplerJ   rK   rL   s      r   (test_metadata_propagation_indiv_resamplez6TestDataFrame.test_metadata_propagation_indiv_resampleL   sg    IOOD!$$ZC@@@
 
 
 T""
%b&11111r   c                 6   dd}|                                 5 }|                    t          ddg           |                    t          d|           t          j                            d           t          t          j                            ddd          d	d
g          }t          t          j                            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          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 )Nc           
         | j         D ]͊|dk    rQ|j        |j        }}t          |d          dz   t          |d          z   }t                              | |           Y|dk    rCd                    fd|j        D                       }t                              | |           t                              | t          |d                     | S )Nmerge |concat+c                 R    g | ]#}t          |d           t          |          $S N)getattr)r%   or   s     r   r'   zSTestDataFrame.test_metadata_propagation_indiv.<locals>.finalize.<locals>.<listcomp>a   s5    XXXaDRVAWAWXD))XXXr   )	_metadataleftrightr_   object__setattr__joinobjs)r   othermethodkwargsrb   rc   valuer   s          @r   finalizez?TestDataFrame.test_metadata_propagation_indiv.<locals>.finalizeY   s     M MW$$"'*ek%D#D$33c9GE4QS<T<TTE&&tT59999x''HHXXXX5:XXX E &&tT59999&&tT75$3K3KLLLLKr   ra   filename__finalize__
   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+foor^   )contextsetattrr   r.   rF   seedrandintra   rm   rX   r1   pdr[   )r   monkeypatchrl   mdf1df2r   s          r   test_metadata_propagation_indivz-TestDataFrame.test_metadata_propagation_indivU   s   	 	 	 	    "" 	0aIIizl;;;IIi:::INN2BI--aF;;c3ZPPPCBI--aF;;c3ZPPPC#-,I'CL'CLYYsSESEwYOOF?&===== BI--aF;;T$ZZPPPC CLYSz**F?i/////)	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0s   E'FFF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   rJ   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   rJ   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   r4   r=   rM   rU   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(0 (0 (0T
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 )	TestDataFrame2rk   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   )rp   rq   z8For argument "inplace" expected type bool, received typer6   r&   r   r   )mapperr   inplacerp   )r   r   r   )rk   r      )
to_replacerk   r   r   r   )condr   )r   r9   r:   r;   copyr   dropfillnareplaceinterpolate_whererp   mask)r   rk   r   r<   s       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   jimjoerr   r   r   zunexpected keywordr6   r   T)r   in_placer   )r   )r   r.   rF   rG   r~   Categoricalnanr   r9   r:   	TypeErrorr   reindexr   )r   r   catsr<   s        r   test_unexpected_keywordz&TestDataFrame2.test_unexpected_keyword   s   ryq!,,uenEEE^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'DDD3EEE=F""F&)F&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   numpyr.   r9   pandasr~   r   r   r   r   pandas._testing_testingrJ   r
   r   r#   r   r   <module>r      s          ! ! ! ! ! !                          @= @= @= @= @= @= @= @=H-( -( -( -( -( -( -( -( -( -(r   