Ë
|
oñúhe' ã ó d Z ddlmZ ddlZddlmZ ddlZddlZddl m
|
Z
|
ddl m Z ddl ZddlmZ ddlmZ ddlmZ dd lmZ dd
|
lmZ dd lmZ d Z ej: d dg¬« d « Z ej: d « Z ej: d « Z ej: d « Z!ej: d « Z"ej: d « Z#ej: d « Z$ G d dejJ « Z& G d dejN « Z(d Z)d Z*y) 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/`.
|
|
é )Ú annotationsN)Úcast)Ú HAS_PYARROW)ÚStorageExtensionDtype©Úis_string_dtype)ÚArrowStringArray©Ú StringDtype)Ú string_dtype_highest_priority)Úbasec ó@ |s| S | j j dk7 r| S t j d« }| j }t |« dz }|j g |d | j ¢||d j ¢« }|j dk( sJ t | « |« S )NÚpyarrowé )
|
ÚdtypeÚstorageÚpytestÚ importorskipÚ _pa_arrayÚlenÚ chunked_arrayÚchunksÚ
|
num_chunksÚtype)ÚarrÚchunkedÚpaÚ arrow_arrayÚsplits úUH:\Change_password\venv_build\Lib\site-packages\pandas/tests/extension/test_string.pyÚmaybe_split_arrayr! % s§ Ù Ø
|
Ø × Ñ iÒ 'Ø
|
ä × Ñ YÓ 'Bà--KÜ Ó Ñ !EØ×"Ñ"ØB+fuÐ
|
×
|
$Ñ
|
$ÐB {°5°6Ð':×'AÑ'AÐBóKð × !Ñ ! QÒ &Ð&Ð &Ø 49[Ó !Ð!ó TF)Úparamsc ó | j S ©N)Úparam)Úrequests r r r 6 s à ==Ðr" c ó& | \ }}t ||¬« S )N)r Úna_valuer
|
)Ústring_dtype_argumentsr r) s r r r ; s à.ÑGXÜ w°Ô :Ð:r" c ó¦ t j j d« j t t
|
j « d¬« }|d |d k( rSt j j d« j t t
|
j « d¬« }|d |d k( rS| j « j || ¬« }t ||« S )Nr éd )Úsizer é ©r )
|
ÚnpÚrandomÚ default_rngÚchoiceÚlistÚstringÚ ascii_lettersÚconstruct_array_typeÚ_from_sequencer! )r r Ústringsr s r Údatar: A s° äii×#Ñ# AÓ&×-Ñ-¬d´6×3GÑ3GÓ.HÈsÐ-ÓSGØ
|
!*
|
Ò
|
"Ü))×'Ñ'¨Ó*×1Ñ1´$´v×7KÑ7KÓ2LÐSVÐ1ÓWð !*
|
Ó
|
"ð ×
|
$Ñ
|
$Ó
|
&×
|
5Ñ
|
5°gÀUÐ
|
5Ó
|
KCÜ S 'Ó *Ð*r" c ó| | j « j t j dg| ¬« }t ||« S )z Length 2 array with [NA, Valid]ÚAr/ ©r7 r8 ÚpdÚNAr! ©r r r s r Ú data_missingrA K s7 ð ×
|
$Ñ
|
$Ó
|
&×
|
5Ñ
|
5´r·u±u¸c°lÈ%Ð
|
5Ó
|
PCÜ S 'Ó *Ð*r" c ó` | j « j g d¢| ¬« }t ||« S )N)ÚBÚCr< r/ )r7 r8 r! r@ s r Údata_for_sortingrE R s- à
|
×
|
$Ñ
|
$Ó
|
&×
|
5Ñ
|
5²oÈUÐ
|
5Ó
|
SCÜ S 'Ó *Ð*r" c ó~ | j « j dt j dg| ¬« }t ||« S )NrC r< r/ r= r@ s r Údata_missing_for_sortingrG X s8 à
|
×
|
$Ñ
|
$Ó
|
&×
|
5Ñ
|
5°s¼B¿E¹EÀ3Ð6GÈuÐ
|
5Ó
|
UCÜ S 'Ó *Ð*r" c
|
ó¤ | j « j ddt j t j ddddg| ¬« }t ||« S )NrC r< rD r/ r= r@ s r Údata_for_groupingrI ^ sM à
|
×
|
$Ñ
|
$Ó
|
&×
|
5Ñ
|
5Ø c255"%% c¨3°Ð4¸Eð 6ó Cô S 'Ó *Ð*r" c óª e Zd Z fdZd Z fdZ fdZ fdZd Z fdZ fdZ
|
d Z d
|
Z dd Z dd Zdd ZddZd Z fdZ fdZ xZS )ÚTestStringArrayc ó t | |« |j t j u r|d|j
|
dk( sJ y |j
|
dk( r+t j t « 5 |dk( sJ d d d « y y # 1 sw Y y xY w)Nzstring[ú]r zstring[pyarrow_numpy]) ÚsuperÚtest_eq_with_strr) r> r? r ÚtmÚassert_produces_warningÚ FutureWarning)Úselfr Ú __class__s r rO z TestStringArray.test_eq_with_strg s ø Ü Ñ Ô'à >>RUUÑ "àg e§m¡m _°AÐ 6Ò6Ð 6Ñ6Ø ]]iÒ 'Ü×+Ñ+¬MÓ:ñ 8Ø Ð 7Ò7Ð7Ñ7÷ 8ð 8ð (÷ 8ð 8ús Á+A>Á>Bc ó t |« sJ y r% r )rS r s r Útest_is_not_string_typez'TestStringArray.test_is_not_string_typeq s ô uÔ%Ð%Ñ%r" c ó¶ |j t j u r-|s+t |« j |j
|
« }|du sJ y t | |« y )NF)r) r0 Únanr Úis_dtypeÚnamerN Útest_is_dtype_from_name)rS r Úusing_infer_stringÚresultrT s r r[ z'TestStringArray.test_is_dtype_from_namev sG ø Ø >>RVVÑ #Ñ,>Ü%[×)Ñ)¨%¯*©*Ó5FØU?Ð "?ä GÑ +¨EÕ 2r" c óø |j t j u rB|s@t j t
|
d¬« 5 |j |j « d d d « y t | % |« y # 1 sw Y y xY w)Nz Cannot construct a 'StringDtype'©Úmatch)
|
r) r0 rX r ÚraisesÚ TypeErrorÚconstruct_from_stringrZ rN Ú#test_construct_from_string_own_name)rS r r\ rT s r rd z3TestStringArray.test_construct_from_string_own_name} s` ø Ø >>RVVÑ #Ñ,>Ü yÐ0RÔSñ 8Ø×+Ñ+¨E¯J©JÔ7÷ 8ð 8ô GÑ 7¸Õ >÷ 8ð 8ús » A0Á0A9c ó |j j dk( rt j d¬« t | |« y ©Nr ú/2D support not implemented for ArrowStringArray©Úreason)r r r ÚskiprN Ú test_view©rS r: rT s r rk zTestStringArray.test_view s/ ø Ø ::× Ñ Ò *Ü KKÐ PÕ QÜ Ñ$Õ r" c ó y r% © ©rS r: s r Útest_from_dtypez TestStringArray.test_from_dtype s à r" c ó |j j dk( rt j d¬« t | |« y rf )r r r rj rN Útest_transposerl s r rr z TestStringArray.test_transpose s/ ø Ø ::× Ñ Ò *Ü KKÐ PÕ QÜ Ñ tÕ$r" c ó |j j dk( rt j d¬« t | |« y rf )r r r rj rN Ú test_setitem_preserves_viewsrl s r rt z,TestStringArray.test_setitem_preserves_views s/ ø Ø ::× Ñ Ò *Ü KKÐ PÕ QÜ Ñ,¨TÕ2r" c ó\ |j « }|dg }t j ||« y )Nr. )ÚdropnarP Ú assert_extension_array_equal)rS rA r] Úexpecteds r Útest_dropna_arrayz!TestStringArray.test_dropna_array s, Ø ×$Ñ$Ó&Ø Ñ$Ü
|
×'Ñ'¨°Õ9r" c óì ||j « }|d }|j |« }||usJ t j ||« |j d¬« }||usJ t j ||« y )Nr Úbackfill)Úmethod)ÚisnaÚfillnarP rw )rS r: Úvalidr] s r Ú test_fillna_no_op_returns_copyz.TestStringArray.test_fillna_no_op_returns_copy sw ØTYY[LÑ!àQØUÓ#ØTÑ!Ð!Ð!Ü
|
×'Ñ'¨°Ô5à JÓ/ØTÑ!Ð!Ð!Ü
|
×'Ñ'¨°Õ5r" c ó@ |dv rt S |dv rt S |dv rt S y )N)Ú__mod__Ú__rmod__Ú
|
__divmod__Ú __rdivmod__Ú__pow__Ú__rpow__)Ú__mul__Ú__rmul__)Ú __truediv__Ú __rtruediv__Ú __floordiv__Ú __rfloordiv__Ú__sub__Ú__rsub__)rb )rS Úop_nameÚobjÚothers r Ú_get_expected_exceptionz'TestStringArray._get_expected_exception¨ s@ ð ð
|
ñ
|
ô Ð Ø Ð/Ñ /ä Ð Ø ð
|
ñ
|
ô Ð àr" c óf |dv xs, |j j t j u xr |dv S )N)ÚminÚmaxÚsum)ÚanyÚall)r r) r0 rX ©rS Úserr s r Ú_supports_reductionz#TestStringArray._supports_reductionà s8 à Ð,Ð ,ò *Øyy×!Ñ!¤R§V¡VÐ+ò *Ø>Ð)ð
|
r" c óB t |j t « sJ |dv S )N)ÚcumminÚcummaxÚcumsum)Ú
|
isinstancer r r s r Ú_supports_accumulationz&TestStringArray._supports_accumulationÊ s# Ü#))Ô%:Ô;Ð;Ð;ØÐ8Ð8Ð8r" c ó t t t j |« « }|dv rF|}t |t
|
« st j |« nd }t |t « rNt ||« }nA|j t j u rt j }n|j dk( rd}nd}|j |« S )N)Ú__add__Ú__radd__r ú bool[pyarrow]Úboolean) r r rP Ú get_dtyper¡ Ústrr r) r0 rX Úbool_r Úastype)rS r r r Úpointwise_resultr Úcast_toÚ dtype_others r Ú_cast_pointwise_resultz&TestStringArray._cast_pointwise_resultÎ s Ü[¤"§,¡,¨sÓ"3Ó4Ø Ð-Ñ -ØGÜ5?ÀÄsÔ5K",, uÔ-ÐQUKÜ+¤{Ô3Ü7¸¸{ÓKØ ^^rvvÑ %ÜhhGØ ]]iÒ 'Ø%Gà GØ ×&Ñ& wÓ/Ð/r" c óV t j |« }| j |||d« y )NÚabc)r> ÚSeriesÚ_compare_other)rS r: Ú comparison_opr s r Útest_compare_scalarz#TestStringArray.test_compare_scalarÝ s# ÜiioØ ×ÑC }°eÕ<r" c ó t |d« « j }|r\|j t j u r@|j
|
dk( r1t j j d¬« }|j |« t | - |« y )Nr. ÚpythonúmThe pointwise operation result will be inferred to string[nan, pyarrow], which does not match the input dtyperh ) Únextr r) r> r? r r ÚmarkÚxfailÚ applymarkerrN Útest_combine_add)rS Ú data_repeatedr\ r' r rº rT s r r½ z TestStringArray.test_combine_addá sr ø Ü] 1Ó%Ó&×,Ñ,Ù Ø ^^ruuÑ $¨%¯-©-¸8Ò*Cä;;×$Ñ$ðMð %ó Dð × Ñ Ô %Ü Ñ Õ/r" c ó |j }|rg|dk( rb|j t j u rFt s|j
|
dk( r1t j j d¬« }|j |« t | - ||« y )Nr¥ r r¸ rh ) r r) r> r? r r r rº r» r¼ rN Ú test_arith_series_with_array)rS r: Úall_arithmetic_operatorsr\ r' r rº rT s r rÀ z,TestStringArray.test_arith_series_with_arrayí sx ø ð
|
|
á Ø(¨JÒ6ؤ"§%¡%Ñ'Ý § ¡ °Ò :ô ;;×$Ñ$ðMð %ó Dð × Ñ Ô %Ü Ñ,¨TÐ3KÕLr" )r r© Úreturnz4type[Exception] | tuple[type[Exception], ...] | None)r z pd.Seriesr r© r Úbool)r r© )Ú__name__Ú
|
__module__Ú __qualname__rO rV r[ rd rk rp rr rt ry r r r r¢ r¯ rµ r½ rÀ Ú __classcell__)rT s @r rK rK f so ø ô8ò&ô
|
3ô?ô ò
|
ô%ô
|
3ò
|
:ò
|
|
6ðØðà =óó6
|
ó9ó 0ò =ô
|
0÷Mð Mr" rK c ó> e Zd Z ej d¬« d « Zy)Ú Test2DCompatT)Úautousec óR t |t « rt j d¬« y y )Nrg rh )r¡ r r rj ro s r Úarrow_not_supportedz Test2DCompat.arrow_not_supported s ä dÔ ,Ô -Ü KKÐ PÖ Qð .r" N)rÄ rÅ rÆ r ÚfixturerÌ rn r" r rÉ rÉ s! ØV^^DÔ!ñRó "ñRr" rÉ c ó | \ }}}| j g d¢« }t j |d< |rt j |« }d}t j
|
t |¬« 5 |j |« d d d « y # 1 sw Y y xY w)N)r r r. éÿÿÿÿzOsearchsorted requires array to be sorted, which is impossible with NAs present.r_ )Útaker> r? r² r ra Ú
|
ValueErrorÚ searchsorted)rE Ú as_seriesÚbÚcÚar Úmsgs r Ú test_searchsorted_with_na_raisesrØ sw à GAq!Ø
|
×
|
Ñ
|
¢ Ó
|
*CÜeeCGáÜiinð 0ð ô
|
z¨Ô -ñ Ø ×ÑÔ÷ ÷ ñ ús Á!A<Á<Bc óÒ t j ddg| ¬« }t j ddgt ¬« }||k( }t j ddgt ¬« }| j dk( r.| j
|
t j u r|j d« }n<| j d k( r-| j
|
t j u r|j d
|
« }t j ||« y ) NrÖ rÔ r/ r. FTr· r§ r r¦ )
|
r> r² Úobjectrà r r) r? r« rP Úassert_series_equal)r r Úmixedr] rx s r Ú test_mixed_object_comparisonrÝ s© ä
|
))S#J eÔ
|
,Cä IIq#h¤fÔ -Eà E\FÜyy% ¬dÔ3HØ }}Ò U§^¡^´r·u±uÑ%<Ø?? 9Ó-Ø )Ò #¨¯©¼"¿%¹%Ñ(?Ø?? ?Ó3ä×Ñ6 8Õ,r" )+Ú__doc__Ú
|
__future__r r5 Útypingr Únumpyr0 r Ú pandas.compatr Úpandas.core.dtypes.baser Úpandasr> Úpandas._testingÚ_testingrP Úpandas.api.typesr Úpandas.core.arraysr Úpandas.core.arrays.string_r Ú'pandas.tests.arrays.string_.test_stringr Úpandas.tests.extensionr r! rÍ r r r: rA rE rG rI ÚExtensionTestsrK ÚDim2CompatTestsrÉ rØ rÝ rn r" r ú<module>rî s# ðñõ #ã Ý ã Û å %å 9ã Ý Ý ,Ý /Ý 2Ý QÝ 'ò"ð" e}Ô%ñó &ðð ñ;ó ð;ð
|
ñ+ó ð+ð ñ+ó ð+ð ñ+ó ð+ð
|
ñ+ó ð+ð
|
ñ+ó ð+ôWMd×)Ñ)ô WMôtR4×'Ñ'ô Rò ó" -r"
|
