U
|
ý°dÀ ã @ s` d Z ddlZddlZddlmZ ddlZddlm Z
|
ddl m Z ddl mZmZ ddlmZ dd Zejeegd d
|
d Zejd d Zejdd Zejdd Zejdd Zejdd Zejdd Zejdd Zejdd Z G d d d ej Z G d d d ej Z G d d! d!ej!Z"G d"d# d#ej#Z$G d$d% d%ej%Z&G d&d' d'ej'Z(G d(d) d)ej)Z*G d*d+ d+ej+Z,G d,d- d-ej-Z.G d.d/ d/ej/Z0G d0d1 d1ej1Z2G d2d3 d3ej3Z4G d4d5 d5ej5Z6G d6d7 d7ej7Z8ej9j:d8d9G d:d; d;ej;Z<G d<d= d=ej=Z>G d>d? d?ej?Z@ej9 Ad@¡G dAdB dBejBZCG dCdD dDejDZEdS )Eaj
|
This file contains a minimal set of tests for compliance with the extension
|
array interface test suite, and should contain no other tests.
|
The test suite for the full functionality of the array is located in
|
`pandas/tests/arrays/`.
|
|
The tests in this file are inherited from the BaseExtensionTests, and only
|
minimal tweaks should be applied to get the tests passing (by overwriting a
|
parent method).
|
|
Additional tests should either be added to one of the BaseExtensionTests
|
classes (if they are relevant for the extension interface for all dtypes), or
|
be added to the array-specific tests in `pandas/tests/arrays/`.
|
|
é N)Úis_extension_array_dtype)Úis_float_dtype)Ú Float32DtypeÚ Float64Dtype)Úbasec C s< t t ddd¡tjg t t ddd¡ tjg ddg S )Nç¹?gÍÌÌÌÌÌì?é g#@gÍÌÌÌÌÌ#@g $@)ÚlistÚnpZarangeÚpdÚNA© r r ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/extension/test_floating.pyÚ make_data s ÿþýüÿr )Úparamsc C s | ¡ S ©N)Úparam)Úrequestr r r Údtype) s r c C s t jt | dS )N©r )r Úarrayr r r r r Údata. s r c C s t jt d¡d | dS )Néd é r )r r r
|
Zonesr r r r Ú data_for_twos3 s r c C s t jt jdg| dS )Nr r ©r r r r r r r Ú data_missing8 s r c C s t jdddg| dS )Nr çÉ?ç r )r r r r r r Údata_for_sorting= s r c C s t jdt jdg| dS )Nr r r r r r r r Údata_missing_for_sortingB s r c C s dd S )Nc S s | t jko|t jkS r ©r r )ÚxÚyr r r Ú<lambda>J ó zna_cmp.<locals>.<lambda>r r r r r Úna_cmpG s r& c C s t jS r r! r r r r Úna_valueM s r' c C s0 d}d}d}t j}t j||||||||g| dS )Nr r r r )r r r )r ÚbÚaÚcÚnar r r Údata_for_groupingR s
|
r, c @ s e Zd ZdS )Ú TestDtypeN©Ú__name__Ú
|
__module__Ú __qualname__r r r r r- [ s r- c s8 e Zd Zd fdd ZefddZd fdd Z ZS )
|
ÚTestArithmeticOpsNc s t j|||d d d S ©N)Úexc©ÚsuperÚ check_opname©ÚselfÚsÚop_nameÚotherr4 ©Ú __class__r r r7 ` s z TestArithmeticOps.check_opnamec C s |d krlt |¡}t|dr<t|js<t|jr<| |j¡}|||}| |||¡}| |¡}| ||¡ n t
|
|¡ ||| W 5 Q R X d S )Nr ) ÚtmZ get_dtypeÚhasattrr r r ÚastypeÚ numpy_dtypeZ_combineZ assert_equalÚpytestÚraises) r9 r: Úopr< r; r4 ZsdtypeÚresultÚexpectedr r r Ú _check_opd s
|
ÿþý
|
|
zTestArithmeticOps._check_opc s t |||d ¡ d S r )r6 Ú_check_divmod_op)r9 r: rE r< r4 r= r r rI { s z"TestArithmeticOps._check_divmod_op)N)N)r/ r0 r1 r7 ÚNotImplementedErrorrH rI Ú __classcell__r r r= r r2 _ s r2 c s2 e Zd ZefddZd fdd Zdd Z ZS ) ÚTestComparisonOpsc C sV |d kr2|||}| ||¡ d¡}| ||¡ n t |¡ ||| W 5 Q R X d S )NÚboolean)ÚcombinerA Zassert_series_equalrC rD )r9 r: rE r< r; r4 rF rG r r r rH s
|
zTestComparisonOps._check_opNc s t j|||d d d S r3 r5 r8 r= r r r7 s z TestComparisonOps.check_opnamec C s d|j d}| |||¡ d S )NÚ__)r/ r7 )r9 r: r rE r< r; r r r Ú_compare_other s z TestComparisonOps._compare_other)N)r/ r0 r1 rJ rH r7 rP rK r r r= r rL s
|
rL c @ s e Zd ZdS )Ú TestInterfaceNr. r r r r rQ s rQ c @ s e Zd ZdS )ÚTestConstructorsNr. r r r r rR s rR c @ s e Zd ZdS )Ú TestReshapingNr. r r r r rS s rS c @ s e Zd ZdS )Ú TestGetitemNr. r r r r rT s rT c @ s e Zd ZdS )Ú TestSetitemNr. r r r r rU £ s rU c @ s e Zd ZdS )Ú TestIndexNr. r r r r rV § s rV c @ s e Zd ZdS )Ú TestMissingNr. r r r r rW « s rW c @ s e Zd ZeZdS )Ú TestMethodsN)r/ r0 r1 ÚobjectZ_combine_le_expected_dtyper r r r rX ¯ s rX c @ s e Zd ZdS )Ú TestCastingNr. r r r r rZ ³ s rZ c @ s e Zd ZdS )Ú TestGroupbyNr. r r r r r[ · s r[ c @ s e Zd Zdd ZdS )ÚTestNumericReducec C s |dkr0t || }t | ¡ |jj¡| }nDt |||d}t | ¡ |jj¡||d}|st| ¡ ¡ rttj}t
|
||¡ d S )NÚcount)Úskipna) ÚgetattrZdropnarA r rB ZisnaÚanyr r r? Zassert_almost_equal)r9 r: r; r^ rF rG r r r Ú check_reduce¼ s ÿz TestNumericReduce.check_reduceN)r/ r0 r1 ra r r r r r\ » s r\ z-Tested in tests/reductions/test_reductions.py)Úreasonc @ s e Zd ZdS )ÚTestBooleanReduceNr. r r r r rc Ì s rc c @ s e Zd ZdS )Ú TestPrintingNr. r r r r rd Ñ s rd c @ s e Zd ZdS )Ú TestParsingNr. r r r r re Õ s re z4ignore:overflow encountered in reduce:RuntimeWarningc @ s e Zd ZdS )Ú Test2DCompatNr. r r r r rf Ù s rf c @ s& e Zd Zej dddg¡dd ZdS )ÚTestAccumulationr^ TFc C s d S r r )r9 r Zall_numeric_accumulationsr^ r r r Ú test_accumulate_series_raisesß s z.TestAccumulation.test_accumulate_series_raisesN)r/ r0 r1 rC ÚmarkZ parametrizerh r r r r rg Þ s rg )FÚ__doc__Únumpyr
|
rC Zpandas.core.dtypes.commonr Zpandasr Zpandas._testingZ_testingr? Zpandas.api.typesr Zpandas.core.arrays.floatingr r Zpandas.tests.extensionr r Zfixturer r r r r r r& r' r, ZBaseDtypeTestsr- ZBaseArithmeticOpsTestsr2 ZBaseComparisonOpsTestsrL ZBaseInterfaceTestsrQ ZBaseConstructorsTestsrR ZBaseReshapingTestsrS ZBaseGetitemTestsrT ZBaseSetitemTestsrU ZBaseIndexTestsrV ZBaseMissingTestsrW ZBaseMethodsTestsrX ZBaseCastingTestsrZ ZBaseGroupbyTestsr[ ZBaseNumericReduceTestsr\ ri ÚskipZBaseBooleanReduceTestsrc ZBasePrintingTestsrd ZBaseParsingTestsre ÚfilterwarningsZDim2CompatTestsrf ZBaseAccumulateTestsrg r r r r Ú<module> s`
|
|
|
|
|
|
|
|
|
|
|
|
