U
|
ý°d¬ ã @ st d dl Z d dlZd dlZd dlmZ d dlmZ G dd deZ G dd de Z
|
G dd de Z G d d
|
d
|
e Z dS ) é N)ÚBaseExtensionTestsc @ s e Zd ZdZdd ZdS )ÚBaseReduceTestszg
|
Reduction specific tests. Generally these only
|
make sense for numeric/boolean operations.
|
c C sT t ||}t | d¡|}|dkr0| }| }n||d}||d}t ||¡ d S )NÚfloat64Úcount©Úskipna)ÚgetattrZastypeÚtmZassert_almost_equal)ÚselfÚsÚop_namer Zres_opZexp_opÚresultÚexpected© r úYd:\z\workplace\vscode\pyvenv\venv\Lib\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 Zd ZdZej dddg¡dd Zej dddg¡dd Zd S )
|
ÚBaseNoReduceTestsz we don't define any reductionsr TFc C s@ |}t |¡}d}tjt|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ÚpytestZraisesÚ 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}tjt|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 Zd Zej dddg¡dd ZdS )ÚBaseNumericReduceTestsr TFc C s@ |}t |¡}t ¡ t dt¡ | |||¡ 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 Zd Zej dddg¡dd ZdS )Ú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 Zpandasr Zpandas._testingZ_testingr Z pandas.tests.extension.base.baser r r r& r. r r r r Ú<module> s
|
