U
|
ý°dÛ ã @ sô d Z ddlZddlZddlmZ ddlZddlm Z ddl
|
m Z ej dgddd Z ej d
|
d Zej d d Zej dd Zej dd Zej dd Zej dd Zej dd ZG dd dZG dd dee jZG d d d ee jZG d d d ee jZG d d! d!e j Z G d"d# d#ee j Z G d$d% d%ee j Z!G d&d' d'ee j"Z#G d(d) d)ee j$Z%G d*d+ d+ee j&Z'G d,d- d-ee j(Z)G d.d/ d/ee j*Z+G d0d1 d1ee j,Z-G d2d3 d3ee j.Z/G d4d5 d5ee j0Z1G d6d7 d7ee j2Z3dS )8aj
|
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)ÚDatetimeTZDtype)Ú DatetimeArray)Úbasez
|
US/Central)Úparamsc C s t d| jdS )NÚns)ÚunitÚtz)r Úparam)Úrequest© r ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/extension/test_datetime.pyÚdtype s r c C s t tjdd| jd| d}|S )NÚ2000éd )Zperiodsr ©r )r ÚpdZ
|
date_ranger )r Údatar r r r s r c C s t tjddgdd| dS )NÚNaTú
|
2000-01-01údatetime64[ns]r )r ÚnpÚarrayr r r r Ú data_missing% s ÿr c C s: t d¡}t d¡}t d¡}ttj|||gdd| dS )Nr ú
|
2000-01-02ú
|
2000-01-03r r ©r Ú Timestampr r r )r ÚaÚbÚcr r r Údata_for_sorting, s
|
|
|
r c C s0 t d¡}t d¡}ttj|d|gdd| dS )Nr r r r r r )r r r r r r Údata_missing_for_sorting4 s
|
|
r! c
|
C sH t d¡}t d¡}t d¡}d}ttj||||||||gdd| dS )z_
|
Expected to be like [B, B, NA, NA, A, A, B, C]
|
|
Where A < B < C and NA is missing
|
r r r r r r r )r r r r Únar r r Údata_for_grouping; s
|
|
|
ÿr# c C s dd } | S )Nc S s | t jko| |kS ©N©r r )r r r r r ÚcmpM s zna_cmp.<locals>.cmpr )r& r r r Úna_cmpK s r' c C s t jS r$ r% r r r r Úna_valueS s r( c @ s e Zd ZdS )ÚBaseDatetimeTestsN©Ú__name__Ú
|
__module__Ú __qualname__r r r r r) Y s r) c @ s e Zd ZdS )ÚTestDatetimeDtypeNr* r r r r r. _ s r. c s e Zd Z fddZ ZS )ÚTestConstructorsc s | d ¡}t |¡ d S r$ )Z
|
_with_freqÚsuperÚtest_series_constructor)Úselfr ©Ú __class__r r r1 d s
|
z(TestConstructors.test_series_constructor)r+ r, r- r1 Ú __classcell__r r r3 r r/ c s r/ c @ s e Zd ZdS )Ú TestGetitemNr* r r r r r6 j s r6 c @ s e Zd ZdS )Ú TestIndexNr* r r r r r7 n s r7 c @ s e Zd Zdd ZdS )Ú TestMethodsc C s d S r$ r )r2 Z data_repeatedr r r Útest_combine_adds s z TestMethods.test_combine_addN)r+ r, r- r9 r r r r r8 r s r8 c @ s e Zd ZdS )Ú TestInterfaceNr* r r r r r: x s r: c sL e Zd ZddhZ fddZ fddZdd Z fd d
|
Zd d Z Z S ) ÚTestArithmeticOpsÚ__sub__Ú__rsub__c sB || j kr0t d|i¡}| j|||d d d nt ||¡ d S )NÚAr ©Úexc)Ú
|
implementsr Z DataFrameÚ check_opnamer0 Ú test_arith_frame_with_scalar)r2 r Úall_arithmetic_operatorsZdfr3 r r rC s
|
z.TestArithmeticOps.test_arith_frame_with_scalarc s@ || j kr.t |¡}| j|||jd d d nt ||¡ d S ©Nr r? ©rA r ÚSeriesrB Zilocr0 Ú test_arith_series_with_scalar©r2 r rD Úserr3 r r rH s
|
|
z/TestArithmeticOps.test_arith_series_with_scalarc C s4 t |¡}d}tjt|d || W 5 Q R X d S )Nz*cannot add DatetimeArray and DatetimeArray)Úmatch)r rG ÚpytestZraisesÚ TypeError)r2 r rJ Úmsgr r r Ú$test_add_series_with_extension_array s
|
z6TestArithmeticOps.test_add_series_with_extension_arrayc s@ || j kr.t |¡}| j|||jd d d nt ||¡ d S rE rF rI r3 r r Ú test_arith_series_with_array s
|
|
z.TestArithmeticOps.test_arith_series_with_arrayc C s d S r$ r )r2 r r r Útest_divmod_series_array s z*TestArithmeticOps.test_divmod_series_array)
|
r+ r, r- rA rC rH rO rP rQ r5 r r r3 r r; | s r; c @ s e Zd ZdS )Ú TestCastingNr* r r r r rR ¥ s rR c @ s e Zd ZdS )ÚTestComparisonOpsNr* r r r r rS © s rS c @ s e Zd ZdS )Ú TestMissingNr* r r r r rT s rT c @ s e Zd ZdS )Ú TestReshapingNr* r r r r rU ± s rU c @ s e Zd ZdS )Ú TestSetitemNr* r r r r rV µ s rV c @ s e Zd ZdS )Ú TestGroupbyNr* r r r r rW ¹ s rW c @ s e Zd ZdS )Ú TestPrintingNr* r r r r rX ½ s rX c @ s e Zd ZdS )Ú Test2DCompatNr* r r r r rY Á s rY )4Ú__doc__Únumpyr rL Zpandas.core.dtypes.dtypesr Zpandasr Zpandas.core.arraysr Zpandas.tests.extensionr Zfixturer r r r r! r# r' r( r) ZBaseDtypeTestsr. ZBaseConstructorsTestsr/ ZBaseGetitemTestsr6 ZBaseIndexTestsr7 ZBaseMethodsTestsr8 ZBaseInterfaceTestsr: ZBaseArithmeticOpsTestsr; ZBaseCastingTestsrR ZBaseComparisonOpsTestsrS ZBaseMissingTestsrT ZBaseReshapingTestsrU ZBaseSetitemTestsrV ZBaseGroupbyTestsrW ZBasePrintingTestsrX ZNDArrayBacked2DTestsrY r r r r Ú<module> sL
|
|
|
|
|
|
|
|
)
|
