U
|
ý°d5 ã @ s> d Z ddlZddlZddlmZ ddlZddlm Z
|
ddl m Z ddl mZ dd Zejdd Zejd
|
d Zejd d Zejdd Zejdd Zejdd Zejdd Zejdd Zejdd ZG dd dejZG 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.Z/G d0d1 d1ej0Z1G d2d3 d3ej2Z3G d4d5 d5ej4Z5G d6d7 d7ej6Z7G d8d9 d9ej8Z9G d:d; d;ej:Z;G d<d= d=ej<Z=G d>d? d?ej>Z?G d@dA dAej@ZAdS )Baj
|
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_bool_dtype©Ú BooleanDtype)Úbasec C s0 ddgd t jg ddgd t jg ddg S )NTFé é, )ÚnpÚnan© r
|
r
|
úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/extension/test_boolean.pyÚ make_data s r c C s t S ©Nr r
|
r
|
r
|
r Údtype s r c C s t jt | dS )N©r )ÚpdÚarrayr r r
|
r
|
r Údata$ s r c C s t jt d¡| dS )Néd r )r r r Zonesr r
|
r
|
r Ú data_for_twos) s r c C s t jtjdg| dS )NTr ©r r r r r r
|
r
|
r Ú data_missing. s r c C s t jdddg| dS ©NTFr )r r r r
|
r
|
r Údata_for_sorting3 s r c C s t jdtjdg| dS r r r r
|
r
|
r Údata_missing_for_sorting8 s r c C s dd S )Nc S s | t jko|t jkS r ©r ÚNA)ÚxÚyr
|
r
|
r Ú<lambda>@ ó zna_cmp.<locals>.<lambda>r
|
r
|
r
|
r
|
r Úna_cmp= s r c C s t jS r r r
|
r
|
r
|
r Úna_valueC s r! c C s* d}d}t j}tj|||||||g| dS r )r r r r )r ÚbÚaÚnar
|
r
|
r Údata_for_groupingH s r% c @ s e Zd ZdS )Ú TestDtypeN©Ú__name__Ú
|
__module__Ú __qualname__r
|
r
|
r
|
r r& P s r& c @ s e Zd ZdS )Ú TestInterfaceNr' r
|
r
|
r
|
r r+ T s r+ c @ s e Zd ZdS )ÚTestConstructorsNr' r
|
r
|
r
|
r r, X s r, c @ s e Zd ZdS )Ú TestGetitemNr' 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 )Ú TestIndexNr' r
|
r
|
r
|
r r/ d s r/ c @ s e Zd ZdS )Ú TestMissingNr' r
|
r
|
r
|
r r0 h s r0 c sf e Zd ZddhZd fdd ZefddZejj dd fd
|
d Z
|
ejj dd fd d Z Z S )ÚTestArithmeticOpsÚ__sub__Ú__rsub__Nc s4 d }| d¡ d¡dkr t}t j||||d d S )NÚ_Úr)ÚpowÚtruedivÚfloordiv©Úexc)ÚstripÚlstripÚNotImplementedErrorÚsuperÚ check_opname©ÚselfÚsÚop_nameÚotherr: ©Ú __class__r
|
r r? o s z TestArithmeticOps.check_opnamec C sØ |d kr´|| j kr>d}tjt|d ||| W 5 Q R X d S |||}| |||¡}|dkrj| d¡}n&|dkr| | t¡||¡}| d¡}|dkr¦tj|| ¡ < |
|
||¡ n t |¡ ||| W 5 Q R X d S )Nznumpy boolean subtract)Úmatch)Ú __floordiv__Ú __rfloordiv__Ú__pow__Ú__rpow__Ú__mod__Ú__rmod__ZInt8)Ú __truediv__Ú __rtruediv__ZFloat64rK ) Ú
|
implementsÚpytestZraisesÚ TypeErrorZ_combineÚastypeÚfloatr r ÚisnaZ assert_equal) rA ÚobjÚoprD rC r: ÚmsgÚresultÚexpectedr
|
r
|
r Ú _check_opw s$
|
|
|
zTestArithmeticOps._check_opzInconsistency between floordiv and divmod; we raise for floordiv but not for divmod. This matches what we do for non-masked bool dtype.)Úreasonc s t ||¡ d S r )r> Útest_divmod_series_array)rA r r rE r
|
r r] s z*TestArithmeticOps.test_divmod_series_arrayc s t |¡ d S r )r> Ú test_divmod)rA r rE r
|
r r^ s z TestArithmeticOps.test_divmod)N) r( r) r* rP r? r= r[ rQ ÚmarkZxfailr] r^ Ú __classcell__r
|
r
|
rE r r1 l s "ÿÿr1 c s e Zd Zd fdd Z ZS )ÚTestComparisonOpsNc s t j|||d d d S )Nr9 )r> r? r@ rE r
|
r r? © s z TestComparisonOps.check_opname)N)r( r) r* r? r` r
|
r
|
rE r ra ¨ s ra c @ s e Zd ZdS )Ú TestReshapingNr' r
|
r
|
r
|
r rb ® s rb c @ s( e Zd ZdZdd Zdd Zdd ZdS ) Ú TestMethodsÚbooleanc C sZ t j|dd\}}tjdddddddgtjd}| ddg¡}t ||¡ | ||¡ d S )NT)Zuse_na_sentinelr éÿÿÿÿé r r ) r Ú factorizer r ÚintpÚtakeÚtmÚassert_numpy_array_equalÚ assert_extension_array_equal)rA r% ÚlabelsÚuniquesZexpected_labelsZexpected_uniquesr
|
r
|
r Útest_factorizeµ s
|
zTestMethods.test_factorizec C sæ t jddgdd}|\}}t| ||g¡}|r:t |¡}| |¡dksLt|j|dddksbt| |¡dkstt|j|ddd kst| | ddg¡¡}tjddgtj d}t
|
||¡ t ddg¡}|j||d
|
dksâtd S ) NTFrd r r Úright)Zsiderf é )Úsorter) r r ÚtypeZ_from_sequenceÚSeriesZ searchsortedÚAssertionErrorri r rh rj rk ) rA r Z as_seriesr" r# ZarrrY rZ rr r
|
r
|
r Útest_searchsorted¾ s
|
z TestMethods.test_searchsortedc C sz | ¡ dkst| ¡ dks t| ddddddg¡}| ¡ dksFt| ¡ dksVt| ¡ dksft| ¡ dksvtd S )Nr rq rf )Zargmaxru Zargminri )rA r r r r
|
r
|
r Útest_argmin_argmaxÕ s z TestMethods.test_argmin_argmaxN)r( r) r* Z_combine_le_expected_dtypero rv rw r
|
r
|
r
|
r rc ² s rc c @ s e Zd ZdS )Ú TestCastingNr' r
|
r
|
r
|
r rx ç s rx c @ s| e Zd ZdZdd Zej dddg¡dd Zd d
|
Z d d Z
|
d d Z dd Z dd Z dd Zej dddg¡dd ZdS )Ú TestGroupbyzw
|
Groupby-specific tests are overridden because boolean only has 2
|
unique values, base tests uses 3 groups.
|
c C sf t ddd d dddg|d¡}| d¡jjd }| d¡jjd }t |j|jj ¡ t
|
|j|¡ d S )NÚBÚA©r{ rz r ) r Ú DataFrameÚgroupbyZgrouperZ groupingsrj rk Zgrouping_vectorr{ Úvaluesrl )rA r% ÚdfZgr1Zgr2r
|
r
|
r Útest_grouping_grouperñ s ÿz!TestGroupby.test_grouping_grouperÚas_indexTFc C s t dddddddg|d¡}|jd|dj ¡ }t j|dd\}}|rxt j|dd }t jd
|
d g|d d }| ||¡ n t |d
|
d gd¡}| ||¡ d S )Nrf rq é r| rz )r T©Úsort©Únameç @ç ð?r{ ©Úindexr )rz r{ )
|
r r} r~ r{ Úmeanrg ÚIndexrt Úassert_series_equalÚassert_frame_equal) rA r r% r rY r4 rn r rZ r
|
r
|
r Útest_groupby_extension_aggû s z&TestGroupby.test_groupby_extension_aggc C s t dddddddg|d¡}|jdddg }| d¡}| d¡ dd i¡}| ||¡ | d¡ d ¡}| ||¡ | d¡ ¡ }| ||¡ d S )
|
Nrf rq r r| r r r{ rz Úfirst)r r} ÚilocZ set_indexr~ Zaggr r )rA r% r rZ rY r
|
r
|
r Útest_groupby_agg_extension s
|
z&TestGroupby.test_groupby_agg_extensionc C sv t dddddddg|d¡}|jdddj ¡ }t j|dd\}}t j|dd}t jd d
|
g|d d }| ||¡ d S ) Nrf rq r r| rz Fr r r r r{ r ) r r} r~ r{ r rg r rt r )rA r% r rY r4 r rZ r
|
r
|
r Ú test_groupby_extension_no_sort s z*TestGroupby.test_groupby_extension_no_sortc C sb || ¡ }t dddddg|d¡}| d¡j t¡}tjdddddgdd}| ||¡ d S )Nrf r r| rz rq r{ r ) rU r r} r~ r{ Z transformÚlenrt r )rA r% Zvalidr rY rZ r
|
r
|
r Ú test_groupby_extension_transform" s
|
z,TestGroupby.test_groupby_extension_transformc C sv t dddddddg|d¡}|jddd |¡ |jdddj |¡ |jddd |¡ |jdddj |¡ d S ) Nrf rq r r| rz F)Z
|
group_keysr{ )r r} r~ Úapplyr{ rz )rA r% Zgroupby_apply_opr r
|
r
|
r Ú test_groupby_extension_apply+ s
|
z(TestGroupby.test_groupby_extension_applyc C s t dddddddg|d¡}| d¡j dd ¡}t j|jjddd g j|jjddg j|jjd
|
d g jgt jdddgdd d d}| ||¡ d S )Nrf rq r r| r{ c S s | j S r )r )r r
|
r
|
r r 4 r z9TestGroupby.test_groupby_apply_identity.<locals>.<lambda>r é r é r rz r )
|
r r} r~ rz r rt r r r r ©rA r% r rY rZ r
|
r
|
r Útest_groupby_apply_identity2 s ýù z'TestGroupby.test_groupby_apply_identityc C sr t dddddddg|dddddddgd¡}| d¡ ¡ j}|jjrVt ddg¡}n t dg¡}t ||¡ d S )Nrf rq r )r{ rz ÚCr{ rz r )
|
r r} r~ ÚsumÚcolumnsr Z _is_numericr rj Zassert_index_equalr r
|
r
|
r Útest_in_numeric_groupby@ s ýÿ z#TestGroupby.test_in_numeric_groupbyÚ min_countr é
|
c C s¼ t dddddddg|d¡}| d¡j|d}|dkrxt jdt jdddgd d
|
it jdddgdd d }t ||¡ n@t jdt jt jgd d d
|
it jdddgdd d }t ||¡ d S ) Nrf rq r r| r{ )r¡ r rz ZInt64r r )r ) r r} r~ r r r rj r r )rA r% r¡ r rY rZ r
|
r
|
r Útest_groupby_sum_mincountQ s þþz%TestGroupby.test_groupby_sum_mincountN)r( r) r* Ú__doc__r rQ r_ Ú parametrizer r r r r r r r£ r
|
r
|
r
|
r ry ë s
|
|
ry c @ s e Zd Zdd ZdS )ÚTestNumericReducec C s |dkr(t || }t | d¡| }n&t |||d}t | d¡||d}t |¡r`tj}n|dkrpt|}t ||¡ d S )NÚcountÚfloat64©Úskipna)ÚminÚmax) ÚgetattrrS r Úisnanr r Úboolrj Zassert_almost_equal©rA rB rC rª rY rZ r
|
r
|
r Ú check_reduced s
|
z TestNumericReduce.check_reduceN)r( r) r* r± r
|
r
|
r
|
r r¦ c s r¦ c @ s e Zd ZdS )ÚTestBooleanReduceNr' r
|
r
|
r
|
r r² s s r² c @ s e Zd ZdS )Ú TestPrintingNr' r
|
r
|
r
|
r r³ w s r³ c @ s e Zd ZdS )Ú TestUnaryOpsNr' r
|
r
|
r
|
r r´ { s r´ c @ s. e Zd Zdd Zej dddg¡dd ZdS ) ÚTestAccumulationc C sT t |||d}t t | d¡¡||d}tj||dd |dkrPt|sPtd S )Nr© r¨ F)Z check_dtype)ZcumminZcummax)r r rt rS rj r r ru r° r
|
r
|
r Úcheck_accumulate s
|
z!TestAccumulation.check_accumulaterª TFc C s d S r r
|
)rA r Zall_numeric_accumulationsrª r
|
r
|
r Ú test_accumulate_series_raises s z.TestAccumulation.test_accumulate_series_raisesN)r( r) r* r¶ rQ r_ r¥ r· r
|
r
|
r
|
r rµ s rµ c @ s e Zd ZdS )Ú TestParsingNr' r
|
r
|
r
|
r r¸ s r¸ c @ s e Zd ZdS )Ú Test2DCompatNr' r
|
r
|
r
|
r r¹ s r¹ )Br¤ Únumpyr rQ Zpandas.core.dtypes.commonr Zpandasr Zpandas._testingZ_testingrj Zpandas.core.arrays.booleanr Zpandas.tests.extensionr r Zfixturer r r r r r r r! r% ZBaseDtypeTestsr& ZBaseInterfaceTestsr+ ZBaseConstructorsTestsr, ZBaseGetitemTestsr- ZBaseSetitemTestsr. ZBaseIndexTestsr/ ZBaseMissingTestsr0 ZBaseArithmeticOpsTestsr1 ZBaseComparisonOpsTestsra ZBaseReshapingTestsrb ZBaseMethodsTestsrc ZBaseCastingTestsrx ZBaseGroupbyTestsry ZBaseNumericReduceTestsr¦ ZBaseBooleanReduceTestsr² ZBasePrintingTestsr³ ZBaseUnaryOpsTestsr´ ZBaseAccumulateTestsrµ ZBaseParsingTestsr¸ ZDim2CompatTestsr¹ r
|
r
|
r
|
r Ú<module> s\
|
|
|
|
|
|
|
|
|
<5x
|
