a oÝEb<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 úc/home/ja/django-apps/lartica_env/lib/python3.9/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,srcCstjgd¢|dS)N)TTFr)rrrr r r Údata_for_sorting1srcCstjdtjdg|dS©NTFrrrr r r Údata_missing_for_sorting6srcCsdd„S)NcSs|tjuo|tjuSr©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|durÈ||jvrRd}tjt|d|||ƒWdƒn1sD0YdS|||ƒ}| |||¡}|dvr~| d¡}n&|dvr¤| | t¡||¡}| d¡}|dkrºtj|| ¡<| ||¡n4t |¡|||ƒWdƒn1sò0YdS)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 Ú ¶rz/TestMethods.test_combine_le..ÚbooleanrrcSs||kSrr rhr r r r¼rcsg|]}|ˆk‘qSr r )rjr"©Úvalr r rk½r)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 |¡}| |¡dksLJ‚|j|dddksbJ‚| |¡dkstJ‚|j|ddd ksŠJ‚| | ddg¡¡}tjddgtjd}t ||¡t ddg¡}|j||d dksâJ‚dS)NTFrlrrÚright)Zsider]rg)Úsorter)rrÚtypeZ_from_sequenceroZsearchsortedr`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_normalizecCsr| ¡dksJ‚| ¡dks J‚| gd¢¡}| ¡dks>J‚| ¡dksNJ‚| ¡dks^J‚| ¡dksnJ‚dS)Nrrg)rgrrr]r]rgr])ZargmaxZargminr`)r9rrrr r r Útest_argmin_argmaxßszTestMethods.test_argmin_argmax)r'r(r)rIÚmarkÚparametrizerfrwr{Zxfailr}r~rrVr 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. cCs\t gd¢|dœ¡}| d¡jjd}| d¡jjd}t |j|jj ¡t |j|¡dS)N)ÚBr„NNÚAr…r„©r…r„r…rr„)rÚ DataFrameÚgroupbyZgrouperZ groupingsrarbZgrouping_vectorr…Úvaluesrc)r9r$ÚdfZgr1Zgr2r r r Útest_grouping_grouperûsÿz!TestGroupby.test_grouping_grouperÚas_indexTFc Cs’t gd¢|dœ¡}|jd|dj ¡}tj|dd\}}|rntj|dd}tjdd g|d d}| ||¡n t |dd gdœ¡}| ||¡dS) N©r]r]rgrgérŽr]r†r„)rŒT©Úsort©Únameç@çð?r…©Úindexr’)r„r…) rr‡rˆr…Úmeanr^ÚIndexrorsÚassert_frame_equal) r9rŒr$rŠrQÚ_rer–rRr r r Útest_groupby_extension_aggsz&TestGroupby.test_groupby_extension_aggcCs†t gd¢|dœ¡}|jgd¢}| d¡}| d¡ ddi¡}| ||¡| d¡ d¡}| ||¡| d¡ ¡}| ||¡dS)Nrr†)rrgrr…r„Úfirst)rr‡rtZ set_indexrˆZaggr™rœ)r9r$rŠrRrQr r r Útest_groupby_agg_extensions z&TestGroupby.test_groupby_agg_extensioncCslt gd¢|dœ¡}|jdddj ¡}tj|dd\}}tj|dd}tjddg|d d }| ||¡dS)Nrr†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_sortcCsV|| ¡}t gd¢|dœ¡}| d¡j t¡}tjgd¢dd}| ||¡dS)N)r]r]rŽrŽr]r†r„)rŽrŽrgrgrŽr…r‘) rMrr‡rˆr…Z transformÚlenrors)r9r$ZvalidrŠrQrRr r r Ú test_groupby_extension_transform,s z,TestGroupby.test_groupby_extension_transformcCs\t gd¢|dœ¡}| d¡ |¡| d¡j |¡| d¡ |¡| d¡j |¡dS)Nrr†r„r…)rr‡rˆÚapplyr…r„)r9r$Zgroupby_apply_oprŠr r r Útest_groupby_extension_apply5s z(TestGroupby.test_groupby_extension_applycCsˆt gd¢|dœ¡}| d¡j dd„¡}tj|jjgd¢j|jjddgj|jjd d gjgtjgd¢ddd d}| ||¡dS)Nrr†r…cSs|jSr)r)rr r r r>rz9TestGroupby.test_groupby_apply_identity..)rr]érgrŽré©r]rgrŽr‘r„r•) rr‡rˆr„r¡rortrr˜rs©r9r$rŠrQrRr r r Útest_groupby_apply_identity<sýù z'TestGroupby.test_groupby_apply_identitycCs^t gd¢|gd¢dœ¡}| d¡ ¡j}|jjrBt ddg¡}nt dg¡}t ||¡dS)Nr)r]r]r]r]r]r]r])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é cCs¬t gd¢|dœ¡}| d¡j|d}|dkrjtjdtjgd¢dd itjgd ¢ddd}t ||¡n>tjdtjtjgd dd itjgd ¢ddd}t ||¡dS)Nrr†r…)r¬rr„)rŽrrZInt64rr¥r‘)r–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‹rIr€rr›rržr r¢r§r«r®r r r r rƒõs rƒc@seZdZdd„ZdS)ÚTestNumericReducecCsXt||ƒ|d}t| d¡|ƒ|d}t |¡r8tj}n|dvrHt|ƒ}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