o
    Eh                     @   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 sJ |jjdks!J t|ddd|}|jjd u s2J |jjdks:J d S )N            fooZcolsaxis)r   r   indexnamecolumns)selfr	   dfresult r   ?lib/python3.10/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s1J |j	jd d gks;J t||d
d|}|j	jdd	gksNJ |jjd d gksXJ d S )N)r   r   c                 S      g | ]}d |fqS )Ar   .0xr   r   r   
<listcomp>"       z7TestDataFrame.test_set_axis_name_mi.<locals>.<listcomp>ZaBcc                 S   r   )Cr   r   r   r   r   r    #   r!   Zxyz)r   r   ZL1ZL2r   r   )
r   npemptyr   Z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   sR  d}t dgg}t dgg}tjt|d | sJ W d    n1 s&w   Y  tjt|d | r9J W d    n1 sCw   Y  t ddgg}d}tjt|d t| W d    n1 sgw   Y  tjt|d, tjt|d |  W d    n1 sw   Y  W d    d S W d    d S 1 sw   Y  d S )NzPDataFrame.bool is now deprecated and will be removed in future version of pandasTFmatchz+The truth value of a DataFrame is ambiguous)r   tmZassert_produces_warningFutureWarningboolpytestraises
ValueError)r   Zmsg_warnr   df1Zmsg_errr   r   r   test_nonzero_single_element0   s,   

"z)TestDataFrame.test_nonzero_single_elementc                 C   sT   t g dg dtjddtjddd}|d }t|| d S )N)r   barr   r2   r   r2   r   r   )oner3   twothreer4   r4   r3   r5   r      )r   Br"   Dr   )	r   r#   randomdefault_rngstandard_normalgroupbysumr*   assert_metadata_equivalentr   r   r   r   r   r   'test_metadata_propagation_indiv_groupbyG   s   z5TestDataFrame.test_metadata_propagation_indiv_groupbyc                 C   s>   t tjddtddddd}|d}t|| d S )	Nr   )  r   Z20130101rA   s)ZperiodsZfreq)r   Z1min)	r   r#   r9   r:   r;   r   Zresampler*   r>   r?   r   r   r   (test_metadata_propagation_indiv_resampleT   s   
z6TestDataFrame.test_metadata_propagation_indiv_resamplec                 C   s  ddd}|  x}|tddg |td| ttjdddd	d
dgd}ttjdddd	ddgd}dgt_d|_d|_|j	|d
gdgdd}|jdksWJ ttjdddd	t
dd}d|_t||g}|jdksyJ W d    d S 1 sw   Y  d S )Nc              	      s   | j D ]G |dkr&|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    i   s   " zSTestDataFrame.test_metadata_propagation_indiv.<locals>.finalize.<locals>.<listcomp>)	_metadataleftrightrJ   object__setattr__joinZobjs)r   othermethodkwargsrN   rO   valuer   rL   r   finalizea   s   
z?TestDataFrame.test_metadata_propagation_indiv.<locals>.finalizerM   filenameZ__finalize__r   r   r   )r   r   abr   cdz
fname1.csvz
fname2.csvinner)Zleft_onZright_onZhowzfname1.csv|fname2.csvabr   zfoo+foorI   )contextsetattrr   r#   r9   r:   ZintegersrM   rX   rD   r%   pdrG   )r   ZmonkeypatchrW   mr0   Zdf2r   r   r   r   test_metadata_propagation_indiv]   s,   

"z-TestDataFrame.test_metadata_propagation_indivc                 C   sV   t dg di}d|_g d|d< d|_|jdksJ t|d tg ddd d S )Nr   r   r   r   r   )r   r      y   rL   )r   rg   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'   r1   r@   rC   rd   ri   rk   r   r   r   r   r      s    

	/r   c                   @   s6   e Zd Zejdddg ddgdd Zdd	 Zd
S )TestDataFrame2rV   r   Truere   g      @c                 C   s  t g dg dd}d}tjt|d | jdddd|d	 W d    n1 s,w   Y  tjt|d | jd
d|d W d    n1 sMw   Y  tjt|d | jd|d W d    n1 smw   Y  tjt|d | jdd|d W d    n1 sw   Y  tjt|d | j	|d W d    n1 sw   Y  tjt|d | j
|jdk|d W d    n1 sw   Y  tjt|d | j|jdk|d W d    d S 1 sw   Y  d S )Nre   )r   rh   rf   )rY   rZ   z8For argument "inplace" expected type bool, received typer(   r   rg   r   )Zmapperr   inplacerY   )r   rs   r   )rV   rs      )Z
to_replacerV   rs   rs   r   )Zcondrs   )r   r-   r.   r/   copyr
   dropfillnareplaceZinterpolateZ_whererY   mask)r   rV   r   msgr   r   r   test_validate_bool_args   s.   "z&TestDataFrame2.test_validate_bool_argsc                 C   sL  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 sBw   Y  t	j
t|d	 |jd
dgdd W d    n1 sbw   Y  t	j
t|d	 |jddd W d    n1 sw   Y  t	j
t|d	 |jddd W d    d S 1 sw   Y  d S )Nr   )rh   r   ZjimZjoer[   r   r   zunexpected keywordr(   r   T)r   in_placeru   )r}   )r   r#   r9   r:   r;   rb   ZCategoricalnanrv   r-   r.   	TypeErrorrw   Zreindexrx   )r   r   catsr{   r   r   r   test_unexpected_keyword   s&   
"z&TestDataFrame2.test_unexpected_keywordN)rl   rm   rn   r-   ro   rp   r|   r   r   r   r   r   rq      s    
rq   )rv   r   operatorr   Znumpyr#   r-   Zpandasrb   r   r   r   r   Zpandas._testingZ_testingr*   r   rq   r   r   r   r   <module>   s     