gL i dddlmZddlmZmZddlmZddlZddlZddlZddl Z ddl Z ddl m Z ddlmcmZddlZddlmZmZmZmZddlmZddlmZddlmZm Z e jBd Z"e jBd dd gd d gdZ#GddZ$GddZ%GddZ&GddZ'GddZ(dZ)dZ*dZ+GddZ,dZ-d Z.d!Z/e j`jcd"ed#d$gd%d&gge2d'(d)fed#d*gd%d*gge2d'(jgd*e4id+fgd,Z5d-Z6e j`jcd.ed/d/gged/d/ged/d/gggd0Z7d1Z8d2Z9d3Z:d4Z;e j`jyd5d6Z=d7Z>d8Z?y)9)deque)datetimetimezone)EnumN) HAS_PYARROW) DataFrameIndex MultiIndexSeries) expressions)_check_mixed_float_check_mixed_intcbtjgdgdgdg}t|gdgdS)z Fixture for simple 3x3 DataFrame Columns are ['one', 'two', 'three'], index is ['a', 'b', 'c']. one two three a 1.0 2.0 3.0 b 4.0 5.0 6.0 c 7.0 8.0 9.0 ?@@)@@g@)g@g @g"@onetwothreeabccolumnsindex)nparrayr)arrs h/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/pandas/tests/frame/test_arithmetic.py simple_framer% s) ((O_oF GC S"9 QQTdnumexprpython)autouseparamsidsc#K|j5}|jtd|j|jdddy#1swYyxYww)N _MIN_ELEMENTS)contextsetattrexprparam)request monkeypatchms r$switch_numexpr_min_elementsr61sF    ! $7mmsA0A  A AAcJeZdZd dZd dZdefdZdefdZd dZdZ d d Z y) DummyElementreturnNcF||_tj||_yN)valuer!dtype)selfr<r=s r$__init__zDummyElement.__init__9s XXe_ r&cXtj|j|jS)Nr=)r!r"r<r=r>r=copys r$ __array__zDummyElement.__array__=sxx $**55r&c<d|jd|jdS)Nz DummyElement(z, ))r<r=r>s r$__str__zDummyElement.__str__@stzzl"TZZL::r&ct|Sr;)strrGs r$__repr__zDummyElement.__repr__Cs 4yr&c||_|Sr;rArBs r$astypezDummyElement.astypeFs  r&cXt||jj||Sr;)typer<view)r>r=s r$rPzDummyElement.viewJs"tDz$**//%0%88r&c,t|jSr;)boolr<)r>axiss r$anyzDummyElement.anyMsDJJr&r9N)NN)Fr;) __name__ __module__ __qualname__r?rDrJrHrKrMrPrTr&r$r8r88s3%6;;#9 r&r8c eZdZdZdZej jdejjdjdde jddd ejjdjddejjdjddd gejjdjddejjdjddd ejjdjdde jddd ge jdde jddd ejjdjddejjdjddd gejjdjdde jddd e jdde jddd ggd Zej jd gd d ZdZdZdZdZy)TestFrameComparisonsctdgdi}tdgdi}|dk(}tj|||djd|d<|dk(}tj||y)NA)foobarbaz)TFFr^category)rtmassert_frame_equalrM)r>dfexpress r$&test_comparison_with_categorical_dtypez;TestFrameComparisons.test_comparison_with_categorical_dtypeXsn23 4234Ek c3'S'..,3Ek c3'r&cttjjdj dt d}d}t jt|5|dvdddy#1swYyxYw)N)ABCDrz+The truth value of a DataFrame is ambiguousmatchr;) rr!random default_rngstandard_normallistpytestraises ValueError)r>rdmsgs r$test_frame_in_listz'TestFrameComparisons.test_frame_in_listgs`  II ! !! $ 4 4V and do not have a common DType.|rn) rrr rbrcjoinrtru TypeError) r>argarg2xyresultcolexpectedmsgsrws r$test_comparison_invalidz,TestFrameComparisons.test_comparison_invalidps` cN dOa./ii 8sS!C&AcF" " 8''II  fh/a./ii 8sS!C&AcF" " 8''II  fh/ hhtn ]]9C 0  F  ]]9C 0  E  ]]9C 0  E  ]]9C 0  F  = 9 9         s;FFF.F%F1F=F"%F.1F:=Gz left, right))gtlt)rr)gele)rr)eqr)nerc "ttjddtjddtjj dj ddtjj djdtdDcgc]}td|zc}d }tj|jtjj dj t|d kDd f<tt|}tt|}|d vrO||tjd }|tjd |}t!j"||nd} t%j&t(| 5||tjd dddt%j&t(| 5|tjd |ddd|d vrO||tjd}|tjd|}t!j"||yd} t%j&t(| 5||tjddddt%j&t(| 5|tjd|dddycc}w#1swY%xYw#1swYxYw#1swYhxYw#1swYyxYw)Nr|ryr}20010102riiʚ;rzr')dates1dates2intcolfloatcol stringcolg?r)rr20010109zL'(<|>)=?' not supported between instances of 'numpy.ndarray' and 'Timestamp'rnnatzJ'(<|>)=?' not supported between instances of 'numpy.ndarray' and 'NaTType')rpd date_ranger!rprqintegersrrrangechrNaTloclengetattroperator Timestamprbrcrtrur) r>leftrightirdleft_fright_frrrws r$test_timestamp_comparez+TestFrameComparisons.test_timestamp_comparesa-- B?-- B?))//2;;JR;PII11!4DDRH49"I>qc#'l>   LN66ryy$$Q'..s2w7#=xGH4((E* < b",,z":;HR\\*5r:F  ! !&( 3? y4 5r2<< 34 5y4 6 Z0"5 6 < b",,u"56HR\\%0"5F  ! !&( 3= y4 0r2<<./ 0y4 1 U+R0 1 1C?" 5 5 6 6 0 0 1 1s6K K  K-5K95L K*-K69LLctddgddgg}tddgddgg}||k(}|jjrJ||k7}|jjsJy)Nz 1989-08-01rirrrd)rrTall)r>rdotherrs r$test_mixed_comparisonz*TestFrameComparisons.test_mixed_comparisonsvq)L!+<= >C:Sz23u::<##%%%uzz|!!!r&cttjdjd}tddgddgddgg}|dk(}t j |||ddgk(}t j ||y)NrjriFTririri)rr!arangereshaperbrc)r>rdrrs r$ test_df_boolean_comparison_errorz5TestFrameComparisons.test_df_boolean_comparison_error stryy|++F3 4uentUmeU^LMv fh/1v fh/r&cttjjdj dt dgd}|j d}|jjrJy)Nri)rrr]BCr)rr!rprqrrr__eq__rTr>rdrs r$test_df_float_none_comparisonz2TestFrameComparisons.test_df_float_none_comparisons]  II ! !! $ 4 4V <(#  4::<##%%%%r&ctddddddg}|jdkD}tj|||jddddftj|| |jddddf|j dk(}tj|||jddddftj|| |jddddfy)Nrr^rrir_r)rrrbrcrr)r>rdmask_amask_bs r$test_df_string_comparisonz.TestFrameComparisons.test_df_string_comparison$s ae,AE.BC D bj"&&1a.9 b&k266!A#q&>: bj"&&1a.9 b&k266!A#q&>:r&N)rVrWrXrgrxrtmark parametrizer!rprqrrrrrrrrrrYr&r$r[r[Usi ( [[..q1::2B:G&z2> ..q1::2B:G..q1::2B:G ..q1::2B:G..q1::2B:G ..q1::2B:G&z2> 'z2>&z2> ..q1::2B:G..q1::2B:G ..q1::2B:G&z2> 'z2>&z2> ?) ,Z([,Z(T [[  +1 +1Z " 0&;r&r[c^eZdZejj dgddZejj deje gdZ dZ dZ dZ d Zejj d gdd Zejj d gdd Zd Zy)TestFrameFlexComparisonsop)rrrrrrctjjdjd}tjjdjd}t |}t |}tj |j dz}|j|jjsJ|j|jjrJt||}tt|}tj||||||j ddddfj#} || } ||| j%|j&|j(} tj| | tj||j|||jtj|d||dd} tj|tj*||tj*t-j.t0| 5||dddy#1swYyxYw) Nrir)rrrrrr$Unable to coerce to Series/DataFramern)r!rprqrrronesshapervaluesrrrTrrrbrcrrCreindexr rnanrtrurv) r>rdata other_datardrndim_5fopart_orsxprws r$test_bool_flex_framez-TestFrameFlexComparisons.test_bool_flex_frame1syy$$Q'77?YY**1-==fE t_*%F*+uuRy##%%%559##''))) BO Hb ! ah"e 512qr6"'') vY r6>>"**>E F b"% a oqU\\/BC adAb!H-4 ai2rvv7 ]]:S 1  fI   s  I!!I*boxc$tjjdjd}t |}|tjjdjd}|tjjdjd}|j |d}|j |}|j |d}|j |} tj||t|k(tj|| tj|| tj||jj |jtj||j t|tj||j t|dtj||j t|d|j|d} |j|} |j|d} |j|} tj| |t|kDtj| | tj| | tj| |jj|j|j|d}|j|}|j|d}|j|}tj||t|k\tj|| tj|| tj||jj|jttjjdjd}ttjjdjd}y)NrirrrrrS)r!rprqrrrrrrbrcr Trsrrrr)r>rrrdidx_sercol_seridx_eqcol_eqidx_necol_neidx_gtcol_gtidx_lecol_leidx_gecol_geidx_ltcol_lts r$test_bool_flex_seriesz.TestFrameFlexComparisons.test_bool_flex_seriesOsyy$$Q'77? t_bii++A.>>qABbii++A.>>qABwQ'wwQ'w fbF7O&;< fvg. fvg. fbddggg&6&8&89 fbeeDM&:; fbeeF7O!e&DE fbeeDMe&BCwQ'wwQ'w fb6'?&:; fvg. fvg. fbddggg&6&8&89wQ'wwQ'w fbF7O&;< fvg. fvg. fbddggg&6&8&89..q1AA!DE..q1AA!DEr&cDttjjdj d}tj |j d<|j|}|j drJ|j|}|j dsJ|j|}|j drJ|j|}|j drJ|j|}|j drJ|j|}|j drJy)Nrirrr) rr!rprqrrrrrrrrrr)r>rdrs r$test_bool_flex_frame_naz0TestFrameFlexComparisons.test_bool_flex_frame_nazs ryy,,Q/??G Hvvt UU2Y66$< UU2Yvvd|| UU2Y66$< UU2Y66$< UU2Y66$< UU2Y66$<' not supported between instances of '.*' and 'complex'z unorderable types: .*complex\(\)rn) r!r"rrrrtrurrrrr) r>r#arr2rddf2rwrarr3df3s r$"test_bool_flex_frame_complex_dtypez;TestFrameFlexComparisons.test_bool_flex_frame_complex_dtypeshh1bff-.xxRVVQ-. Sz "d $hhK3   ]]9C 0  EE#J ]]9C 0 ! sGJJs3x  !]]9C 0 # II " #UU3Zyy}}xxRVVT*+d $ ]]9C 0  FF2J ]]9C 0  HKKO ]]9C 0  JJO  -   ! ! # #      sH%H8II2I'I+I78IIII(+I47Jctddtjdgit}tddt j dgit}|j |}tdgdi}tj||y)Nrr^r_rA)FTF) rr!robjectrnowrrbrc)r>df1rrres r$!test_bool_flex_frame_object_dtypez:TestFrameFlexComparisons.test_bool_flex_frame_object_dtypesi 67vF  >?vN 456 fc*r&cttjg}|tjk(}|jdj dusJ|j tj}|jdj dusJ|tjk7}|jdj dusJ|j tj}|jdj dusJy)NrFT)rrrilocitemrrrs r$test_flex_comparison_natz1TestFrameFlexComparisons.test_flex_comparison_natsx rvv{{4 %%'5000rvv{{4 %%'5000rvv{{4 %%'4///rvv{{4 %%'4///r&opnamec tgdgdd}d}t|||jj}t j |t dgtjtgdy)Nrrirrrrricountr name) rrdtypes value_countsrbassert_series_equalr r!r=rR)r>rrdconstrs r$&test_df_flex_cmp_constant_return_typesz?TestFrameFlexComparisons.test_df_flex_cmp_constant_return_typesscY_= >$V$U+22??A  FA3rxx~&6WE r&c tgdgdd}d}|jdd}t|||jj }t j |tdgtjtgdy)Nrrrrirrr ) rrrr r rbr r r!r=rR)r>rrdremptyrs r$,test_df_flex_cmp_constant_return_types_emptyzETestFrameFlexComparisons.test_df_flex_cmp_constant_return_types_emptysrY_= > ''.55BBD  FA3rxx~&6WE r&cftjjgd}t||d}t ddg}|j |d}tddgddgd}t j||t ddgdd g }|j |d}t j||y) Nrr]rrrFrrir]rr )r IntervalIndex from_breaksrr rrbrc)r>iirdserrfrser2res2s r$-test_df_flex_cmp_ea_dtype_with_ndarray_serieszFTestFrameFlexComparisons.test_df_flex_cmp_ea_dtype_with_ndarray_seriess    ) )) 4 Rb) *aVneeCae E5>GH c8,q!fS#J/uuTu" dH-r&N)rVrWrXrtrrrr!r"r rrrrrrrrrYr&r$rr/s [[T#GHI: [[URXXv$67(F8(FT "%N+0$ [[X'KL M  [[X'KL  M   .r&rceZdZdZdZdZdZejjdgddZ ejjde d d d Z d Z ejjdgd dZdZejjdddgdZdZdZdZejjdgddZejjdddgdZdZdZdZd Zy)!TestFrameFlexArithmeticcvtjd}t|}t||d}|j |d}t|j Dcic] }||||z c}}t j|||j |jd}t j||ycc}w)Nrrrr) r!rr rfloordivrrbrcr)r>r#rrdrrrresult2s r$test_floordiv_axis0z+TestFrameFlexArithmetic.test_floordiv_axis0siilSk Ss+ ,Sq)RZZHcc2c7c>1HI fh/++cjjq+1 gx0 IsB6c tjdd}tjdd}t|}t ||d}|j |d}t ||z||zd}t j||y)N 2016-01-01ryr}1rrrr)rrtimedelta_ranger raddrbrc)r>dtitditserrdrrs r$test_df_add_td64_columnwisez3TestFrameFlexArithmetic.test_df_add_td64_columnwisesummL"5  b1c{ 33' (1%ssSy9: fh/r&c^tjdd}tgdd}t||d}t||d}tjd j }|j || }ttgd d |d zd}tj||y)Nr$rr})z1 Dayrz2 Daystimedelta64[ns]rArr)days fill_value)z 2016-01-02z 2016-01-03z 2016-01-05zdatetime64[ns]ri) rrr r Timedeltato_timedelta64r(rbrc)r>r)rrdrfillrrs r$$test_df_add_flex_filled_mixed_dtypeszFV1W    fh/r&c|fd}t|d|z}||d|z}tj||t|d|z}||d|z}tj||t|ddiy)Ncjdr'ttjdd||Stt||S)N__r__) startswithrrreplace)rrrs r$rz8TestFrameFlexArithmetic.test_arith_flex_frame..f*sF}}U#AwxE4)@A!QGG(78R(A. .r&rirrA)rrbrcr )r>all_arithmetic_operators float_framemixed_float_framerrrrs @r$test_arith_flex_framez-TestFrameFlexArithmetic.test_arith_flex_frame$s& / *b)!k/:[!k/2 fh/0*B/4E0EF&,=(=> fh/6#t5r&r)__add____sub____mul__c"tt|}t||d|z}||d|z}d} |dvrddd} n|dvrddi} tjr|dk(rd|zdj| d<t j ||t|| t||d|z}||d|z}t j ||t|ddi t||d|z}||d|z}t j ||y) Nri)rAuint64)rr)r@rBrrr]rA) rrr1 USE_NUMEXPRr=rbrcrr ) r>r int_framemixed_int_framer>r6rrrr=s r$test_arith_flex_frame_mixedz3TestFrameFlexArithmetic.test_arith_flex_frame_mixed:s2 Hb !."-a/.AB_a/&9:  ".E ) )$KE    ;q @ o-s399E#J fh/u-0*B/4E0EF&,=(=> fh/6#t5(B'I 6YI . fh/r&dimrrjc|}tjd|z}d}tjt|5t |||dddy#1swYyxYw)N)rrrn)r!rrtrurvr)r>r<r=rIrr#rws r$test_arith_flex_frame_raisez3TestFrameFlexArithmetic.test_arith_flex_frame_raisecsV&ggdSj!4 ]]:S 1 * $GK $S ) * * *s AAc|jd}tj||dz|j|dd}|jtj z}tj|||ddj|}|jtj z}tj||t jtd5|j|jdddddt jtd5|j|jddddddy#1swYOxYw#1swYyxYw) Nrrr1rnrr0r rSr1) r(rbrc sort_indexr!rrtruNotImplementedErrorr)r>r= const_addrrs r$test_arith_flex_frame_cornerz4TestFrameFlexArithmetic.test_arith_flex_frame_cornerns*OOA&  iq9Ra1))+bff4 fh/Ra$$[1))+bff4 fh/ ]].l C ? OOK,,Q/AO > ?]].l C M OOK,,Q/g!O L M M ? ? M Ms!E  "E EE )r(submulmodc$|}|jd}|d}t||}tt|}tj|||||tj||d||j |j y)Nrrrr)xsrrrbrcr)r>r%rrdrowrrs r$test_arith_flex_series_opsz2TestFrameFlexArithmetic.test_arith_flex_series_opssteeCji BO Xr " afbSk2 a!nbsmoo>r&cT|}|jd}|d}tj|j|d||ztj|j |||z tj|j |d|j |z j y)Nrrrr)rVrbrcr(divr)r>r%rdrWrs r$test_arith_flex_seriesz.TestFrameFlexArithmetic.test_arith_flex_seriess eeCji bffStf4b3h? bffSk284 bffSqf1BDD3J>>Br&r=int64float64cttjdjd|}ttjtj gddgddgg}|j |dd }tj||y) NrjrrAr?g?rr r) rr!rrrinfrZrbrc)r>r=rdrrs r$#test_arith_flex_series_broadcastingz;TestFrameFlexArithmetic.test_arith_flex_series_broadcastingspryy'//7u Ervvrvv.c S$KHI1G, fh/r&ctgt}tddg}tddgddggddg}tjt d 5|j |d dddtjt d 5|j|ddd dddy#1swYExYw#1swYyxYw)NrAr]rrmrrirrkr1rnEr0rM)r rrrtrurOr(rR)r>ser_len0df_len0rds r$test_arith_flex_zero_len_raisesz7TestFrameFlexArithmetic.test_arith_flex_zero_len_raisess"F+S#J/ AA'#s < ]].l C - FF8F , -]].l C : KK3dqK 9 : : - - : :sB(B4(B14B=ctjddtjdddgd}td|it d  }|j dj d }|j d d }tj||y) NrrrrkrfloatrAr^rjrrir0) r!r"rrrfillnar(rbrc)r>datrdrerfs r$test_flex_add_scalar_fill_valuez7TestFrameFlexArithmetic.test_flex_add_scalar_fill_valuesqhh1bffaA.g> s|58 4iilq!ffQ1f% c3'r&ctgdgd}tgdgd}tgdgd}|j|}tj||y)N)rrirrkr)rrirrirrr)rrirriri)rrririr)rrRrbrc)r>rrrrs r$'test_sub_alignment_with_duplicate_indexz?TestFrameFlexArithmetic.test_sub_alignment_with_duplicate_indexsC? 3_OD fh/r&)r@rBrA __truediv__c6ttjdtjj djdd}t |||}ddg|_ddg|_t |||}tj||y)Nryrirr]) rr!rrprqrrrbrc)r>rrdrrs r$&test_arithmetic_with_duplicate_columnsz>TestFrameFlexArithmetic.test_arithmetic_with_duplicate_columnssRYYr]1F1Fq1I1P1PQS1TU V"72r?2&:3Z  R$ fh/r&levelrNctgdgdd}|jjd|_tgdgdd}|jjddg|_|j||}tgdgdd}|jjddg|_t j ||y) NrrrirrL1rrr))r]r)r]DL2)rq)rr set_namesr(rbrc)r>rqrrrrs r$test_broadcast_multiindexz1TestFrameFlexArithmetic.test_broadcast_multiindexsii89kk++D1 YIFGkk++T4L9 E*)KL#++55tTlC fh/r&c ptgdgddtjdgdggdggd }td gtjdgdggd d g }tgd gddtjdgdggdggd }|j |d}t j ||y)NrrrkriirrrsscenrTidnamesr皙?rTr~)ffffff?333333@333333 @)r皙@g@rrrr from_productr r(rbrcr>rdseriesrrs r$ test_frame_multiindex_operationsz8TestFrameFlexArithmetic.test_frame_multiindex_operationss I .))y)1F  E))C53%.P  "/ :))y)1F  Q' fh/r&c ntdgdgdtjdgdggddg }tgd tjdgdggd ggd  }tgd gddtjdgdggd ggd  }|j |d}t j ||y)Nrrr|rrr~rTrr)$@g4@g>@rsr})g&@g?@)g*@g7@g@@rrrrs r$))y)1F  Q' fh/r&c tgdgddtjdgdggdggd }td gtjd gdggd d g }ttj tj dtj ddddd tj fggd }|j|d}tj||y)Nrr{r|rrrsr}rrrrrTr~rrrrrr)rrrirr) rr rr r!r from_tuplesr(rbrcrs r$)test_frame_multiindex_operations_no_alignzATestFrameFlexArithmetic.test_frame_multiindex_operations_no_aligns I .))y)1F  E))C53%.P  66 (((!!!#rvv&  ,  Q' fh/r&ctgdgddtjgdgd}tdgtjd gd ggd d g}td dt j gddt j gdtjgdgd}|j|d}tj||y)Nrr{r|)rr)rrrir}rrrrrrTr~rrrrrr) rr rr rr!rr(rbrcrs r$+test_frame_multiindex_operations_part_alignzCTestFrameFlexArithmetic.test_frame_multiindex_operations_part_align%s I .(( ,    E))C53%.P  bff%c3-? @(( ,   Q' fh/r&)rVrWrXr"r,r5r?rtrrrHrrKrQrXr[rarfrkrmrpryrrrrrYr&r$rrs9 1 00&6, [[T#DE&0F&0P [[UE!QK0*1*M& [[T#?@ ?A ? C [[Ww &:;0<0 :(0 [[T#ST0U0 [[Wq$i0 01 000000:0r&rceZdZdZdZdZdZdZdZdZ e jjdd d gd Z e jjd d dgdejd dged ded dggdZdZdZdZe jjdddddddej,dddfej.dddfge jjd ej2ej4ej6ej8ej:ej<gd!"d#Zd$Z d%Z!y&)'TestFrameArithmeticctddgd}tddgddgg}||z}t||d}tj||y) Nrr.rArrirrkr&r rrbrc)r>rrdrrs r$test_td64_op_nat_castingz,TestFrameArithmetic.test_td64_op_nat_castingHsTeU^+<= AA' (c-. fh/r&ctjdjdd}t|ddggd}|dgddf}|jd|jdfk(sJtdd gd dgdd gg|j |j |j }||z}tj||||z}tj||y) NrjrriTFrrrrkrrr r= r!rrrrrr r=rbrc)r>r#rdrowlikerrs r$'test_df_add_2d_array_rowlike_broadcastsz;TestFrameArithmetic.test_df_add_2d_array_rowlike_broadcastsRsiil""1a( sT5M Iqc1f+}}BHHQK 0000VaVaV $JJ(())  g fh/2 fh/r&ctjdjdd}t|ddggd}|dddgf}|j|jd dfk(sJtddgd dgd d gg|j |j |j }||z}tj||||z}tj||y)NrjrriTFrrrrr ryrr)r>r#rdcollikerrs r$'test_df_add_2d_array_collike_broadcastsz;TestFrameArithmetic.test_df_add_2d_array_collike_broadcastsgsiil""1a( sT5M Ia!f+}}!a 0000VaVaW %JJ(())  g fh/2 fh/r&c|}|r|dvrtj|tjdj dd}t |ddggd}|d gddf}|j d |j d fk(sJt|jd ||jt|jd ||jt|jd ||jg}t ||j|j} t|||} tj| | y) N__rmod__ __rfloordiv__rjrriTFrrrr]rr)td&mark_array_manager_not_yet_implementedr!rrrrrrsqueezerr rbrc) r>r3r<using_array_managerrr#rdrexvalsrrs r$)test_df_arith_2d_array_rowlike_broadcastsz=TestFrameArithmetic.test_df_arith_2d_array_rowlike_broadcasts|s * 6-J#J  5 5g >iil""1a( sT5M Iqc1f+}}BHHQK 0000 )GBFF3K (): ; (GBFF3K (): ; (GBFF3K (): ;  VRZZrxxH$V$W- fh/r&c~|}|r|dvrtj|tjdj dd}t |ddggd}|ddd gf}|j |j d d fk(sJt|d||jt|d||jd }d} |dvr'tjd |jD} t ||j|j| } t|||} tj| | y)NrrjrriTFrrrr)TFc34K|]}|jywr;)r).0rs r$ zPTestFrameArithmetic.test_df_arith_2d_array_collike_broadcasts..s$G!QXX$Gsr)rrr!rrrrrr common_typerrr rbrc) r>r3r<rrr#rdrrr=rrs r$)test_df_arith_2d_array_collike_broadcastsz=TestFrameArithmetic.test_df_arith_2d_array_collike_broadcastss$* 6-J#J  5 5g >iil""1a( sT5M Ia!f+}}!a 0000,'"T(F+GOO,=>-72e9f-goo.?@   2 2NN$Gv}}$GHEVRZZrxxuU$V$W- fh/r&ctddgddgg}|dz}|jjd}|dk(jsJd|z}|jjd}|dk(jsJy)NFTrc|jSr;kindrs r$z:TestFrameArithmetic.test_df_bool_mul_int.. affr&rc|jSr;rrs r$rz:TestFrameArithmetic.test_df_bool_mul_int..rr&)rr applyr)r>rdrkindss r$test_df_bool_mul_intz(TestFrameArithmetic.test_df_bool_mul_ints u~6 7a ##$45 !!###R ##$45 !!###r&ctgdgdd}||z}tgdgdd}tj||y)Nrrr)aabbccrirkrjrrbrc)r>rrrs r$test_arith_mixedz$TestFrameArithmetic.test_arith_mixeds:Y?@#5IFG fh/r&rr]rctddgddgd}||d|}|||d}tj||y)Ng?gffffff @g@g333333rr)rrbr )r>all_arithmetic_functionsrrdrrs r$test_arith_getitem_commutez.TestFrameArithmetic.test_arith_getitem_commutesJ c3ZsDk: ;)"a05+BsGQ7 vx0r&rrrirrirctddgddgd}tddgddgd}||z}tj||y)Nrrrirr)r>rrdrrs r$&test_arith_alignment_non_pandas_objectz:TestFrameArithmetic.test_arith_alignment_non_pandas_objectsK aV1a&1 2Aq6A78f fh/r&cttjdddjddgdgd}|j d j }t|j |z|j |j }tj||z|t|j j|zj|j |j }tj|j|d |t|d }t|j |z|j |j }tj||z|t|j j|zj|j |j }tj|j|d |tjjdj|j}t|j |z|j |j }tj|j||y)Nrryf8rArrrrrrrrrr ri)rr!rrrVrr rrbrcrr(rsrprqr)r>rdval1addedval2val3s r$test_arith_non_pandas_objectz0TestFrameArithmetic.test_arith_non_pandas_objects  IIa4 ( 0 0A 6+!  uuSz  "))d*"((BJJO b4i/299;;-00"**U bffTf2E:BuI"))d*"((BJJO b4i/299;;-00"**U bffTf8%@yy$$Q'..rxx8"))d*"((BJJO bffTlE2r&c 6|}tjtjdd}ddg}t|g|}d}t |||}t|Dcgc]}t |||c}g|} t j || ycc}w)Nr)startendrrirmry)rCategoricalIndexinterval_rangerrrbrc) r>r<rindrrdnumrnrs r$.test_operations_with_interval_categories_indexzBTestFrameArithmetic.test_operations_with_interval_categories_indexs %!!""3"3#3"GH1v vs + R%DAq~wq"~c2ABCP fh/BsBcttjdtjdgtjdtjdgdddgd }|dg}||z }ttjd tjd gtj tj gdddg }t j||y) N2019202020182021)r^r_r^r_M8[ns]rr=rrm)rrrr2r!rrbrc)r>rdrrrs r$test_frame_with_frame_reindexz1TestFrameArithmetic.test_frame_with_frame_reindexs  V,bll6.BC V,bll6.BC EN  %kc\\!_bll1o6?O PEN  fh/r&z value, dtype)ri8)rr)r)y? complex128)rr)TrRns.s ajjr&)r,c tjdftjdftjdftjdfh}t ||}t d|j|jgi|j}tjdftjdftjdftjdftjdftjdftjdftjdftjdfh }||f|vrd} |dk(r|tjk(s|dk(r|tjk(rd} nM|dk(rd} nE|tjur#d} |dk(r,tjr|d k(rt} nd |jd } tj t"| 5t%j&| 5|||jddddddy||f|vr|tjtjfvrPtjr |d k(rt} nd} t%j&| 5|||jdddyd } tj t(| 5|||jdddyt%j&d5|||jj*} |||j*} dddt%j,  y#1swY*xYw#1swYyxYw#1swYyxYw#1swYyxYw#1swYQxYw)NrRr]rArrrz3ufunc 'remainder' not supported for the input typesz-numpy boolean subtract, the `-` operator, is rzcannot perform __z7__ with this index type: (DatetimeArray|TimedeltaArray)rnz+operator '.*' not implemented for .* dtypes)rtruedivpowr(rSr8rr<r=rTrRr1rE UserWarningrVrtrurrbassert_produces_warningrOr r ) r>rr<r=r6skipelemrdinvalidwarnrwrrs r$test_binop_otherz$TestFrameArithmetic.test_binop_others6  v & \\6 " \\6 " \\6 "  E5) djj$**56djj I\\9 % \\9 %   y ) \\9 % \\9 % \\9 % \\9 % \\6 " \\< (   ;' !D"rX\\'9"rX\\'9,&Kx||#EVO((3q8&D( }5AA y4 '//5'r4::&' ' '%[D hllHLL11##(Cq(H&DD//5'r4::&''D]]#6cB'r4::&''++D1 0B +22b%=// 0  " "68 4-'' ' ''' '' 0 0sHLL*LL+L71ML LL(+L47MM cdtjtddgtddgg}tjtddgdtddgg}tddgddgg|}tddgddgg|}||z }tddgddgg|}t j ||y) NrrirrkInt8rArmrr from_arraysr rrbrcr>midxmidx2rrrrs r$*test_arithmetic_midx_cols_different_dtypesz>TestFrameArithmetic.test_arithmetic_midx_cols_different_dtypesvs%%vq!f~vq!f~&FG&&1vV(DfaQRVn'UV1a&1a&)48Aq6Aq6*E:q!fq!f-t< fh/r&cdtjtddgtddgg}tjtddgdtddgg}tddgddgg|}tddgddgg|}||z }tddgddgg|}t j ||y) NrrirrkrrArmrrs r$:test_arithmetic_midx_cols_different_dtypes_different_orderzNTestFrameArithmetic.test_arithmetic_midx_cols_different_dtypes_different_orders%%vq!f~vq!f~&FG&&1vV(DfaQRVn'UV1a&1a&)48Aq6Aq6*E:r1gAw/> fh/r&N)"rVrWrXrrrrrrrrtrrrr!r"rrrrrr timedelta64 datetime64rr(rRrSrrTrrrrrYr&r$rrGs}00*0*060@ $0 [[US#J/101  [[Aq668288QF#3U1a[%A-P0034 00( [[     !  R^^B %y 1 R]]2t $i 0   [[ LL LL LL    LL LL   ! D5  2D5L00r&rcttjjdj dj ddddg}t tj}||z}t|jtjz|j}tj||tjtd 5||k(dddt|jj!d |j}tjtd 5||k(dddy#1swYcxYw#1swYyxYw) Nrirjrr]rrmrAz not alignedrnr)rr!rprqrrrr r]rrrrbrcrtrurvrP)rdrrrrs r$,test_frame_with_zero_len_series_corner_casesrs  a 003;;AqACQT: B rzz "C #XFRVV+RZZ@H&(+ z 7 c  BIINN8,bjj AC z 7 s   sD6'E6D?E ctddgtj}tddgddg}||z}|}t j ||y)Nr]rrrrir)rr!r]r rbrc)rdrrrs r$,test_zero_len_frame_with_series_corner_casesrsH C:RZZ 8B !QSz *C #XFH&(+r&cdtddtjgi}t|}|j d}tgd}t j ||y)NOner]333333?rr)r]r r)r r!rrsumrbr )rrdrrs r$+test_frame_single_columns_object_sum_axis_1r sT vsC() D 4B VVV^Fm$H68,r&creZdZdZdZej jdejejejejgdZ ej jdddgej jdd Zej jdgd d Zd Zd ZdZdZdZdZej jdej.ej0ej2ej4ej6ej8gdZdZdZdZ dZ!dZ"ej jdddejFdej jIe%ddd d!d"d#d$d%d&g d'Z&ej jd(gd)d)e'jPgd)e'jR*e*d+d,gd-Z+ej jd(d+d.gd/e'jPd+d.ge*d+d0gd1Z,d2Z-d3Z.d4Z/d5Z0y6)7TestFrameArithmeticUnsortedctjdddd}ttjj dj t||dg}|jd }||z}|jjtjusJ||z}|jjtjusJy) Nz1/1/2011ryh US/Eastern)r~freqtzrirrz Europe/Moscow) rrrr!rprqrrr tz_convertr rrutc)r>rngrd df_moscowrs r$*test_frame_add_tz_mismatch_converts_to_utczFTestFrameArithmeticUnsorted.test_frame_add_tz_mismatch_converts_to_utcsmmJN  II ! !! $ 4 4SX >cTWSX MM/2 i||(,,...R||(,,...r&ctjddd}ttjj dj t|df|}||dddz}||z}tj|jddd<tj|||ddd}||jtjj djt|z}tj||y) Nz1/1/2000z1/1/2010Y)rrirrr)r period_rangerr!rprqrrrrrrbrctake permutation)r>rtsrrhalfs r$test_align_framez,TestFrameArithmeticUnsorted.test_align_framesooj*3?  II ! !! $ 4 4c#h] C3 b1g7 ff add fh/#A#wdii 5 5a 8 D DSY OPP fh/r&rctgdgddt}d}tjt|5|j t j}ddd||d}|djt}t j|tj|<tj|||||}|||jt}t j|tj|<tj||d}tjt|5|||j d}dddtj||d}tjt|5||j d|}dddtj||y#1swYuxYw#1swYxYw#1swY9xYw) N)rir{N)rrirrk)col1col2rAzDowncasting object dtype arraysrnrr) rrrbr FutureWarningrir!rrMrisnarc)r>rrdrwfilledrrs r$test_operators_none_as_naz5TestFrameArithmeticUnsorted.test_operators_none_as_nas(, ?v  0  ' ' S A 'YYrvv&F 'Bfa=''/&(ff"# fh/Bff%,,V4&(ff"# fh//  ' ' S A *BIIaL)F * fh//  ' ' S A * ! b)F * fh/) ' ' * *  * *s# G =GG" GG"G+zop,res)rF)__ne__Tz ignore:elementwise:FutureWarningc~t||d}t|jj|usJy)Nr^)rrRr)r>rrfr=rs r$%test_logical_typeerror_with_non_validzATestFrameArithmeticUnsorted.test_logical_typeerror_with_non_valids9 *b)%0FJJL$$&'3...r&)r(rRrSrZrc tjtdgdgdggd}tt j dj dd|gd  j}tj}tt|d}|ytgd gd}t|||d d }tj|jD cgc])\}} ||j|dddd|fddf| +c} }j} t!j"|| tddgddg}t|||dd }tj|jD cgc]&\}} ||j|dd|fddf| (c} }j%|j} t!j"|| ycc} }wcc} }w)Nabcrr)firstsecondthirdrQr)value1value2value3r)rrgY@r1r)rqrSrrrrr0)r rrsrr!rrrNr IndexSlicerrr concatitemsrrbrc reindex_like) r>rr rdidxoparrrvrs r$test_binary_ops_alignz1TestFrameArithmeticUnsorted.test_binary_ops_aligns '' %[19 =.   IIf  % %b! ,2  *, mmhD) ;  %y 1 R':999: CASAq!G a(! , C *,  fh/ C; 0 1 R(; II J1s266#ad)Q,/3J K \"  Z\  fh/ DKs .G 7+G$ ctjddgddgg}ttjdd|}t d d d }|j }d d g|j_|j }d |j_ |j|d d }|j|d d }|j|d d }|j|d d } |j|d d } |j|d d } ttjgdgdgd|} ||fD]} tj| | d d g| j_|| | | fD]} tj| | y)Nr]rrr)rirkr\rArmrrirlvl0lvl1)rSrq)rrirri)r rrr!rr rCrrr r rSr"rbrc)r>rrdsrs2res1rres3res4res5res6rerfs r$&test_binary_ops_align_series_dataframezBTestFrameArithmeticUnsorted.test_binary_ops_align_series_dataframe*sr&&c S#J'?@ rwwvW5t D # $ggi#V,  VVX vvaaqv)vvbqv*wwqqw*wwrw+wwqqw/wwrw0 HHlL1 A4 $< ,C  ! !#s + ,$V, $d+ ,C  ! !#s + ,r&cXtjgdd}|jddd}|jddd}tdddgi| }tdddgi| }tdtj d tj gi| }t j||z|y) N)z 2011-01-01z 2011-01-02z 2011-01-03UTC)rz Asia/Tokyorirrr]rr)r DatetimeIndexrrr!rrbrc)r>baseidx1idx2rrres r$ test_add_with_dti_mismatched_tzszr=r>rG frame_copyrindexerre self_added added_rev plus_empty empty_plus empty_emptyreverses r$test_combineFramez-TestFrameArithmeticUnsorted.test_combineFrameSs (():):3Q3)?@ sO57VV ,))!,,c12j(*##%++3!#))+ uSz002CGGG4DE,.FF(() uSz3775:3C3C+DExxc **:+;+;r=r>rGrrkeyrB larger_series larger_addeds r$test_combine_seriesz/TestFrameArithmeticUnsorted.test_combine_seriess 1 1! 45f$kkm FFC  " "1k#&6&D E F(  c}- "]2 !'') GFC  " "<#4a&+o F Gl"""xx S)*..000"F*vvellfll2333"FMM)$<<5d 4!FMM)$<<5d 4 3<"7"7"@@ wYW7S  3<"7"7"@@ wYW7S r&c|d}|j|d}|jD]r\}}||z}tj|||d||j|k(sJ|j|jk(r|jdk(rcJ|jrJ|dd}|j|d}tj |j |j |dd} |j| d} tj|| |j|ddd}ttj|j |j} tj|| |ddj|d}ttj|j |j} tj|| |dd jg } | j|d}t|t|k(sJy) Nr]r rF) check_namesrrrrm)r(r9rbr r rUr rcrr!rrrrSr) r>datetime_framerrrdrr smaller_frame smaller_added smaller_tssmaller_added2rframes r$test_combine_timeseriesz3TestFrameArithmeticUnsorted.test_combine_timeseriess C  ""2G"4&,,. +HC2XF  " "5:v5 I:??c) ))xx277"{{c))){{*** +'s+ %))"7);  m11>3G3GHW '++JW+E m^< ##BrF#9 FF...8N8N  fh/ #'''9 FF...8N8N  fh/r"**2*62G,6{c"g%%%r&c|dz}tj|j|jdz|dz}|jD]5\}}tj|j||jdz7t |ddit dz}|j jt j sJt|jdk(sJy)NrirrAr) rbassert_numpy_array_equalrr9r rr equalsrr)r>r=r>rrrBs r$test_combineFuncz,TestFrameArithmeticUnsorted.test_combineFuncsq ##FMM;3E3E3IJ#Q&LLN SDAq  ' '2CA2F2M2MPQ2Q R S6#t5q||""9;#4#45556>>"a'''r&funcc ttjjdj dt t dttjddd }|j}|jd }tj|jd z}|||}tj|j ||j |j d } t#j$t&t)j*|  5|||ddd|||} tj| j ||j |j ||d} tj| j ||j dd} t#j$t&| 5|||dddddy#1swYxYw#1swYyxYw)Nri)rkrlrAz 2000-01-01rxr)r~rrrrrrzNUnable to coerce to Series/DataFrame, dimension must be <= 2: (30, 4, 1, 1, 1)rnrzQCan only compare identically-labeled \(both index and columns\) DataFrame objects)rr!rprqrrr rsrrrrCrVrrrbrsrrtrurvreescape) r>r%r=rvrrrWrrrwr!result3s r$test_comparisonsz,TestFrameArithmeticUnsorted.test_comparisonss  II ! !! $ 4 4W =$v,f5-- bsC  hhjooc"Y./c3 ##FMM4 CJJ3OP 7 ]]:RYYs^ <  f  |S) ## NND!4!4cjjA {A& ##GNND9K9KQ4OP  ]]:S 1 1 |BQ/ 0 1 1   1 1s G# G/#G,/G8c tdDcic] }|dddd c}}tt|}djddg}t j t | 5||d dddycc}w#1swYyxYw) Nrr^r_r`)rrzrz;'[<>]=?' not supported between instances of 'str' and 'int'z,Invalid comparison between dtype=str and intrnr)rrrrrtrur)r>compare_operators_no_eq_nerrdrrws r$*test_strings_to_numbers_comparisons_raiseszFTestFrameArithmeticUnsorted.test_strings_to_numbers_comparisons_raisess >M NQe%e4 4 N  H8 9hhM>   ]]9C 0  b!H   O  sA0 A55A>cttjdtjt t dt }tj|j|jddf<tjd5|jdk}dddtjd 5|dkj}dddtjy#1swYNxYw#1swY,xYw) N)ryrkrArlrmrr]ignore)rraise)rr!rr]r rsrrrr errstaterrbrs)r> missing_dfrrs r$'test_comparison_protected_from_errstatezCTestFrameArithmeticUnsorted.test_comparison_protected_from_errstate,s GGG2:: .$v,f5 4666 z''*C/0 [[ * -!((1,H - [[ ) - 1n,,F - ##FH5  - - - -s C(9C4(C14C=c,ttjdjd}tjddg}tj ddg}|j }gd}t|}tddgddgddgg}||kD}tj|||j|kD}tj||jd} d} d } tjt| 5||kDdddtjt| 5||kDddd||kD}tj|||j|kD}tj||jtjt| 5||kDdddtjt| 5|j|kDdddtddgddgddgg}||k(}tj||tjt| 5||k(dddtjt| 5||k(ddd||k(}tj|||j|k(}tj||jtjt| 5||k(ddd|jj|jk7sJttjdjdt!d t!d  }|j"|_|j$|_tjt| 5||k(dddtjt| 5||k(dddy#1swYxYw#1swYxYw#1swY'xYw#1swYxYw#1swYxYw#1swYxYw#1swY4xYw#1swYxYw#1swYyxYw)NrjrririririFTz5Unable to coerce to Series, length must be 2: given 3,Unable to coerce to DataFrame, shape must bez4operands could not be broadcast together with shapesrnABr.r)rr!rrr" atleast_2drtuplerbrcrrsrtrurvrrsr r) r>rdrb_rb_clsttuprrmsg1dmsg2dmsg2dbs r$test_boolean_comparisonz3TestFrameArithmeticUnsorted.test_boolean_comparison8sdryy|++F3 4 HHaV mmQF#eeCjuenudmdD\JKa fh/Q ##FHOO<G>G ]]:U 3  H ]]:U 3  H  c fh/S ##FHOO< ]]:U 3  H ]]:V 4  IIO uentUmeU^LMq fh/ ]]:U 3  #I ]]:U 3  #I s fh/c! ##FHOO< ]]:U 3  #I yy#))+++ IIaL  ($t*DK :: ]]:U 3  #I ]]:U 3  #I  m                  sl6N#N0N=;O OO$9O1+O>P #N-0N:=O OO!$O.1O;>P Pctd}ttjdt |zj dt ||t d}d|jddddfjz}td}td}|j}|||zj| }||xx|z cc<|j| }|j}||||zj| } ||xx||z cc<|j| } tj||tj|| tj|| |j}|||z j| }||xx|zcc<|j| }|j}||||z j| } ||xx||zcc<|j| } tj||tj|| tj|| y) Nabcdefgryrrr'rbedcfbcdefrm) rsrr!rrrrrrCrrbrc) r>rX_origZblock1subsXresult1r!r|result4s r$test_inplace_ops_alignmentz6TestFrameArithmeticUnsorted.test_inplace_ops_alignmentsy/ IIb3w<' ( 0 0S\ B)  &++a2g&++- -gG} KKMV9q=))$)7 & Q ))D)) KKMV9qy(11$1? & QvY ))D)) gw/ gw/ gw/ KKMV9q=))$)7 & Q ))D)) KKMV9qy(11$1? & QvY ))D)) gw/ gw/ gw/r&cLtgd}ttjj dj dddj dd}|j}|}|dz }tj||tj|dz|||usJ|j|jusJ|j}|}|dz }tj||tj|dz|||usJ|j|jusJ|j}|}|d z }tj||tj|d z||j}|}|d z }tj||tj|d z|||usJ|j|jusJtjj dj ddd}t|jd d }|j}|}|d xxdz cc<t|jdzd d }tj||tj|||j|jusJ|j}|}|d xxd z cc<t|jd zd d }tj||tj|||j|jusJy) Nrrirrryrzrrr_r^rr]) r rr!rprqrrrCrbr _mgrrc) r>s_origdf_origrBrCrdrr#rs r$test_inplace_ops_identityz5TestFrameArithmeticUnsorted.test_inplace_ops_identitys " II ! !! $ - -a - < D DR K  KKM  Q q"% vz1-Bwwvv   \\^ a b#& gk2.Syyww#((""" KKM  S q"% v|Q/ \\^ c  b#& gmR0Syyww#(("""ii##A&//2A/>#((*59: \\^ 31 388:>>? b(+ c8,ww#((""" \\^ 33388:#3%@A b(+ c8,ww#(("""r&r(andrZz__idiv__ not implemented)rureason)marksr rTrSorrrRrxorc$tgdgdd}d}|dvrgd|d<|j}d|d }d |d }t|||t|||}tj||t |}t ||k(sJy) Nrrrri)rrr)TFTr__ir9)rrCrrbrcr)r>rrdoperanddf_copyioprs r$test_inplace_ops_identity2z6TestFrameArithmeticUnsorted.test_inplace_ops_identity2s,_9= > % %)BsG'')B4rl"R[ C!'77B'0 b(+b6"v!!!r&valrrArrkcgd}gd}ttjjdj d||}tj }t|||d|j }tj|||dd |tgd gd gd d|j }tj|||d d |y) Nrrrrrirrrrrrrryrrrr) rr!rprqrr _align_for_opr rbrc)r>rr rrdalignrs r$test_alignment_non_pandasz5TestFrameArithmeticUnsorted.test_alignment_non_pandass !  II ! !! $ 4 4V <  ''3Ss;288L eB!4Q7B)) rr rrdrrws r$)test_alignment_non_pandas_length_mismatchzETestFrameArithmeticUnsorted.test_alignment_non_pandas_length_mismatch5s!  II ! !! $ 4 4V <  ''E ]]:S 1 # "c " #]]:S 1 # "c " # # # # # #s0 B4 C4B=C cRgd}gd}ttjjdj d||}tj }tj gdgdgdg}tj|||d d t||j|jtj|||d d t||j|jd }tj gdgdg}tjt| 5|||d dddtjt| 5|||d dddtjd}tj d}tjt| 5|||d dddtjt| 5|||d dddy#1swYxYw#1swYxYw#1swYQxYw#1swYyxYw)Nrrrirrrrkrrj)rrrrrrrrnrzGUnable to coerce to Series/DataFrame, dimension must be <= 2: (3, 3, 3))rr!rprqrrrr"rbrcr rrtrurvzerosrzr{)r>r rrdrrrws r$'test_alignment_non_pandas_index_columnszCTestFrameArithmeticUnsorted.test_alignment_non_pandas_index_columnsHs!  II ! !! $ 4 4V <  ''hh 9i89  "c "1 % c2:: >   "c "1 % c2:: > =hh 9-. ]]:S 1 # "c " #]]:S 1 # "c " #hhy!ii U ]]:S 1 # "c " # ]]:S 1 # "c " # # # # # # # # # #s0- G9 H5 H$ H9HHHH&ctddgddgd}|d}tjd5t|||dddy#1swYyxYw)Nrrr)rrbrrr>r<rdrs r$test_no_warningz+TestFrameArithmeticUnsorted.test_no_warningnsW c3ZsDk: ; sG  ' ' - 5 1GB0 1! 4 5 5 5s A  Actddgddgd}|d}tjtd5t |||ddddy#1swYyxYw)Nrrrztakes 2 positional argumentsrnr)rrtrurrrs r$test_dunder_methods_binaryz6TestFrameArithmeticUnsorted.test_dunder_methods_binarytsZ c3ZsDk: ; sG ]]9,J K 8 1GB0 1!Q 7 8 8 8s AAcftjddjdd}tjdt}t |}|j |d<|jt}||jz }||jz }tj||y)Nr'r]rAry)ryrz0.X) r!rrrintrrrMrhmeanrbrc)r>rrrrrrs r$test_align_int_fill_bugz3TestFrameArithmeticUnsorted.test_align_int_fill_bug{s IIgY / 7 7B ? GGG3 'lYY[E jjsxxz!# fh/r&N)1rVrWrXrr!rtrrrr(rRrSrr)filterwarningsr,r>rIrPr^rgrqrurrrrrrr}rrrrrr2xfailAttributeErrorrr!r"r\rrrrrrrrYr&r$rrsq / 0 [[ x||X\\8<<9I9IJ00< [[X(9;K'LM [[ BC/DN/  [[T#JK$0L$0L,>.;3z& P,&\ ( [[ hkk8;; X[[(++V"1 "1H 6L\.0`9#v [[   FLLkk'')2L(         ! *"+*"& [[   BHHYbhh / !QK  CC& [[UaVVXRXXq!f5EuQPQ{$ST#U#$$#L5 8 0r&rctdgdi}tgd}||z}tdtjdtjgi}tj||y)Nr]rsrrrr!rrbrc)rrrrs r$test_pow_with_realignmentrsR c9% &D I &E 5[F#RVV456H&(+r&cttjjdj dddgd}t gdgd}|j d |j d jd d z}t||jd }|jd }||z}tj||||jd z}tj||y)Nrirr')ryrrrmrrr\rrInt64r) rr!rprqrr to_numpyrrrMrbrc)rdrrdf_ears r$&test_dataframe_series_extension_dtypesrs  a ))!S':O B / 2C{{7#cll7&;&C&CB&JJH2::WEH IIg E S[F&(+ SZZ( (F&(+r&c tjjdjddd}t |}|j j dddd}tj|jdgdf<|j j ddi}tj|jddgf<|j j dddd}tj|jdtjddf<|j j dddd }tj|jdtjd d f<||f||f||ffD]M\}}||z}t |jD cic]} | || || zc} } tj|| Oycc} w) Nriri)r'ryrh)rrrrrirrkrrkr) r!rprqrrrCrMrrrrrbrc) r#rrrdf4df5rrrfrrs r$"test_dataframe_blockwise_slicelikers ))   " + +AtY ?C C.C ((*  G@ ACVVCHHQ \ ((*  QL )CvvCHHQV ((*  G@ AC#%66CHHQ !Q  ((*  G@ AC#%66CHHQ !Q  c S#Jc ;- eUlT\\Jaa58!33JK c8, -KsG z df, col_dtyperrrrabrmr]rrctdtjgdtjggtd}|j d|i}|t dgtdz}t j||y) Nrrrrmrgrr)rr!rrsrMr rbrc)rd col_dtyperrs r$/test_dataframe_operation_with_non_numeric_typesrsc3-#rvv7dLHY/0H &$tCy1 1F&(+r&ctdggdg}tgdggd}||z}ttjddgggd}tj||y)Nrr0)rrru)r/r0r0rmr)rrrrs r$"test_arith_reindex_with_duplicatesrsV 1# 3C ).K LC 3YF2661a.)3PQH&(+r&to_addrc tddgddgd}dt|dd}tjt|5||zdddtjt|5||zdddy#1swY3xYw#1swYyxYw)NrrirzUnable to coerce list of rz to Series/DataFramern)rrOrtrurv)rrdrws r$"test_arith_list_of_arraylike_raisers !Qq!f- .B %d6!9o%66J KC z - V  z - sA8)B8BB c~tdgdi}|j}|d}|j}tj|5|dz }ddd|r'|j|usJtj ||y|j|usJtdgdi}tj ||y#1swYhxYw)Nr]rrr)rrC_valuesrbassert_cow_warningrc)using_copy_on_writewarn_copy_on_writerdrrvalsrs r$%test_inplace_arithmetic_series_updaters C# $BggiG WF >>D  1 2! ~~T))) b'*~~%%%c9-. b(+s B33B<c $tdggdgtjdgdggddg}tdggdgtdgd}tdggdgtjdgdggddg}||z }t j ||y) zP Regression test for: https://github.com/pandas-dev/pandas/issues/33765 rrrrrrr N)rr rr rbrc)rrrrs r$ test_arithmetic_multiindex_alignrs  e''!qc 3*E C aSE#qc0D EC e''!qc 3*EH 3YF&(+r&ctdtdtd}|dz}ttjdtdtd}t j ||y)NTrcdrr)rrsr!rrbrc)rdrrs r$test_bool_frame_mult_floatrsO 4dT$Z 0B #XF$t*d4jAH&(+r&ctgd|}tgd|}tgd|}|j|jz }tj||y)N)rriNrAr)rrN)r rto_framerbrc)any_int_ea_dtypeseries1series2rrs r$test_frame_sub_nullable_intrsU\)9:GY&67G-=>H    '"2"2"4 4F&(+r&zMignore:Passing a BlockManager|Passing a SingleBlockManager:DeprecationWarningcGfddtGfddtdgdgdd}|d z}dgd gd d}tj||||z}tj||y) Nc6eZdZefdZefdZy)Etest_frame_op_subclass_nonclass_constructor..SubclassedSeriescSr;rYr>SubclassedSeriess r$ _constructorzRtest_frame_op_subclass_nonclass_constructor..SubclassedSeries._constructor& # #r&cSr;rY)r>SubclassedDataFrames r$_constructor_expanddimz\test_frame_op_subclass_nonclass_constructor..SubclassedSeries._constructor_expanddim*s & &r&N)rVrWrXpropertyrr)rrsr$rr%s)  $  $  '  'r&rcLeZdZdgZdfd ZedZefdZxZS)Htest_frame_op_subclass_nonclass_constructor..SubclassedDataFrame my_extra_datac2||_t||i|yr;)rsuperr?)r>rargskwargs __class__s r$r?zQtest_frame_op_subclass_nonclass_constructor..SubclassedDataFrame.__init__1s!.D  G d -f -r&cTtjt||jSr;) functoolspartialrOrrGs r$rzUtest_frame_op_subclass_nonclass_constructor..SubclassedDataFrame._constructor5s$$T$Z1C1CD Dr&cSr;rYrs r$_constructor_slicedz\test_frame_op_subclass_nonclass_constructor..SubclassedDataFrame._constructor_sliced9rr&rU) rVrWrX _metadatar?rrr  __classcell__)rrs@r$rr.s:$%  .  E  E  $  $r&r some_datarrrrir)rry r)sdfrrrrs @@r$+test_frame_op_subclass_nonclass_constructorrsq '6' $i $ k+K LC 1WF";ik0RSH&(+ 3YF&(+r&ctdd}t|jgdi}t|jgdi}||j||jk(}tgd|j}t j ||y)NColsz col1 col2r)TTTr)rrr$r rbr )rq1q2rrs r$test_enum_column_equalityrFso  $D DIIy) *B DIIy) *B  ]bm +F(tyy9H68,r&cdttdddg}ttdddg}|jjd|_||z}ttdddg}|r+trd}nd}|jj||_t j ||y)Nr.rr)rrr rstringzstring[pyarrow])rrsrrMrrbrc)using_infer_stringrrrrr=s r$test_mixed_col_index_dtyperRs DKc! =C DKc! =C++$$X.CK 3YFe3qcBH %EE#++2259&(+r&)@ collectionsrrrenumrr rrznumpyr!rt pandas.compatrpandas.util._test_decoratorsutil_test_decoratorsrpandasrrr r r pandas._testing_testingrbpandas.core.computationr r1pandas.tests.frame.commonr rfixturer%r6r8r[rrrrrr rrrrrrrsrMrrrrrrrrrrrrrYr&r$r*s   %)) 7  R R aXIx3HIJ   :W;W;t@.@.NN0N0b A0A0H .,-"J 0J 0Z,,"-4 S#Jc +T$Z @)L SzC:.T C J Jf      , ,,VQF^$4vq!f~vqRSf~6V#WXY,&,&,,S!,!,H -,r&