U
    >vfa	                     @   st   d dl Z d dlZd dlZd dlmZ d dlmZ G dd deZ	G dd de	Z
G dd de	ZG d	d
 d
e	ZdS )    N)BaseExtensionTestsc                   @   s   e Zd ZdZdd ZdS )BaseReduceTestszg
    Reduction specific tests. Generally these only
    make sense for numeric/boolean operations.
    c                 C   sT   t ||}t |d|}|dkr0| }| }n||d}||d}t|| d S )NZfloat64countskipna)getattrZastypetmZassert_almost_equal)selfsop_namer   Zres_opZexp_opresultexpected r   F/tmp/pip-unpacked-wheel-vdrwu74i/pandas/tests/extension/base/reduce.pycheck_reduce   s    


zBaseReduceTests.check_reduceN)__name__
__module____qualname____doc__r   r   r   r   r   r   
   s   r   c                   @   sD   e Zd ZdZejdddgdd Zejdddgdd Zd	S )
BaseNoReduceTestszwe don't define any reductionsr   TFc              	   C   s@   |}t |}d}tjt|d t|||d W 5 Q R X d S NzV[Cc]annot perform|Categorical is not ordered for operation|does not support reduction|)matchr   pdSeriespytestZraises	TypeErrorr   )r	   dataall_numeric_reductionsr   r   r
   msgr   r   r   test_reduce_series_numeric   s    
z,BaseNoReduceTests.test_reduce_series_numericc              	   C   s@   |}t |}d}tjt|d t|||d W 5 Q R X d S r   r   )r	   r   all_boolean_reductionsr   r   r
   r   r   r   r   test_reduce_series_boolean,   s    
z,BaseNoReduceTests.test_reduce_series_booleanN)	r   r   r   r   r   markparametrizer    r"   r   r   r   r   r      s
   
r   c                   @   s&   e Zd Zejdddgdd ZdS )BaseNumericReduceTestsr   TFc              	   C   s@   |}t |}t   tdt | ||| W 5 Q R X d S )Nignore)r   r   warningscatch_warningssimplefilterRuntimeWarningr   )r	   r   r   r   r   r
   r   r   r   test_reduce_series;   s
    

z)BaseNumericReduceTests.test_reduce_seriesNr   r   r   r   r#   r$   r+   r   r   r   r   r%   :   s   r%   c                   @   s&   e Zd Zejdddgdd ZdS )BaseBooleanReduceTestsr   TFc                 C   s    |}t |}| ||| d S )N)r   r   r   )r	   r   r!   r   r   r
   r   r   r   r+   G   s    
z)BaseBooleanReduceTests.test_reduce_seriesNr,   r   r   r   r   r-   F   s   r-   )r'   r   Zpandasr   Zpandas._testingZ_testingr   Z pandas.tests.extension.base.baser   r   r   r%   r-   r   r   r   r   <module>   s   