o[we<dZddlZddlZddlmZddlZddlmZddl m Z ddl m Z ddlmZdZejdZejdejg d Zejd Zejdejg d Zejdejg d Zejdejg dZejdejg dZejdZejdejg dZejdejg dZGddZGddeejZGddeej Z!Gddeej"Z#Gddeej$Z%Gddeej&Z'Gdd eej(Z)Gd!d"ej*Z+Gd#d$eej,Z-Gd%d&eej.Z/Gd'd(eej0Z1Gd)d*eej2Z3Gd+d,eZ4Gd-d.eej5Z6Gd/d0eej7Z8Gd1d2ej9Z:dS)3aj 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)basec*tjd}tj|r|d}n@|dddt }|d|dkr|dxxdz cc<||ddd<|S)Nd)size)r dtyper)nprandom default_rngisnanuniformintegersint) fill_valuerngdatas Blib/python3.11/site-packages/pandas/tests/extension/test_sparse.py make_datars )   " "C x {{{$$||AsC|88 7d1g   GGGqLGGGDAJ KctSNrrrr r *s ==r)paramscVtt|j|j}|S)z*Length-100 PeriodArray for semantics test.r)rrparam)requestress rrr/s' i ..7= I I IC JrcJttjddzS)Nr r )rronesrrr data_for_twosr'6s rws||a' ( ((rcFttjdg|jS)zLength 2 array with [NA, Valid]r r!rrnanr"r#s r data_missingr,;s {w} = = ==rc#Kfd}|VdS)z1Return different versions of data for count timesc3Kt|D],}ttjjV-dS)Nr!)rangerrr")count_r#s rgenzdata_repeated..genEsRu R RAi 667=QQQ Q Q Q Q R RrNr)r#r2s` r data_repeatedr3As4RRRRR IIIIIrc2tgd|jS)N)r rr r!)rr"r+s rdata_for_sortingr5Ls yyyW] ; ; ;;rcHtdtjdg|jS)Nr r r!r)r+s rdata_missing_for_sortingr7Qs 261~'- @ @ @@rcdS)NcRtj|otj|Sr)pdisna)leftrights rzna_cmp..Xsrwt}}?rrrrrna_cmpr?Vs ? ??rc ftddtjtjddddg|jS)Nr r rr!r)r+s rdata_for_groupingrA[s, 1bfbfaAq9gm T T TTrc Vtddtjdddddddg |jS)Nrr rr!r)r+s rdata_for_comparerF`s. 1bfb"aAq!< W W WWrceZdZdZdS)BaseSparseTestscn|jttdkrtjddSdS)NrzCan't store nan in int array.)r rrpytestskipselfrs r_check_unsupportedz"BaseSparseTests._check_unsupportedfs8 :S!,, , , K7 8 8 8 8 8 - ,rN)__name__ __module__ __qualname__rNrrrrHrHes#99999rrHceZdZdZdS) TestDtypec@|tusJdSr)construct_array_typer)rMrr s rtest_array_type_with_argz"TestDtype.test_array_type_with_argls%))++{::::::rN)rOrPrQrVrrrrSrSks#;;;;;rrSceZdZdS) TestInterfaceNrOrPrQrrrrXrXpDrrXceZdZdS)TestConstructorsNrYrrrr\r\trZrr\c ,eZdZdZejdddgej ddgddg gejd d d gfd Z fdZ fdZ fdZ fdZfdZfdZxZS) TestReshapingc`tjd|ddi}tjdgdi}tjdgdid}|||g}tj|}tjd|D}t j||dS)NAr)r r r)abccategoryc:g|]}|dS)cZtj|tSr)rasarrayastypeobject)ss rr>zCTestReshaping.test_concat_mixed_dtypes...srz!}}33F;;r)apply).0xs r z:TestReshaping.test_concat_mixed_dtypes..s( J J JQWW;; < < J J Jr)r: DataFramerhconcattmassert_frame_equal)rMrdf1df2df3dfsresultexpecteds rtest_concat_mixed_dtypesz&TestReshaping.test_concat_mixed_dtypesyslCbqb?++lC+,,lC12299*EECo39 J Jc J J J   fh/////rcolumnsr`B)r`ra)r`rbouterinner)names future_stackTFcNt|||dSr)super test_stack)rMrrzr __class__s rrzTestReshaping.test_stacks' 4,77777rcv||t||dSr)rNrtest_concat_columnsrMrna_valuers rrz!TestReshaping.test_concat_columnss7 %%% ##D(33333rcv||t||dSr)rNr'test_concat_extension_arrays_copy_falsers rrz5TestReshaping.test_concat_extension_arrays_copy_falses7 %%% 77hGGGGGrcv||t||dSr)rNr test_alignrs rrzTestReshaping.test_align7 %%% 4*****rcv||t||dSr)rNrtest_align_framers rrzTestReshaping.test_align_frames7 %%%   x00000rcv||t||dSr)rNrtest_align_series_framers rrz%TestReshaping.test_align_series_frames7 %%% ''h77777rcv||t||dSr)rNr test_mergers rrzTestReshaping.test_mergerr)rOrPrQryrJmark parametrizer: MultiIndex from_tuplesrrrrrrr __classcell__rs@rr^r^xs^ 0 0 0 [ #J M % %Z('0B &    [^dE];;8888<;844444HHHHH+++++1111188888+++++++++rr^c$eZdZdZfdZxZS) TestGetitemc tj|dtt|D}t j|jjrIt j|drt j|j dsJn&|d|j dksJ|d|j dksJdS)Ncg|]}d|zS)r r)rlis rrnz(TestGetitem.test_get..s$E$E$EqQU$E$E$Er)indexrEr r ) r:Seriesr/lenrrvaluesrgetiloc)rMrsers rtest_getzTestGetitem.test_getsi$E$EE#d))4D4D$E$E$EFFF 8CJ) * * -8CGGAJJ'' ABHSXa[,A,A A A A A771::!,,,,wwqzzSXa[((((((rcv||t||dSr)rNr test_reindexrs rrzTestGetitem.test_reindexs7 %%% T8,,,,,r)rOrPrQrrrrs@rrrsG)))---------rrceZdZdS) TestSetitemNrYrrrrrrZrrceZdZdS) TestIndexNrYrrrrrrZrrceZdZdZfdZfdZejdfdZ dZ xZ S) TestMissingct|}tttj|jj}tddg|}|}tj||| d}tttj|jj}tddgd|}tj ||dS)NTFr rrr ) rrboolr:r;r rrqassert_sp_array_equalfillna assert_equal)rMr,sarrexpected_dtyperxrws r test_isnazTestMissing.test_isnas<(($T27<3E3P+Q+QRRe}NCCC  222{{1~~$T27<3E3P+Q+QRRu~%~VVV  X.....rcttf}tj|d5t |ddddS#1swxYwYdSNF)check_stacklevel)r FutureWarningrqassert_produces_warningrtest_fillna_limit_backfill)rMr,warnsrs rrz&TestMissing.test_fillna_limit_backfills#]3  ' F F F = = GG . .| < < < = = = = = = = = = = = = = = = = = =s"AAActj|jr8|jt jdt |dS)Nz'returns array with different fill valuereason) rrrnode add_markerrJrxfailrtest_fillna_no_op_returns_copy)rMrr#rs rrz*TestMissing.test_fillna_no_op_returns_copysh 8DO $ $  L # # !!)R!SS    ..t44444r UnsupportedrcHtdSr)rtest_fillna_series)rMrs rrzTestMissing.test_fillna_seriess! ""$$$$$rcd|d}tj|ddgd|}tj|jrt |j|}n|j}tj|||g|ddgd}tj ||dS)Nr r )r`r{r) r:rorr;rrr _from_sequencerqrr)rMr,rrwr rxs rtest_fillna_framezTestMissing.test_fillna_frames!!_ L1v>>??FFzRR 7<* + + ' 2J??EE &E<!00*j1IQV0WWV     fh/////r) rOrPrQrrrrJrrrrrrs@rrrs / / /===== 55555 [m,,%%%%-,%0000000rrc xeZdZdZdZdZejdfdZ dZ fdZ d Z ej d d d gfd Zej dejejejgfdZej dddedejgfddedejgfededdgdfededejgfgdZej dddgdZxZS) TestMethodsz Sparse[bool]c|ddg}tjd|id}|jd}||}t |jdr7|r|jj|jjusJn|jj|jjusJ|j j | usJdS)Nr r`Fcopy)rrblocks) taker:rorrhasattr_mgrrrr`_valuesto_dense)rMr,using_copy_on_writearrdf filled_valrws rtest_fillna_copy_framez"TestMethods.test_fillna_copy_framesA'' \3*5 1 1 1WT] :&& 27H % % @" @y~);;;;;;y~V]-?????t|$$&&#,,..888888rc>|ddg}tj|d}|d}||}|r|j|jusJn|j|jusJ|j|usJdS)Nr Frr)rr:rrrr)rMr,rrrrrws rtest_fillna_copy_seriesz#TestMethods.test_fillna_copy_series sA''i%(((V J''  5;&.00000;fn4444{##%%777777rzNot ApplicablercJt|dSr)rtest_fillna_length_mismatch)rMr,rs rrz'TestMethods.test_fillna_length_mismatchs! ++L99999rc|d|dksJt|}|dd\}}tj|||||g|j}t jgd}||}tdd} tj|||||g| } tj || |||||g|j} t jgd}||| }tj|||||g|j} tj || dS) Nrr r r)TTFFfloatg)TFTT) typer:rrr rarraywhererrqassert_series_equal) rMrrclsrarbrcondrw new_dtyperxothers rtest_where_serieszTestMethods.test_where_seriesslAw$q'!!!!4jjBQBx1i**Aq!QzTestMethods.JqrNg?c|Srrrs rr>zTestMethods.Krrignorez1.0r*r!ctdtjg}|||}t j||dS)Nr  na_action)rrr*maprqassert_extension_array_equal)rMfuncrrxrrws rtest_mapzTestMethods.test_mapGsEArv;''$)44 '99999rrcd}tjt|5|d|ddddS#1swxYwYdS)Nz-fill value in the sparse values not supportedmatchctjSr)rr*rs rr>z-TestMethods.test_map_raises..[srvrr)rJraises ValueErrorr)rMrrmsgs rtest_map_raiseszTestMethods.test_map_raisesVs> ]:S 1 1 1 < < HH%%H ; ; ; < < < < < < < < < < < < < < < < < $> ? $ UENuEEE F (KK88 9  :::  [[4*:;;<<<;<<<<s rtest_sparse_arrayz#TestComparisonOps.test_sparse_arraysN" ,mSAAA" ,mSAAAAArN) rOrPrQrr6r8rJrrr?rArrrr.r.s3{3333*EKEEEE  [o..B;BBB/.B [o..B+BBB/.BBBrr.c`eZdZejdfdZxZS) TestPrintingzDifferent reprrcLt||dSr)rtest_array_repr)rMrr rs rrEzTestPrinting.test_array_reprs# d+++++r)rOrPrQrJrrrErrs@rrCrCsW [.//,,,,0/,,,,,rrCcdeZdZejdddgfdZxZS) TestParsingenginercpythoncd}tjt|5t||ddddS#1swxYwYdS)Nz,.*must implement _from_sequence_of_strings.*r)rJr NotImplementedErrorr test_EA_types)rMrHr expected_msgrs rrLzTestParsing.test_EA_typessF ].l C C C 0 0 GG ! !&$ / / / 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0s#AAA)rOrPrQrJrrrLrrs@rrGrGsY [XX7700008700000rrGceZdZdS)TestNoNumericAccumulationsNrYrrrrOrOrZrrO);__doc__numpyrrJ pandas.errorsrpandasr:rpandas._testing_testingrq pandas.arraysrpandas.tests.extensionrrfixturer r*rr'r,r3r5r7r?rArFrHBaseDtypeTestsrSBaseInterfaceTestsrXBaseConstructorsTestsr\BaseReshapingTestsr^BaseGetitemTestsrBaseSetitemTestsrBaseIndexTestsrBaseMissingTestsrBaseMethodsTestsrBaseCastingTestsrBaseArithmeticOpsTestsrr.BasePrintingTestsrCBaseParsingTestsrGBaseAccumulateTestsrOrrrrgs\  ,,,,,,%%%%%%''''''   26{###$# )))26{###>>$#> 26{###$#26{###<<$#<26{###AA$#A@@@26{###UU$#U26{###XX$#X99999999 ;;;;;!4;;;      OT%<        (B   3+3+3+3+3+OT%<3+3+3+l - - - - -/4#8 - - -     /4#8        #   1010101010/4#8101010hc<c<c<c<c