U
|
ý°dB ã @ sp d Z ddlZddlZddlmZ ddlZddlmZ ddl m
|
Z ddl m Z ddlmZ dd Zejd d
|
Zejdejgd d d Zejdd Zejdejgd dd Zejdejgd dd Zejdejgd dd Zejdejgd dd Zejdd Zejdd Zejdejgd d d Z ejdejgd d 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#Z$G d(d) d)e ej%Z&G d*d+ d+e ej'Z(G d,d- d-ej)Z*G d.d/ d/e ej+Z,G d0d1 d1e ej-Z.G d2d3 d3e ej/Z0G d4d5 d5e ej1Z2G d6d7 d7e Z3G d8d9 d9e ej4Z5G d:d; d;e ej6Z7G d<d= d=ej8Z9dS )>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)ÚPerformanceWarning©Ú SparseDtype)Ú SparseArray)Úbasec C s^ t | ¡rt jjdd}n2t jjdddd}|d |d krL|d d7 < | |dd d
< |S )Néd )Úsizeé r é é )ÚnpÚisnanÚrandomÚuniformÚrandint)Ú
|
fill_valueÚdata© r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/extension/test_sparse.pyÚ make_data s
|
r c C s t S ©Nr r r r r Údtype) s r )Úparamsc C s t t| j| jd}|S )z*Length-100 PeriodArray for semantics test.©r )r r Úparam)ÚrequestÚresr r r r . s r c C s t t d¡d S )Nr r
|
)r r Zonesr r r r Ú data_for_twos5 s r c C s t tjdg| jdS )z Length 2 array with [NA, Valid]r r ©r r Únanr ©r r r r Ú data_missing: s r! c # s fdd}|V dS )z1Return different versions of data for count timesc 3 s( t | D ]}tt j jdV qd S )Nr )Úranger r r )ÚcountÚ_r r r ÚgenD s zdata_repeated.<locals>.genNr )r r% r r r Ú data_repeated@ s r& c C s t dddg| jdS )Nr
|
r r r )r r r r r r Údata_for_sortingK s r' c C s t dtjdg| jdS )Nr
|
r r r r r r r Údata_missing_for_sortingP s r( c C s t jS r )r r r r r r Úna_valueU s r) c C s dd S )Nc S s t | ¡ot |¡S r )ÚpdÚisna)ÚleftÚrightr r r Ú<lambda>\ ó zna_cmp.<locals>.<lambda>r r r r r Úna_cmpZ s r0 c C s" t ddtjtjddddg| jdS )Nr r
|
r r r r r r r Údata_for_grouping_ s r1 c C s$ t ddtjdddddddg
|
| jdS )Nr éþÿÿÿéÿÿÿÿé r
|
r r r r r r r Údata_for_compared s r5 c s2 e Zd Zdd Zejjdd fddZ ZS )ÚBaseSparseTestsc C s |j ttdkrt d¡ d S )Nr z Can't store nan in int array.)r r ÚintÚpytestÚskip©Úselfr r r r Ú_check_unsupportedj s z"BaseSparseTests._check_unsupportedú$SparseArray does not support setitem©Úreasonc s t |¡ d S r )ÚsuperÚ
|
test_ravelr: ©Ú __class__r r rA n s zBaseSparseTests.test_ravel) Ú__name__Ú
|
__module__Ú __qualname__r< r8 ÚmarkÚxfailrA Ú __classcell__r r rB r r6 i s r6 c @ s e Zd Zdd ZdS )Ú TestDtypec C s | ¡ tkstd S r )Zconstruct_array_typer ÚAssertionError)r; r r r r r Útest_array_type_with_argt s z"TestDtype.test_array_type_with_argN)rD rE rF rL r r r r rJ s s rJ c @ s e Zd Zdd Zdd ZdS )Ú TestInterfacec C s | ¡ d S r ©Úcopyr: r r r Ú test_copyy s zTestInterface.test_copyc C s | ¡ d S r )Úviewr: r r r Ú test_view} s zTestInterface.test_viewN)rD rE rF rP rR r r r r rM x s rM c @ s e Zd ZdS )ÚTestConstructorsN©rD rE rF r r r r rS s rS c s° e Zd Zdd Zej dddgejj ddgdd gd
|
g¡ fd d Z
|
fd dZ fddZ fddZ fddZ fddZ fddZejjdd fdd Z ZS ) Ú TestReshapingc C s| t d|d d
i¡}t ddddgi¡}t ddddgi¡ d¡}|||g}t |¡}t d d
|
|D ¡}| ||¡ d S ) NÚAr r r
|
ÚaÚbÚcÚcategoryc S s g | ]}| d d ¡qS )c S s t | ¡ t¡S r )r ÚasarrayÚastypeÚobject)Úsr r r r. r/ zCTestReshaping.test_concat_mixed_dtypes.<locals>.<listcomp>.<lambda>)Úapply©Ú.0Úxr r r Ú
|
<listcomp> s z:TestReshaping.test_concat_mixed_dtypes.<locals>.<listcomp>)r* Ú DataFramer\ ÚconcatÚassert_frame_equal)r; r Zdf1Zdf2Zdf3ÚdfsÚresultÚexpectedr r r Útest_concat_mixed_dtypes s
|
|
ÿz&TestReshaping.test_concat_mixed_dtypesÚcolumnsrV ÚB)rV rW )rV rX ÚouterÚinner)Únamesc s t ||¡ d S r )r@ Ú
|
test_stack)r; r rk rB r r rp s
|
zTestReshaping.test_stackc s | |¡ t ||¡ d S r )r< r@ Útest_concat_columns©r; r r) rB r r rq ¢ s
|
z!TestReshaping.test_concat_columnsc s | |¡ t ||¡ d S r )r< r@ Ú'test_concat_extension_arrays_copy_falserr rB r r rs ¦ s
|
z5TestReshaping.test_concat_extension_arrays_copy_falsec s | |¡ t ||¡ d S r )r< r@ Ú
|
test_alignrr rB r r rt ª s
|
zTestReshaping.test_alignc s | |¡ t ||¡ d S r )r< r@ Útest_align_framerr rB r r ru ® s
|
z TestReshaping.test_align_framec s | |¡ t ||¡ d S r )r< r@ Útest_align_series_framerr rB r r rv ² s
|
z%TestReshaping.test_align_series_framec s | |¡ t ||¡ d S r )r< r@ Ú
|
test_mergerr rB r r rw ¶ s
|
zTestReshaping.test_merger= r> c s t |¡ d S r )r@ Útest_transposer: rB r r rx º s z TestReshaping.test_transpose)rD rE rF rj r8 rG Ú parametrizer* Z
|
MultiIndexÚ from_tuplesrp rq rs rt ru rv rw rH rx rI r r rB r rU s& ÿþþ rU c s$ e Zd Zdd Z fddZ ZS )Ú TestGetitemc C s t j|dd tt|D d}t |jj¡rTt | d¡¡rNt |j d ¡slt
|
n| d¡|j d kslt
|
| d¡|j d kst
|
d S )Nc S s g | ] }d | qS )r
|
r )ra Úir r r rc Á s z(TestGetitem.test_get.<locals>.<listcomp>)Úindexr4 r
|
r ) r* ÚSeriesr" Úlenr r Úvaluesr ÚgetÚilocrK )r; r Úserr r r Útest_getÀ s
|
&zTestGetitem.test_getc s | |¡ t ||¡ d S r )r< r@ Ú test_reindexrr rB r r r
È s
|
zTestGetitem.test_reindex)rD rE rF r r
rI r r rB r r{ ¿ s r{ c @ s e Zd ZdS )Ú TestIndexNrT r r r r r Ð s r c sj e Zd Zdd Z fddZ fddZ fddZ fd d
|
Zej j
|
d d fd dZ dd Z Z S )Ú TestMissingc C s t |}ttt |jj¡}t ddg|d}| ¡ }t ||¡ | d¡}ttt |jj¡}t ddgd|d}|
|
| ¡ |¡ d S )NTF©r r ©r r ) r r Úboolr* r+ r r ÚtmÚassert_sp_array_equalÚfillnaZ assert_equal)r; r! ZsarrZexpected_dtyperi rh r r r Ú test_isnaÕ s
|
zTestMissing.test_isnac s* t jtdd t |¡ W 5 Q R X d S ©NF©Zcheck_stacklevel)r Úassert_produces_warningr r@ Útest_fillna_limit_pad©r; r! rB r r r â s z!TestMissing.test_fillna_limit_padc s* t jtdd t |¡ W 5 Q R X d S r ©r r r r@ Útest_fillna_limit_backfillr rB r r r æ s z&TestMissing.test_fillna_limit_backfillc sL t |j¡r"|j tjjdd¡ tj t
|
dd t |¡ W 5 Q R X d S )Nz'returns array with different fill valuer> Fr ) r r r ÚnodeÚ
|
add_markerr8 rG rH r r r r@ Ú test_fillna_no_op_returns_copy)r; r r rB r r r ê s ÿz*TestMissing.test_fillna_no_op_returns_copyc s* t jtdd t |¡ W 5 Q R X d S r r r rB r r Útest_fillna_series_methodò s z%TestMissing.test_fillna_series_methodZ Unsupportedr> c s t ¡ d S r )r@ Útest_fillna_series)r; rB r r r ö s z TestMissing.test_fillna_seriesc C st |d }t |ddgd¡ |¡}t |j¡r<t|j|}n|j}t |j||g|dddgd¡}| ||¡ d S )Nr r
|
)rV rl r ) r* rd r r+ r r r Ú_from_sequencerf )r; r! r rh r ri r r r Útest_fillna_frameû s þÿz TestMissing.test_fillna_frame)rD rE rF r r r r r r8 rG rH r r rI r r rB r r Ô s r c sª e Zd ZdZdd Zdd Zejjdd fdd Z d
|
d Z
|
fd d Z fddZ dd Z ej dddg¡ fddZej dejejejg¡ fddZ ZS )Ú TestMethodsz Sparse[bool]c C s | ddg¡}tjd|idd}|jd }| |¡}t|jdrn|rZ|jj|jjksnt n|jj|jjk snt |j
|
j ¡ | ¡ kst d S )Nr rV FrN )r r Úblocks) Útaker* rd r r ÚhasattrZ_mgrr r rK rV Ú_valuesÚto_dense)r; r! Úusing_copy_on_writeÚarrZdfÚ
|
filled_valrh r r r Útest_fillna_copy_frame s
|
|
z"TestMethods.test_fillna_copy_framec C sn | ddg¡}tj|dd}|d }| |¡}|rD|j|jksTtn|j|jk sTt|j ¡ | ¡ ksjtd S )Nr FrN r )r r* r~ r r¡ rK r¢ )r; r! r£ r¤ r r¥ rh r r r Útest_fillna_copy_series! s
|
z#TestMethods.test_fillna_copy_serieszNot Applicabler> c s t |¡ d S r )r@ Útest_fillna_length_mismatchr rB r r r¨ / s z'TestMethods.test_fillna_length_mismatchc C sü |d |d kst t|}|d d
\}}t |j||||g|jd¡}t ddddg¡}| |¡}t dd} t |j||||g| d¡}
|
|
|
||
|
¡ |j||||g|jd} t ddddg¡}| || ¡}t |j||||g|jd¡}
|
|
|
||
|
¡ d S ) Nr r r
|
r TFÚfloatg ) rK Útyper* r~ r r r ÚarrayÚwherer Úassert_series_equal) r; r r) ÚclsrW rX r Zcondrh Z new_dtyperi Úotherr r r Útest_where_series3 s
|
|
ÿ z TestMethods.test_where_seriesc s6 |j jdkr&tjjdd}|j |¡ t |¡ d S )Nr7 z2TODO(SparseArray.__setitem__) will preserve dtype.r> ) r Úsubtyper8 rG rH r r r@ Útest_combine_first)r; r r rG rB r r r² I s ÿ z TestMethods.test_combine_firstc s, t jtdd t ||¡ W 5 Q R X d S r )r r r r@ Útest_searchsorted)r; r' Ú as_seriesrB r r r³ S s z TestMethods.test_searchsortedc C s6 | d¡}|jd |jd< |jd |jd ks2td S )Nr r )ÚshiftZ_sparse_valuesrK )r; r rh r r r Útest_shift_0_periodsW s
|
z TestMethods.test_shift_0_periodsÚmethodZargmaxZargminc s | |¡ t |||¡ d S r )r< r@ Útest_argmin_argmax_all_na)r; r· r r) rB r r r¸ ^ s
|
z%TestMethods.test_argmin_argmax_all_naÚboxc s | |¡ t ||||¡ d S r )r< r@ Ú test_equals)r; r r) r´ r¹ rB r r rº d s
|
zTestMethods.test_equals)rD rE rF Z_combine_le_expected_dtyper¦ r§ r8 rG rH r¨ r° r² r³ r¶ ry r¸ r* r« r~ rd rº rI r r rB r r s
|
r c s4 e Zd Zdd Zejjedd fddZ Z S )Ú TestCastingc C sH t |d d
¡ t¡}t jdd |d d
D td}| ||¡ d S )Né c S s g | ] }t |qS r )Ústrr` r r r rc o s z/TestCasting.test_astype_str.<locals>.<listcomp>r )r* r~ r\ r½ r] r )r; r rh ri r r r Útest_astype_strk s zTestCasting.test_astype_strzno sparse StringDtype)Úraisesr? c s t |¡ d S r )r@ Útest_astype_stringr: rB r r rÀ r s z TestCasting.test_astype_string)
|
rD rE rF r¾ r8 rG rH Ú TypeErrorrÀ rI r r rB r r» j s r» c s\ e Zd ZdZdZdZdZdd Z fddZ fddZ fdd Z
|
e f fd
|
d Z Z S ) ÚTestArithmeticOpsNc C s |j dkrt d¡ d S )Nr z=Incorrected expected from Series.combine and tested elsewhere)r r8 r9 r: r r r Ú_skip_if_different_combine} s
|
z,TestArithmeticOps._skip_if_different_combinec s | |¡ t ||¡ d S r )rà r@ Ú test_arith_series_with_scalar©r; r Úall_arithmetic_operatorsrB r r rÄ s
|
z/TestArithmeticOps.test_arith_series_with_scalarc s | |¡ t ||¡ d S r )rà r@ Ú test_arith_series_with_arrayrÅ rB r r rÇ s
|
z.TestArithmeticOps.test_arith_series_with_arrayc sH |j jdkrn(| d¡dkr6tjjdd}|j |¡ t ||¡ d S )Nr r$ )ÚmulZrmulÚfloordivZ rfloordivÚpowÚmodZrmodz result dtype.fill_value mismatchr> )
|
r r Ústripr8 rG rH r r r@ Ú test_arith_frame_with_scalar)r; r rÆ r rG rB r r rÍ s z.TestArithmeticOps.test_arith_frame_with_scalarc s t j|||d d d S )N)Úexc)r@ Ú_check_divmod_op)r; r Úopr¯ rÎ rB r r rÏ s z"TestArithmeticOps._check_divmod_op)rD rE rF Zseries_scalar_excZframe_scalar_excZ
|
divmod_excZseries_array_excrà rÄ rÇ rÍ ÚNotImplementedErrorrÏ rI r r rB r r w s r c @ s` e Zd ZedddZedddZejjddeddd Z ejjddedd
|
d Z
|
d S ) ÚTestComparisonOps)r5 c C s |}|||}t |ts t|jjtjks.tt |trH||j|j}n>t |t |j¡t |¡¡}t||
|
¡ t |¡|tjd}t ||¡ d S )Nr ) Ú
|
isinstancer rK r r± r Zbool_r Úallr[ r¢ r r )r; r5 Ú comparison_opr¯ rÐ rh r ri r r r Ú_compare_other¢ s
|
|
ÿýz TestComparisonOps._compare_otherc C s> | ||d¡ | ||d¡ | ||d¡ | ||tj¡ d S )Nr r r3 )rÖ r r )r; r5 rÕ r r r Ú test_scalar· s z TestComparisonOps.test_scalarz Wrong indicesr> c C s t ddd¡}| |||¡ d S )Néüÿÿÿr¼ é
|
)r ZlinspacerÖ ©r; r5 rÕ r¤ r r r Ú
|
test_array½ s z TestComparisonOps.test_arrayc C s0 |d }| |||¡ |d }| |||¡ d S )Nr r
|
)rÖ rÚ r r r Útest_sparse_array s z#TestComparisonOps.test_sparse_arrayN) rD rE rF r rÖ r× r8 rG rH rÛ rÜ r r r r rÒ ¡ s rÒ c s* e Zd Zejjdd fddZ ZS )Ú TestPrintingzDifferent reprr> c s t ||¡ d S r )r@ Útest_array_repr)r; r r rB r r rÞ Ë s z TestPrinting.test_array_repr)rD rE rF r8 rG rH rÞ rI r r rB r rÝ Ê s rÝ c s. e Zd Zej dddg¡ fddZ ZS )Ú TestParsingÚenginerY Úpythonc s0 d}t jt|d t ||¡ W 5 Q R X d S )Nz,.*must implement _from_sequence_of_strings.*)Úmatch)r8 r¿ rÑ r@ Ú test_EA_types)r; rà r Z expected_msgrB r r rã Ñ s zTestParsing.test_EA_types)rD rE rF r8 rG ry rã rI r r rB r rß Ð s rß c @ s& e Zd Zej dddg¡dd ZdS )ÚTestNoNumericAccumulationsÚskipnaTFc C s d S r r )r; r Zall_numeric_accumulationsrå r r r Útest_accumulate_seriesÙ s z1TestNoNumericAccumulations.test_accumulate_seriesN)rD rE rF r8 rG ry ræ r r r r rä Ø s rä ):Ú__doc__Únumpyr r8 Z pandas.errorsr Zpandasr* r Zpandas._testingZ_testingr Z pandas.arraysr Zpandas.tests.extensionr r Zfixturer r r r r! r& r' r( r) r0 r1 r5 r6 ZBaseDtypeTestsrJ ZBaseInterfaceTestsrM ZBaseConstructorsTestsrS ZBaseReshapingTestsrU ZBaseGetitemTestsr{ ZBaseIndexTestsr ZBaseMissingTestsr ZBaseMethodsTestsr ZBaseCastingTestsr» ZBaseArithmeticOpsTestsr rÒ ZBasePrintingTestsrÝ ZBaseParsingTestsrß ZBaseAccumulateTestsrä r r r r Ú<module> s\
|
|
|
|
|
|
|
|
|
|
|
|
|
|
9<Z *)
|
