a
    j=ic                     @   st   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 dZG dd dZdS )	    )deepcopy)methodcallerN)	DataFrame
MultiIndexSeries
date_rangec                   @   sp   e Zd Zejdddgdd Zejdddgdd Zdd	 Zd
d Z	dd Z
dd Zdd Zdd ZdS )TestDataFramefuncZ_set_axis_namerename_axisc                 C   sx   t ddgddgg}t|d|}|jjd u s2J |jjdksBJ t|ddd|}|jjd u sdJ |jjdkstJ d S )N            foocolsaxis)r   r   indexnamecolumns)selfr	   dfresult r   p/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/pandas/tests/generic/test_frame.pytest_set_axis_name   s    z TestDataFrame.test_set_axis_namec                 C   s   t tdtdd tdD tdd tdD d}dd	g}t|||}|jj|ksbJ |j	jd d gksvJ t||d
d|}|j	jdd	gksJ |jjd d gksJ d S )N)r   r   c                 S   s   g | ]}d |fqS )Ar   .0xr   r   r   
<listcomp>"       z7TestDataFrame.test_set_axis_name_mi.<locals>.<listcomp>ZaBcc                 S   s   g | ]}d |fqS )Cr   r   r   r   r   r    #   r!   Zxyz)r   r   ZL1ZL2r   r   )
r   npemptyr   from_tupleslistr   r   namesr   )r   r	   r   Zlevel_namesr   r   r   r   test_set_axis_name_mi   s    z#TestDataFrame.test_set_axis_name_mic                 C   s   t dgg}| sJ t dgg}| r0J t ddgg}d}tjt|d |  W d    n1 sn0    Y  tjt|d t| W d    n1 s0    Y  d S )NTFz+The truth value of a DataFrame is ambiguousmatch)r   boolpytestraises
ValueError)r   r   msgr   r   r   test_nonzero_single_element0   s    &z)TestDataFrame.test_nonzero_single_elementc                 C   sz   t g dg dtjdtjdd}d}tjt|d |d }W d    n1 s`0    Y  t	|| d S )N)r   barr   r1   r   r1   r   r   )oner2   twothreer3   r3   r2   r4      )r   Br"   Dz!The default value of numeric_onlyr)   r   )
r   r#   randomrandntmZassert_produces_warningFutureWarninggroupbysumassert_metadata_equivalent)r   r   r/   r   r   r   r   'test_metadata_propagation_indiv_groupby@   s    

,z5TestDataFrame.test_metadata_propagation_indiv_groupbyc                 C   s:   t tjddtddddd}|d}t|| d S )Ni  r   Z20130101s)Zperiodsfreq)r   Z1T)r   r#   r8   r9   r   Zresampler:   r>   )r   r   r   r   r   r   (test_metadata_propagation_indiv_resampleO   s    
z6TestDataFrame.test_metadata_propagation_indiv_resamplec                 C   s  ddd}|  }|tddg |td| tjd ttjddd	d
dgd}ttjddd	ddgd}dgt_d|_d|_|j	|d
gdgdd}|jdksJ ttjddd	t
dd}d|_t||g}|jdksJ W d    n1 s0    Y  d S )Nc              	      s   | j D ] |dkrL|j|j }}t| dd t| d }t|  | q|dkr~d fdd|jD }t|  | qt|  t| d q| S )Nmerge |concat+c                    s"   g | ]}t | d rt | qS )N)getattr)r   or   r   r   r    e   r!   zSTestDataFrame.test_metadata_propagation_indiv.<locals>.finalize.<locals>.<listcomp>)	_metadataleftrightrH   object__setattr__joinZobjs)r   othermethodkwargsrL   rM   valuer   rJ   r   finalize\   s    
z?TestDataFrame.test_metadata_propagation_indiv.<locals>.finalizerK   filenameZ__finalize__
   r   r   )r   r   abr   cdz
fname1.csvz
fname2.csvinner)Zleft_onZright_onhowzfname1.csv|fname2.csvabr   zfoo+foo)N)contextsetattrr   r#   r8   seedrandintrK   rV   rC   r&   pdrF   )r   ZmonkeypatchrU   mZdf1Zdf2r   r   r   r   test_metadata_propagation_indivX   s     

z-TestDataFrame.test_metadata_propagation_indivc                 C   sV   t dg di}d|_g d|d< d|_|jdks6J t|d tg ddd d S )Nr   r   r   r   r   )r   r      y   rJ   )r   ri   r:   Zassert_series_equalr   )r   r   r   r   r   test_set_attribute   s    z TestDataFrame.test_set_attributec                 C   s(   t g g dgd}t|}t|| d S )Nr   )datar   r   )r   r   r:   Zassert_frame_equal)r   Zempty_frameZempty_frame_copyr   r   r   test_deepcopy_empty   s    z!TestDataFrame.test_deepcopy_emptyN)__name__
__module____qualname__r,   markparametrizer   r(   r0   r?   rB   rf   rk   rm   r   r   r   r   r      s   

	+r   c                       s>   e Zd Zejdddg ddg fddZdd	 Z  ZS )
TestDataFrame2rT   r   Truerg   g      @c                    s
  t g dg dd}d}tjt|d, tt |jdddd|d	 W d    n1 sZ0    Y  tjt|d& tt |jd
d|d W d    n1 s0    Y  tjt|d$ tt |jd|d W d    n1 s0    Y  tjt|d& tt |jdd|d W d    n1 s&0    Y  tjt|d" tt |j	|d W d    n1 sh0    Y  tjt|d* tt |j
|jdk|d W d    n1 s0    Y  tjt|d* tt |j|jdk|d W d    n1 s0    Y  d S )Nrg   )r   rj   rh   )rX   rY   z8For argument "inplace" expected type bool, received typer)   r   ri   r   )Zmapperr   inplacerX   )r   ru   r   )rT   ru      )Z
to_replacerT   ru   ru   r   )Zcondru   )r   r,   r-   r.   superr
   dropfillnareplaceinterpolateZ_whererX   mask)r   rT   r   r/   	__class__r   r   test_validate_bool_args   s$    
$4262:z&TestDataFrame2.test_validate_bool_argsc                 C   sH  t tjddddgd}tdddddtjg}|d  }tj|d< d}tj	t
|d	  |jdd
dd W d    n1 s0    Y  tj	t
|d	" |jd
dgdd W d    n1 s0    Y  tj	t
|d	 |jddd W d    n1 s0    Y  tj	t
|d	 |jddd W d    n1 s:0    Y  d S )Nrj   r   ZjimZjoerZ   r   r   zunexpected keywordr)   r   T)r   in_placerw   )r   )r   r#   r8   r9   rd   ZCategoricalnancopyr,   r-   	TypeErrorry   Zreindexrz   )r   r   catsr/   r   r   r   test_unexpected_keyword   s    
.0,z&TestDataFrame2.test_unexpected_keyword)	rn   ro   rp   r,   rq   rr   r   r   __classcell__r   r   r~   r   rs      s   rs   )r   r   operatorr   numpyr#   r,   Zpandasrd   r   r   r   r   Zpandas._testingZ_testingr:   r   rs   r   r   r   r   <module>   s    	