U
    MZf                     @   sd   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S )    N)np_version_gte1p25c                  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   R/tmp/pip-unpacked-wheel-nbcvw55c/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-   z%name of min now equals name of np.min)reasonc               
   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%   r3   )r   r   r   r+   r    r   r   r   r   r!   r   r,   r   r   r   r   "test_agg_relabel_partial_functions0   s.    * 
r4   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!   )r7   )r   r   r#   fftr   r   r   r#   r8   r   r
   r   r/   g        r'   r(   )
r   r   r   ZNamedAggr   r   r   r   r+   r,   r   r   r   r   test_agg_namedtupleM   s,    

 
r9   )Znumpyr+   ZpytestZpandas.compat.numpyr   Zpandasr   Zpandas._testingZ_testingr   r   r-   markZxfailr4   r9   r   r   r   r   <module>   s   
