U
|
ý°dÐ= ã @ s d Z ddlZddlZddlmZ ddlmZmZ ddl Z
|
ddl m Z ddlmZ ddlmZ ddlmZ ddlmZ ed d
|
d Ze jZdIeed ddZej ddgddd Z ej dd Z ej dd Z ej dd Z ej dd Z!ej d d Z"ej d d Z#ej d!d" Z$ej d#d$ Z%ej d%d& Z&ej' (d&¡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/Z0G d/d0 d0e*ej1Z2G d1d2 d2e*ej3Z4G d3d4 d4e*ej5Z6G d5d6 d6e*ej7Z8G d7d8 d8e*ej9Z:G d9d: d:e*ej;Z<G d;d< d<e*ej=Z>e)G d=d> d>e*ej?Z@G d?d@ d@e*ejAZBG dAdB dBe*ejCZDG dCdD dDe*ejEZFe)G dEdF dFe*ejGZHG dGdH dHe*ejIZJdS )JaÎ
|
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/`.
|
|
Note: we do not bother with base.BaseIndexTests because PandasArray
|
will never be held in an Index.
|
é N)Úcan_hold_element)ÚExtensionDtypeÚ PandasDtype)Úis_object_dtype)Ú PandasArray)Úblocks)Úbase)Úreturnc C s t |tr| ¡ }t| |S ©N)Ú
|
isinstancer Úto_numpyr )ÚobjÚelement© r úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/extension/test_numpy.pyÚ_can_hold_element_patched# s
|
r Ú
|
Attributes)Úattrr c C st | dkrbt |dd}t |dd}t|trBt|tsB| |j¡}n t|trbt|tsb| |j¡}t| ||| dS )zh
|
patch tm.assert_attr_equal so PandasDtype("object") is closed enough to
|
np.dtype("object")
|
ÚdtypeN)Úgetattrr r ÚastypeÚ numpy_dtypeÚorig_assert_attr_equal)r ÚleftÚrightr ZlattrZrattrr r r Ú_assert_attr_equal, s r ÚfloatÚobject)Úparamsc C s t t | j¡S r
|
)r Únpr Úparam)Úrequestr r r r < s r c c sJ | ¡ 8}| tdd¡ | tdt¡ | tjdt¡ dV W 5 Q R X dS )aó
|
A monkeypatch to tells pandas to let us in.
|
|
By default, passing a PandasArray to an index / series / frame
|
constructor will unbox that PandasArray to an ndarray, and treat
|
it as a non-EA column. We don't want people using EAs without
|
reason.
|
|
The mechanism for this is a check against ABCPandasArray
|
in each constructor.
|
|
But, for testing, we need to allow them in pandas. So we patch
|
the _typ of PandasArray, so that we evade the ABCPandasArray
|
check.
|
Z_typÚ extensionr Úassert_attr_equalN)ÚcontextÚsetattrr r r ÚtmZ assertersr )Z monkeypatchÚmr r r Úallow_in_pandasA s
|
|
r( c C s: |j dkr$t dd tdD ¡jS ttjdd|jdS )Nr c S s g | ]
|
}|fqS r r )Ú.0Úir r r Ú
|
<listcomp>\ s zdata.<locals>.<listcomp>éd é ée ©r ) r ÚpdÚSeriesÚrangeÚarrayr r ÚarangeÚ_dtype©r( r r r r ÚdataY s
|
r7 c C s6 |j dkr"ttjtjdgtdS tt tjdg¡S )Nr ©r- r/ g ð?©r r r r3 Únanr r6 r r r Ú data_missing` s
|
r; c C s t jS r
|
)r r: r r r r Úna_valueg s r< c C s dd } | S )Nc S s t | ¡ot |¡S r
|
)r Úisnan)ÚaÚbr r r Úcmpn s zna_cmp.<locals>.cmpr )r@ r r r Úna_cmpl s rA c C s@ |j dkr,ttjddddgtddd
S tt dd d
|
g¡S ) ziLength-3 array with a known sort order.
|
|
This should be three items [B, C, A] with
|
A < B < C
|
r r ©é ©é r8 r/ r- NrC r )r r r r3 r r6 r r r Údata_for_sortingt s
|
"rF c C s: |j dkr$ttjdtjdgtdS tt dtjdg¡S )zvLength-3 array with a known sort order.
|
|
This should be three items [B, NA, A] with
|
A < B and NA missing.
|
r r8 )r r/ r- r r9 r6 r r r Údata_missing_for_sorting s
|
rG c
|
C sN |j dkrd\}}}nt d¡\}}}ttj||tjtj||||g|j dS )zData for factorization, grouping, and unique tests.
|
|
Expected to be like [B, B, NA, NA, A, A, B, C]
|
|
Where A < B < C and NA is missing
|
r )r8 rB rD rE r/ )r r r4 r r3 r: )r( r r> r? Úcr r r Údata_for_grouping s
|
"ÿrI c C s& | dkr"t jjdd}|j |¡ dS )a
|
Tests for PandasArray with nested data. Users typically won't create
|
these objects via `pd.array`, but they can show up through `.array`
|
on a Series with nested data. Many of the base tests fail, as they aren't
|
appropriate for nested data.
|
|
This fixture allows these tests to be skipped when used as a usefixtures
|
marker to either an individual test or a test class.
|
r úFails for object dtype©ÚreasonN)ÚpytestÚmarkÚxfailÚnodeÚ
|
add_marker)r r! rN r r r Úskip_numpy_object s rR c @ s e Zd Zedd ZdS )ÚBaseNumPyTestsc O sH t |tjr4t |jts4t |jtr4| t|j¡}tj||f||S r
|
) r r0 r1 r r r r r& Úassert_series_equal)Úclsr r ÚargsÚkwargsr r r rT ³ s
|
ÿ
|
þ
|
ýz"BaseNumPyTests.assert_series_equalN)Ú__name__Ú
|
__module__Ú __qualname__Ú classmethodrT r r r r rS ² s rS c @ s e Zd ZdS )Ú TestCastingN©rX rY rZ r r r r r\ À s r\ c s6 e Zd Zejjdddd Ze fddZ Z S )ÚTestConstructorszWe don't register our dtyperK c C s d S r
|
r ©Úselfr7 r r r Útest_from_dtypeÅ s z TestConstructors.test_from_dtypec s t ||¡ d S r
|
)ÚsuperÚ)test_series_constructor_scalar_with_index)r` r7 r ©Ú __class__r r rc Ê s z:TestConstructors.test_series_constructor_scalar_with_index)
|
rX rY rZ rM rN Úskipra Ú skip_nestedrc Ú __classcell__r r rd r r^ Ä s
|
r^ c s( e Zd Z fddZ fddZ ZS )Ú TestDtypec s< |j jdkr,|j tjjd|j j d¡ t |¡ d S )Nr z2PandasArray expectedly clashes with a NumPy name: rK ) r r rP rQ rM rN rO rb Útest_check_dtype)r` r7 r! rd r r rj Ñ s ÿÿzTestDtype.test_check_dtypec s( |j dkrt|s$tn t |¡ d S )Nr )r r ÚAssertionErrorrb Útest_is_not_object_type)r` r r! rd r r rl Û s
|
z!TestDtype.test_is_not_object_type)rX rY rZ rj rl rh r r rd r ri Ð s
|
ri c s e Zd Ze fddZ ZS )Ú TestGetitemc s t |¡ d S r
|
)rb Útest_getitem_scalarr_ rd r r rn å s z TestGetitem.test_getitem_scalar)rX rY rZ rg rn rh r r rd r rm ä s rm c @ s e Zd ZdS )Ú TestGroupbyNr] r r r r ro ë s ro c s e Zd Ze fddZ ZS )Ú TestInterfacec s t |¡ d S r
|
)rb Útest_array_interfacer_ rd r r rq ð s z"TestInterface.test_array_interface)rX rY rZ rg rq rh r r rd r rp ï s rp c s e Zd Ze fddZe fddZe fddZe fddZej j
|
d d
|
fd d Z fd dZ e fddZ ZS )Ú TestMethodsc s t |¡ d S r
|
)rb Útest_shift_fill_valuer_ rd r r rs ÷ s z!TestMethods.test_shift_fill_valuec s t |¡ d S r
|
)rb Útest_fillna_copy_frame©r` r; rd r r rt ü s z"TestMethods.test_fillna_copy_framec s t |¡ d S r
|
)rb Útest_fillna_copy_seriesru rd r r rv s z#TestMethods.test_fillna_copy_seriesc s t ||¡ d S r
|
)rb Útest_searchsorted)r` rF Z as_seriesrd r r rw s z TestMethods.test_searchsortedz"PandasArray.diff may fail on dtyperK c s t ||¡S r
|
)rb Ú test_diff)r` r7 Zperiodsrd r r rx s zTestMethods.test_diffc s6 |j jtkr&tjjdd}|j |¡ t |¡ d S )Nz$Dimension mismatch in np.concatenaterK )
|
r r r rM rN rO rP rQ rb Ú test_insert)r` r7 r! rN rd r r ry s zTestMethods.test_insertc s t ||¡ d S r
|
)rb Útest_insert_invalid©r` r7 Zinvalid_scalarrd r r rz s z TestMethods.test_insert_invalid)rX rY rZ rg rs rt rv rw rM rN rO rx ry rz rh r r rd r rr ö s rr c sh e Zd ZdZdZdZdZe fddZedd Z e fddZ
|
fdd Z e fd
|
d Z Z S ) ÚTestArithmeticsNc s t |¡ d S r
|
)rb Ú test_divmodr_ rd r r r} " s zTestArithmetics.test_divmodc C s t |¡}| j|t|d d d S )N)Úexc)r0 r1 Z_check_divmod_opÚdivmod)r` r7 Zserr r r Útest_divmod_series_array& s
|
z(TestArithmetics.test_divmod_series_arrayc s t ||¡ d S r
|
)rb Ú test_arith_series_with_scalar©r` r7 Úall_arithmetic_operatorsrd r r r + s z-TestArithmetics.test_arith_series_with_scalarc sD |}|j jtkr2|dkr2tjjdd}|j |¡ t ||¡ d S )N)Ú__add__Ú__radd__rJ rK )
|
r r r rM rN rO rP rQ rb Ú test_arith_series_with_array)r` r7 r r! ÚopnamerN rd r r r / s
|
z,TestArithmetics.test_arith_series_with_arrayc s t ||¡ d S r
|
)rb Ú test_arith_frame_with_scalarr rd r r r 6 s z,TestArithmetics.test_arith_frame_with_scalar)rX rY rZ Z
|
divmod_excZseries_scalar_excZframe_scalar_excZseries_array_excrg r} r r r r rh r r rd r r| s
|
r| c @ s e Zd ZdS )Ú TestPrintingNr] r r r r r ; s r c s6 e Zd Zdd Zej dddg¡ fddZ ZS )ÚTestNumericReducec C s: t |||d}t | |jj¡||d}t ||¡ d S )N)Úskipna)r r r r5 r& Zassert_almost_equal)r` ÚsZop_namer ÚresultÚexpectedr r r Ú check_reduce@ s z TestNumericReduce.check_reducer TFc s t |||¡ d S r
|
)rb Útest_reduce_series)r` r7 Zall_boolean_reductionsr rd r r r F s z$TestNumericReduce.test_reduce_series) rX rY rZ r rM rN Ú parametrizer rh r r rd r r ? s r c @ s e Zd ZdS )ÚTestBooleanReduceNr] r r r r r K s r c s0 e Zd Ze fddZe fddZ ZS )Ú TestMissingc s t |¡ d S r
|
)rb Útest_fillna_seriesru rd r r r Q s z TestMissing.test_fillna_seriesc s t |¡ d S r
|
)rb Útest_fillna_frameru rd r r r V s z TestMissing.test_fillna_frame)rX rY rZ rg r r rh r r rd r r P s r c sB e Zd Zej ddejdejjdddg¡ fddZ Z S ) Ú TestReshapingÚin_frameTFz$PandasArray inconsistently extractedrK ©Zmarksc s t ||¡ d S r
|
)rb Ú test_concat)r` r7 r rd r r r ] s zTestReshaping.test_concat)
|
rX rY rZ rM rN r r rO r rh r r rd r r \ s þþþ
|
r c
|
s® e Zd Ze fddZe fddZeej dddg¡ fdd Z e fd
|
d Z
|
eejjd e d d d ddg¡e j d d d ddgddgddgd fddZeejjddddge j dddgdde dddg¡gdd dgd fd d Zejjd ddde jgdfejddde jgd ejjd e j ddde jgdddfe j ddde jgdddfgd!d"d#d$gd fd%d&Ze fd'd(Ze fd)d*Zd+d, Z ZS )-Ú TestSetitemc s t ||¡ d S r
|
)rb Útest_setitem_invalidr{ rd r r r l s z TestSetitem.test_setitem_invalidc s t ||¡ d S r
|
)rb Ú test_setitem_sequence_broadcasts©r` r7 Ú box_in_seriesrd r r r q s z,TestSetitem.test_setitem_sequence_broadcastsÚsetterÚlocNc s t ||¡ d S r
|
)rb Útest_setitem_mask_broadcast)r` r7 r rd r r r¡ w s z'TestSetitem.test_setitem_mask_broadcastc s t |¡ d S r
|
)rb Ú&test_setitem_scalar_key_sequence_raiser_ rd r r r¢ ~ s z2TestSetitem.test_setitem_scalar_key_sequence_raiseÚmaskTFÚbooleanr/ z numpy-arrayz boolean-array)Úidsc s t |||¡ d S r
|
)rb Útest_setitem_mask)r` r7 r£ r rd r r r¦ s
|
z TestSetitem.test_setitem_maskÚidxr r- rC ZInt64Úlistz integer-arrayc s t |||¡ d S r
|
)rb Útest_setitem_integer_array©r` r7 r§ r rd r r r© s z&TestSetitem.test_setitem_integer_arrayzidx, box_in_seriesr z
|
list-Falsez list-Truezinteger-array-Falsezinteger-array-Truec s t |||¡ d S r
|
)rb Ú(test_setitem_integer_with_missing_raisesrª rd r r r« s z4TestSetitem.test_setitem_integer_with_missing_raisesc s t ||¡ d S r
|
)rb Útest_setitem_slicer rd r r r¬ ¨ s z TestSetitem.test_setitem_slicec s t |¡ d S r
|
)rb Útest_setitem_loc_iloc_slicer_ rd r r r ¬ s z'TestSetitem.test_setitem_loc_iloc_slicec C s t dt |¡i¡ }}t j|jd}||}|d |j|df< |jjtkrtt|t rb|t d krtt d|
|
¡ i¡}| ||¡ d S )Nr7 )Úindex) r0 Z DataFramer1 r® r r r r r Úslicer Zassert_frame_equal)r` r7 Z full_indexerZdfr r Úkeyr r r Ú,test_setitem_with_expansion_dataframe_column° s z8TestSetitem.test_setitem_with_expansion_dataframe_column)rX rY rZ rg r r rM rN r r¡ r¢ r r3 r0 r¦ r© ZNAr rO r« r¬ r r± rh r r rd r r k sP þú*ý ü
|
ø
|
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³ )r )KÚ__doc__Únumpyr rM Zpandas.core.dtypes.castr Zpandas.core.dtypes.dtypesr r Zpandasr0 Zpandas._testingZ_testingr& Zpandas.api.typesr Zpandas.core.arrays.numpy_r Zpandas.core.internalsr Zpandas.tests.extensionr Úboolr r# r Ústrr Zfixturer r( r7 r; r< rA rF rG rI rR rN Z usefixturesrg rS ZBaseCastingTestsr\ ZBaseConstructorsTestsr^ ZBaseDtypeTestsri ZBaseGetitemTestsrm ZBaseGroupbyTestsro ZBaseInterfaceTestsrp ZBaseMethodsTestsrr ZBaseArithmeticOpsTestsr| ZBasePrintingTestsr ZBaseNumericReduceTestsr ZBaseBooleanReduceTestsr ZBaseMissingTestsr ZBaseReshapingTestsr ZBaseSetitemTestsr ZBaseParsingTestsr² ZNDArrayBacked2DTestsr³ r r r r Ú<module> sj
|
|
|
|
|
|
|
|
|
|
& W
|
