o
    d5                     @   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            foocolsaxis)r   r   indexnamecolumns)selfr	   dfresult r   S/app/.heroku/python/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   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}| rJ t ddgg}d}tjt|d |  W d    n1 s7w   Y  tjt|d t| W d    d S 1 sSw   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   sH   t g dg dtjdtjdd}|d }t|| d S )N)r   barr   r2   r   r2   r   r   )oner3   twothreer4   r4   r3   r5      )r   Br#   Dr   )r   r$   randomrandngroupbysumtmassert_metadata_equivalent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$   r9   r:   r   Zresampler=   r>   r?   r   r   r   (test_metadata_propagation_indiv_resampleL   s   
z6TestDataFrame.test_metadata_propagation_indiv_resamplec                 C   s  ddd}|  u}|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sWJ ttjddd	t
dd}d|_t||g}|jdksvJ 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!   a   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   finalizeY   s   
z?TestDataFrame.test_metadata_propagation_indiv.<locals>.finalizerM   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+foorI   )contextsetattrr   r$   r9   seedrandintrM   rX   rD   r'   pdrG   )r   ZmonkeypatchrW   mZdf1Zdf2r   r   r   r   test_metadata_propagation_indivU   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   rk   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   rh   rm   ro   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   Trueri   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 )Nri   )r   rl   rj   )rZ   r[   z8For argument "inplace" expected type bool, received typer*   r    rk   r   )Zmapperr   inplacerZ   )r   rw   r   )rV   rw      )Z
to_replacerV   rw   rw   r   )Zcondrw   )r   r-   r.   r/   copyr
   dropfillnareplaceZinterpolateZ_whererZ   mask)r   rV   r   r0   r   r   r   test_validate_bool_args   s.   "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 s@w   Y  tj	t
|d	 |jd
dgdd W d    n1 s`w   Y  tj	t
|d	 |jddd W d    n1 s~w   Y  tj	t
|d	 |jddd W d    d S 1 sw   Y  d S )Nrl   r   ZjimZjoer\   r   r   zunexpected keywordr*   r   T)r   in_placery   )r   )r   r$   r9   r:   rf   ZCategoricalnanrz   r-   r.   	TypeErrorr{   Zreindexr|   )r   r   catsr0   r   r   r   test_unexpected_keyword   s"   
"z&TestDataFrame2.test_unexpected_keywordN)rp   rq   rr   r-   rs   rt   r   r   r   r   r   r   ru      s    
ru   )rz   r   operatorr   numpyr$   r-   Zpandasrf   r   r   r   r   Zpandas._testingZ_testingr=   r   ru   r   r   r   r   <module>   s     