o ýòTiÃ<ã@svdZddlZddlZddlmZddlZddlmZddl m Zddlm Z ddlmZdd„Zejd d „ƒZejdejgddd „ƒZejdd„ƒZejdejgddd„ƒZejdejgddd„ƒZejdejgddd„ƒZejdejgddd„ƒZejdd„ƒZejdejgddd„ƒZejdejgddd„ƒZGdd„dƒZGd d!„d!eejƒZGd"d#„d#eej ƒZ!Gd$d%„d%eej"ƒZ#Gd&d'„d'eej$ƒZ%Gd(d)„d)eej&ƒZ'Gd*d+„d+eej(ƒZ)Gd,d-„d-ej*ƒZ+Gd.d/„d/eej,ƒZ-Gd0d1„d1eej.ƒZ/Gd2d3„d3eej0ƒZ1Gd4d5„d5eej2ƒZ3Gd6d7„d7eƒZ4Gd8d9„d9eej5ƒZ6Gd:d;„d;eej7ƒZ8Gdaj 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)ÚbasecCshtj d¡}t |¡r|jdd}n|jdddtd}|d|dkr+|dd7<||ddd…<|S)Nééd)Úsizeé)r Údtyperé)ÚnpÚrandomZdefault_rngÚisnanÚuniformZintegersÚint)Ú fill_valueÚrngÚdata©rúb/var/www/html/evchargy.com/venv/lib/python3.10/site-packages/pandas/tests/extension/test_sparse.pyÚ make_datas rcCstƒS©Nrrrrrr*sr)ÚparamscCstt|jƒ|jd}|S)z*Length-100 PeriodArray for semantics test.©r)rrÚparam)ÚrequestÚresrrrr/srcCstt d¡dƒS)Nrr)rr ZonesrrrrÚ data_for_twos6órcCsttjdg|jdS)zLength 2 array with [NA, Valid]r r©rr Únanr©rrrrÚdata_missing;sr#c#s‡fdd„}|VdS)z1Return different versions of data for count timesc3s*t|ƒD] }ttˆjƒˆjdVqdS)Nr)Úrangerrr)ÚcountÚ_r"rrÚgenEs€ÿzdata_repeated..genNr)rr'rr"rÚ data_repeatedAs€ r(cCstgd¢|jdS)N)rrr r)rrr"rrrÚdata_for_sortingLrr)cCstdtjdg|jdS)Nrr rr r"rrrÚdata_missing_for_sortingQsr*cCsdd„S)NcSst |¡o t |¡Sr)ÚpdÚisna)ÚleftÚrightrrrÚXózna_cmp..rrrrrÚna_cmpVsr1c Cs"tddtjtjddddg|jdS)Nr rrrr r"rrrÚdata_for_grouping[s"r2cCs$tddtjdddddddg |jdS)Nréþÿÿÿéÿÿÿÿérrrr r"rrrÚdata_for_compare`s$r6c@óeZdZdd„ZdS)ÚBaseSparseTestscCs"|jttdƒkrt d¡dSdS)NrzCan't store nan in int array.)rrrÚpytestÚskip©ÚselfrrrrÚ_check_unsupportedfsÿz"BaseSparseTests._check_unsupportedN)Ú__name__Ú __module__Ú__qualname__r=rrrrr8eór8c@r7)Ú TestDtypecCs| ¡tusJ‚dSr)Zconstruct_array_typer)r<rrrrrÚtest_array_type_with_arglsz"TestDtype.test_array_type_with_argN)r>r?r@rCrrrrrBkrArBc@óeZdZdS)Ú TestInterfaceN©r>r?r@rrrrrEpórEc@rD)ÚTestConstructorsNrFrrrrrHtrGrHcs¨eZdZdd„Zej dddgejj ddgdd gd g¡ej ddd g¡‡fdd„ƒƒZ ‡fdd„Z‡fdd„Z‡fdd„Z ‡fdd„Z‡fdd„Z‡fdd„Z‡ZS)Ú TestReshapingcCsxt d|dd…i¡}t dgd¢i¡}t dgd¢i¡ d¡}|||g}t |¡}t dd„|Dƒ¡}t ||¡dS)NÚAr)r rr)ÚaÚbÚcÚcategorycSsg|] }| dd„¡‘qS)cSst |¡ t¡Sr)r ÚasarrayÚastypeÚobject)Úsrrrr/„szCTestReshaping.test_concat_mixed_dtypes...)Úapply)Ú.0ÚxrrrÚ „sz:TestReshaping.test_concat_mixed_dtypes..)r+Ú DataFramerPÚconcatÚtmÚassert_frame_equal)r<rZdf1Zdf2Zdf3ÚdfsÚresultÚexpectedrrrÚtest_concat_mixed_dtypesys ÿz&TestReshaping.test_concat_mixed_dtypesÚcolumnsrJÚB)rJrK)rJrLÚouterÚinner)ÚnamesÚfuture_stackTFcstƒ |||¡dSr)ÚsuperÚ test_stack)r<rr_rd©Ú __class__rrrfˆszTestReshaping.test_stackcó| |¡tƒ ||¡dSr)r=reÚtest_concat_columns©r<rÚna_valuergrrrj•ó z!TestReshaping.test_concat_columnscrir)r=reÚ'test_concat_extension_arrays_copy_falserkrgrrrn™rmz5TestReshaping.test_concat_extension_arrays_copy_falsecrir)r=reÚ test_alignrkrgrrrormzTestReshaping.test_aligncrir)r=reÚtest_align_framerkrgrrrp¡rmzTestReshaping.test_align_framecrir)r=reÚtest_align_series_framerkrgrrrq¥rmz%TestReshaping.test_align_series_framecrir)r=reÚ test_mergerkrgrrrr©rmzTestReshaping.test_merge)r>r?r@r^r9ÚmarkÚparametrizer+Z MultiIndexÚfrom_tuplesrfrjrnrorprqrrÚ __classcell__rrrgrrIxs$ÿþþ rIcs$eZdZdd„Z‡fdd„Z‡ZS)ÚTestGetitemcCsˆtj|dd„tt|ƒƒDƒd}t |jj¡r*t | d¡¡r't |j d¡s)J‚n| d¡|j dks6J‚| d¡|j dksBJ‚dS)NcSsg|]}d|‘qS)rr)rTÚirrrrV°r0z(TestGetitem.test_get..)Úindexr5rr ) r+ÚSeriesr$Úlenr rÚvaluesrÚgetÚiloc)r<rÚserrrrÚtest_get¯s &zTestGetitem.test_getcrir)r=reÚtest_reindexrkrgrrr·rmzTestGetitem.test_reindex)r>r?r@r€rrvrrrgrrw®srwc@rD)ÚTestSetitemNrFrrrrr‚¼rGr‚c@rD)Ú TestIndexNrFrrrrrƒÀrGrƒcsReZdZdd„Z‡fdd„Z‡fdd„Zejjdd‡fd d „ƒZ dd„Z ‡ZS) ÚTestMissingcCs„t|ƒ}ttt |jj¡ƒ}tddg|d}| ¡}t ||¡| d¡}ttt |jj¡ƒ}tddgd|d}t | ¡|¡dS)NTF©rr©rr)rrÚboolr+r,rrrYÚassert_sp_array_equalÚfillnaZassert_equal)r<r#ZsarrZexpected_dtyper]r\rrrÚ test_isnaÅs zTestMissing.test_isnacsHttf}tj|ddtƒ |¡WdƒdS1swYdS©NF)Zcheck_stacklevel)rÚ FutureWarningrYÚassert_produces_warningreÚtest_fillna_limit_backfill)r<r#ZwarnsrgrrrŽÒs"ÿz&TestMissing.test_fillna_limit_backfillcs2t |j¡r|j tjjdd¡tƒ |¡dS)Nz'returns array with different fill value©Úreason) r rrÚnodeÚ add_markerr9rsÚxfailreÚtest_fillna_no_op_returns_copy)r<rrrgrrr”×s ÿz*TestMissing.test_fillna_no_op_returns_copyZUnsupportedrcstƒ ¡dSr)reÚtest_fillna_series)r<rgrrr•ÞszTestMissing.test_fillna_seriescCst|d}t |ddgdœ¡ |¡}t |j¡rt|j|ƒ}n|j}t |j||g|dddgdœ¡}t ||¡dS)Nr r)rJr`r…) r+rWr‰r,rrrÚ_from_sequencerYrZ)r<r#rr\rr]rrrÚtest_fillna_frameãsþÿzTestMissing.test_fillna_frame)r>r?r@rŠrŽr”r9rsr“r•r—rvrrrgrr„Äs r„cs eZdZdZdd„Zdd„Zejjdd‡fdd „ƒZ d d„Z ‡fdd „Zdd„Zej dddg¡‡fdd„ƒZej dejejejg¡‡fdd„ƒZej ddd„dedejgƒfdd„dedejgƒfededd gd d!fededejgƒfg¡d"d#„ƒZej d$ddg¡d%d&„ƒZ‡ZS)'ÚTestMethodszSparse[bool]cCsŠ| ddg¡}tjd|idd}|jd}| |¡}t|jdƒr7|r-|jj|jjus,J‚n |jj|jjus7J‚|j j ¡| ¡usCJ‚dS)Nr rJF©Úcopy)rrÚblocks)Útaker+rWr~r‰ÚhasattrZ_mgrr|rrJÚ_valuesÚto_dense)r<r#Úusing_copy_on_writeÚarrZdfÚ filled_valr\rrrÚtest_fillna_copy_frameûs z"TestMethods.test_fillna_copy_framecCsn| ddg¡}tj|dd}|d}| |¡}|r"|j|jus!J‚n|j|jus*J‚|j ¡| ¡us5J‚dS)Nr Fr™r)rœr+rzr‰ržrŸ)r<r#r r¡rr¢r\rrrÚtest_fillna_copy_series s z#TestMethods.test_fillna_copy_serieszNot Applicablercótƒ |¡dSr)reÚtest_fillna_length_mismatch)r<r#rgrrr¦óz'TestMethods.test_fillna_length_mismatchcCsô|d|dks J‚t|ƒ}|dd…\}}t |j||||g|jd¡}t gd¢¡}| |¡}tddƒ} t |j||||g| d¡} t || ¡|j||||g|jd}t gd¢¡}| ||¡}t |j||||g|jd¡} t || ¡dS) Nrr rr…)TTFFÚfloatg)TFTT)Útyper+rzr–rr ÚarrayÚwhererrYZassert_series_equal)r<rrlÚclsrKrLrZcondr\Z new_dtyper]ÚotherrrrÚtest_where_seriess ÿzTestMethods.test_where_seriescsBtjtddtƒ ||¡WdƒdS1swYdSr‹)rYrrreÚtest_searchsorted)r<r)Ú as_seriesrgrrr¯1s"ÿzTestMethods.test_searchsortedcCs6| d¡}|jd|jd<|jd|jdksJ‚dS)Nrr )ÚshiftZ_sparse_values)r<rr\rrrÚtest_shift_0_periods5s z TestMethods.test_shift_0_periodsÚmethodZargmaxZargmincs| |¡tƒ |||¡dSr)r=reÚtest_argmin_argmax_all_na)r<r³rrlrgrrr´<s z%TestMethods.test_argmin_argmax_all_naÚboxcs | |¡tƒ ||||¡dSr)r=reÚtest_equals)r<rrlr°rµrgrrr¶Bs zTestMethods.test_equalszfunc, na_action, expectedcCó|Srr©rUrrrr/JózTestMethods.Ngð?cCr·rrr¸rrrr/Kr¹Úignorez1.0r!rcCs,tdtjgƒ}|j||d}t ||¡dS)Nr ©Ú na_action)rr r!ÚmaprYZassert_extension_array_equal)r<Úfuncr¼r]rr\rrrÚtest_mapGszTestMethods.test_mapr¼cCsJd}tjt|d|jdd„|dWdƒdS1swYdS)Nz-fill value in the sparse values not supported©ÚmatchcSstjSr)r r!r¸rrrr/[sz-TestMethods.test_map_raises..r»)r9ÚraisesÚ ValueErrorr½)r<rr¼ÚmsgrrrÚtest_map_raisesVs"ÿzTestMethods.test_map_raises)r>r?r@Z_combine_le_expected_dtyper£r¤r9rsr“r¦r®r¯r²rtr´r+rªrzrWr¶rr r!Ústrr¿rÅrvrrrgrr˜øs0üþ r˜cs,eZdZejjedd‡fdd„ƒZ‡ZS)ÚTestCastingzno sparse StringDtype)rÂrcr¥r)reÚtest_astype_stringr;rgrrrÈ_r§zTestCasting.test_astype_string) r>r?r@r9rsr“Ú TypeErrorrÈrvrrrgrrÇ^srÇcsLeZdZdZdZdZdZdd„Z‡fdd„Z‡fdd„Z ‡fdd „Z ‡ZS) ÚTestArithmeticOpsNcCs|jdkrt d¡dSdS)Nrz=Incorrected expected from Series.combine and tested elsewhere)rr9r:r;rrrÚ_skip_if_different_combinejs üz,TestArithmeticOps._skip_if_different_combinecrir)rËreÚtest_arith_series_with_scalar©r<rÚall_arithmetic_operatorsrgrrrÌqrmz/TestArithmeticOps.test_arith_series_with_scalarcrir)rËreÚtest_arith_series_with_arrayrÍrgrrrÏurmz.TestArithmeticOps.test_arith_series_with_arraycsH|jjdkrn| d¡dvrtjjdd}|j |¡tƒ ||¡dS)Nrr&)ÚmulZrmulÚfloordivZ rfloordivÚpowÚmodZrmodz result dtype.fill_value mismatchr) rrÚstripr9rsr“r‘r’reÚtest_arith_frame_with_scalar)r<rrÎrrsrgrrrÕys z.TestArithmeticOps.test_arith_frame_with_scalar)r>r?r@Zseries_scalar_excZframe_scalar_excZ divmod_excZseries_array_excrËrÌrÏrÕrvrrrgrrÊdsrÊc@s`eZdZdefdd„Zdefdd„Zejjdddefdd „ƒZ ejjdddefd d„ƒZ dS) ÚTestComparisonOpsr6cCs–|}|||ƒ}t|tƒsJ‚|jjtjksJ‚t|tƒr$||j|jƒ}nt |t |j¡t |¡ƒ¡}t|| ¡t |¡ƒ|tjd}t ||¡dS)Nr†)Ú isinstancerrÚsubtyper Zbool_rÚallrOrŸrYrˆ)r<r6Ú comparison_oprÚopr\rr]rrrÚ_compare_other‹s ÿýz TestComparisonOps._compare_othercCs>| ||d¡| ||d¡| ||d¡| ||tj¡dS)Nrr r4)rÜr r!)r<r6rÚrrrÚtest_scalar szTestComparisonOps.test_scalarz Wrong indicesrcCs t ddd¡}| |||¡dS)Néüÿÿÿéé )r ZlinspacerÜ©r<r6rÚr¡rrrÚ test_array¦szTestComparisonOps.test_arraycCs0|d}| |||¡|d}| |||¡dS)Nr r)rÜrárrrÚtest_sparse_array«sz#TestComparisonOps.test_sparse_arrayN)r>r?r@rrÜrÝr9rsr“rârãrrrrrÖŠsrÖcs*eZdZejjdd‡fdd„ƒZ‡ZS)ÚTestPrintingzDifferent reprrcstƒ ||¡dSr)reÚtest_array_repr)r<rr rgrrrå´rzTestPrinting.test_array_repr)r>r?r@r9rsr“rårvrrrgrrä³sräcs.eZdZej dddg¡‡fdd„ƒZ‡ZS)ÚTestParsingÚenginerMÚpythoncsFd}tjt|dtƒ ||¡WdƒdS1swYdS)Nz,.*must implement _from_sequence_of_strings.*rÀ)r9rÂÚNotImplementedErrorreÚ test_EA_types)r<rçrZexpected_msgrgrrrêºs"ÿzTestParsing.test_EA_types)r>r?r@r9rsrtrêrvrrrgrræ¹sræc@rD)ÚTestNoNumericAccumulationsNrFrrrrrëÁrGrë);Ú__doc__Únumpyr r9Z pandas.errorsrZpandasr+rZpandas._testingZ_testingrYZ pandas.arraysrZpandas.tests.extensionrrZfixturerr!rrr#r(r)r*r1r2r6r8ZBaseDtypeTestsrBZBaseInterfaceTestsrEZBaseConstructorsTestsrHZBaseReshapingTestsrIZBaseGetitemTestsrwZBaseSetitemTestsr‚ZBaseIndexTestsrƒZBaseMissingTestsr„ZBaseMethodsTestsr˜ZBaseCastingTestsrÇZBaseArithmeticOpsTestsrÊrÖZBasePrintingTestsräZBaseParsingTestsræZBaseAccumulateTestsrërrrrÚs\ 64f&)