U
|
ý°d ' ã @ sê d Z ddlZddlZddlZddlZddlmZmZm Z ddl
|
m Z ddl mZ ddlmZ dd Zejdd Zejd
|
d Zejd d Zejdd Zejdd Zejdd Zejdd ZG dd dejZG dd dej Z G dd dej 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-ej0Z1G d.d/ d/ej2Z3G d0d1 d1ej4Z5G d2d3 d3ej6Z7G d4d5 d5ej8Z9dS )6aj
|
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)Ú CategoricalÚCategoricalIndexÚ Timestamp©ÚCategoricalDtype)Úbasec C s. t jjttjdd} | d | d kr q*q | S )Néd )Úsizer é )ÚnpÚrandomÚchoiceÚlistÚstringÚ ascii_letters)Úvalues© r ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/extension/test_categorical.pyÚ make_data s r c C s t S ©Nr r r r r Údtype+ s r c C s
|
t t S )zLength-100 array for this type.
|
|
* data[0] and data[1] should both be non missing
|
* data[0] and data[1] should not be equal
|
)r r r r r r Údata0 s r c C s t tjdgS )z Length 2 array with [NA, Valid]ÚA)r r Únanr r r r Ú data_missing: s r c C s t dddgdddgddS )Nr ÚBÚCT©Ú
|
categoriesZordered©r r r r r Údata_for_sorting@ s r c C s t dd dgddgddS )Nr r Tr r r r r r Údata_missing_for_sortingE s r! c C s t jS r )r r r r r r Úna_valueJ s r" c C s t ddd d ddddgS )NÚaÚbÚcr r r r r Údata_for_groupingO s r& c @ s e Zd ZdS )Ú TestDtypeN©Ú__name__Ú
|
__module__Ú __qualname__r r r r r' T s r' c s2 e Zd Zejjdd fddZdd Z ZS )Ú TestInterfacezMemory usage doesn't match©Úreasonc s t |¡ d S r )ÚsuperÚtest_memory_usage©Úselfr ©Ú __class__r r r0 Y s z TestInterface.test_memory_usagec C s |j j}|| ¡ }|d |ks&t|d |ks6t||ksBt||ksNttjD ]&}||krbqT||ksnt||ksTtqTd S )Nr )r r" ZisnaÚAssertionErrorÚtmZ NULL_OBJECTS)r2 r r r" Z na_value_objr r r Ú test_contains^ s
|
zTestInterface.test_contains) r) r* r+ ÚpytestÚmarkÚxfailr0 r7 Ú __classcell__r r r3 r r, X s r, c @ s e Zd Zdd ZdS )ÚTestConstructorsc C s: | ¡ }|jd|d}t||s$t|jtg ks6td S )N)é ©r )Zconstruct_array_typeÚ_emptyÚ
|
isinstancer5 r r )r2 r ÚclsÚresultr r r Ú
|
test_emptyx s zTestConstructors.test_emptyN)r) r* r+ rC r r r r r< w s r< c @ s e Zd ZdS )Ú TestReshapingNr( r r r r rD s rD c s* e Zd Zejjdd fddZ ZS )Ú TestGetitemzBackwards compatibilityr- c s t |¡ d S r )r/ Útest_getitem_scalarr1 r3 r r rF s z TestGetitem.test_getitem_scalar)r) r* r+ r8 r9 ÚskiprF r; r r r3 r rE s rE c @ s e Zd ZdS )Ú TestSetitemNr( r r r r rH s rH c @ s e Zd ZdS )Ú TestIndexNr( r r r r rI s rI c @ s e Zd ZdS )Ú TestMissingNr( r r r r rJ s rJ c @ s e Zd ZdS )Ú
|
TestReduceNr( r r r r rK s rK c @ s& e Zd Zej dddg¡dd ZdS )ÚTestAccumulateÚskipnaTFc C s d S r r )r2 r Zall_numeric_accumulationsrM r r r Útest_accumulate_series s z%TestAccumulate.test_accumulate_seriesN)r) r* r+ r8 r9 Ú parametrizerN r r r r rL s rL c s2 e Zd Zejjdd fddZdd Z ZS )Ú TestMethodsz Unobserved categories includedr- c s t ||¡S r )r/ Útest_value_counts)r2 Zall_dataZdropnar3 r r rQ ¦ s z TestMethods.test_value_countsc s¤ |d\}}t |¡}t |¡}| |dd ¡}t dd tt|t|D ¡}| ||¡ |jd | dd ¡}t fddt|D ¡}| ||¡ d S ) Né c S s | | S r r ©Úx1Zx2r r r Ú<lambda>° ó z.TestMethods.test_combine_add.<locals>.<lambda>c S s g | ]\}}|| qS r r )Ú.0r# r$ r r r Ú
|
<listcomp>² s z0TestMethods.test_combine_add.<locals>.<listcomp>r c S s | | S r r rS r r r rU · rV c s g | ] }| qS r r )rW r# ©Úvalr r rX ¸ s )ÚpdÚSeriesÚcombineÚzipr Úassert_series_equalZiloc)r2 Z data_repeatedZ
|
orig_data1Z
|
orig_data2Ús1Ús2rB Úexpectedr rY r Útest_combine_addª s
|
|
ÿ
|
z TestMethods.test_combine_add) r) r* r+ r8 r9 r: rQ rc r; r r r3 r rP ¥ s rP c @ s e Zd Zej deeg¡ej ddej ge
|
de j gg¡dd Z ej de jdd gd
|
d e jd d gd d e je d¡e d ¡gdd e je d d¡e dd¡gdd e jdej gdd g¡dd Zej ddejdgdd fde e
|
dddg¡jfg¡dd Zd S ) Ú TestCastingrA r r
|
Ú2000c C s8 ||}d}t jttf|d | t¡ W 5 Q R X d S )NzCannot (cast|convert)©Úmatch)r8 ÚraisesÚ
|
ValueErrorÚ TypeErrorÚastypeÚint)r2 rA r ÚsÚmsgr r r Útest_cast_nan_to_int½ s z TestCasting.test_cast_nan_to_intrb Ú2019Ú2020zdatetime64[ns, UTC]r> r ztimedelta64[ns]z period[A-DEC]rR ÚintervalZInt64c C s" | d¡ |j¡}t ||¡ d S )NÚcategory)rk r r6 r_ )r2 rb rB r r r Ú%test_cast_category_to_extension_dtypeÇ s z1TestCasting.test_cast_category_to_extension_dtypezdtype, expectedzdatetime64[ns]z 2015-01-01T00:00:00.000000000zdatetime64[ns, MET]z2015-01-01 00:00:00+0100ZMET)Útzc C s t dg |¡}||ks td S )Nz
|
2015-01-01)r rk r5 )r2 r rb rB r r r Útest_consistent_casting× s z#TestCasting.test_consistent_castingN)r) r* r+ r8 r9 rO r r r r r r[ ZNaTro r\ ZPeriodZIntervalrt ÚarrayZ DatetimeIndexrv r r r r rd ¼ s4 ûþ
|
|
þ ÿþûþrd c sH e Zd Z fddZ fddZdd Zdd Zef fd d
|
Z Z S ) ÚTestArithmeticOpsc s4 |}|dkr"|j tjjdd¡ t ||¡ d S ©NÚ__rmod__z/rmod never called when string is first argumentr- )ÚnodeÚ
|
add_markerr8 r9 r: r/ Ú test_arith_frame_with_scalar©r2 r Zall_arithmetic_operatorsÚrequestÚop_namer3 r r r} í s ÿÿz.TestArithmeticOps.test_arith_frame_with_scalarc s4 |}|dkr"|j tjjdd¡ t ||¡ d S ry )r{ r| r8 r9 r: r/ Ú test_arith_series_with_scalarr~ r3 r r r ø s ÿÿz/TestArithmeticOps.test_arith_series_with_scalarc C s0 t |¡}tjtdd || W 5 Q R X d S )Nz"cannot perform|unsupported operandrf )r[ r\ r8 rh rj )r2 r Zserr r r Ú$test_add_series_with_extension_array s
|
z6TestArithmeticOps.test_add_series_with_extension_arrayc C s d S r r )r2 r r r Útest_divmod_series_array s z*TestArithmeticOps.test_divmod_series_arrayc s t j|||tdS )N)Úexc)r/ Ú_check_divmod_oprj )r2 rm ÚopÚotherr r3 r r r
s z"TestArithmeticOps._check_divmod_op)
|
r) r* r+ r} r r r ÚNotImplementedErrorr
r; r r r3 r rx ì s
|
|
rx c @ sD e Zd Zdd Zej dddgddgeded gg¡d
|
d Zd S ) ÚTestComparisonOpsc C s¢ d|j d}|dkrB|||}| |dd ¡}||k ¡ stn\|dkrv|||}| |dd ¡}||k ¡ stn(d}tjt|d ||| W 5 Q R X d S ) NÚ__Ú__eq__c S s | |kS r r ©ÚxÚyr r r rU rV z2TestComparisonOps._compare_other.<locals>.<lambda>Ú__ne__c S s | |kS r r r r r r rU rV z7Unordered Categoricals can only compare equality or notrf )r) r] Úallr5 r8 rh rj ) r2 rm r r r r rB rb rn r r r Ú_compare_other s
|
|
z TestComparisonOps._compare_otherr r# r$ r r
|
rp rq c C s< t jddg|d}t jddg|d}||k}| ¡ s8td S )Néÿÿÿÿr )r r
|
)r Z
|
from_codesr r5 )r2 r Zc1Úc2rB r r r Útest_not_equal_with_na" s z(TestComparisonOps.test_not_equal_with_naN) r) r* r+ r r8 r9 rO r r r r r r r s þr c @ s e Zd ZdS )Ú TestParsingNr( r r r r r 0 s r c @ s e Zd Zdd ZdS )Ú Test2DCompatc C sH t | dd¡}| d¡dks"tt | dd¡}| d¡dksDtd S )Nr
|
r z
|
Categories)ÚreprZreshapeÚcountr5 )r2 r Úresr r r Ú test_repr_2d5 s zTest2DCompat.test_repr_2dN)r) r* r+ r r r r r r 4 s r ):Ú__doc__r Únumpyr r8 Zpandasr[ r r r Zpandas._testingZ_testingr6 Zpandas.api.typesr Zpandas.tests.extensionr r Zfixturer r r r r! r" r& ZBaseDtypeTestsr' ZBaseInterfaceTestsr, ZBaseConstructorsTestsr< ZBaseReshapingTestsrD ZBaseGetitemTestsrE ZBaseSetitemTestsrH ZBaseIndexTestsrI ZBaseMissingTestsrJ ZBaseNoReduceTestsrK ZBaseAccumulateTestsrL ZBaseMethodsTestsrP ZBaseCastingTestsrd ZBaseArithmeticOpsTestsrx ZBaseComparisonOpsTestsr ZBaseParsingTestsr ZNDArrayBacked2DTestsr r r r r Ú<module> sN
|
|
|
|
|
|
|
0$
|
