Ë
|
oñúhâ< ã ó d Z ddlZddlZddlmZ ddlZddlm Z
|
ddl m Z ddl mZ ddlmZ e
|
j" Zd dedefd Z ej* d
|
d g¬ « d « Zej* d « Zej* d « Zej* d « Zej* d « Zej* d « Zej* d « Z ej* d « Z ej* d « Z ej* d « Z ej@ jC d« Z" G d dejF « Z$ G d dejJ « Z&y) aÖ
|
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 NumpyExtensionArray
|
will never be held in an Index.
|
é N)Ú NumpyEADtype)Úis_object_dtype)ÚNumpyExtensionArray)ÚbaseÚattrÚobjc óL | dk( rt |dd« }t |dd« }t |t « r,t |t « s |j |j « }n;t |t « r+t |t « s|j |j « }t | |||« y)zi
|
patch tm.assert_attr_equal so NumpyEADtype("object") is closed enough to
|
np.dtype("object")
|
ÚdtypeN)ÚgetattrÚ
|
isinstancer ÚastypeÚ numpy_dtypeÚorig_assert_attr_equal)r ÚleftÚrightr ÚlattrÚrattrs úTH:\Change_password\venv_build\Lib\site-packages\pandas/tests/extension/test_numpy.pyÚ_assert_attr_equalr s ð
|
wÜg tÓ,Üw¨Ó-Ü e\Ô *´:¸eÄ\Ô3RØ;;u× 0Ñ 0Ó1DÜ |Ô ,´ZÀÄ|Ô5TØLL ×!2Ñ!2Ó3Eä4 u¨cÕ2ó ÚfloatÚobject)Úparamsc óR t t j | j « « S ©N)r Únpr
|
Úparam)Úrequests r r
|
r
|
0 s ä § ¡ Ó/Ó 0Ð0r c # óÖ K | j « 5 }|j t dd« |j t j dt
|
« d ddd« y# 1 sw Y yxY ww)a
|
A monkeypatch to tells pandas to let us in.
|
|
By default, passing a NumpyExtensionArray to an index / series / frame
|
constructor will unbox that NumpyExtensionArray 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 ABCNumpyExtensionArray
|
in each constructor.
|
|
But, for testing, we need to allow them in pandas. So we patch
|
the _typ of NumpyExtensionArray, so that we evade the ABCNumpyExtensionArray
|
check.
|
Ú_typÚ extensionÚassert_attr_equalN)ÚcontextÚsetattrr ÚtmÚ assertersr )Ú monkeypatchÚms r Úallow_in_pandasr) 5 sT è ø ð"
|
× Ñ Ó ð !Ø Ô% v¨{Ô;Ø ",,Ð 3Ô5GÔHÛ ÷÷ ñ üs A)AA Á A)Á A&Á"A)c óî |j dk( r7t j t d« D cg c] }|f c}« j S t t j dd|j ¬« « S c c}w )Nr éd é ée ©r
|
) r ÚpdÚSeriesÚrangeÚarrayr r ÚarangeÚ_dtype)r) r
|
Úis r Údatar6 L sW à ×ÑHÒ$Üyy¬¨c«
|
Ö3 11$Ò3Ó4×:Ñ:Ð:Ü ryy¨¨C°u·|±|Ô DÓ EÐEùò 4s ¬
|
A2c óä |j dk( r4t t j t j dgt
|
¬« « S t t j t j dg« « S )Nr ©r, r. g ð?©r r r r2 Únanr ©r) r
|
s r Ú data_missingr< S sJ à ×ÑHÒ$Ü"¤2§8¡8¬R¯V©V°T¨NÄ&Ô#IÓJÐJÜ rxx¬¯©°¨ Ó 6Ó 7Ð7r c ó d } | S )Nc óZ t j | « xr t j |« S r )r Úisnan)ÚaÚbs r Úcmpzna_cmp.<locals>.cmp\ s Üxx{Ò*rxx¨{Ð*r © )rB s r Úna_cmprD Z s ò+ð Jr c ó² |j dk( r)t t j g d¢t ¬« dd « S t t j g d¢« « S )ziLength-3 array with a known sort order.
|
|
This should be three items [B, C, A] with
|
A < B < C
|
r )rC ©é ©é r8 r. r, N)r, rG r )r r r r2 r r; s r Údata_for_sortingrJ b sH ð ×ÑHÒ$ô #¤2§8¡8Ò,BÌ&Ô#QÐRSÐRTÐ#UÓVÐVÜ rxxª Ó 2Ó 3Ð3r c óè |j dk( r5t t j dt j dgt
|
¬« « S t t j dt j dg« « 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 r; s r Údata_missing_for_sortingrL p sQ ð ×ÑHÒ$Ü"¤2§8¡8¨T´2·6±6¸4Ð,@ÌÔ#OÓPÐPÜ rxx¨¬B¯F©F°A¨Ó 7Ó 8Ð8r c óü |j dk( rd\ }}}nt j d« \ }}}t t j ||t j
|
t j
|
||||g|j ¬« « S )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 rF rH rI r. )r r r3 r r2 r: )r) r
|
r@ rA Úcs r Údata_for_groupingrO | sj ð ×ÑHÒ$Ø"1aä))A,1aÜ Ü
|
!Q¤§¡¨¨1¨a°Ð3¸5×;LÑ;LÔMó ð r c ó® | j dk( rt j | d« t j d« dz }t j || ¬« S )NÚOz is not a numeric dtyper+ rG r. )ÚkindÚpytestÚskipr Úonesr Ú_from_sequence)r
|
Úarrs r Ú data_for_twosrX sG à zzSÒÜ ugÐ 4Ð5Ô6Ü
|
''#,Ñ
|
CÜ × -Ñ -¨c¸Ô ?Ð?r c ór | dk( r2t j j d¬« }|j |« yy)a
|
Tests for NumpyExtensionArray 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 zFails for object dtype©ÚreasonN)rS ÚmarkÚxfailÚ applymarker)r
|
r r\ s r Úskip_numpy_objectr_ s7 ð ÒÜ{{× Ñ Ð(@Ð ÓAØ×ÑDÕ!ð r r_ c ó¦ e Zd Zej j d¬« d « Ze fd« Z fdZ fdZ
|
e fd« Z e fd« Z e fd « Z e fd
|
« Ze fd « Zej j! d ¬« fd « Z fdZe fd« ZdZdZdZdZ fdZd Z fdZ fdZ fdZ de j< de de fdZ!de j< de de fdZ"ej j d « ej jG dd d g« d « « Z$e fd « Z%e fd!« Z&e fd"« Z'e fd#« Z(eej jG d$d%dg« fd&« « Z)e fd'« Z*eej jG d( e+jX g d)¢« e jX g d)¢d*¬+« gd,d-g¬.« fd/« « Z-eej jG d0g d1¢ e jX g d1¢d2¬+« e+jX g d1¢« gg d3¢¬.« fd4« « Z.ej jG d5d6d7d8e j^ gd f ej` d6d7d8e j^ gd ej j ¬9« e jX d6d7d8e j^ gd2¬+« d f e jX d6d7d8e j^ gd2¬+« d fgg d:¢¬.« fd;« Z1e fd<« Z2e fd=« Z3d> Z4ej j! d?¬« fd@« Z5eej jG dAdBdCg« fdD« « Z6 xZ7S )EÚTestNumpyExtensionArrayzWe don't register our dtyperZ c ó y r rC )Úselfr6 s r Útest_from_dtypez'TestNumpyExtensionArray.test_from_dtype© ó ð r c ó& t | ||« y r )ÚsuperÚ)test_series_constructor_scalar_with_index)rc r6 r
|
Ú __class__s r rh zATestNumpyExtensionArray.test_series_constructor_scalar_with_index® s ø ô Ñ9¸$ÀÕFr c óâ |j j dk( rF|j t j j d|j j ¬« « t | |« y )Nr z:NumpyExtensionArray expectedly clashes with a NumPy name: rZ )r
|
r r^ rS r\ r] rg Útest_check_dtype)rc r6 r Úusing_infer_stringri s r rk z(TestNumpyExtensionArray.test_check_dtype³ sc ø Ø ::× !Ñ ! XÒ -Ø × Ñ Ü ×!Ñ!ð #Ø#'§:¡:×#9Ñ#9Ð":ð <ð "ó ô ô Ñ Õ&r c ó^ |j dk( rt |« sJ y t | |« y )Nr )r r rg Útest_is_not_object_type)rc r
|
r ri s r rn z/TestNumpyExtensionArray.test_is_not_object_type½ s. ø Ø × Ñ Ò (ô # 5Ô)Ð )Ñ)ä GÑ +¨EÕ 2r c ó$ t | |« y r )rg Útest_getitem_scalar©rc r6 ri s r rp z+TestNumpyExtensionArray.test_getitem_scalarÅ s ø ô Ñ# DÕ)r c ó$ t | |« y r )rg Útest_shift_fill_valuerq s r rs z-TestNumpyExtensionArray.test_shift_fill_valueÊ s ø ô Ñ% dÕ+r c ó$ t | |« y r )rg Útest_fillna_copy_frame©rc r< ri s r ru z.TestNumpyExtensionArray.test_fillna_copy_frameÏ s ø ô Ñ& |Õ4r c ó$ t | |« y r )rg Útest_fillna_copy_seriesrv s r rx z/TestNumpyExtensionArray.test_fillna_copy_seriesÔ s ø ô Ñ'¨ Õ5r c ó& t | ||« y r )rg Útest_searchsorted)rc rJ Ú as_seriesri s r rz z)TestNumpyExtensionArray.test_searchsortedÙ s ø ô
|
Ñ!Ð"2°IÕ>r z*NumpyExtensionArray.diff may fail on dtypec ó$ t | ||« S r )rg Ú test_diff)rc r6 Úperiodsri s r r} z!TestNumpyExtensionArray.test_diffà s ø äwÑ wÓ/Ð/r c óÀ |j j t k( r1t j j d¬« }|j |« t | ! |« y )Nz$Dimension mismatch in np.concatenaterZ ) r
|
r r rS r\ r] r^ rg Ú test_insert)rc r6 r r\ ri s r r z#TestNumpyExtensionArray.test_insertä sH ø Ø ::× !Ñ !¤VÒ +Ü;;×$Ñ$Ð,RÐ$ÓSDØ × Ñ Ô %ä ÑDÕ!r c ó& t | ||« y r )rg Útest_insert_invalid©rc r6 Úinvalid_scalarri s r r z+TestNumpyExtensionArray.test_insert_invalidë s ø ô Ñ# D¨.Õ9r Nc ót d }|j j dk( rt }|| _ t | |« y ©NrQ )r
|
rR Ú TypeErrorÚ
|
divmod_excrg Ú test_divmod)rc r6 r ri s r r z#TestNumpyExtensionArray.test_divmodõ s1 ø Ø
|
Ø ::??cÒ !Ü"JØ$Ü ÑDÕ!r c ó¬ t j |« }d }|j j dk( r t }|| _ | j |t |« y r )r/ r0 r
|
rR r r Ú_check_divmod_opÚdivmod)rc r6 ÚserÚexcs r Útest_divmod_series_arrayz0TestNumpyExtensionArray.test_divmod_series_arrayü sB ÜiioØØ ::??cÒ !ÜCØ !DOØ × Ñ c¤6¨4Õ0r c ó |}d }|j j t k( rE|dv r;t j j d¬« }|j j |« t }|| _ t | - ||« y ©N)Ú__mul__Ú__rmul__z9the Series.combine step raises but not the Series method.rZ ) r
|
r r rS r\ r] ÚnodeÚ
|
add_markerr Úseries_scalar_excrg Ú test_arith_series_with_scalar)rc r6 Úall_arithmetic_operatorsr Úopnamer r\ ri s r r z5TestNumpyExtensionArray.test_arith_series_with_scalar sw ø Ø)Ø ÐØ ::× !Ñ !¤VÒ +ØÐ0Ñ0Ü {{×(Ñ(ØVð )ó ð ×'Ñ'¨Ô-Ü )Ð Ø!2Ô Ü Ñ-¨dÐ4LÕMr c ó |}d }|j j t k( r
|
|dvrt }|| _ t
|
| ||« y )N)Ú__add__Ú__radd__)r
|
r r r Úseries_array_excrg Ú test_arith_series_with_array)rc r6 r r r ri s r r z4TestNumpyExtensionArray.test_arith_series_with_array sF ø Ø)Ø ÐØ ::× !Ñ !¤VÒ +°Ð>UÑ0UÜ (Ð Ø 0Ô Ü Ñ,¨TÐ3KÕLr c ó |}d }|j j t k( rE|dv r;t j j d¬« }|j j |« t }|| _ t | - ||« y r ) r
|
r r rS r\ r] r r r Úframe_scalar_excrg Ú test_arith_frame_with_scalar)rc r6 r r r r r\ ri s r r¡ z4TestNumpyExtensionArray.test_arith_frame_with_scalar sw ø Ø)Ø ÐØ ::× !Ñ !¤VÒ +ØÐ0Ñ0Ü {{×(Ñ(ØVð )ó ð ×'Ñ'¨Ô-Ü (Ð Ø 0Ô Ü Ñ,¨TÐ3KÕLr r Úop_nameÚreturnc ó> |j j dk( r|dv S y)NrQ )ÚsumÚminÚmaxÚanyÚallT)r
|
rR )rc r r¢ s r Ú_supports_reductionz+TestNumpyExtensionArray._supports_reduction& s! Ø 99>>SÒ ØÐ AÐAÐ AØr Úskipnac óú t ||« }|j j }|j |« }t ||« }|dk( r |« } |« } n ||¬« } ||¬« } t j
|
|| « y )NÚcount)r« )r r
|
r r r% Úassert_almost_equal)
|
rc r r¢ r« Úres_opÚ cmp_dtypeÚaltÚexp_opÚresultÚexpecteds
|
r Ú check_reducez$TestNumpyExtensionArray.check_reduce+ ss ÜgÓ&ð II×)Ñ) ØjjÓ#ÜgÓ&Ø gÒ ÙXFÙ xHá 6Ô*FÙ VÔ,HÜ
|
× Ñ v xÕ0r zTODO: tests not written yetTFc ó y r rC )rc r6 Úall_numeric_reductionsr« s r Útest_reduce_framez)TestNumpyExtensionArray.test_reduce_frame; re r c ó$ t | |« y r )rg Útest_fillna_seriesrv s r rº z*TestNumpyExtensionArray.test_fillna_series@ s ø ô Ñ" <Õ0r c ó$ t | |« y r )rg Útest_fillna_framerv s r r¼ z)TestNumpyExtensionArray.test_fillna_frameE s ø ô Ñ! ,Õ/r c ó& t | ||« y r )rg Útest_setitem_invalidr s r r¾ z,TestNumpyExtensionArray.test_setitem_invalidJ s ø ô Ñ$ T¨>Õ:r c ó& t | ||« y r )rg Ú test_setitem_sequence_broadcasts©rc r6 Ú box_in_seriesri s r rÀ z8TestNumpyExtensionArray.test_setitem_sequence_broadcastsO s ø ô Ñ0°°}ÕEr ÚsetterÚlocc ó& t | ||« y r )rg Útest_setitem_mask_broadcast)rc r6 rà ri s r rÆ z3TestNumpyExtensionArray.test_setitem_mask_broadcastU s ø ô
|
Ñ+¨D°&Õ9r c ó$ t | |« y r )rg Ú&test_setitem_scalar_key_sequence_raiserq s r rÈ z>TestNumpyExtensionArray.test_setitem_scalar_key_sequence_raise\ s ø ô Ñ6°tÕ<r Úmask)TTTFFÚbooleanr. ú numpy-arrayz boolean-array)Úidsc ó( t | |||« y r )rg Útest_setitem_mask)rc r6 rÉ r ri s r rÎ z)TestNumpyExtensionArray.test_setitem_maskd s ø ô Ñ! $¨¨mÕ<r Úidx)r r, rG ÚInt64)Úlistz integer-arrayrË c ó( t | |||« y r )rg Útest_setitem_integer_array©rc r6 rÏ r ri s r rÓ z2TestNumpyExtensionArray.test_setitem_integer_arrayp s ø ô Ñ*¨4°°mÕDr zidx, box_in_seriesr r, rG )Úmarks)z
|
list-Falsez list-Truezinteger-array-Falsezinteger-array-Truec ó( t | |||« y r )rg Ú(test_setitem_integer_with_missing_raisesrÔ s r r× z@TestNumpyExtensionArray.test_setitem_integer_with_missing_raisesy s ø ô Ñ8¸¸sÀMÕRr c ó& t | ||« y r )rg Útest_setitem_slicerÁ s r rÙ z*TestNumpyExtensionArray.test_setitem_slice s ø ä Ñ" 4¨ Õ7r c ó$ t | |« y r )rg Útest_setitem_loc_iloc_slicerq s r rÛ z3TestNumpyExtensionArray.test_setitem_loc_iloc_slice s ø ä Ñ+¨DÕ1r c óÄ t j dt j |« i« x}}t j |j ¬« } ||« }|d |j |df< |j
|
j t k7 rCt |t « r|t d « k7 r%t j d|j « i« }t j ||d¬« y )Nr6 )ÚindexF)Úcheck_column_type) r/ Ú DataFramer0 rÝ rÄ r
|
r r r ÚsliceÚto_numpyr% Úassert_frame_equal)rc r6 Ú full_indexerÚdfr´ r³ Úkeys r Ú,test_setitem_with_expansion_dataframe_columnzDTestNumpyExtensionArray.test_setitem_with_expansion_dataframe_column sª ä f¬b¯i©i¸«oÐ%>Ó?Ð?XÜ B§H¡HÔ-ñ 2Ó Ø"$ V¡*
|
|
3;Ñ ð ::× !Ñ !¤VÒ +Ü c¤5Ô)¨S´E¸$³KÒ-?Ü <<¨°· ± ³ Ð(AÓBÜ
|
× Ñ f hÀ%ÖHr zNumpyEADtype is unpackedc ó$ t | |« y r )rg Ú#test_index_from_listlike_with_dtyperq s r rè z;TestNumpyExtensionArray.test_index_from_listlike_with_dtype s ø ä Ñ3°DÕ9r ÚenginerN Úpythonc ó( t | |||« y r )rg Ú test_EA_types)rc ré r6 r ri s r rì z%TestNumpyExtensionArray.test_EA_types£ s ø ô Ñ f d¨GÕ4r )8Ú__name__Ú
|
__module__Ú __qualname__rS r\ rT rd Ú skip_nestedrh rk rn rp rs ru rx rz r] r} r r r r r r r r r r r¡ r/ r0 ÚstrÚboolrª rµ Ú parametrizer¸ rº r¼ r¾ rÀ rÆ rÈ r r2 rÎ rÓ ÚNAr r× rÙ rÛ ræ rè rì Ú __classcell__)ri s @r ra ra ¨ s2 ø Ø [[×ÑÐ :ÐÓ;ñ ó <ð ð óGó ðGô'ô3ð ó*ó ð*ð ó,ó ð,ð ó5ó ð5ð ó6ó ð6ð ó?ó ð?ð [[×ÑÐ JÐÓKó0ó Lð0ô"ð ó:ó ð:ð JØ ÐØÐØÐô"ò1ô NôMô Mð r§y¡yð ¸3ð À4ó ð
|
1 § ¡ ð 1°Cð 1Àó 1ð [[×ÑÐ3Ó4Ø [[× Ñ X¨¨e }Ó5ñ ó 6ó 5ð ð ó1ó ð1ð ó0ó ð0ð ó;ó ð;ð óFó ðFð
|
Ø [[× Ñ X¨¨t }Ó5ó:ó 6ó ð:ð
|
ó=ó ð=ð Ø [[× Ñ Øà BHHÒ5Ó 6Ø BHHÒ5¸YÔ Gð
|
ð OÐ ,ð ó ó=óó ð=ð Ø [[× Ñ Ø Ú HBHHY¨gÔ6¸¸¿¹ÂÓ8KÐLÚ 4ð ó ó
|
Eó ó ð Eð [[× Ñ Ø àAruuÐ uÐ %Ø FLL!Q 2§5¡5Ð)¨4°v·{±{×7HÑ7HÔ IØ RXXq!Q §¡Ð&¨gÔ 6¸Ð >Ø RXXq!Q §¡Ð&¨gÔ 6¸Ð >ð
|
ò Uð ó óSó ðSð ó8ó ð8ð ó2ó ð2òIð" [[×ÑÐ 8ÐÓ9ó:ó :ð:ð Ø [[× Ñ X¨¨X Ó7ó5ó 8ó ô5r ra c ó e Zd Zy)Ú Test2DCompatN)rí rî rï rC r r r÷ r÷ © s Ør r÷ )Ú
|
Attributes)'Ú__doc__Únumpyr rS Úpandas.core.dtypes.dtypesr Úpandasr/ Úpandas._testingÚ_testingr% Úpandas.api.typesr Úpandas.core.arrays.numpy_r Úpandas.tests.extensionr r" r rñ r Úfixturer
|
r) r6 r< rD rJ rL rO rX r_ r\ Ú usefixturesrð ÚExtensionTestsra ÚNDArrayBacked2DTestsr÷ rC r r ú<module>r s ðñó" Û å 2ã Ý Ý ,Ý 9Ý 'à×-Ñ-Ð ñ 3Sð 3°Có 3ð Ð*Ô+ñ1ó ,ð1ð ñó ðð, ñFó ðFð ñ8ó ð8ð ñó ðð ñ
|
4ó ð
|
4ð ñ9ó ð9ð ñ ó ð ð ñ@ó ð@ð ñ "ó ð "ð kk×%Ñ%Ð&9Ó: ô~5d× 1Ñ 1ô ~5ôB 4×,Ñ,õ r
|
