o ҷh<@svdZddlZddlZddlmZddlZddlmZddl m Z ddl m Z ddlmZddZejd d Zejdejgd d d ZejddZejdejgd ddZejdejgd ddZejdejgd ddZejdejgd ddZejddZejdejgd ddZejdejgd ddZGdddZGd d!d!eejZGd"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-Gd0d1d1eej.Z/Gd2d3d3eej0Z1Gd4d5d5eej2Z3Gd6d7d7eZ4Gd8d9d9eej5Z6Gd:d;d;eej7Z8Gdaj 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)basecCshtjd}t|r|jdd}n|jdddtd}|d|dkr+|dd7<||ddd<|S)Nd)size)r dtyper)nprandom default_rngisnanuniformintegersint) fill_valuerngdatarU/var/www/html/venv/lib/python3.10/site-packages/pandas/tests/extension/test_sparse.py make_datas  rcCstSNrrrrrr *sr )paramscCstt|j|jd}|S)z*Length-100 PeriodArray for semantics test.r)rrparam)requestresrrrr/srcCsttddS)Nrr)rr onesrrrr data_for_twos6r!cCsttjdg|jdS)zLength 2 array with [NA, Valid]r rrr nanrrrrr data_missing;sr&c#sfdd}|VdS)z1Return different versions of data for count timesc3s*t|D] }ttjjdVqdS)Nr)rangerrr)count_r%rrgenEs zdata_repeated..genNr)rr*rr%r data_repeatedAs  r+cCstgd|jdS)N)rr r r)rrr%rrrdata_for_sortingLr"r,cCstdtjdg|jdS)Nrr rr#r%rrrdata_missing_for_sortingQsr-cCsddS)NcSst|o t|Sr)pdisna)leftrightrrrXzna_cmp..rrrrrna_cmpVsr4c Cs"tddtjtjddddg|jdS)Nr rr rr#r%rrrdata_for_grouping[s"r5c Cs$tddtjdddddddg |jdS)Nrrr rr#r%rrrdata_for_compare`s$r9c@eZdZddZdS)BaseSparseTestscCs"|jttdkrtddSdS)NrzCan't store nan in int array.)r rrpytestskipselfrrrr_check_unsupportedfsz"BaseSparseTests._check_unsupportedN)__name__ __module__ __qualname__r@rrrrr;e r;c@r:) TestDtypecCs|tusJdSr)construct_array_typer)r?rr rrrtest_array_type_with_arglsz"TestDtype.test_array_type_with_argN)rArBrCrGrrrrrEkrDrEc@ eZdZdS) TestInterfaceNrArBrCrrrrrIprIc@rH)TestConstructorsNrJrrrrrLtrKrLcseZdZddZejdddgejj ddgdd gd gejd d d gfddZ fddZ fddZ fddZ fddZfddZfddZZS) TestReshapingcCsxtd|ddi}tdgdi}tdgdid}|||g}t|}tdd|D}t||dS)NAr )r rr )abccategorycSsg|] }|ddqS)cSst|tSr)r asarrayastypeobject)srrrr2szCTestReshaping.test_concat_mixed_dtypes...)apply).0xrrr sz:TestReshaping.test_concat_mixed_dtypes..)r. DataFramerTconcattmassert_frame_equal)r?rdf1df2df3dfsresultexpectedrrrtest_concat_mixed_dtypesys   z&TestReshaping.test_concat_mixed_dtypescolumnsrNB)rNrO)rNrPouterinner)names future_stackTFcst|||dSr)super test_stack)r?rrfrk __class__rrrms zTestReshaping.test_stackc||t||dSr)r@rltest_concat_columnsr?rna_valuernrrrq z!TestReshaping.test_concat_columnscrpr)r@rl'test_concat_extension_arrays_copy_falserrrnrrrurtz5TestReshaping.test_concat_extension_arrays_copy_falsecrpr)r@rl test_alignrrrnrrrvrtzTestReshaping.test_aligncrpr)r@rltest_align_framerrrnrrrwrtzTestReshaping.test_align_framecrpr)r@rltest_align_series_framerrrnrrrxrtz%TestReshaping.test_align_series_framecrpr)r@rl test_mergerrrnrrryrtzTestReshaping.test_merge)rArBrCrer<mark parametrizer. MultiIndex from_tuplesrmrqrurvrwrxry __classcell__rrrnrrMxs$       rMcs$eZdZddZfddZZS) TestGetitemcCstj|ddtt|Dd}t|jjr*t|dr't|j ds)Jn |d|j dks6J|d|j dksBJdS)NcSsg|]}d|qS)rr)rXirrrrZr3z(TestGetitem.test_get..)indexr8rr ) r.Seriesr'lenr rvaluesrgetiloc)r?rserrrrtest_gets &zTestGetitem.test_getcrpr)r@rl test_reindexrrrnrrrrtzTestGetitem.test_reindex)rArBrCrrr~rrrnrrsrc@rH) TestSetitemNrJrrrrrrKrc@rH) TestIndexNrJrrrrrrKrcsReZdZddZfddZfddZejjddfd d Z d d Z Z S) TestMissingcCst|}ttt|jj}tddg|d}|}t||| d}ttt|jj}tddgd|d}t ||dS)NTFr rrr ) rrboolr.r/r rr]assert_sp_array_equalfillna assert_equal)r?r&sarrexpected_dtyperdrcrrr test_isnas  zTestMissing.test_isnacsHttf}tj|ddt|WddS1swYdSNF)check_stacklevel)r FutureWarningr]assert_produces_warningrltest_fillna_limit_backfill)r?r&warnsrnrrrs"z&TestMissing.test_fillna_limit_backfillcs2t|jr|jtjjddt |dS)Nz'returns array with different fill valuereason) r rrnode add_markerr<rzxfailrltest_fillna_no_op_returns_copy)r?rrrnrrrs  z*TestMissing.test_fillna_no_op_returns_copy UnsupportedrcstdSr)rltest_fillna_series)r?rnrrrszTestMissing.test_fillna_seriescCst|d}t|ddgd|}t|jrt|j|}n|j}t|j||g|dddgd}t ||dS)Nr r)rNrgr) r.r[rr/rrr _from_sequencer]r^)r?r&rrcr rdrrrtest_fillna_frames zTestMissing.test_fillna_frame) rArBrCrrrr<rzrrrr~rrrnrrs  rc s eZdZdZddZddZejjddfdd Z d d Z fd d Z ddZ ej dddgfddZej dejejejgfddZej ddddedejgfdddedejgfededd gd d!fededejgfgd"d#Zej d$ddgd%d&ZZS)' TestMethodsz Sparse[bool]cCs|ddg}tjd|idd}|jd}||}t|jdr7|r-|jj|jjus,Jn |jj|jjus7J|j j | usCJdS)Nr rNFcopy)rrblocks) taker.r[rrhasattr_mgrrrrN_valuesto_dense)r?r&using_copy_on_writearrdf filled_valrcrrrtest_fillna_copy_frames   z"TestMethods.test_fillna_copy_framecCsn|ddg}tj|dd}|d}||}|r"|j|jus!Jn|j|jus*J|j|us5JdS)Nr Frr)rr.rrrr)r?r&rrrrrcrrrtest_fillna_copy_series s z#TestMethods.test_fillna_copy_serieszNot Applicablerct|dSr)rltest_fillna_length_mismatch)r?r&rnrrrz'TestMethods.test_fillna_length_mismatchc Cs|d|dks Jt|}|dd\}}t|j||||g|jd}tgd}||}tdd} t|j||||g| d} t || |j||||g|jd} tgd}||| }t|j||||g|jd} t || dS) Nrr rr)TTFFfloatg)TFTT) typer.rrr r arraywhererr]assert_series_equal) r?rrsclsrOrPrcondrc new_dtyperdotherrrrtest_where_seriess     zTestMethods.test_where_seriescsBtjtddt||WddS1swYdSr)r]rrrltest_searchsorted)r?r, as_seriesrnrrr1s"zTestMethods.test_searchsortedcCs6|d}|jd|jd<|jd|jdksJdS)Nrr )shift_sparse_values)r?rrcrrrtest_shift_0_periods5s z TestMethods.test_shift_0_periodsmethodargmaxargmincs||t|||dSr)r@rltest_argmin_argmax_all_na)r?rrrsrnrrr<s z%TestMethods.test_argmin_argmax_all_naboxcs ||t||||dSr)r@rl test_equals)r?rrsrrrnrrrBs zTestMethods.test_equalszfunc, na_action, expectedcC|SrrrYrrrr2JzTestMethods.Ng?cCrrrrrrrr2Krignorez1.0r$rcCs,tdtjg}|j||d}t||dS)Nr  na_action)rr r$mapr]assert_extension_array_equal)r?funcrrdrrcrrrtest_mapGs zTestMethods.test_maprcCsJd}tjt|d|jdd|dWddS1swYdS)Nz-fill value in the sparse values not supportedmatchcSstjSr)r r$rrrrr2[sz-TestMethods.test_map_raises..r)r<raises ValueErrorr)r?rrmsgrrrtest_map_raisesVs"zTestMethods.test_map_raises)rArBrC_combine_le_expected_dtyperrr<rzrrrrrr{rr.rrr[rrr r$strrrr~rrrnrrs0   rcs,eZdZejjeddfddZZS) TestCastingzno sparse StringDtype)rrcrr)rltest_astype_stringr>rnrrr_rzTestCasting.test_astype_string) rArBrCr<rzr TypeErrorrr~rrrnrr^srcsLeZdZdZdZdZdZddZfddZfddZ fdd Z Z S) TestArithmeticOpsNcCs|jdkr tddSdS)Nrz=Incorrected expected from Series.combine and tested elsewhere)rr<r=r>rrr_skip_if_different_combinejs z,TestArithmeticOps._skip_if_different_combinecrpr)rrltest_arith_series_with_scalarr?rall_arithmetic_operatorsrnrrrqrtz/TestArithmeticOps.test_arith_series_with_scalarcrpr)rrltest_arith_series_with_arrayrrnrrrurtz.TestArithmeticOps.test_arith_series_with_arraycsH|jjdkrn|ddvrtjjdd}|j|t ||dS)Nrr))mulrmulfloordiv rfloordivpowmodrmodz result dtype.fill_value mismatchr) r rstripr<rzrrrrltest_arith_frame_with_scalar)r?rrrrzrnrrrys  z.TestArithmeticOps.test_arith_frame_with_scalar) rArBrCseries_scalar_excframe_scalar_exc divmod_excseries_array_excrrrrr~rrrnrrds  rc@s`eZdZdefddZdefddZejjdddefdd Z ejjdddefd d Z d S) TestComparisonOpsr9cCs|}|||}t|tsJ|jjtjksJt|tr$||j|j}nt|t|jt|}t|| t||tjd}t ||dS)Nr) isinstancerr subtyper bool_rallrSrr]r)r?r9 comparison_oproprcrrdrrr_compare_others  z TestComparisonOps._compare_othercCs>|||d|||d|||d|||tjdS)Nrr r7)rr r$)r?r9rrrr test_scalarszTestComparisonOps.test_scalarz Wrong indicesrcCs tddd}||||dS)N )r linspacerr?r9rrrrr test_arrayszTestComparisonOps.test_arraycCs0|d}|||||d}||||dS)Nr r)rrrrrtest_sparse_arraysz#TestComparisonOps.test_sparse_arrayN) rArBrCrrrr<rzrrrrrrrrs  rcs*eZdZejjddfddZZS) TestPrintingzDifferent reprrcst||dSr)rltest_array_repr)r?rr rnrrrr"zTestPrinting.test_array_repr)rArBrCr<rzrrr~rrrnrrs rcs.eZdZejdddgfddZZS) TestParsingenginerQpythoncsFd}tjt|dt||WddS1swYdS)Nz,.*must implement _from_sequence_of_strings.*r)r<rNotImplementedErrorrl test_EA_types)r?rr expected_msgrnrrr s"zTestParsing.test_EA_types)rArBrCr<rzr{r r~rrrnrrsrc@rH)TestNoNumericAccumulationsNrJrrrrr rKr );__doc__numpyr r< pandas.errorsrpandasr.rpandas._testing_testingr] pandas.arraysrpandas.tests.extensionrrfixturer r$rr!r&r+r,r-r4r5r9r;BaseDtypeTestsrEBaseInterfaceTestsrIBaseConstructorsTestsrLBaseReshapingTestsrMBaseGetitemTestsrBaseSetitemTestsrBaseIndexTestsrBaseMissingTestsrBaseMethodsTestsrBaseCastingTestsrBaseArithmeticOpsTestsrrBasePrintingTestsrBaseParsingTestsrBaseAccumulateTestsr rrrrs\               64f&)