a j=ic6<ã@sŠdZddlZddlZddlmZddlmZmZddl Z ddlmZ ddlmZddlmZddlmZedœd d „Ze jZdHeedœd d„Zejddgddd„ƒZejdd„ƒZejdd„ƒZejdd„ƒZejdd„ƒZejdd„ƒZ ejdd„ƒZ!ejd d!„ƒZ"ejd"d#„ƒZ#ejd$d%„ƒZ$ej% &d%¡Z'Gd&d'„d'ƒZ(Gd(d)„d)e(ej)ƒZ*Gd*d+„d+e(ej+ƒZ,Gd,d-„d-e(ej-ƒZ.Gd.d/„d/e(ej/ƒZ0Gd0d1„d1e(ej1ƒZ2Gd2d3„d3e(ej3ƒZ4Gd4d5„d5e(ej5ƒZ6Gd6d7„d7e(ej7ƒZ8Gd8d9„d9e(ej9ƒZ:Gd:d;„d;e(ej;ƒZGd>d?„d?e(ej?ƒZ@Gd@dA„dAe(ejAƒZBGdBdC„dCe(ejCƒZDe'GdDdE„dEe(ejEƒƒZFGdFdG„dGe(ejGƒZHdS)IaÎ 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/`. Note: we do not bother with base.BaseIndexTests because PandasArray will never be held in an Index. éN)Úcan_hold_element)ÚExtensionDtypeÚPandasDtype)ÚPandasArray)Úblocks)Úbase)ÚreturncCst|tƒr| ¡}t||ƒS©N)Ú isinstancerÚto_numpyr)ÚobjÚelement©rúr/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/pandas/tests/extension/test_numpy.pyÚ_can_hold_element_patched"s rÚ Attributes)ÚattrrcCst|dkrbt|ddƒ}t|ddƒ}t|tƒrBt|tƒsB| |j¡}n t|tƒrbt|tƒsb| |j¡}t||||ƒdS)zh patch tm.assert_attr_equal so PandasDtype("object") is closed enough to np.dtype("object") ÚdtypeN)Úgetattrr rÚastypeÚnumpy_dtypeÚorig_assert_attr_equal)rÚleftÚrightrZlattrZrattrrrrÚ_assert_attr_equal+srÚfloatÚobject)ÚparamscCstt |j¡ƒSr )rÚnprÚparam)Úrequestrrrr;srccs^| ¡B}| tdd¡| tdt¡| tjdt¡dVWdƒn1sP0YdS)aó A monkeypatch to tells pandas to let us in. By default, passing a PandasArray to an index / series / frame constructor will unbox that PandasArray to an ndarray, and treat it as a non-EA column. We don't want people using EAs without reason. The mechanism for this is a check against ABCPandasArray in each constructor. But, for testing, we need to allow them in pandas. So we patch the _typ of PandasArray, so that we evade the ABCPandasArray check. Z_typÚ extensionrÚassert_attr_equalN)ÚcontextÚsetattrrrrÚtmZ assertersr)ZmonkeypatchÚmrrrÚallow_in_pandas@s r'cCs:|jdkr$t dd„tdƒDƒ¡jSttjdd|jdƒS)NrcSsg|] }|f‘qSrr)Ú.0ÚirrrÚ [ózdata..édéée©r) rÚpdÚSeriesÚrangeÚarrayrrÚarangeÚ_dtype©r'rrrrÚdataXs r7cCs6|jdkr"ttjtjdgtdƒStt tjdg¡ƒS)Nr©r-r/gð?©rrrr3Únanrr6rrrÚdata_missing_s r;cCstjSr )rr:rrrrÚna_valuefsr<cCsdd„}|S)NcSst |¡ot |¡Sr )rÚisnan)ÚaÚbrrrÚcmpmszna_cmp..cmpr)r@rrrÚna_cmpksrAcCs:|jdkr(ttjgd¢tddd…ƒStt gd¢¡ƒS)ziLength-3 array with a known sort order. This should be three items [B, C, A] with A < B < C r)r©é©ér8r/r-N)r-rCr)rrrr3rr6rrrÚdata_for_sortingss rFcCs:|jdkr$ttjdtjdgtdƒStt dtjdg¡ƒS)zvLength-3 array with a known sort order. This should be three items [B, NA, A] with A < B and NA missing. rr8)rr/r-rr9r6rrrÚdata_missing_for_sortings rGc CsN|jdkrd\}}}nt d¡\}}}ttj||tjtj||||g|jdƒS)z“Data for factorization, grouping, and unique tests. Expected to be like [B, B, NA, NA, A, A, B, C] Where A < B < C and NA is missing r)r8rBrDrEr/)rrr4rr3r:)r'rr>r?ÚcrrrÚdata_for_groupings "ÿrIcCs&|dkr"tjjdd}|j |¡dS)aŒ Tests for PandasArray with nested data. Users typically won't create these objects via `pd.array`, but they can show up through `.array` on a Series with nested data. Many of the base tests fail, as they aren't appropriate for nested data. This fixture allows these tests to be skipped when used as a usefixtures marker to either an individual test or a test class. rúFails for object dtype©ÚreasonN)ÚpytestÚmarkÚxfailÚnodeÚ add_marker)rr rNrrrÚskip_numpy_objectžsrRc@seZdZedd„ƒZdS)ÚBaseNumPyTestscOsNt|tjƒr4t|jtƒs4t|jtƒr4| t|jƒ¡}tj||g|¢Ri|¤ŽSr ) r r0r1rrrrr%Úassert_series_equal)ÚclsrrÚargsÚkwargsrrrrT²s ÿ þ ýz"BaseNumPyTests.assert_series_equalN)Ú__name__Ú __module__Ú__qualname__ÚclassmethodrTrrrrrS±srScs eZdZe‡fdd„ƒZ‡ZS)ÚTestCastingcstƒ |¡dSr )ÚsuperÚtest_astype_str©Úselfr7©Ú __class__rrr^ÀszTestCasting.test_astype_str)rXrYrZÚskip_nestedr^Ú __classcell__rrrarr\¿sr\cs6eZdZejjdddd„ƒZe‡fdd„ƒZ‡Z S)ÚTestConstructorszWe don't register our dtyperKcCsdSr rr_rrrÚtest_from_dtypeÇsz TestConstructors.test_from_dtypecstƒ ||¡dSr )r]Ú)test_series_constructor_scalar_with_index)r`r7rrarrrgÌsz:TestConstructors.test_series_constructor_scalar_with_index) rXrYrZrMrNÚskiprfrcrgrdrrrarreÆs recseZdZ‡fdd„Z‡ZS)Ú TestDtypecs<|jjdkr,|j tjjd|jj›d¡tƒ |¡dS)Nrz2PandasArray expectedly clashes with a NumPy name: rK) rrrPrQrMrNrOr]Útest_check_dtype)r`r7r rarrrjÓsÿÿÿzTestDtype.test_check_dtype)rXrYrZrjrdrrrarriÒsrics eZdZe‡fdd„ƒZ‡ZS)ÚTestGetitemcstƒ |¡dSr )r]Útest_getitem_scalarr_rarrrlßszTestGetitem.test_getitem_scalar)rXrYrZrcrlrdrrrarrkÞsrkcseZdZ‡fdd„Z‡ZS)ÚTestGroupbycstƒ ||¡dSr )r]Útest_groupby_extension_apply)r`rIZgroupby_apply_oprarrrnæsz(TestGroupby.test_groupby_extension_apply)rXrYrZrnrdrrrarrmåsrmcs eZdZe‡fdd„ƒZ‡ZS)Ú TestInterfacecstƒ |¡dSr )r]Útest_array_interfacer_rarrrpësz"TestInterface.test_array_interface)rXrYrZrcrprdrrrarroêsrocs†eZdZe‡fdd„ƒZe‡fdd„ƒZe‡fdd„ƒZe‡fdd„ƒZej j d d ‡fdd„ƒZ‡fd d„Ze‡fdd„ƒZ ‡ZS)ÚTestMethodscstƒ |¡dSr )r]Útest_shift_fill_valuer_rarrrròsz!TestMethods.test_shift_fill_valuecstƒ |¡dSr )r]Útest_fillna_copy_frame©r`r;rarrrs÷sz"TestMethods.test_fillna_copy_framecstƒ |¡dSr )r]Útest_fillna_copy_seriesrtrarrruüsz#TestMethods.test_fillna_copy_seriescstƒ ||¡dSr )r]Útest_searchsorted)r`rFZ as_seriesrarrrvszTestMethods.test_searchsortedz"PandasArray.diff may fail on dtyperKcstƒ ||¡Sr )r]Ú test_diff)r`r7ZperiodsrarrrwszTestMethods.test_diffcs6|jjtkr&tjjdd}|j |¡tƒ |¡dS)Nz$Dimension mismatch in np.concatenaterK) rrrrMrNrOrPrQr]Útest_insert)r`r7r rNrarrrx szTestMethods.test_insertcstƒ ||¡dSr )r]Útest_insert_invalid©r`r7Zinvalid_scalarrarrryszTestMethods.test_insert_invalid)rXrYrZrcrrrsrurvrMrNrOrwrxryrdrrrarrqñsrqcsheZdZdZdZdZdZe‡fdd„ƒZedd„ƒZ e‡fdd„ƒZ ‡fdd „Ze‡fd d„ƒZ‡Z S)ÚTestArithmeticsNcstƒ |¡dSr )r]Útest_divmodr_rarrr|szTestArithmetics.test_divmodcCs t |¡}|j|t|dddS)N)Úexc)r0r1Z_check_divmod_opÚdivmod)r`r7ZserrrrÚtest_divmod_series_array!s z(TestArithmetics.test_divmod_series_arraycstƒ ||¡dSr )r]Útest_arith_series_with_scalar©r`r7Úall_arithmetic_operatorsrarrr€&sz-TestArithmetics.test_arith_series_with_scalarcsD|}|jjtkr2|dvr2tjjdd}|j |¡tƒ ||¡dS)N)Ú__add__Ú__radd__rJrK) rrrrMrNrOrPrQr]Útest_arith_series_with_array)r`r7r‚r ÚopnamerNrarrr…*s z,TestArithmetics.test_arith_series_with_arraycstƒ ||¡dSr )r]Útest_arith_frame_with_scalarrrarrr‡1sz,TestArithmetics.test_arith_frame_with_scalar)rXrYrZZ divmod_excZseries_scalar_excZframe_scalar_excZseries_array_excrcr|rr€r…r‡rdrrrarr{s r{c@seZdZdS)ÚTestPrintingN©rXrYrZrrrrrˆ6srˆcs6eZdZdd„Zej dddg¡‡fdd„ƒZ‡ZS)ÚTestNumericReducecCs:t||ƒ|d}t| |jj¡|ƒ|d}t ||¡dS)N)Úskipna)rrrr5r%Zassert_almost_equal)r`ÚsZop_namer‹ÚresultÚexpectedrrrÚcheck_reduce;szTestNumericReduce.check_reducer‹TFcstƒ |||¡dSr )r]Útest_reduce_series)r`r7Zall_boolean_reductionsr‹rarrrAsz$TestNumericReduce.test_reduce_series) rXrYrZrrMrNÚparametrizerrdrrrarrŠ:srŠc@seZdZdS)ÚTestBooleanReduceNr‰rrrrr’Fsr’cs0eZdZe‡fdd„ƒZe‡fdd„ƒZ‡ZS)ÚTestMissingcstƒ |¡dSr )r]Útest_fillna_seriesrtrarrr”LszTestMissing.test_fillna_seriescstƒ |¡dSr )r]Útest_fillna_framertrarrr•QszTestMissing.test_fillna_frame)rXrYrZrcr”r•rdrrrarr“Ksr“c sBeZdZej ddejdejjdddg¡‡fdd„ƒZ‡Z S) Ú TestReshapingÚin_frameTFz$PandasArray inconsistently extractedrK©Zmarkscstƒ ||¡dSr )r]Útest_concat)r`r7r—rarrr™XszTestReshaping.test_concat) rXrYrZrMrNr‘rrOr™rdrrrarr–Wsþþþ r–c s¢eZdZe‡fdd„ƒZe‡fdd„ƒZeej dddg¡‡fdd „ƒƒZ e‡fd d„ƒZ eejjde gd ¢¡e jgd ¢ddgddgd‡fdd„ƒƒZ‡fdd„Zeejjdgd¢e jgd¢dde gd¢¡ggd¢d‡fdd„ƒƒZejjdddd e jgd!fejddd e jgd"ejjd#e jddd e jgddd!fe jddd e jgddd!fggd$¢d‡fd%d&„ƒZe‡fd'd(„ƒZe‡fd)d*„ƒZd+d,„Z‡ZS)-ÚTestSetitemcstƒ ||¡dSr )r]Útest_setitem_invalidrzrarrr›gsz TestSetitem.test_setitem_invalidcstƒ ||¡dSr )r]Ú test_setitem_sequence_broadcasts©r`r7Ú box_in_seriesrarrrœlsz,TestSetitem.test_setitem_sequence_broadcastsÚsetterÚlocNcstƒ ||¡dSr )r]Útest_setitem_mask_broadcast)r`r7rŸrarrr¡rsz'TestSetitem.test_setitem_mask_broadcastcstƒ |¡dSr )r]Ú&test_setitem_scalar_key_sequence_raiser_rarrr¢ysz2TestSetitem.test_setitem_scalar_key_sequence_raiseÚmask)TTTFFÚbooleanr/únumpy-arrayz boolean-array)Zidscstƒ |||¡dSr )r]Útest_setitem_mask)r`r7r£ržrarrr¦s zTestSetitem.test_setitem_maskcstƒ ||¡dSr )r]Útest_setitem_mask_raisesrrarrr§sz$TestSetitem.test_setitem_mask_raisesÚidx)rr-rCZInt64)Úlistz integer-arrayr¥cstƒ |||¡dSr )r]Útest_setitem_integer_array©r`r7r¨ržrarrrªsz&TestSetitem.test_setitem_integer_arrayzidx, box_in_seriesrr-rCFTr˜)z list-Falsez list-Truezinteger-array-Falsezinteger-array-Truecstƒ |||¡dSr )r]Ú(test_setitem_integer_with_missing_raisesr«rarrr¬™sz4TestSetitem.test_setitem_integer_with_missing_raisescstƒ ||¡dSr )r]Útest_setitem_slicerrarrr¦szTestSetitem.test_setitem_slicecstƒ |¡dSr )r]Útest_setitem_loc_iloc_slicer_rarrr®ªsz'TestSetitem.test_setitem_loc_iloc_slicecCs„t dt |¡i¡}}tj|jd}||ƒ}|d|j|df<|jjtkrtt|t ƒrb|t dƒkrtt d| ¡i¡}| ||¡dS)Nr7)Úindex)r0Z DataFramer1r¯r rrrr ÚslicerZassert_frame_equal)r`r7Zfull_indexerÚdfrŽrÚkeyrrrÚ,test_setitem_with_expansion_dataframe_column®sz8TestSetitem.test_setitem_with_expansion_dataframe_column)rXrYrZrcr›rœrMrNr‘r¡r¢rr3r0r¦r§rªZNArrOr¬rr®r³rdrrrarršfsRþú$ýüø ršc@seZdZdS)ÚTestParsingNr‰rrrrr´Àsr´c@seZdZdS)ÚTest2DCompatNr‰rrrrrµÅsrµ)r)IÚ__doc__ÚnumpyrrMZpandas.core.dtypes.castrZpandas.core.dtypes.dtypesrrZpandasr0Zpandas._testingZ_testingr%Zpandas.core.arrays.numpy_rZpandas.core.internalsrZpandas.tests.extensionrÚboolrr"rÚstrrZfixturerr'r7r;r<rArFrGrIrRrNZusefixturesrcrSZBaseCastingTestsr\ZBaseConstructorsTestsreZBaseDtypeTestsriZBaseGetitemTestsrkZBaseGroupbyTestsrmZBaseInterfaceTestsroZBaseMethodsTestsrqZBaseArithmeticOpsTestsr{ZBasePrintingTestsrˆZBaseNumericReduceTestsrŠZBaseBooleanReduceTestsr’ZBaseMissingTestsr“ZBaseReshapingTestsr–ZBaseSetitemTestsršZBaseParsingTestsr´ZNDArrayBacked2DTestsrµrrrrÚsh &Z