gL i0ddlmZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl Zddl mZmZmZmZddlmZGddZy) )deepcopyN)using_string_dtype)option_context) HAS_PYARROW) DataFrameSeries date_rangetimedelta_rangec eZdZdZdZdZdZdZdZe jje xre dd Zd Zd Zd Zd ZdZdZdZdZe jj/deedgedgedgigdZdZdZdZdZdZdZe jj/dgddZ d Z!d!Z"y")#TestDataFrameMiscc|d}|jdk(sJ|jd}|jdk(sJ|jdddf}|jdk(sJ|jdd}|jdk(sJy)NAB)namepoploc)self float_framess2s a/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/pandas/tests/frame/test_api.pytest_getitem_pop_assign_namez.TestDataFrameMisc.test_getitem_pop_assign_namesy  vv}} OOC vv}} OOAsF #vv}} UU1Xww#~~cD|}|jddk(sJ|jddk(sJ|jddk(sJ|jddk(sJ|jddk(sJ|jddk(sJ|jddk(sJ|jddk(sJ|jddk(sJ|jddk(sJ|jd|jusJ|jd|jusJt j td5|jddddt j td 5|jd dddt j td 5|jddddt j td5|jddddy#1swYxYw#1swYxYw#1swYWxYw#1swYyxYw) Nrindexrowscolumnsz No axis namedmatchz No axis.*foofooz No axis.*None)_get_axis_number_get_axis_name _get_axisrrpytestraises ValueError)rrfs r test_get_axiszTestDataFrameMisc.test_get_axis%s !!!$)))!!!$)))!!'*a///!!&)Q...!!),111"g---"i///(G333'7222 *i777{{1~((({{1~*** ]]:_ = "  q ! "]]:^ < $  U # $]]:_ = #  T " #]]:_ = %  t $ % % " " $ $ # # % %s08G2-G>"H H2G;>H HHctjtd5|j|vdddy#1swYyxYw)Nzunhashable type: 'Index'r)r&r' TypeErrorr)rrs rtest_column_contains_raisesz-TestDataFrameMisc.test_column_contains_raisesBs4 ]]9,F G /   ; . / / /s4=c 8ttdtdgtd}tdD]}|t|vrJt|j dt sJttdtdgt jjttdd}tdD]}|t|vrJtdD]}|t|vrJt|j dtsJy)NabcdefghABCDrrEFGH) rlistdir isinstance __getitem__rpd MultiIndex from_tupleszip)rdfkeys rtest_tab_completionz%TestDataFrameMisc.test_tab_completionFs V d6l3T&\ J< "C#b'> !> ""..-v666 &\4< (MM--d3vv3F.GH < "C#b'> !> "< &Cc"g% %% &"..-y999rctdDcgc]}dt|z}}tdtdg}t||}dt|vsJdt|vsJt dd5t||}dt|vsJd t|vsJ dddt dd5t||}d t|vsJ dddycc}w#1swYAxYw#1swYyxYw) Niar2a99a100zdisplay.max_dir_itemsi,a299a300a419)rangestrrr5r)rirvaluesr<s rtest_display_max_dir_itemsz,TestDataFrameMisc.test_display_max_dir_itemsYs).s4A3Q<44*eCj) vw /BSW$$$ 3S 9 )673BSW$ $$R( (( ) 3T : %673BSW$ $$ % %5 ) )  % %sC2,C3C+C(+C4c t}tdg}d}tjt|5t |dddtjt|5t |dddy#1swY9xYw#1swYyxYw)Nrzunhashable type: 'DataFrame'r)rr&r'r,hash)r empty_framer<msgs rtest_not_hashablez#TestDataFrameMisc.test_not_hashablelssk s^, ]]9C 0  H  ]]9C 0         s A8# B8BB zsurrogates not allowed)reasonchd}t|gi}|t|vsJ|jd|k(sJy)Nur)rr5r)rcolnamer<s r+test_column_name_contains_unicode_surrogatez=TestDataFrameMisc.test_column_name_contains_unicode_surrogatevs?  } %c"g%%%zz!}'''rc2ttjjdj d}ttjjdj d}d|j _|j j Jy)Nr!)rr")rnprandom default_rngstandard_normalrr)rdf1df2s rtest_new_empty_indexz&TestDataFrameMisc.test_new_empty_indexsi --a0@@HI --a0@@HI yy~~%%%rc|jd}||jusJ|jd}||jusJd}tjt |5|jddddy#1swYyxYw)NrrzAxis must be 0 or 1 \(got 2\)rr!) _get_agg_axisrrr&r'r()rrcolsidxrNs rtest_get_agg_axisz#TestDataFrameMisc.test_get_agg_axiss((+{*****''*k'''''. ]]:S 1 )  % %a ( ) ) )s A;;Bct}|jsJ|jrJ|jrJtgdgddtjd}|d=|jrJy)N)g?g@g@r@bcrrrUrr)remptyrVarange)rrfloat_string_framerMr<s r test_emptyzTestDataFrameMisc.test_emptysgk     $$$$%++++_?C299UV< X sG88|8rct|t|jk(sJ|ddgj}|jddgj}t j ||y)Nrrr2)lenrrIreindextmassert_almost_equal)rrarrexpecteds rtest_lenzTestDataFrameMisc.test_lensc;3{'8'8#99993*%,,&&Sz&:AA sH-rc|}|jd}|jd}tj|||jd}|jd}tj||y)Nr)axisrrr)sumroassert_series_equal)rrr)rrresults rtest_axis_aliasesz#TestDataFrameMisc.test_axis_aliasessg 55a5=G$ vx055a5=I& vx0rctjtjsJtjtjsJyN)pydocgetdocrrr)rs rtest_class_axisz!TestDataFrameMisc.test_class_axiss2||IOO,,,||I--...rcn|j}|jD]\}}|j|k(rJyr{)_seriesitemsr)rrjserieskvs rtest_series_put_namesz'TestDataFrameMisc.test_series_put_namess6#++LLN DAq66Q; ; rctgd}|jrJtdgdg}|jrJtddgddgj}|jsJ|jjsJy)N)rr!rUr)rrr@rdred)rrhdropnaTrr<s rtest_empty_nonzeroz$TestDataFrameMisc.test_empty_nonzerosl y !88| aS1# .88| c3Z#s < C C Exxxttzzzrr<rrgr2cP|jsJ|jjsJyr{)rhrrs rtest_empty_likez!TestDataFrameMisc.test_empty_likesxxxttzzzrcttddtddd}|j}|jj }t tjddid}tj||y) N20130101 )periodsz1 dayrfobjectcount)r) rr r rdtypes value_countsrrVdtyperorw)rr<trxrrs rtest_with_datetimelikesz)TestDataFrameMisc.test_with_datetimelikessl  B7$Wb9   DD&&(288H-r2A vx0rc^t|}d|jd<|j|rJy)Nr)rr)rrequals)rrcps r test_deepcopyzTestDataFrameMisc.test_deepcopys0 k "v%%b))))rctgdgdgdd}d}d}||j|d}||jd|d }||j|d }||j|d }||j|d }||j|d }||j|d}||j||jd}d}||jdd|d}||j|d}||j|d}||j|y)N)r"barbazqux)rrrr)rr!rUrcc||}|Jyr{)baser)rxs r_check_fz._check_fstWF> !>rc(|jddS)Nr@Tinplace) set_indexxs rz.sakk#tk4rc&|jdSNTr reset_indexrs rrz.sammDm1rr@c&|jdSr)drop_duplicatesrs rrz.sa'''5rc(|jddS)NrdTr) sort_valuesrs rrz.sammCm6rc&|jdSr) sort_indexrs rrz.sall4l0rc(|jddSNrTrfillnars rrz.ahhq$h/rc*|jdddSNrrTrreplacers rrz. aii1di3rc,|jddidSNrr"Trrenamers rrz.ahh5z4h8rrec(|jddS)NT)rdroprrs rrz.sammDtm<rc(|jddSrrrs rrz.rrc*|jdddSrrrs rrz.rrc,|jddidSrrrs rrz."rr)rcopyr)rdatarr)rs rtest_inplace_return_selfz*TestDataFrameMisc.test_inplace_return_selfsN.\ U  " 5a  2$a( 6a  7a  1a  0a  4a  9a  IIK  =$S)1- 01 41 91rcVtjddddlm}|turd}nd}|j |t jdd 5|d 5t|jjd d ddddddy#1swYxYw#1swYyxYw) NIPythonz6.0.0) minversionr)provisionalcompleterz/from pandas import DataFrame; obj = DataFrame()z5from pandas import Series; obj = Series(dtype=object)F)raise_on_extra_warningsignorezobj.r) r& importorskipIPython.core.completerrrrun_cellroassert_produces_warningr4 Completer completions)ripframe_or_seriesrcodes rtest_tab_complete_warningz+TestDataFrameMisc.test_tab_complete_warning%sI':? i 'DDJD D ' 'e L :%h/ :R\\--fa89 : : : : : : :s$ B&BBB BB(ctdddgi}|jik(sJd|jd<|jt}|jddik(sJy)Nrr!rUrversionr2rattrsrrGrr<rxs r test_attrszTestDataFrameMisc.test_attrs6sX aV} %xx2~~3'|| 1~---rctdddgi}|jik(sJddh|jd<|jt}|j|jk(sJ|jd|jdusJy)Nrr!rUspamhamtagsr2rrs rtest_attrs_deepcopyz%TestDataFrameMisc.test_attrs_deepcopy>s| aV} %xx2~~"E?3'||rxx'''||F#288F+;;;;rallows_duplicate_labels)TFNctdddgi}d}|tur|d}d}|j|}||jjdusJ|jj|usJ||usJ|jjdusJ|tur,t j |j|js4Jt j |dj|djsJtj|5d|j|<ddd|r|j|dk(sCJ|j|dk(sJtj|5d|j|<ddd|jd|}d |j|<|j|dk(sJy#1swYxYw#1swYLxYw) Nrrr!)rrr)rT)rrr) rr set_flagsflagsrrVmay_share_memoryrIroassert_cow_warningiloc)rrrusing_copy_on_writewarn_copy_on_writeobjr=rxs rtest_set_flagsz TestDataFrameMisc.test_set_flagsGsq!f & f $c(CC7NO " *<<774? ??<<77;RR RR&   yy00D888 f $&&szz6==A AA&&s3xs 8J8JK KK  " "#5 6 ! FKK  ! 88C=A% %%88C=A% %%&&'9: %#$ C  %/F  Cxx}!!! ! ! % %sF*F6*F36F?ct}d}tjt|5|j t j djddddddy#1swYyxYw)Nz<'DataFrame' object has no attribute '_constructor_expanddim'rrU)rr&r'AttributeError_constructor_expanddimrVrireshaperr<rNs rtest_constructor_expanddimz,TestDataFrameMisc.test_constructor_expanddimzs\[L ]]> 5 F  % %biim&;&;Aq!&D E F F Fs 6A''A0ct}d}tjt|d5t j |dddy#1swYyxYw)NzDataFrame._data is deprecatedF)r check_stacklevel)rrorDeprecationWarninginspect getmembersrs rtest_inspect_getmembersz)TestDataFrameMisc.test_inspect_getmemberssH [-  ' ' cE  #   r " # # #s AAN)#__name__ __module__ __qualname__rr*r-r>rJrOr&markxfailrrrSr\rarkrsryr~rr parametrizerrrrrrrrrrrrrrr r s! %:/:&%& [[,5M((& ) . 1/   [[ K QC qc " q"g    1* >@:".< [[68KL0"M0"dF#rr )rrrr|numpyrVr&pandas._configrpandas._config.configr pandas.compatrpandasr8rrr r pandas._testing_testingror rrrrs;  -0% t#t#r