U
    *if                     @   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 ddddgddddgddddg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                  ABC)r
   sum)foor
   
   r   index)r
   min)r   barr   )pd	DataFrameaggIndextmassert_frame_equalZdfresultexpected r   a/home/mars/bis/venv/lib/python3.8/site-packages/pandas/tests/apply/test_frame_apply_relabeling.pytest_agg_relabel   s    * r   c               
   C   s   t ddddgddddgddddg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 ddddddgd}t|| d S )Nr   r   r   r   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#   r$   r%   r   )r   r   r   npnanr   r   r   r   r   r   r   ,test_agg_relabel_multi_columns_multi_methods   s"    *r,   c               
   C   s  t ddddgddddgddddgd} | jdtjfd	dtfd
}t jddddgit dddg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 ddddddgd}t|| d S )Nr   r   r   r   r   r   r   r	   )r	   r   r   r   r"   g      ?r&   r   r   r"   r   r
   r!   c                 S   s   t | S )N)r   )xr   r   r   <lambda><       z4test_agg_relabel_partial_functions.<locals>.<lambda>)r   r   r"   r#   r$   kkr(   g      $@r)   r#   r$   r1   )r   r   r   r*   r   r   r   r   r   r    r   r+   r   r   r   r   "test_agg_relabel_partial_functions-   s.    * 
r2   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ddddgit ddddg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 dddgd}t|| d S )Nr   r   r   )r	   r
   r
   r   count)columnaggfuncr    )r5   )r   r   r"   fftr   r   r   r"   r6   r   r	   r   r-   g        r&   r'   )
r   r   r   ZNamedAggr   r   r   r   r*   r+   r   r   r   r   test_agg_namedtupleI   s,    

 
r7   )numpyr*   Zpandasr   Zpandas._testingZ_testingr   r   r,   r2   r7   r   r   r   r   <module>   s   