U
|
ý°dº ã @ s: d Z ddlZddlZddlmZ ddlmZ ddlm Z ddl
|
m Z ddl Z ddlmZ ddlmZ ddlmZ ejd d
|
gd d d Zejdd Zejdd Zejdd Zejdd Zejdd Zejdd Z ejdd Z G d d d Z G d d d e ej Z G d d! d!e ej!Z"G d"d# d#e ej#Z$G d$d% d%ej%Z&G d&d' d'e ej'Z(G d(d) d)e ej)Z*G d*d+ d+e ej+Z,G d,d- d-e ej-Z.G d.d/ d/e ej/Z0G d0d1 d1e ej1Z2G d2d3 d3e ej3Z4G d4d5 d5e ej5Z6G d6d7 d7e ej7Z8G d8d9 d9e ej9Z:G d:d; d;e ej;Z<G d<d= d=e ej=Z>dS )>aj
|
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)ÚiNaT)Úis_platform_windows)Únp_version_gte1p24)Ú PeriodDtype)Ú PeriodArray)ÚbaseÚDZ2D)Úparamsc C s t | jdS )N©Úfreq)r Úparam)Úrequest© r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/extension/test_period.pyÚdtype s r c C s t t dd¡| jdS )Ni² i r
|
)r ÚnpZaranger ©r r r r Údata$ s r c C s t t d¡d | jdS )Néd é r
|
)r r Zonesr r r r r Ú data_for_twos) s r c C s t dddg| jdS )Néâ éã éá r
|
)r r r r r r Údata_for_sorting. s r c C s t tdg| jdS )Nr r
|
©r r r r r r r Ú data_missing3 s r c C s t dtdg| jdS )Nr r r
|
r r r r r Údata_missing_for_sorting8 s r c C s. d}t }d}d}t||||||||g| jdS )Nr r r r
|
)r r r )r ÚBZNAÚAÚCr r r Údata_for_grouping= s
|
r! c C s t jS ©N)ÚpdZNaTr r r r Úna_valueF s r$ c @ s e Zd ZdS )ÚBasePeriodTestsN©Ú__name__Ú
|
__module__Ú __qualname__r r r r r% K s r% c @ s e Zd ZdS )ÚTestPeriodDtypeNr& r r r r r* O s r* c @ s e Zd ZdS )ÚTestConstructorsNr& r r r r r+ S s r+ c @ s e Zd ZdS )Ú TestGetitemNr& r r r r r, W s r, c @ s e Zd ZdS )Ú TestIndexNr& r r r r r- [ s r- c s6 e Zd Zdd Zej dddg¡ fddZ ZS )Ú TestMethodsc C s d S r" r )ÚselfZ data_repeatedr r r Útest_combine_add` s z TestMethods.test_combine_addÚperiodsé éþÿÿÿc sF t r4tr4tjtdd t ||¡ W 5 Q R X nt ||¡ d S )NF)Zcheck_stacklevel)r r ÚtmZassert_produces_warningÚRuntimeWarningÚsuperÚ test_diff)r/ r r1 ©Ú __class__r r r7 d s
|
zTestMethods.test_diff) r' r( r) r0 ÚpytestÚmarkÚ parametrizer7 Ú __classcell__r r r8 r r. _ s r. c @ s e Zd ZdS )Ú TestInterfaceNr& r r r r r> m s r> c s\ e Zd ZddhZ fddZ fddZ fddZef fd d
|
Zd d Z d d Z
|
Z S )ÚTestArithmeticOpsÚ__sub__Ú__rsub__c sB || j kr0t d|i¡}| j|||d d d nt ||¡ d S )Nr r ©Úexc)Ú
|
implementsr# Ú DataFrameÚ check_opnamer6 Ú test_arith_frame_with_scalar)r/ r Úall_arithmetic_operatorsZdfr8 r r rG t s
|
z.TestArithmeticOps.test_arith_frame_with_scalarc s@ || j kr.t |¡}| j|||jd d d nt ||¡ d S ©Nr rB ©rD r# ÚSeriesrF Zilocr6 Ú test_arith_series_with_scalar©r/ r rH Úsr8 r r rL } s
|
|
z/TestArithmeticOps.test_arith_series_with_scalarc s@ || j kr.t |¡}| j|||jd d d nt ||¡ d S rI rJ rM r8 r r Ú test_arith_series_with_array s
|
|
z.TestArithmeticOps.test_arith_series_with_arrayc s t j|||td d S )NrB )r6 Ú_check_divmod_opÚ TypeError)r/ rN ÚopÚotherrC r8 r r rP s z"TestArithmeticOps._check_divmod_opc C s4 t |¡}d}tjt|d || W 5 Q R X d S )NzIunsupported operand type\(s\) for \+: \'PeriodArray\' and \'PeriodArray\')Úmatch)r# rK r: ZraisesrQ )r/ r rN Úmsgr r r Ú$test_add_series_with_extension_array s
|
|
ÿz6TestArithmeticOps.test_add_series_with_extension_arrayc C s6 t |¡}|t jkr | ¡ }| |¡}|tks2td S r" )r# rK rE Zto_framer@ ÚNotImplementedÚAssertionError)r/ r Zframe_or_seriesrS Úresultr r r Ú6test_direct_arith_with_ndframe_returns_not_implemented s
|
|
|
|
zHTestArithmeticOps.test_direct_arith_with_ndframe_returns_not_implemented) r' r( r) rD rG rL rO ÚNotImplementedErrorrP rV rZ r= r r r8 r r? q s
|
r? c @ s e Zd ZdS )Ú TestCastingNr& r r r r r\ § s r\ c @ s e Zd ZdS )ÚTestComparisonOpsNr& r r r r r] « s r] c @ s e Zd ZdS )Ú TestMissingNr& r r r r r^ ¯ s r^ c @ s e Zd ZdS )Ú TestReshapingNr& r r r r r_ ³ s r_ c @ s e Zd ZdS )Ú TestSetitemNr& r r r r r` · s r` c @ s e Zd ZdS )Ú TestGroupbyNr& r r r r ra » s ra c @ s e Zd ZdS )Ú TestPrintingNr& r r r r rb ¿ s rb c s. e Zd Zej dddg¡ fddZ ZS )Ú TestParsingÚengineÚcÚpythonc s t ||¡ d S r" )r6 Ú test_EA_types)r/ rd r r8 r r rg Ä s zTestParsing.test_EA_types)r' r( r) r: r; r< rg r= r r r8 r rc à s rc c @ s e Zd ZdS )Ú Test2DCompatNr& r r r r rh É s rh )?Ú__doc__Únumpyr r: Z pandas._libsr Z pandas.compatr Zpandas.compat.numpyr Zpandas.core.dtypes.dtypesr Zpandasr# Zpandas._testingZ_testingr4 Zpandas.core.arraysr Zpandas.tests.extensionr Zfixturer r r r r r r! r$ r% ZBaseDtypeTestsr* ZBaseConstructorsTestsr+ ZBaseGetitemTestsr, ZBaseIndexTestsr- ZBaseMethodsTestsr. ZBaseInterfaceTestsr> ZBaseArithmeticOpsTestsr? ZBaseCastingTestsr\ ZBaseComparisonOpsTestsr] ZBaseMissingTestsr^ ZBaseReshapingTestsr_ ZBaseSetitemTestsr` ZBaseGroupbyTestsra ZBasePrintingTestsrb ZBaseParsingTestsrc ZNDArrayBacked2DTestsrh r r r r Ú<module> sV
|
|
|
|
|
|
|
|
6
|
