o
    i                     @   s@   d dl Zd dlZd dlmZ dd Zdd Zdd Z	dd	 Z
dS )
    Nc                  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    i/var/www/edux/Edux_v2/venv/lib/python3.10/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"   r0   c               	   C   s  t g dg dg dd} | jdtjfddtfd}t jdg dit g dd	}t|| | jdtfdtjfd
t	fdd
tj
fd
dd fd}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   r   )r   r#   r   r   r&   )      ?r2   r*   r   r   r%   c                 S   s   t | S )N)r   )xr    r    r!   <lambda><   s    z4test_agg_relabel_partial_functions.<locals>.<lambda>)r   r   r&   r'   r(   kkr*   r,   g      $@r-   )r   r   r   r.   r#   r   r   r   r   r$   r   r/   r   r    r    r!   "test_agg_relabel_partial_functions-   s,   
r6   c                  C   s   t ddgddgd} | jt ddt dt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   count)columnaggfuncr$   )r9   )r   r   r&   fft)r   r   r   r   r   r   r   r1   g        r*   r+   )
r   r   r   NamedAggr   r   r   r   r.   r/   r   r    r    r!   test_agg_namedtupleI   s*   


r<   )numpyr.   pandasr   pandas._testing_testingr   r"   r0   r6   r<   r    r    r    r!   <module>   s    