3
Udr                 @   sr   d dl Z d dlZd dlZd dlj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   s6   t |||d}t |jd||d}tj|| d S )N)skipnaZfloat64)getattrZastypetmZassert_almost_equal)selfsop_namer   resultexpected r   V/var/www/html/virt/lib64/python3.6/site-packages/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jdddgdd Zejjdddgdd Zd	S )
BaseNoReduceTestsz  we don't define any reductions r   TFc          
   C   s8   |}t j|}tjt t|||d W d Q R X d S )N)r   )pdSeriespytestraises	TypeErrorr   )r   dataall_numeric_reductionsr   r
   r	   r   r   r   test_reduce_series_numeric   s    
z,BaseNoReduceTests.test_reduce_series_numericc          
   C   s8   |}t j|}tjt t|||d W d Q R X d S )N)r   )r   r   r   r   r   r   )r   r   all_boolean_reductionsr   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jdddgdd ZdS )BaseNumericReduceTestsr   TFc             C   s@   |}t j|}tj   tjdt | j||| W d 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jdddgdd ZdS )BaseBooleanReduceTestsr   TFc             C   s    |}t j|}| j||| d S )N)r   r   r   )r   r   r   r   r
   r	   r   r   r   r'   8   s    
z)BaseBooleanReduceTests.test_reduce_seriesN)r   r   r   r   r   r    r'   r   r   r   r   r(   7   s   r(   )r#   r   Zpandasr   Zpandas._testingZ_testingr   baser   r   r   r!   r(   r   r   r   r   <module>   s   
