3
Ud_                 @   s4   d dl Zd dlZd dlZd dljZG dd dZdS )    Nc               @   s4   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdS )TestDataFrameNamedAggregatec             C   s   t j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 jdgd}tj|| |jddd}t jdddgit jddgd}tj|| d S )N                  )ABCr
   sum)foo
   r   )indexmin)r   barr   )r
   r   )r
   r   )r
   r   )pd	DataFrameaggIndextmassert_frame_equal)selfdfresultexpected r   b/var/www/html/virt/lib64/python3.6/site-packages/pandas/tests/frame/apply/test_apply_relabeling.pytest_agg_relabel	   s    * z,TestDataFrameNamedAggregate.test_agg_relabelc          	   C   s   t j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 jddddddgd}tj|| d S )#Nr   r   r   r   r   r   )r	   r
   r   r	   r   r
   meanr   maxr   )r   r   catdatfgg      @g      ?g       @g      @g      @g      @r   r   r!   r"   r#   r$   )r   )r	   r   )r
   r   )r	   r   )r
   r    )r	   r    )r   r   )r   r   r   npnanr   r   r   )r   r   r   r   r   r   r   ,test_agg_relabel_multi_columns_multi_methods   s    *zHTestDataFrameNamedAggregate.test_agg_relabel_multi_columns_multi_methodsc          	   C   s  t j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 jdddgd}tj|| |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 jddddddgd}tj|| d S )Nr   r   r   r   r   r   )r	   r
   r   r	   r   )r   r   r!   g      ?g      ?r   r   r!   )r   r
   r   r   c             S   s   t | S )N)r   )xr   r   r   <lambda><   s    zPTestDataFrameNamedAggregate.test_agg_relabel_partial_functions.<locals>.<lambda>)r   r   r!   r"   r#   kkg      @g      $@g      @r"   r#   r*   )r	   r   )r   r   )r   r   r   r%   r   r   r   r   r   r    r   r&   )r   r   r   r   r   r   r   "test_agg_relabel_partial_functions-   s$    * z>TestDataFrameNamedAggregate.test_agg_relabel_partial_functionsc             C   s   t jddgddgd}|jt jddt jdtt jdddt jdd	d
d}t jdddddgit jddddgd}tj|| |jt j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 jdddgd}tj|| d S )Nr   r   r   )r	   r
   r
   r   count)columnaggfuncr    )r.   )r   r   r!   fftr   r   r   r!   r/   )r   r	   r   )r   r   r!   g        g      ?g       @)
r   r   r   ZNamedAggr   r   r   r   r%   r&   )r   r   r   r   r   r   r   test_agg_namedtupleH   s"    

$
z/TestDataFrameNamedAggregate.test_agg_namedtuplec          	   C   sB   t jddgddgd}d}tjt|d |j  W d Q R X d S )Nr   r   r   )r	   r
   zMust provide)match)r   r   pytestZraises	TypeErrorr   )r   r   msgr   r   r   test_agg_raisesb   s    z+TestDataFrameNamedAggregate.test_agg_raisesN)__name__
__module____qualname__r   r'   r+   r0   r5   r   r   r   r   r      s
   r   )	Znumpyr%   r2   Zpandasr   Zpandas._testingZ_testingr   r   r   r   r   r   <module>   s   
