U *¿if<2ã@s dZddlZddlZddlZddlmZddl m Z ddlmZdd„Z ejdd„ƒZejd d „ƒZejdd„ƒZejd d„ƒZejdd„ƒZejdd„ƒZejdd„ƒZejdd„ƒZejdd„ƒZGdd„dejƒZGdd„dejƒZGdd„dejƒZGdd „d ejƒZGd!d"„d"ej ƒZ!Gd#d$„d$ej"ƒZ#Gd%d&„d&ej$ƒZ%Gd'd(„d(ej&ƒZ'Gd)d*„d*ej(ƒZ)Gd+d,„d,ej*ƒZ+Gd-d.„d.ej,ƒZ-Gd/d0„d0ej.ƒZ/Gd1d2„d2ej0ƒZ1Gd3d4„d4ej2ƒZ3Gd5d6„d6ej4ƒZ5Gd7d8„d8ej6ƒZ7Gd9d:„d:ej8ƒZ9Gd;d<„d„d>ej<ƒZ=dS)?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©ÚBooleanDtype)ÚbasecCs0ddgdtjgddgdtjgddgS)NTFéé,)ÚnpÚnan©r r úV/home/mars/bis/venv/lib/python3.8/site-packages/pandas/tests/extension/test_boolean.pyÚ make_datasrcCstƒS©Nrr r r r Údtypesr cCstjtƒ|dS)N©r )ÚpdÚarrayrrr r r Údata"srcCstjt d¡|dS)Nédr)rrrZonesrr r r Ú data_for_twos'srcCstjtjdg|dS)NTr©rrrrrr r r Údata_missing,srcCstjdddg|dS©NTFr)rrrr r r Údata_for_sorting1srcCstjdtjdg|dSrrrr r r Údata_missing_for_sorting6srcCsdd„S)NcSs|tjko|tjkSr©rÚNA)ÚxÚyr r r Ú>ózna_cmp..r r r r r Úna_cmp;srcCstjSrrr r r r Úna_valueAsr cCs*d}d}tj}tj|||||||g|dSr)rrrr)r ÚbÚaÚnar r r Údata_for_groupingFsr$c@seZdZdS)Ú TestDtypeN©Ú__name__Ú __module__Ú__qualname__r r r r r%Nsr%c@seZdZdS)Ú TestInterfaceNr&r r r r r*Rsr*c@seZdZdS)ÚTestConstructorsNr&r r r r r+Vsr+c@seZdZdS)ÚTestGetitemNr&r r r r r,Zsr,c@seZdZdS)ÚTestSetitemNr&r r r r r-^sr-c@seZdZdS)Ú TestIndexNr&r r r r r.bsr.c@seZdZdS)ÚTestMissingNr&r r r r r/fsr/cs@eZdZddhZd ‡fdd„ Zefdd„Zd‡fdd „ Z‡ZS)ÚTestArithmeticOpsÚ__sub__Ú__rsub__Ncstƒj|||dddS©N)Úexc©ÚsuperÚcheck_opname©ÚselfÚsÚop_nameÚotherr4©Ú __class__r r r7mszTestArithmeticOps.check_opnamec CsØ|dkr´||jkr>d}tjt|d|||ƒW5QRXdS|||ƒ}| |||¡}|dkrj| d¡}n&|dkr| | t¡||¡}| d¡}|dkr¦tj|| ¡<| ||¡n t |¡|||ƒW5QRXdS)Nznumpy boolean subtract)Úmatch)Ú__floordiv__Ú __rfloordiv__Ú__pow__Ú__rpow__Ú__mod__Ú__rmod__ZInt8)Ú__truediv__Ú__rtruediv__ZFloat64rC)Ú implementsÚpytestZraisesÚ TypeErrorZ_combineÚastypeÚfloatrrÚisnaZassert_equal) r9ÚobjÚopr<r;r4ÚmsgÚresultÚexpectedr r r Ú _check_opqs$ zTestArithmeticOps._check_opcstƒ |||d¡dSr)r6Ú_check_divmod_op)r9r:rOr<r4r=r r rT“sz"TestArithmeticOps._check_divmod_op)N)N) r'r(r)rHr7ÚNotImplementedErrorrSrTÚ __classcell__r r r=r r0js"r0cseZdZd‡fdd„ Z‡ZS)ÚTestComparisonOpsNcstƒj|||dddSr3r5r8r=r r r7™szTestComparisonOps.check_opname)N)r'r(r)r7rVr r r=r rW˜srWc@seZdZdS)Ú TestReshapingNr&r r r r rXžsrXcsveZdZej dddg¡dd„ƒZdd„Zdd „Zejj d d‡fdd „ƒZ ejj d d‡fdd„ƒZdd„Z‡Z S)ÚTestMethodsÚna_sentineléÿÿÿÿéþÿÿÿcCsZtj||d\}}tjdd||dddgtjd}| ddg¡}t ||¡| ||¡dS)N)rZrérr) rÚ factorizerrÚintpÚtakeÚtmÚassert_numpy_array_equalÚassert_extension_array_equal)r9r$rZÚlabelsÚuniquesZexpected_labelsZexpected_uniquesr r r Útest_factorize£sÿzTestMethods.test_factorizecs¬|dƒ\}}t |¡}t |¡}| |dd„¡}tjdd„tt|ƒt|ƒƒDƒdd}| ||¡|jd‰| ˆd d„¡}tj‡fd d„t|ƒDƒdd}| ||¡dS)NécSs||kSrr ©Úx1Zx2r r r r´rz-TestMethods.test_combine_le..cSsg|]\}}||k‘qSr r )Ú.0r"r!r r r Ú ¶sz/TestMethods.test_combine_le..ÚbooleanrrcSs||kSrr rhr r r r¼rcsg|]}|ˆk‘qSr r )rjr"©Úvalr r rk½s)rÚSeriesÚcombineÚzipÚlistÚassert_series_equalÚiloc)r9Z data_repeatedZ orig_data1Z orig_data2Ús1Ús2rQrRr rmr Útest_combine_le¯s þ zTestMethods.test_combine_lec Csætjddgdd}|\}}t|ƒ ||g¡}|r:t |¡}| |¡dksLt‚|j|dddksbt‚| |¡dkstt‚|j|ddd ksŠt‚| | ddg¡¡}tjddgtj d}t ||¡t ddg¡}|j||d dksât‚dS)NTFrlrrÚright)Zsider]rg)Úsorter)rrÚtypeZ_from_sequenceroZsearchsortedÚAssertionErrorr`rr_rarb) r9rZ as_seriesr!r"ZarrrQrRryr r r Útest_searchsortedÀs zTestMethods.test_searchsortedzuses nullable integer)Úreasoncstƒ ||¡Sr)r6Útest_value_counts)r9Zall_dataZdropnar=r r r~×szTestMethods.test_value_countscstƒ |¡dSr)r6Ú test_value_counts_with_normalize)r9rr=r r rÛsz,TestMethods.test_value_counts_with_normalizecCsz| ¡dkst‚| ¡dks t‚| ddddddg¡}| ¡dksFt‚| ¡dksVt‚| ¡dksft‚| ¡dksvt‚dS)Nrrgr])Zargmaxr{Zargminr`)r9rrrr r r Útest_argmin_argmaxßszTestMethods.test_argmin_argmax)r'r(r)rIÚmarkÚparametrizerfrwr|Zxfailr~rr€rVr r r=r rY¢s rYc@seZdZdS)ÚTestCastingNr&r r r r rƒñsrƒc@s|eZdZdZdd„Zej dddg¡dd„ƒZd d „Z dd„Z d d„Zdd„Zdd„Z dd„Zej dddg¡dd„ƒZdS)ÚTestGroupbyzw Groupby-specific tests are overridden because boolean only has 2 unique values, base tests uses 3 groups. c Csft dddddddg|dœ¡}| d¡jjd}| d¡jjd}t |j|jj ¡t |j|¡dS)NÚBÚA©r†r…r)rÚ DataFrameÚgroupbyZgrouperZ groupingsrarbZgrouping_vectorr†Úvaluesrc)r9r$ÚdfZgr1Zgr2r r r Útest_grouping_grouperûsÿz!TestGroupby.test_grouping_grouperÚas_indexTFc Csœt dddddddg|dœ¡}|jd|dj ¡}tj|dd\}}|rxtj|dd }tjd dg|dd }| ||¡n t |d dgdœ¡}| ||¡dS)Nr]rgér‡r…)rT©Úsort©Únameç@çð?r†©Úindexr’)r…r†) rrˆr‰r†Úmeanr^ÚIndexrorsÚassert_frame_equal) r9rr$r‹rQÚ_rer–rRr r r Útest_groupby_extension_aggsz&TestGroupby.test_groupby_extension_aggc Cs’t dddddddg|dœ¡}|jdddg}| d¡}| d¡ dd i¡}| ||¡| d¡ d ¡}| ||¡| d¡ ¡}| ||¡dS) Nr]rgrŽr‡rrr†r…Úfirst)rrˆrtZ set_indexr‰Zaggr™rœ)r9r$r‹rRrQr r r Útest_groupby_agg_extensions z&TestGroupby.test_groupby_agg_extensionc Csvt dddddddg|dœ¡}|jdddj ¡}tj|dd\}}tj|dd}tjd d g|dd}| ||¡dS) Nr]rgrŽr‡r…Frr‘r”r“r†r•) rrˆr‰r†r—r^r˜rors)r9r$r‹rQršr–rRr r r Útest_groupby_extension_no_sort#sz*TestGroupby.test_groupby_extension_no_sortcCsb|| ¡}t dddddg|dœ¡}| d¡j t¡}tjdddddgdd}| ||¡dS)Nr]rŽr‡r…rgr†r‘) rMrrˆr‰r†Z transformÚlenrors)r9r$Úvalidr‹rQrRr r r Ú test_groupby_extension_transform,s z,TestGroupby.test_groupby_extension_transformc Csft dddddddg|dœ¡}| d¡ |¡| d¡j |¡| d¡ |¡| d¡j |¡dS)Nr]rgrŽr‡r…r†)rrˆr‰Úapplyr†r…)r9r$Zgroupby_apply_opr‹r r r Útest_groupby_extension_apply5s z(TestGroupby.test_groupby_extension_applyc Cs–t dddddddg|dœ¡}| d¡j dd„¡}tj|jjddd gj|jjddgj|jjd dgjgtjdddgddd d}| ||¡dS)Nr]rgrŽr‡r†cSs|jSr)r)rr r r r>rz9TestGroupby.test_groupby_apply_identity..rérér‘r…r•) rrˆr‰r…r¢rortrr˜rs©r9r$r‹rQrRr r r Útest_groupby_apply_identity<sýù z'TestGroupby.test_groupby_apply_identitycCsrt dddddddg|dddddddgdœ¡}| d¡ ¡j}|jjrVt ddg¡}nt dg¡}t ||¡dS)Nr]rgrŽ)r†r…ÚCr†r…r¨) rrˆr‰ÚsumÚcolumnsr Z_is_numericr˜raZassert_index_equalr¦r r r Útest_in_numeric_groupbyJsýÿz#TestGroupby.test_in_numeric_groupbyÚ min_countré c Cs¼t dddddddg|dœ¡}| d¡j|d}|dkrxtjdtjdddgd d itjdddgddd}t ||¡n@tjdtjtjgdd d itjdddgddd}t ||¡dS) Nr]rgrŽr‡r†)r¬rr…ZInt64rr‘)r–) rrˆr‰r©rr˜rar™r)r9r$r¬r‹rQrRr r r Útest_groupby_sum_mincount[sþþz%TestGroupby.test_groupby_sum_mincountN)r'r(r)Ú__doc__rŒrIrr‚r›rržr¡r£r§r«r®r r r r r„õs r„c@seZdZdd„ZdS)ÚTestNumericReducecCsXt||ƒ|d}t| d¡|ƒ|d}t |¡r8tj}n|dkrHt|ƒ}t ||¡dS)N)ÚskipnaZfloat64)ÚminÚmax) ÚgetattrrKrÚisnanrrÚboolraZassert_almost_equal)r9r:r;r±rQrRr r r Úcheck_reducens zTestNumericReduce.check_reduceN)r'r(r)r·r r r r r°msr°c@seZdZdS)ÚTestBooleanReduceNr&r r r r r¸ysr¸c@seZdZdS)ÚTestPrintingNr&r r r r r¹}sr¹c@seZdZdS)ÚTestUnaryOpsNr&r r r r rºsrºc@seZdZdS)ÚTestParsingNr&r r r r r»…sr»c@seZdZdS)ÚTest2DCompatNr&r r r r r¼‰sr¼)>r¯ÚnumpyrrIZpandasrZpandas._testingZ_testingraZpandas.core.arrays.booleanrZpandas.tests.extensionrrZfixturer rrrrrrr r$ZBaseDtypeTestsr%ZBaseInterfaceTestsr*ZBaseConstructorsTestsr+ZBaseGetitemTestsr,ZBaseSetitemTestsr-ZBaseIndexTestsr.ZBaseMissingTestsr/ZBaseArithmeticOpsTestsr0ZBaseComparisonOpsTestsrWZBaseReshapingTestsrXZBaseMethodsTestsrYZBaseCastingTestsrƒZBaseGroupbyTestsr„ZBaseNumericReduceTestsr°ZBaseBooleanReduceTestsr¸ZBasePrintingTestsr¹ZBaseUnaryOpsTestsrºZBaseParsingTestsr»ZDim2CompatTestsr¼r r r r ÚsX .Ox