a þdÃ<ã@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}n2|jdddtd}|d|dkrV|dd7<||ddd…<|S)Nééd)Úsizeé)r Údtyperé)ÚnpÚrandomZdefault_rngÚisnanÚuniformZintegersÚint)Ú fill_valueÚrngÚdata©rúk/var/www/html/stable-diffusion-webui/venv/lib/python3.9/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_twos6srcCsttjdg|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ÚgenEszdata_repeated..genNr)rr&rr!rÚ data_repeatedAsr'cCstgd¢|jdS)N)rrr r)rrr!rrrÚdata_for_sortingLsr(cCstdtjdg|jdS)Nrr rrr!rrrÚdata_missing_for_sortingQsr)cCsdd„S)NcSst |¡ot |¡Sr)ÚpdÚisna)ÚleftÚrightrrrÚXózna_cmp..rrrrrÚna_cmpVsr0c Cs"tddtjtjddddg|jdS)Nr rrrrr!rrrÚdata_for_grouping[sr1cCs$tddtjdddddddg |jdS)Nréþÿÿÿéÿÿÿÿérrrrr!rrrÚdata_for_compare`sr5c@seZdZdd„ZdS)ÚBaseSparseTestscCs|jttdƒkrt d¡dS)NrzCan't store nan in int array.)rrrÚpytestÚskip©ÚselfrrrrÚ_check_unsupportedfsz"BaseSparseTests._check_unsupportedN)Ú__name__Ú __module__Ú__qualname__r;rrrrr6esr6c@seZdZdd„ZdS)Ú TestDtypecCs| ¡tusJ‚dSr)Zconstruct_array_typer)r:rrrrrÚtest_array_type_with_arglsz"TestDtype.test_array_type_with_argN)r<r=r>r@rrrrr?ksr?c@seZdZdS)Ú TestInterfaceN©r<r=r>rrrrrApsrAc@seZdZdS)ÚTestConstructorsNrBrrrrrCtsrCcs¨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.„r/zCTestReshaping.test_concat_mixed_dtypes...)Úapply)Ú.0ÚxrrrÚ „r/z:TestReshaping.test_concat_mixed_dtypes..)r*Ú DataFramerKÚconcatÚtmÚassert_frame_equal)r:rZdf1Zdf2Zdf3ÚdfsÚresultÚexpectedrrrÚtest_concat_mixed_dtypesys ÿz&TestReshaping.test_concat_mixed_dtypesÚcolumnsrEÚB)rErF)rErGÚouterÚinner)ÚnamesÚfuture_stackTFcstƒ |||¡dSr)ÚsuperÚ test_stack)r:rrZr_©Ú __class__rrraˆszTestReshaping.test_stackcs| |¡tƒ ||¡dSr)r;r`Útest_concat_columns©r:rÚna_valuerbrrrd•s z!TestReshaping.test_concat_columnscs| |¡tƒ ||¡dSr)r;r`Ú'test_concat_extension_arrays_copy_falsererbrrrg™s z5TestReshaping.test_concat_extension_arrays_copy_falsecs| |¡tƒ ||¡dSr)r;r`Ú test_alignrerbrrrhs zTestReshaping.test_aligncs| |¡tƒ ||¡dSr)r;r`Útest_align_framererbrrri¡s zTestReshaping.test_align_framecs| |¡tƒ ||¡dSr)r;r`Útest_align_series_framererbrrrj¥s z%TestReshaping.test_align_series_framecs| |¡tƒ ||¡dSr)r;r`Ú test_mergererbrrrk©s zTestReshaping.test_merge)r<r=r>rYr7ÚmarkÚparametrizer*Z MultiIndexÚfrom_tuplesrardrgrhrirjrkÚ __classcell__rrrbrrDxs"ÿþþ rDcs$eZdZdd„Z‡fdd„Z‡ZS)ÚTestGetitemcCsˆtj|dd„tt|ƒƒDƒd}t |jj¡rTt | d¡¡rNt |j d¡slJ‚n| d¡|j dkslJ‚| d¡|j dks„J‚dS)NcSsg|]}d|‘qS)rr)rOÚirrrrQ°r/z(TestGetitem.test_get..)Úindexr4rr ) r*ÚSeriesr#Úlenr rÚvaluesrÚgetÚiloc)r:rÚserrrrÚtest_get¯s &zTestGetitem.test_getcs| |¡tƒ ||¡dSr)r;r`Útest_reindexrerbrrrz·s zTestGetitem.test_reindex)r<r=r>ryrzrorrrbrrp®srpc@seZdZdS)ÚTestSetitemNrBrrrrr{¼sr{c@seZdZdS)Ú TestIndexNrBrrrrr|Àsr|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+rrrTÚassert_sp_array_equalÚfillnaZassert_equal)r:r"ZsarrZexpected_dtyperXrWrrrÚ test_isnaÅs zTestMissing.test_isnacsFttf}tj|ddtƒ |¡Wdƒn1s80YdS©NF)Zcheck_stacklevel)rÚ FutureWarningrTÚassert_produces_warningr`Útest_fillna_limit_backfill)r:r"Zwarnsrbrrr‡Òsz&TestMissing.test_fillna_limit_backfillcs2t |j¡r"|j tjjdd¡tƒ |¡dS)Nz'returns array with different fill value©Úreason) r rrÚnodeÚ add_markerr7rlÚxfailr`Útest_fillna_no_op_returns_copy)r:rrrbrrr×s ÿz*TestMissing.test_fillna_no_op_returns_copyZUnsupportedrˆcstƒ ¡dSr)r`Útest_fillna_series)r:rbrrrŽÞszTestMissing.test_fillna_seriescCst|d}t |ddgdœ¡ |¡}t |j¡rrƒr‡rr7rlrŒrŽrrorrrbrr}Ä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ƒrn|rZ|jj|jjusnJ‚n|jj|jjusnJ‚|j j ¡| ¡us†J‚dS)Nr rEF©Úcopy)rrÚblocks)Útaker*rRrwr‚ÚhasattrZ_mgrrurrEÚ_valuesÚto_dense)r:r"Úusing_copy_on_writeÚarrZdfÚ filled_valrWrrrÚtest_fillna_copy_frameûs z"TestMethods.test_fillna_copy_framecCsn| ddg¡}tj|dd}|d}| |¡}|rD|j|jusTJ‚n|j|jusTJ‚|j ¡| ¡usjJ‚dS)Nr Fr’r)r•r*rsr‚r—r˜)r:r"r™ršrxr›rWrrrÚtest_fillna_copy_series s z#TestMethods.test_fillna_copy_serieszNot Applicablerˆcstƒ |¡dSr)r`Útest_fillna_length_mismatch)r:r"rbrrržsz'TestMethods.test_fillna_length_mismatchcCsô|d|dksJ‚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*rsrrr ÚarrayÚwhererrTZassert_series_equal)r:rrfÚclsrFrGrxZcondrWZ new_dtyperXÚotherrrrÚtest_where_seriess ÿzTestMethods.test_where_seriescs@tjtddtƒ ||¡Wdƒn1s20YdSr„)rTr†rr`Útest_searchsorted)r:r(Ú as_seriesrbrrr¦1szTestMethods.test_searchsortedcCs6| d¡}|jd|jd<|jd|jdks2J‚dS)Nrr )ÚshiftZ_sparse_values)r:rrWrrrÚtest_shift_0_periods5s z TestMethods.test_shift_0_periodsÚmethodZargmaxZargmincs| |¡tƒ |||¡dSr)r;r`Útest_argmin_argmax_all_na)r:rªrrfrbrrr«<s z%TestMethods.test_argmin_argmax_all_naÚboxcs | |¡tƒ ||||¡dSr)r;r`Útest_equals)r:rrfr§r¬rbrrrBs zTestMethods.test_equalszfunc, na_action, expectedcCs|Srr©rPrrrr.Jr/zTestMethods.Ngð?cCs|Srrr®rrrr.Kr/Úignorez1.0r rcCs,tdtjgƒ}|j||d}t ||¡dS)Nr ©Ú na_action)rr r ÚmaprTZassert_extension_array_equal)r:Úfuncr±rXrrWrrrÚtest_mapGszTestMethods.test_mapr±cCsHd}tjt|d"|jdd„|dWdƒn1s:0YdS)Nz-fill value in the sparse values not supported©ÚmatchcSstjSr)r r r®rrrr.[r/z-TestMethods.test_map_raises..r°)r7ÚraisesÚ ValueErrorr²)r:rr±ÚmsgrrrÚtest_map_raisesVszTestMethods.test_map_raises)r<r=r>Z_combine_le_expected_dtyperœrr7rlrŒržr¥r¦r©rmr«r*r¡rsrRrrr r Ústrr´rºrorrrbrr‘øs.üþ r‘cs,eZdZejjedd‡fdd„ƒZ‡ZS)ÚTestCastingzno sparse StringDtype)r·r‰cstƒ |¡dSr)r`Útest_astype_stringr9rbrrr½_szTestCasting.test_astype_string) r<r=r>r7rlrŒÚ TypeErrorr½rorrrbrr¼^sr¼csLeZdZdZdZdZdZdd„Z‡fdd„Z‡fdd„Z ‡fdd „Z ‡ZS) ÚTestArithmeticOpsNcCs|jdkrt d¡dS)Nrz=Incorrected expected from Series.combine and tested elsewhere)rr7r8r9rrrÚ_skip_if_different_combinejs z,TestArithmeticOps._skip_if_different_combinecs| |¡tƒ ||¡dSr)rÀr`Útest_arith_series_with_scalar©r:rÚall_arithmetic_operatorsrbrrrÁqs z/TestArithmeticOps.test_arith_series_with_scalarcs| |¡tƒ ||¡dSr)rÀr`Útest_arith_series_with_arrayrÂrbrrrÄus z.TestArithmeticOps.test_arith_series_with_arraycsH|jjdkrn(| d¡dvr6tjjdd}|j |¡tƒ ||¡dS)Nrr%)ÚmulZrmulÚfloordivZ rfloordivÚpowÚmodZrmodz result dtype.fill_value mismatchrˆ) rrÚstripr7rlrŒrŠr‹r`Útest_arith_frame_with_scalar)r:rrÃrrlrbrrrÊys z.TestArithmeticOps.test_arith_frame_with_scalar)r<r=r>Zseries_scalar_excZframe_scalar_excZ divmod_excZseries_array_excrÀrÁrÄrÊrorrrbrr¿dsr¿c@s`eZdZedœdd„Zedœdd„Zejjddedœdd „ƒZ ejjddedœd d„ƒZ dS) ÚTestComparisonOps)r5cCs–|}|||ƒ}t|tƒsJ‚|jjtjks.J‚t|tƒrH||j|jƒ}n>t |t |j¡t |¡ƒ¡}t|| ¡t |¡ƒ|tjd}t ||¡dS)Nr)Ú isinstancerrÚsubtyper Zbool_rÚallrJr˜rTr)r:r5Ú comparison_opr¤ÚoprWrrXrrrÚ_compare_other‹s ÿýz TestComparisonOps._compare_othercCs>| ||d¡| ||d¡| ||d¡| ||tj¡dS)Nrr r3)rÑr r )r:r5rÏrrrÚtest_scalar szTestComparisonOps.test_scalarz Wrong indicesrˆcCs t ddd¡}| |||¡dS)Néüÿÿÿéé )r ZlinspacerÑ©r:r5rÏ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Òr7rlrŒr×rØrrrrrËŠsrËcs*eZdZejjdd‡fdd„ƒZ‡ZS)ÚTestPrintingzDifferent reprrˆcstƒ ||¡dSr)r`Útest_array_repr)r:rr rbrrrÚ´szTestPrinting.test_array_repr)r<r=r>r7rlrŒrÚrorrrbrrÙ³srÙcs.eZdZej dddg¡‡fdd„ƒZ‡ZS)ÚTestParsingÚenginerHÚpythoncsDd}tjt|dtƒ ||¡Wdƒn1s60YdS)Nz,.*must implement _from_sequence_of_strings.*rµ)r7r·ÚNotImplementedErrorr`Ú test_EA_types)r:rÜrZexpected_msgrbrrrßºszTestParsing.test_EA_types)r<r=r>r7rlrmrßrorrrbrrÛ¹srÛc@seZdZdS)ÚTestNoNumericAccumulationsNrBrrrrràÁsrà);Ú__doc__Únumpyr r7Z pandas.errorsrZpandasr*rZpandas._testingZ_testingrTZ pandas.arraysrZpandas.tests.extensionrrZfixturerr rrr"r'r(r)r0r1r5r6ZBaseDtypeTestsr?ZBaseInterfaceTestsrAZBaseConstructorsTestsrCZBaseReshapingTestsrDZBaseGetitemTestsrpZBaseSetitemTestsr{ZBaseIndexTestsr|ZBaseMissingTestsr}ZBaseMethodsTestsr‘ZBaseCastingTestsr¼ZBaseArithmeticOpsTestsr¿rËZBasePrintingTestsrÙZBaseParsingTestsrÛZBaseAccumulateTestsràrrrrÚsZ 64f&)