a
    Pf                     @   sl   d dl Zd dlZd dlmZ d dlZd dlmZ	 dd Z
dd Zejjeddd	d
 Zdd Zdd ZdS )    N)np_version_gte1p25c                  C   s   t g dg dg dd} | jdd}t jddgit d	gd
}t|| | jddd}t jdddgit d	dgd
}t|| d S )N      r   r   r   r         r   r         ABC)r   sum)foor   
   r   index)r   min)r   barr   r   )pd	DataFrameaggIndextmassert_frame_equaldfresultexpected r!   g/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/apply/test_frame_apply_relabeling.pytest_agg_relabel
   s     r#   c               	   C   s   t g dg dg dd} | jddddd	d
d}t jdtjdtjdtjgtjdtjdtjtjgtjtjtjtjtjdgdt g dd}t|| d S )Nr   r   r	   r   )r   r   )r   mean)r   r   )r   max)r   r%   r   r   )r   r   catdatfgg      @      ?       @g      @      @      @r   )r   r   r   npnanr   r   r   r   r!   r!   r"   ,test_agg_relabel_multi_columns_multi_methods   s"    r1   z%name of min now equals name of np.min)reasonc               
   C   sf  t g dg dg dd} d}tjt|d* | jdtjfddtfd	}W d    n1 s`0    Y  t jdg d
it 	g d	d}t
|| d}tjt|dB | jdtfdtjfdtfddtjfddd fd}W d    n1 s0    Y  t jddtjtjtjtjgtjtjdtjddgtjtjtjdtjtjgdt 	g dd}t
|| d S )Nr   r   r	   r   zusing Series.[mean|min])matchr   )r   r$   r   r   r'   )      ?r5   r+   r   zusing Series.[mean|min|max|sum]r   r&   c                 S   s   t | S )N)r   )xr!   r!   r"   <lambda>D       z4test_agg_relabel_partial_functions.<locals>.<lambda>)r   r   r'   r(   r)   kkr+   r-   g      $@r.   )r   r   r   Zassert_produces_warningFutureWarningr   r/   r$   r   r   r   r%   r   r0   )r   msgr   r    r!   r!   r"   "test_agg_relabel_partial_functions0   s4    8
$r<   c                  C   s   t ddgddgd} | jt ddt ddt jddd	t jdd
dd}t jdg dit g dd}t|| | jt ddt jdd
d	t jdd
d	d}t jdtjdgtjdtjgdt g dd}t|| d S )Nr   r   r   )r   r   r   r   r   count)columnaggfuncr%   )r?   )r   r   r'   Zfft)r   r   r   r   r   r   r4   g        r+   r,   )	r   r   r   ZNamedAggr   r   r   r/   r0   r   r!   r!   r"   test_agg_namedtupleQ   s*    


r@   c                  C   s"   t jjd} d}t| | d S )Nr   )Fr   NN)r   coreapplyZreconstruct_funcr   Zassert_equal)r   r    r!   r!   r"   test_reconstruct_funcl   s    rC   )numpyr/   ZpytestZpandas.compat.numpyr   Zpandasr   Zpandas._testingZ_testingr   r#   r1   markZxfailr<   r@   rC   r!   r!   r!   r"   <module>   s   
 