Ë
|
oñúh8 ã ó d Z ddlZddlZddlZddlmZmZ ddlm Z ddl
|
m Z m Z m Z ddlZddlmZ ddlmZ ddlmZmZ ddlmZmZmZm Z m Z m Z m Z m Z dd l!m"Z" e« xr e xs e Z#ejH jK d
|
« ejH jK d « ejH jK d « gZ&d Z'd Z(d Z) ejT eeee e e e e eeeg ¬« d « Z+ejT d « Z,ejT d « Z-ejT d « Z.ejT d « Z/ejT d « Z0ejT d « Z1ejT d « Z2 G d de"jf « Z4 G d d e"jj « Z6y) aj
|
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)Únp_version_gt2)Úis_float_dtypeÚis_signed_integer_dtypeÚis_unsigned_integer_dtype)Ú BooleanDtype)Ú Float32DtypeÚ Float64Dtype)Ú Int8DtypeÚ
|
Int16DtypeÚ
|
Int32DtypeÚ
|
Int64DtypeÚ
|
UInt8DtypeÚ UInt16DtypeÚ UInt32DtypeÚ UInt64Dtype)Úbasez9ignore:invalid value encountered in divide:RuntimeWarningz)ignore:Mean of empty slice:RuntimeWarningz4ignore:overflow encountered in reduce:RuntimeWarningc óª t t dd« « t j gz t t dd« « z t j gz ddgz S )Né é é
|
éb éc éd )ÚlistÚrangeÚpdÚNA© ó úUH:\Change_password\venv_build\Lib\site-packages\pandas/tests/extension/test_masked.pyÚ make_datar# @ sB Ü a Ó ¤§¡wÑ &¬¬e°B¸«mÓ)<Ñ <ÄÇÁ¸wÑ FÈ"ÈcÈÑ RÐRr! c óÖ t t j ddd« « t j gz t t j ddd« « z t j gz ddgz S )Nç¹?gÍÌÌÌÌÌì?r g#@gÍÌÌÌÌÌ#@g $@)r ÚnpÚaranger r r r! r" Úmake_float_datar( D sb ä RYYsC Ó %Ó&Ü 55'ñ ä
|
ryyC Ó%Ó
|
&ñ 'ô 55'ñ ð +ñ ðr! c ór ddgdz t j gz ddgdz z t j gz ddgz S )NTFé é, )r& Únanr r! r" Úmake_bool_datar- N s? Ø %=1Ñ ¤§¡xÑ '¨4°¨-¸"Ñ*<Ñ <ÄÇÁ¸xÑ GÈ4ÐQVÈ-Ñ WÐWr! )Úparamsc ó" | j « S ©N)Úparam)Úrequests r" Údtyper3 R s ð ==?Ðr! c ó¬ | j dk( r t « }n$| j dk( r t « }n
|
t « }t j
|
|| ¬« S )NÚfÚb©r3 )Úkindr( r- r# r Úarray)r3 Údatas r" r: r: e sB à zzSÒÜ Ó Ø sÒ Ü Ó ä{Ü 88D Ô &Ð&r! c óÎ | j dk( r*t j t j d« | ¬« S t j t j d« dz | ¬« S )Nr6 r r7 é )r8 r r9 r& Úonesr7 s r" Ú data_for_twosr> p sE à zzSÒÜxx ¨EÔ2Ð2Ü 88BGGCL 1Ñ$¨EÔ 2Ð2r! c ó( | j dk( r't j t j dg| ¬« S | j dk( r't j t j
|
dg| ¬« S t j t j dg| ¬« S )Nr5 r% r7 r6 Tr ©r8 r r9 r r& r, r7 s r" Ú data_missingrA w se à zzSÒÜxx ¨EÔ2Ð2Ø sÒ Üxx ¨eÔ4Ð4Ü 88RUUAJ eÔ ,Ð,r! c óÔ | j dk( rt j g d¢| ¬« S | j dk( rt j g d¢| ¬« S t j g d¢| ¬« S )Nr5 )r% çÉ?ç r7 r6 )TTF)r r< r )r8 r r9 r7 s r" Údata_for_sortingrE sN à zzSÒÜxx¨uÔ5Ð5Ø sÒ ÜxxÒ+°5Ô9Ð9Ü 88I UÔ +Ð+r! c ó. | j dk( r(t j dt j dg| ¬« S | j dk( r(t j dt j
|
dg| ¬« S t j dt j d g| ¬« S )
|
Nr5 r% rD r7 r6 TFr r r@ r7 s r" Údata_missing_for_sortingrG sm à zzSÒÜxxbee SÐ)°Ô7Ð7Ø sÒ Üxxrvv uÐ-°UÔ;Ð;Ü 88QqM¨Ô /Ð/r! c ó d S )Nc óN | t j u xr |t j u S r0 )r r )ÚxÚys r" ú<lambda>zna_cmp.<locals>.<lambda> s RUU
|
Ò1 q¬B¯E©E z r! r r r! r" Úna_cmprM s
|
ñ 2Ð1r! c
|
óÄ | j dk( rd}d}d}n | j dk( rd}d}|}nd}d }d
|
}t j }t j ||||||||g| ¬ « S ) Nr5 r% rD rC r6 TFr r r< r7 )r8 r r r9 )r3 r6 ÚaÚcÚnas r" Údata_for_groupingrR sr à zzSÒØ Ø Ø Ø sÒ Ø Ø Ø à Ø Ø ä BÜ 88Q2r 1 a¨¨AÐ.°eÔ <Ð<r! c ó8 e Zd Zej j dddg« d « Zd Zd Zde fdZ
|
fd Z fd
|
Z d e j de d efd Zd e j de defdZde defdZd e j de d efdZd e j de defdZ xZS )ÚTestMaskedArraysÚ na_actionNÚignorec óê |j d |¬« }|j t « k( r"|j dt j
|
¬« }n|j « }t j ||« y )Nc ó | S r0 r )rJ s r" rL z+TestMaskedArrays.test_map.<locals>.<lambda>® s ¨A r! ©rU Úfloat64)r3 Úna_value)Úmapr3 r
|
Úto_numpyr& r, ÚtmÚassert_numpy_array_equal)ÚselfrA rU ÚresultÚexpecteds r" Útest_mapzTestMaskedArrays.test_map¬ s^ à ×!Ñ!¡+¸Ð!ÓCØ × Ñ ¤ £ Ò /à#×,Ñ,°9ÄrÇvÁvÐ,ÓNHà#×,Ñ,Ó.HÜ
|
×#Ñ# F¨HÕ5r! c ó@ |d |j fdd¬« }|j j dk( r-t j dt
|
j dgt ¬« }n&t j t j g« }t j ||« y )Nr< c ó S r0 r )rJ Úzeros r" rL z<TestMaskedArrays.test_map_na_action_ignore.<locals>.<lambda>¸ s ø ¸ r! rV rY r6 Fr7 ) r\ r3 r8 r& r9 r r Úobjectr, r^ r_ )r` rG ra rb rf s @r" Útest_map_na_action_ignorez*TestMaskedArrays.test_map_na_action_ignore¶ sz ø Ø'¨Ñ*Ø)×-Ñ-«nÈÐ-ÓQØ #× )Ñ )× .Ñ .°#Ò 5Üxx ¬¯©¨uÐ 5¼VÔDHäxx ¤r§v¡v¨tÐ 4Ó5HÜ
|
×#Ñ# F¨HÕ5r! c óü t j |« }|j dk( r3|j d« j d« dv rt S |dv rt S y y # t $ r t j |« }Y cw xY w)Nr6 Ú_Úr)ÚpowÚtruedivÚfloordiv)Ú__sub__Ú__rsub__)r^ Ú get_dtypeÚAttributeErrorr8 ÚstripÚlstripÚNotImplementedErrorÚ TypeError)r` Úop_nameÚobjÚotherr3 s r" Ú_get_expected_exceptionz(TestMaskedArrays._get_expected_exception¿ sz ð (ÜLL Ó%Eð
|
::Ò Ø}}SÓ!×(Ñ(¨Ó-Ð1OÑOä*Ð*ØÐ 3Ñ3ä Ð ØØøô ò (äLL Ó'Eð (ús A Á A;Á:A;rw c ó t j |« }|}|dv r|j d« S |j dv r|dv rkt j
|
« 5 t j ddt ¬« |j t j « }d d d « j d« }|S |j |« }|S |j d k( r|d
|
v r|j d « }nL|d v rH| j |« }| j |j t « ||« }|j d« }|d k( r3 t ||« |« } t j || j « < |S |j |« }|S # 1 sw Y íxY w)N)ÚeqÚneÚleÚgeÚltÚgtÚbooleanÚiu)Ú __rtruediv__Ú __truediv__Ú__div__rV z Downcasting object dtype arrays)ÚcategoryÚFloat64r6 )Ú __floordiv__Ú __rfloordiv__Ú__pow__Ú__rpow__Ú__mod__Ú__rmod__ÚInt8)r
r r )r^ rq Úastyper8 ÚwarningsÚcatch_warningsÚfilterwarningsÚ FutureWarningÚfillnar& r, Úget_op_from_nameÚ_combineÚfloatÚgetattrÚisna)
|
r` rw rx ry Úpointwise_resultÚsdtyperb ÚfilledÚopra s
|
r" Ú_cast_pointwise_resultz'TestMaskedArrays._cast_pointwise_resultÐ sw Ü cÓ"Ø#à Ð:Ñ :Ø?? 9Ó-Ð -à ;;$Ñ ØÐDÑDÜ ×,Ñ,Ó.ñ 5Ü ×+Ñ+Ø Ø9Ü!.õð
|
&__¬R¯V©VÓ 4F÷ 5ð "==¨Ó3ð< ð7 $??¨6Ó2ð6 ð5 [[CÒ Øð ñ ð $??¨6Ó2àÐ ;Ñ;ð ×*Ñ*¨7Ó3Ø ==¨¯©´EÓ):¸EÀ2ÓFØ#??¨9Ó5à*Ò$à. gÓ.¨uÓ5Ü*,¯&©& Ñ'ð ð vÓ.HØ÷K5ð 5ús Á<E;Å;Fc óº |j j dk( r1t j j d¬« }|j |« t | ||« y )Nr6 zInconsistency between floordiv and divmod; we raise for floordiv but not for divmod. This matches what we do for non-masked bool dtype.©Úreason)r3 r8 ÚpytestÚmarkÚxfailÚ applymarkerÚsuperÚtest_divmod_series_array)r` r: r> r2 r¤ Ú __class__s r" r¨ z)TestMaskedArrays.test_divmod_series_array sP ø Ø ::??cÒ !Ü;;×$Ñ$ð)ð %ó Dð
|
× Ñ Ô %Ü Ñ(¨¨}Õ=r! c ó |d« \ }}|j j dk( rd| _ n t | _ t | |« y )Nr< r6 r )r3 r8 Ú_combine_le_expected_dtyperg r§ Útest_combine_le)r` Ú data_repeatedÚ
|
orig_data1Ú
|
orig_data2r© s r" r¬ z TestMaskedArrays.test_combine_le
|
sE ø á!.¨qÓ!1Ñ
|
JØ × Ñ × Ñ CÒ 'Ø.7DÕ +ô /5DÔ +Ü Ñ Õ.r! ÚserÚreturnc ój |dv r/|j j dk7 rt j d¬« y)N©ÚanyÚallr6 z-Tested in tests/reductions/test_reductions.pyr¡ T)r3 r8 r£ Úskip©r` r° rw s r" Ú_supports_reductionz$TestMaskedArrays._supports_reduction s) Ø nÑ $¨¯©¯ © ¸3Ò)>Ü KKÐ NÕ OØr! Úskipnac ó d}|j j dk( r|j j }n |j j dk( r|dv rd}|j « j |« }|dk( r# t ||« « } t ||« « }nZ t ||« |¬« } t ||« |¬« }|s2|j « j « r|dvrt j }t j ||« y ) NÚint64r5 r6 ©ÚminÚmaxÚboolÚcount©r¹ r³ ) r3 r8 Ú numpy_dtypeÚdropnar r r r´ r r r^ Úassert_almost_equal)r` r° rw r¹ Ú cmp_dtypeÚaltra rb s r" Ú check_reducez TestMaskedArrays.check_reduce sá ð Ø 99>>SÒ ð ×-Ñ-IØ YY^^sÒ "Ø.Ñ(Ø " ð
|
jjl×!Ñ! )Ó,Ø gÒ Ø*WS 'Ó*Ó,FØ,ws GÓ,Ó.Hà*WS 'Ó*°&Ô9FØ,ws GÓ,°FÔ;HÙchhjnnÔ .°7À.Ñ3PÜ 55Ü
|
× Ñ v xÕ0r! c ó t |j « r|j j }|S |dv rd}|S |dv r|j j }|S |j dv r|j j }|S t |j « r t « rt
|
r|rt sd}|S d}|S t |j « r t « rt
|
r|rt sd}|S d}|S |j j d k( r=|dv rd}|S |d
|
v rd }|S |d v r t « rt
|
r|rt sdnd}|S t d « t d « )N)ÚmeanÚmedianÚvarÚstdÚskewr )r¾ r½ )ÚInt64ÚUInt64ÚInt32rÎ ÚUInt32rÏ r6 r¼ r )ÚsumÚprodznot supposed to reach this)
|
r r3 Únamer r r r r r8 rv )r` Úarrrw r¹ rÅ s r" Ú _get_expected_reduction_dtypez.TestMaskedArrays._get_expected_reduction_dtype4 s Ü #))Ô $Ø IðL ÐðK Ð@Ñ @Ø!IðH ÐðG Ñ &Ø IðD ÐðC YYÐ-Ñ -Ø Ið@ Ðô? % S§Y¡YÔ /ô (Ô)µ>ÉÝð ð ð: Ðð3 ð ð: Ðô/ ' s§y¡yÔ 1ô (Ô)µ>ÉÝð ð ð, Ðð% ð ð, Ðð! YY^^sÒ "ØÐBÑBØ % ð Ðð NÑ*Ø % ð Ðð OÑ+ô ,Ô-µ~ÉVÝ ñ ð !ð ð Ðô Ð <Ó=Ð=äÐ 8Ó9Ð 9r! c ó y)NTr r· s r" Ú_supports_accumulationz'TestMaskedArrays._supports_accumulation^ s Ør! c
|
ó d}t r|j j dk( sd}|j j j d« rd| }n|j j j d« rd| }nW|j j j d« r |j }n%|j j
|
d k( r |d
|
v rd }nd| }d k( r"|d k( r |rt j d| d| « |dk( rt t ||« |¬« }t j t j t |j d« |« |¬« |¬« « }t j ||« y |dv r~ t ||« |¬« }t j t j t |j d« |« |¬« |j ¬« « }t j ||« y |d k( rz t |d d |« |¬« }t j t j t |d d j d« |« |¬« |¬« « }t j ||« y t | d« )Né@ é é ÚUÚUIntÚIÚIntÚFr6 )ÚcumminÚcummaxr ÚFloat32Úcumprodz4Float32 precision lead to large differences with op z and skipna=ÚcumsumrÁ rZ r7 )rã râ é z not supported)Úis_windows_or_32bitr3 ÚitemsizerÔ Ú
|
startswithr8 r£ r¶ r r ÚSeriesr9 r r^ Úassert_series_equalru )r` r° rw r¹ ÚlengthÚexpected_dtypera rb s r" Úcheck_accumulatez!TestMaskedArrays.check_accumulatea s8 ð Ý ð 99×%Ñ%¨Ò*Øà 99>>× $Ñ $ SÔ )Ø # F 8_NØ YY^^× &Ñ & sÔ +Ø " 6 (^NØ YY^^× &Ñ & sÔ +ð !YYNØ YY^^sÒ "ØÐ.Ñ.Ø!*à#& v h à YÒ &¨7°iÒ+?ÁFä KKØFÀwÀið P Ø $Xð'ô ð
|
hÒ Ø*WS 'Ó*°&Ô9FÜyyÜØ;GCJJ yÓ 1°7Ó;À6ÔJØ(ôóHô × "Ñ " 6¨8Õ 4Ø Ð,Ñ ,Ø*WS 'Ó*°&Ô9FÜyyÜØ;GCJJ yÓ 1°7Ó;À6ÔJØ ))ôóHô × "Ñ " 6¨8Õ 4Ø Ò !Ø/WS "X wÓ/°vÔ>FÜyyÜØ@GC HOO¨IÓ 6¸Ó@ÈÔOØ(ôóHô × "Ñ " 6¨8Õ 4ô &¨¨ ° Ð&@ÓAÐ Ar! )Ú__name__Ú
|
__module__Ú __qualname__r£ r¤ Ú parametrizerc rh rz Ústrr r¨ r¬ r rë r¿ r¸ rÇ rÖ rØ rï Ú __classcell__)r© s @r" rT rT « sÜ ø Ø [[× Ñ [¨4°Ð*:Ó;ñ6ó <ð6ò6òð".¨có .ô`>ô/ð r§y¡yð ¸3ð À4ó ð
|
1 § ¡ ð 1°Cð 1Àó 1ð6(¸#ð (Àtó (ðT¨"¯)©)ð ¸cð Àdó ð<B B§I¡Ið <B¸ð <BÀT÷ <Br! rT c ó e Zd Zy)Ú Test2DCompatN)rð rñ rò r r! r" r÷ r÷ s Ør! r÷ )7Ú__doc__r Únumpyr& r£ Ú pandas.compatr r Úpandas.compat.numpyr Úpandas.core.dtypes.commonr r r Úpandasr Úpandas._testingÚ_testingr^ Úpandas.core.arrays.booleanr Úpandas.core.arrays.floatingr
|
r Úpandas.core.arrays.integerr r r r r r r r Úpandas.tests.extensionr rè r¤ r Ú
|
pytestmarkr# r( r- Úfixturer3 r: r> rA rE rG rM rR ÚExtensionTestsrT ÚDim2CompatTestsr÷ r r! r" ú<module>r sÒ ðñó ã Û ÷õ /÷ñ ó Ý Ý 3÷÷ ÷ ó õ (á*Ó,ÒC°^Ð1CÒPÈDÈÐ ð KK× Ñ ØCóð KK× Ñ Ð JÓKà
|
KK× Ñ Ð UÓVð
|
òSòòXð àØØØØØØØØØØð ôñ ó ð ð ñ'ó ð'ð ñ3ó ð3ð ñ-ó ð-ð ñ,ó ð,ð ñ0ó ð0ð ñ2ó ð2ð
|
ñ=ó ð=ô$rBt×*Ñ*ô rBôj 4×'Ñ'õ r!
|
