U
|
ý°d ã @ s d Z ddlZddlZddlmZmZ ddlZddl m
|
Z ddl m Z mZ ddlmZmZmZmZmZmZmZmZ ddlmZ dd Zejeeeeeeeegd d
|
d Z ejd d Z ejdd Z ejdd Z ejdd Z ejdd Z!ejdd Z"ejdd Z#ejdd 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 d(d) d)ej1Z2G d*d+ d+ej3Z4G d,d- d-ej5Z6G d.d/ d/ej7Z8G d0d1 d1ej9Z:G d2d3 d3ej;Z<G d4d5 d5ej=Z>G d6d7 d7ej?Z@ejAjBd8d9G d:d; d;ejCZDG d<d= d=ejEZFG d>d? d?ejGZHG d@dA dAejIZJG dBdC dCejKZLdS )Daj
|
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)ÚIS64Úis_platform_windows)Úis_extension_array_dtypeÚis_integer_dtype)Ú Int8DtypeÚ
|
Int16DtypeÚ
|
Int32DtypeÚ
|
Int64DtypeÚ
|
UInt8DtypeÚ UInt16DtypeÚ UInt32DtypeÚ UInt64Dtype)Úbasec C s4 t tddtjg t tdd tjg ddg S )Né é é
|
éb éc éd )ÚlistÚrangeÚpdÚNA© r r úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/extension/test_integer.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 )Nr é r! )r r" ÚnpZonesr! r r r Ú data_for_twosD s r& c C s t jt jdg| dS )Nr r! ©r r" r r! r r r Ú data_missingI s r( c C s t jdddg| dS )Nr r$ r r! )r r" r! r r r Údata_for_sortingN s r) c C s t jdt jdg| dS )Nr r r! r' r! r r r Údata_missing_for_sortingS s r* c C s dd S )Nc S s | t jko|t jkS r ©r r )ÚxÚyr r r Ú<lambda>[ ó zna_cmp.<locals>.<lambda>r r r r r Úna_cmpX s r0 c C s t jS r r+ r r r r Úna_value^ s r1 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_groupingc s
|
r6 c @ s e Zd ZdS )Ú TestDtypeN©Ú__name__Ú
|
__module__Ú __qualname__r r r r r7 l s r7 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Úotherr> ©Ú __class__r r rA q s z TestArithmeticOps.check_opnamec C s² |d krt |¡}t|drBt|jsBt|jrB|jrB| |j¡}|||}| |||¡}|dkrv|
|
t j ¡ d¡}n
|
| |¡}| ||¡ n t |¡ ||| W 5 Q R X d S )Nr )Ú __rtruediv__Ú __truediv__Z__div__ZFloat64)ÚtmZ get_dtypeÚhasattrr r r Zis_unsigned_integerÚastypeZ numpy_dtypeZ_combineZfillnar% ÚnanZ assert_equalÚpytestÚraises) rC rD ÚoprF rE r> ZsdtypeÚresultÚexpectedr r r Ú _check_opu s&
|
ÿþýü
|
|
zTestArithmeticOps._check_opc s t |||d ¡ d S r )r@ Ú_check_divmod_op)rC rD rQ rF r> rG r r rU s z"TestArithmeticOps._check_divmod_op)N)N)r9 r: r; rA ÚNotImplementedErrorrT rU Ú __classcell__r r rG r r< p s r< 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)ÚcombinerM Úassert_series_equalrO rP )rC rD rQ rF rE r> rR rS r r r rT s
|
zTestComparisonOps._check_opNc s t j|||d d d S r= r? rB rG r r rA £ s z TestComparisonOps.check_opnamec C s d|j d}| |||¡ d S )NÚ__)r9 rA )rC rD r# rQ rF rE r r r Ú_compare_other¦ s z TestComparisonOps._compare_other)N)r9 r: r; rV rT rA r] rW r r rG r rX s
|
rX c @ s e Zd ZdS )Ú TestInterfaceNr8 r r r r r^ « s r^ c @ s e Zd ZdS )ÚTestConstructorsNr8 r r r r r_ ¯ s r_ c @ s e Zd ZdS )Ú TestReshapingNr8 r r r r r` ³ s r` c @ s e Zd ZdS )Ú TestGetitemNr8 r r r r ra » s ra c @ s e Zd ZdS )Ú TestSetitemNr8 r r r r rb ¿ s rb c @ s e Zd ZdS )Ú TestIndexNr8 r r r r rc à s rc c @ s e Zd ZdS )Ú TestMissingNr8 r r r r rd Ç s rd c @ s e Zd ZeZdS )Ú TestMethodsN)r9 r: r; ÚobjectZ_combine_le_expected_dtyper r r r re Ë s re c @ s e Zd ZdS )Ú TestCastingNr8 r r r r rg Ï s rg c @ s e Zd ZdS )Ú TestGroupbyNr8 r r r r rh Ó s rh c @ s e Zd Zdd ZdS )ÚTestNumericReducec C s| |dkr,t || }t | ¡ d¡| }n@t |||d}t | ¡ d¡||d}|sl| ¡ ¡ rltj}t ||¡ d S )NÚcountÚint64©Úskipna) ÚgetattrZdropnarM ZisnaÚanyr r rK Zassert_almost_equal)rC rD rE rm rR rS r r r Ú check_reduceØ s z TestNumericReduce.check_reduceN)r9 r: r; rp r r r r ri × s ri z-Tested in tests/reductions/test_reductions.py)Úreasonc @ s e Zd ZdS )ÚTestBooleanReduceNr8 r r r r rr æ s rr c @ s. e Zd Zdd Zej dddg¡dd ZdS ) ÚTestAccumulationc C sL d}t rt r |jjdks d}|jj d¡r8d| }n
|
d| }|dkrt|||d}t tj t|
|
d ¡||d|d
|
¡}t ||¡ nº|d krÜt|||d}t tj t|
|
d ¡||d|jd
|
¡}t ||¡ nl|d kr:t|d d
||d}t tj t|d d
|
d ¡||d|d
|
¡}t ||¡ nt | dd S )Né@ é é ÚUZUIntZIntZcumsumrl Úfloat64r! )ZcummaxZcumminZcumprodé z not supported)r r r ÚitemsizeÚnameÚ
|
startswithrn r ZSeriesr" rM rK r[ rV )rC rD rE rm ÚlengthZexpected_dtyperR rS r r r Úcheck_accumulateì sF
|
|
þÿþÿ
|
þÿz!TestAccumulation.check_accumulaterm TFc C s d S r r )rC r# Zall_numeric_accumulationsrm r r r Ú test_accumulate_series_raises s z.TestAccumulation.test_accumulate_series_raisesN)r9 r: r; r~ rO ÚmarkZ parametrizer r r r r rs ë s ,rs c @ s e Zd ZdS )Ú TestPrintingNr8 r r r r r s r c @ s e Zd ZdS )Ú TestParsingNr8 r r r r r ! s r c @ s e Zd ZdS )Ú Test2DCompatNr8 r r r r r % s r )MÚ__doc__Únumpyr% rO Z pandas.compatr r Zpandasr Zpandas._testingZ_testingrK Zpandas.api.typesr r Zpandas.core.arrays.integerr r r r r
|
r r r Zpandas.tests.extensionr r Zfixturer r# r& r( r) r* r0 r1 r6 ZBaseDtypeTestsr7 ZBaseArithmeticOpsTestsr< ZBaseComparisonOpsTestsrX ZBaseInterfaceTestsr^ ZBaseConstructorsTestsr_ ZBaseReshapingTestsr` ZBaseGetitemTestsra ZBaseSetitemTestsrb ZBaseIndexTestsrc ZBaseMissingTestsrd ZBaseMethodsTestsre ZBaseCastingTestsrg ZBaseGroupbyTestsrh ZBaseNumericReduceTestsri r ÚskipZBaseBooleanReduceTestsrr ZBaseAccumulateTestsrs ZBasePrintingTestsr ZBaseParsingTestsr ZDim2CompatTestsr r r r r Ú<module> sr (
|
øÿ
|
|
|
|
|
|
|
|
|
( 2
|
