gL i dZddlZddlmZddlZddlZddlZddlZddlm Z ddl m Z m Z m Z ddlZddlmZmZmZmZmZmZmZddlmZddlmZddlmZGd d ZGd d ZGd dZ GddZ!dZ"dZ#ejHjKdedgdfegde&ddfgdZ'dZ(ejHjKddddgejddgejdddggdZ)ejHjKddddgejddgejdddggd Z*ejHjKdd!d"gejd"gejd#d"ggd$Z+ejHjKdd!d"gejd"gejd#d"ggd%Z,d&Z-y)'z test fancy indexing & misc N)datetime) IndexingError)is_float_dtypeis_integer_dtypeis_object_dtype) DataFrameIndexNaTSeries date_rangeoffsetstimedelta_range)_mklbl)gen_objc4eZdZdZdZdZejjddZ ejjddZ dZ d Z d Z ejjd d d gdZdZdZdZdZdZejjdgdedgdZdZdZdZdZdZdZdZdZd Zd!Z d"Z!d#Z"d$Z#d%Z$ejjd&e%jLe%jNgd'Z(y()) TestFancyz"pure get/set item & fancy indexingctttjddtj}tj dtj |d<tj dt|d<d}tjt| 5tjgd |j|jd d df<dddtjgd |j|jd d df<|j|jd d df}tgd gdd}tj ||y#1swYxYw)N dtypeindex foobar@Must have equal len keys and value when setting with an iterablematch)yp= ף@yGz??g@?)r"r#rname)rr nparangeint64zerosfloat64complexpytestraises ValueErrorarraylocrr tmassert_series_equal)selfdfmsgresultexpecteds i/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/pandas/tests/indexing/test_indexing.pytest_setitem_ndarray_1dz!TestFancy.test_setitem_ndarray_1d(sU299Q#3288D EHHRrzz25 HHRw/5 Q ]]:S 1 T+-884R+SBFF288Aa=%' ( T(*xx0N'Orxx!}e#$1 u,- *,U  vx0 T Ts 4EEc~tttjdd}tjdtj |d<tjdt |d<d}tjt| 5tjdd d z|d d dddy#1swYyxYw)Nrrrrrrrrrr%y?r!r") rr r(r)r+r,r-r.r/r0)r5r6r7s r:test_setitem_ndarray_1d_2z#TestFancy.test_setitem_ndarray_1d_2>sU299Q#34 5HHRrzz25 HHRw/5 P ]]:S 1 +ii1o*BqG + + +s B33B<zPignore:Series.__getitem__ treating keys as positions is deprecated:FutureWarningcLt||}||}tjjdj dd}g}|t urF|t jt jfvr$|jd|r|jd|t us|t jur$|jd|r|jd|t jus|t ur#|t jur|jd |tur#|t jur|jd t|tjr#|t jur|jd t|tjtj tj"fr|jd t%|d k(st|tj&r|jd t)|t*urFt|j,tj.s"|jd|jddj1|} t2t4t6f} t9j:| | 5||dddy#1swYyxYw)Nr!r"r!r!r!sizez8Wrong number of dimensions. values.ndim > ndim \[3 > 1\]z$Passed array should be 1-dimensional;Buffer has wrong number of dimensions \(expected 1, got 3\)zindexer should be 1-dimensionalz&Cannot index with multidimensional key Index data must be 1-dimensionalData must be 1-dimensionalrz%positional indexers are out-of-boundszvalues must be a 1D arrayz only handle 1-dimensional arrays|r)rr(random default_rngintegersr r3setitemilocappendr2r isinstancepd IntervalIndexTimedeltaIndex DatetimeIndex PeriodIndexlen MultiIndextyper _valuesndarrayjoin IndexErrorr0NotImplementedErrorr.r/) r5rframe_or_series indexer_sliusing_array_managerobjidxrnd3msgsr7potential_errorss r:test_getitem_ndarray_3dz!TestFancy.test_getitem_ndarray_3dJsou-3ii##A&// /B f $RWW8M)M KKS T" BC f $ rww(> KKV W" => "&& v %+*C KK@ A i 'K2::,E KK: ; eR-- .;"''3I KK: ; eb//1A1A2>>R S KK4 5 u:?j > KK? @ ;%  5=="**(M KK3 4 KK: ;hhtn& 4GH ]]+3 7  I   s JJ#zPignore:Series.__setitem__ treating keys as positions is deprecated:FutureWarningcpt||}||}tjjdj dd}|t j urt}d|j}nt}djgd}tj||5d ||<dddy#1swYyxYw) Nr!r"r?r@zCannot set values with ndim > rE)rBzCannot set values with ndim > 1rCrDz,Array conditional must be same shape as selfrr) rr(rFrGrHr3rJr0ndimrWr.r/) r5rrZr[r]r^r_errr7s r:test_setitem_ndarray_3dz!TestFancy.test_setitem_ndarray_3dxs ou-3ii##A&// /B "'' !C2388*=CC((C]]3c * DI   s B,,B5ctjd}tddgddgg}||}tddgd}t j ||tddg}||}|dk(sJy)Nrrr!r$r%r')r(r1rr r3r4)r5keyr6r8r9sers r:test_getitem_ndarray_0dz!TestFancy.test_getitem_ndarray_0dsuhhqkAA' (C1a&q) vx0aVnS{{cxtdg}d|jd<d|jd<d|jtj<|jtjdfdk(sJ|j}t ddtjgtj }tj||y)Nrcolumnsrr!r$r) rr2r(infrr r,r3assert_index_equalr5r6r8r9s r:test_inf_upcastzTestFancy.test_inf_upcasts s #q q rvvvvbffai A%%%!Qrzz: fh/rlctddidddg}tj|d<|djtjk(sJt j td5d |jd <dddtddgtjdgtd tjgt d }t j||y#1swY`xYw) Narr$r!rubcitem of incompatible dtyperr)rrxrrurwrx) rr(nanrr,r3assert_produces_warning FutureWarningr2r objectassert_frame_equalr5r6r9s r:test_setitem_dtype_upcastz#TestFancy.test_setitem_dtype_upcasts a"23 4&&3#w}} ***  ' ' !=  ##BFF6N #a&{v1V W  b(+ # #s #CCvalgQ @wxyzcttjddjddt dgd}|j }t jtd 5||jd <dddtd |dggd gt dgd}t j||t|dsJt|dsJy#1swY`xYw)Nr#r*rr!r$abrrbazrroryrrurrr$r%r"rr) rr(r)reshapelistcopyr3r|r}r2rr)r5rr6leftrights r:test_setitem_dtype_upcast2z$TestFancy.test_setitem_dtype_upcast2s IIaw ' / /1 5t*)  wwy  ' ' !=  '$'DHHZ  'a[) $t*)  dE*U ,,,U ,,, ' 's *CC"cttjddjdddz t dgd }t j td 5d |jd <dddtgdgdgt dgd }t j||t|dsJt|dsJy#1swY_xYw)Nr#r*rr!r$g$@rrrryrrr)rrg?)g333333?g?g?rr) rr(r)rrr3r|r}r2rr)r5rrs r:test_setitem_dtype_upcast3z$TestFancy.test_setitem_dtype_upcast3s IIaw ' / /1 5 <t*)   ' ' !=  *$*DHHZ  *   /t*)  dE*d5k***d5k*** * *s C  Ccttjdgd}|ddgj}t gd}t j ||y)Nr$)rururwrnrwru)rwruru)rr(eyeror r3rqrrs r:test_dups_fancy_indexingz"TestFancy.test_dups_fancy_indexingsErvvay/ :S#J'') fh/rlctgdgtd}tgdg}td|_|jdddf|jdddft j ||y)N)rr!r @@rraaaaaaarnr%)rrrorJr3rr5r6r8s r:&test_dups_fancy_indexing_across_dtypesz0TestFancy.test_dups_fancy_indexing_across_dtypess` ; :  FF4L . ]]8> :  FF4L        s.C>%D >D Dcttjjdj dt d}|rHt jttjd5|jdgdddyt jttjd5|jdgdddy#1swYyxYw#1swYyxYw) Nr!)r"r$AABCDrz8"None of [Index(['E'], dtype='str')] are in the [index]"rrz;"None of [Index(['E'], dtype='object')] are in the [index]") rr(rFrGstandard_normalrr.r/rreescaper2)r5using_infer_stringdfnus r:+test_dups_fancy_indexing_only_missing_labelz5TestFancy.test_dups_fancy_indexing_only_missing_labels II ! !! $ 4 4V :  FF9    s AA ctdgdigd}tjtd5|jgddddy#1swYyxYw)Nrrrrrr)rrrrr5r6s r:#test_dups_fancy_indexing_non_uniquez-TestFancy.test_dups_fancy_indexing_non_unique;sC  .6J K ]]8> : $ FF? # $ $ $s A  Acttjjdj dgd}t j td5|jddgdfdddy#1swYyxYw)Nr!)r"r")rrrrrrnrr)rrr) rr(rFrGrr.r/rr2rs r:test_dups_fancy_indexing2z#TestFancy.test_dups_fancy_indexing2Asd II ! !! $ 4 4V <-  ]]8> : ' FF1o% & ' ' 's A66A?cttjjdj dgdddg}|j dd}|j d dg}tj|||}|j ddddgf}tj|||j ddddf}|j d dgddgf}tj||y) Nr!)rr!) rrrr!r!r!r$r$r$rurwrrr#r) rr(rFrGrrJr2r3r)r5r6r9r8s r:test_dups_fancy_indexing3z#TestFancy.test_dups_fancy_indexing3Ls  II ! !! $ 4 4V <-#J  771Q<A fh/C: & fh/771Q36?Ac *+ fh/rlcttdgd}ttdddg}||dg}tj||y)Nr$rrr$rr!r)r ranger3r4)r5 indexer_slrjr9r8s r:test_duplicate_int_indexingz%TestFancy.test_duplicate_int_indexing`sFU1XY/%(1a&1C!% vx0rlctddddddddd}|d jd |d <|d d k(}|j|d fjd }||j|d f<|jddk(sJy)Naaabbbcccrr!r$oiMrvruc|dk(rdS|S)Nr_xs r:z9TestFancy.test_indexing_mixed_frame_bug..nsAJSArlrrc|dk(rdS|S)Nr-----rrs r:rz9TestFancy.test_indexing_mixed_frame_bug..rs1:1rl)rr!r)rapplyr2rJ)r5r6idxtemps r:test_indexing_mixed_frame_bugz'TestFancy.test_indexing_mixed_frame_buggs %03337O P  W]]#EF6 jCvvc3h%%&LM"sF{wwt}'''rlc^ttjjdjddggdz}t j t tjd5|gdddd|d jd |jd k(sJy#1swY.xYw) Nr!)rr"ru)rnz'[26, -8] not in index'r)rirr) rr(rFrGr.r/rrrshapers r: test_multitype_list_index_accessz*TestFancy.test_multitype_list_index_accessvs  II ! !! $ + +G 4seFV>V ]]82995N+O P  |  "v||A"((1+---  s .B##B,crtddddddddddd d d dd d d d tjtjtjtjdtjtjtjtjtjtjd ddddddddddddddd dddd d d d d d d d d d d!d d"j}|j gd#jj |j $}tj||y)%NnonQC10111213243546474859)rrrrrr goe@gm77M @g,o@g2?g@?gufn?g%?g TFa?gI"?g+ew7?gy7Z?g9?gGp#?gC?iiiii)PRuidQCdatayear)rrrrn) rr(r{ reset_index set_indexreindexror3rrs r:test_set_index_nanzTestFancy.test_set_index_nansL  ""+**++*++++++++"cA C F +-G L LL0 1 [] WRZZW (  fb)rlc tgdgdttdttdddjddi}tj |j d<|j}|jj}d dg}|d z}tj |j d <td tj d d d d ggdtgddddtj ddgd}|j||f|j||f<tj||td tj d d d d ggdgddddtj ddgd}|j}|j||fj|j||f<tj||y)N)rurwrurwrurw)rrrrrrr# )FCPFcol1col2r r,rrr r!)r$r$rurw)rrr%r#rrrrr)rrrastyper(r{rJrrisnar r2r3rvalues)r5r6df2maskcolsdftr9s r:test_multi_assignzTestFancy.test_multi_assigns4(U1XU1b\*    &&)$ %  ggi ~AgBFFCc37(23QBFFB3   "ggdDj1d  c8, BFFCc37(+QBFFB3   ggi!ggdDj188d  c8,rlctgdgdgdgdd}|j}|ddk(}dD]}|d|j||f<|dj|j|ddk(dd gf<tj||y) N)rr!rrr)rrrrr)r$r%r"r#r)rrrDrrrrrr)rrr2r3r)r5r6r9rcols r:"test_multi_assign_broadcasting_rhsz,TestFancy.test_multi_assign_broadcasting_rhss $&&$   779}! .C&(gHLLs # .,.c7<<>r#w!|c3Z'( b(+rlctddgdg}gd|jd<ddg|jd<tddgdg}ddg|jd<tj||y)Nrrrrr r!)rrJr3rrs r:test_setitem_listzTestFancy.test_setitem_listshaVaS 1! A !Q!5F D fb)rlc|tdgttjdgt}|j j sJtjtd5|ddddtjtd5|jddddy#1swY=xYw#1swYyxYw)Nrz 2011-01-01r'2011'r2011rr) rr rM Timestampr~r _is_all_datesr.r/rr2rs r:test_string_slicezTestFancy.test_string_slicessE2<< #=">fM Nxx%%%% ]]88 4  vJ ]]88 4  FF9       s$B& B2&B/2B;c0t}|jjrJtjt d5|ddddtjt d5|j ddddy#1swY=xYw#1swYyxYw)Nrrrz^0$r)rrr!r.r/rr2rs r:test_string_slice_emptyz!TestFancy.test_string_slice_empty)s~[88)))) ]]88 4  vJ ]]85 1  FF9       sB'B B  Bctgdgtd}|tdjt|td<|j }|j ddddfjt j|j ddddf<tgdgtd}|tdjt|td<|d jt|d <|d jt|d <tj|||j }|jddd fjt j|jddd f<tgd gtd}|tdjt|td<tj|||j }|jddd d gfjt j|jddd d gf<tgd gtd}|tdjt|td<tj||y)N)123.4r"@rABCDEFGrnABCDGrr!)rr!r(r)r"r*rCDGrr)rr'r(r)r"r*rr)r&r!r$r)r"r*r) rrr r~rrJr(r*r3rr2)r5rdf_origr6r9s r:test_astype_assignmentz TestFancy.test_astype_assignment4s 1 2DO ")g!7!>!>v!FW  \\^''!QqS&/00:1Q3 - .Y !)e 5 < ? $ '0(1 (.L*\.-`,& *  $,L , [[YRVV(<=)->)-rlrc 2eZdZdZdZdZdZejjde dde jddze jdd dgd Zd Zd Zd ZejjdgddZdZdZdZy)TestMiscc8ttjjdjdtjjdjdd}d|d<t|d|ddgdzd}t j ||y)Nr!r)rr rurr )rr ru)rr(rFrGr3rrs r:test_float_index_to_mixedz"TestMisc.test_float_index_to_mixeds YY**1-44R8YY**1-44R8   32c7CtbyIJ h+rlctgdgddgd}d|j|jdd<tgdgd d|j}tj||y) Nrrrvr rrrrr!r)rrr")rr2rr3rrs r:&test_float_index_non_scalar_assignmentz/TestMisc.test_float_index_non_scalar_assignmentsP YY7 O rxx|99=RXXN h+rlctgdgddgd}|j}|j|j|j|j<t j ||y)NrrrvrGr)rrr2rr3rr5r6rs r: test_loc_setitem_fullindex_viewsz)TestMisc.test_loc_setitem_fullindex_viewssM YY7 Oggi66"((+rxx b#&rlcd}tjdjdd}gd}t||t dd}d |j d d d d d d fz}|j }gd|d<gd|d<|j dddd fxxd zcc<|j }|j dddd fxxd zcc<|||||||||fD]1} | djd| d<| djd| d<3gd|d<gd|d<|rTtjtd5tjtd5|||||ddddddytjtd5|||||dddy#1swYBxYw#1swYyxYw#1swYyxYw)Nctdgdtdd}}}ddgddgtdd} }}|j} || j||f<t j | ||j} || j ||f<t j | ||j} || j || f<t j | |y) Nbcdrrr%joejolier!r$)rslicerr2r3rrJ) r6rhs right_loc right_iloclbl_oneidx_one slice_onelbl_twoidx_two slice_twors r: run_testsz.TestMisc.test_rhs_alignment..run_testss*.u+y%1+iWG+0'*:QFE!QKiWG779D),DHHWg% &  ! !$ 2779D*-DIIgw& '  ! !$ 3779D.1DIIi* +  ! !$ 3rlrr"r%)jimrOrPjolineabcder*)rorrr$rr!)rrr#rrO)r! rr"rrPrr,c d|S)N@rrs r:rz-TestMisc.test_rhs_alignment..s AaS'rl)r z@-28z@-20z@-12g1@)z@2g:g2g$z@18z Invalid valuerzincompatible dtype)r(r)rrrrJrr mapr.r/ TypeErrorr3r|r}) r5rr[xsrr6rRrTrSframes r:test_rhs_alignmentzTestMisc.test_rhs_alignments 4"YYr] " "1a (0 r4tG}G L2771Qr61Qr6>**WWY . 5/ 7!QqS!R'!GGI qsAaCx B&  "c9j1#y*5 CE <..y9E%L"7^//0ABE'N C@ 5@ 7 y@ >//!)=>b#y*=> > > ++MAUV :"c9j9 : : >> > >  : :s07F, F F, F8 F) %F,,F58Grrrdctj}t|}tt j d|}t j|||ddd|dddt j||d|dd|dddt j|||d|dd|dddt j|||d|dd|ddy)Nrrrr`rrbr)rM IndexSlicer r r(r)r3!assert_indexing_slices_equivalent)r5rSLCrjs r:)test_str_label_slicing_with_negative_stepz2TestMisc.test_str_label_slicing_with_negative_stepsmmCjRYYr]#. ,,S#c!fll2CSBZP ,,S#mAm2Dc%1R%jQ ,, SWs1v*+SAb\  ,,S#c!fs2w6K2LcRTSTgVrlc|tjt||}tjt d5||ddddddy#1swYyxYw)Nrzslice step cannot be zerorr)r(r)rRr.r/r0)r5rrrZr]s r: test_slice_with_zero_step_raisesz)TestMisc.test_slice_with_zero_step_raisessObiiE 35A ]]:-H I ! sOCaC  ! ! !s AA#cztgdd}tgdgdd|}|j}dd d}||jd <dd g|jd <t j ||tgdgd d|}||jd <tgd gd d|}t j ||y)N)rr"zrh)rr!r#)r!r!r)ryrrcr")rrg @)rr!r)rrgX@)r rrr2r3r)r5rr6r9rRs r:8test_loc_setitem_indexing_assignment_dict_already_existszATestMisc.test_loc_setitem_indexing_assignment_dict_already_existssjs+ YY7u E779Bq b' Q b(+Y_=U Kq 93CDER b(+rlctgdgdd}|jgddf}|jdddfjtj k(sJt j|jdddf|jdddfy)Nr)rwb2b3rvrur)rrJr2rr(r*r3r4rJs r:*test_iloc_getitem_indexing_dtypes_on_emptyz3TestMisc.test_iloc_getitem_indexing_dtypes_on_emptyso Y->? @ggb!enwwq#v$$000 swwq#vA?rlrA)r"i?Bi@Bctt|tj}d|jtd<t j |jtdtddgd|jtd<t j |jtdtd ddgy) N)rr*rgE@rr+r!gE@)r rr(r,r2r3r4)r5rAr6s r:!test_loc_range_in_series_indexingz*TestMisc.test_loc_range_in_series_indexings tBJJ 7eAh quuU1XtA30GHeAh quuU1XtAq60JKrlcttjdjddt dt d}tdt dt d}||j }ttj d d tjgd d tjgtjgdzgt dt d}tj||y) Ng"@r$rABCrrrABrr rr) rr(r)rrnotnullr1r{r3r)r5r6index_dfr8r9s r:#test_partial_boolean_frame_indexingz,TestMisc.test_partial_boolean_frame_indexings  IIcN " "1a (U T%[ Qd4j$t*EH$$&' HHsC(3RVV*r@rr(r)linspacerprrrxr|rrrrrrlr:rCrCs ,, ' 4:l [[ sB2!4kbkk#sB6OP W W! ,@ [[V%9:L;L 0 #rlrCc eZdZgdejddgefgdejddgdfedd d edd dedd dgeedd dedd dgdfgd gd dfgZe jjd ed Z e jjd edZ e jjd edZdZy)TestDataframeNoneCoercionrr!r$rGrrNrr)Nrrr9c|\}}}td|i}d|jddgf<td|i}tj||yNrrrr2r3rr5r9 start_dataexpected_resultwarnstart_dataframeexpected_dataframes r:test_coercion_with_locz0TestDataframeNoneCoercion.test_coercion_with_locKsT,4) OT#UJ$78*.AwJ'&'?@ o/ABrlc|\}}}td|i}d||d|ddk(<td|i}tj||yr)rr3rrs r:(test_coercion_with_setitem_and_dataframezBTestDataframeNoneCoercion.test_coercion_with_setitem_and_dataframeUs\,4) OT#UJ$78OS./%2H2KKL&'?@ o/ABrlc|\}}}td|i}d|j|d|ddk(<td|i}tj||yrrrs r:$test_none_coercion_loc_and_dataframez>TestDataframeNoneCoercion.test_none_coercion_loc_and_dataframe_sb,4) OT#UJ$78SWOE2oe6LQ6OOP&'?@ o/ABrlc dtgdgdtdddtdddtdddggdd}d|jd <ttjddgtjd d gt tdddtdddggd d}t j||y) NrrGrrr!r$rz)rurwrxrrrr)Nrwrx)rrrJr(r{r r3r)r5rr8s r:test_none_coercion_mixed_dtypesz9TestDataframeNoneCoercion.test_none_coercion_mixed_dtypesis#$tQ*HT1a,@(4QRTUBVW$   #'Qffa^ffc3'8D!Q/$11EF%    os3rl)r:r;r<r(r{r}rr EXPECTED_SINGLE_ROW_RESULTSr.r>r@rrrrrrlr:rr<s  RVVQNM2 2663,d3dAq !8D!Q#7$19M N (4A&q!(< =   4d; # [[Z)DECFC [[Z)DECFC [[Z)DECFC4rlrc eZdZdZej j deeje je je gej j dddge ddejgdgdZej j d d ej d gd Zej j deeje je je gej j dddge ddejgdgd Zy)TestDatetimelikeCoercioncR|}tdd|}t|jd}|j}d}|j ||||d<|@|j |j k(sJ|jj |j usJy|j|usJy)N 2016-01-01r$periodstzTdeepz 2018-01-01r)r r rrU_validate_setitem_valuer_ndarray)r5tz_naive_fixturer[rdtirjrnewvals r:test_setitem_dt64_string_scalarz8TestDatetimelikeCoercion.test_setitem_dt64_string_scalars qR8SXX4X()&&v.$ C :99 ) ));;''6??: ::;;&( ((rlboxrirrr!)TTFc|}t|tr|tjur tdd}t dd|}t |j d}|j}|dd g} |j| | |||<|@|j|jk(sJ|jj|jusJy|j|usJy) Nrrrr$rTrz 2019-01-01z 2010-01-02) rLrQr3r2r r rrUrrr) r5rr[rirrrrjrnewvalss r:test_setitem_dt64_string_valuesz8TestDatetimelikeCoercion.test_setitem_dt64_string_valuess  c5 !kRVV&;1+CqR8SXX4X()|\23&&w/ ' C :99 ) ));;''6??: ::;;&( ((rlscalarz3 Daysr%ctdd}t|jd}|j}|j ||||d<|jj |j usJy)N1 Dayr$rTrr)rr rrUrr)r5r[rtdirjrs r:test_setitem_td64_scalarz1TestDatetimelikeCoercion.test_setitem_td64_scalarsggq1SXX4X()&&v.$ C{{##v666rlc\t|tr|tjur tdd}t dd}t |j d}|j}|dd g}|j|||||<|jj|jusJy) Nrrrr$rTrz10 Daysz44 hours) rLrQr3r2rr rrUrr)r5r[rirrrjrrs r:test_setitem_td64_string_valuesz8TestDatetimelikeCoercion.test_setitem_td64_string_valuess c5 !kRVV&;1+Cgq1SXX4X()y*-.&&w/ ' C{{##v666rlN)r:r;r<rr.r>r@rr(r1rM Categoricalr rQrr Hourrrrrlr:rrs()* [[UT288RXXr~~u$UV [[ Aa XRXX.A%BC)W)0 [[X,',,q/'BC 7D 7 [[UT288RXXr~~u$UV [[ Aa XRXX.A%BC7W7rlrcfttjddgdtjddgdddd g }ttjddgdd d gd }|jd}t j |||jd}t j ||y)Nrr!Int64rr$r%rrurwrrrr&r)rrMr1r r2r3r4rJ)r6r9r8s r:"test_extension_array_cross_sectionrs 1a&01a&0 Cj  Bbhh1vW5c3ZcRH VVC[F68, WWQZF68,rlc|ttjddgdtjddgddddg }|jd}t ddgdd d gd }t j|||jd }t j||ttjddgdtjddgdddg }|jd}t ddgtd d gd }t j|||jd }t j||y)Nrr!rrr*rrurwrrr)rrr'r) rrMr1r(r2r r3r4rJr~)r6r8r9s r:+test_extension_array_cross_section_convertsrs 1a&01a&0 Cj  BVVC[Fq!fGC:CHH68, WWQZF68, hh1vW-BHHc3Z4HICj BVVC[Fq#hfS#JSIH68, WWQZF68,rlz ser, keysr)rrrrr)rrc&td}}tj||5|||ddd|tjk(rt d}}tj||5d|||<dddy#1swYPxYw#1swYyxYw)NzToo many indexersrztoo many indices for arrayr)rr.r/r3rJrX)rjkeys indexer_liexp_errexp_msgs r:'test_ser_tup_indexer_exceeds_dimensionsrs %&9WG wg .3RWW%'C wg ." ! 3""""s A;& B;BBctdg}||ddg}tddgtddg}tj||y)Nrrr)r r r3r4)rrjresr8s r:(test_ser_list_indexer_exceeds_dimensionsrsI ",C S/1a& !C "b1v /C3$rlvaluerrwctdgdi}djddg}tjt|5||j d<dddtgdt jgdt d }tjtd5||j d<dddy#1swYhxYw#1swYyxYw) NrrrEz"Must have equal len keys and value(setting an array element with a sequencerrrrrurwrr) rrWr.r/r0r2r(r1r~)rr6r7s r:%test_scalar_setitem_with_nested_valuers C# $B (( 0 6  C z -v -v)NO PB z)M Nv  sB5C5B>C crtgd}tjtd5|||d<dddtgdt}|||d<t |t jr#|jd|k(jsJy|jd|k(sJy#1swYyxYw)Nrrrrrr) r r.r/r0r~rLr(rVr2all)rr[rjs r:,test_scalar_setitem_series_with_nested_valuer6s  C z)S T$# C$ f -CKQ%$ e#((***wwqzU"""$$s B--B6)rrrctdgdi}||jd<tgddtjtjgd}t j ||tgdtj gdtd}||jd<t|tjr#|jd|k(jsJy|jd|k(sJy)Nrrrrrrr) rr2r(r{r3rr1r~rLrVr)rr6r9s r:-test_scalar_setitem_with_nested_value_length1rHs C# $BBFF6NyRVVRVV/DEFH"h' -v)NO PBBFF6N%$v%',,...vvf~&&&rlctgd}t|tjr/|||d<tgd}t j ||n/t jtd5|||d<dddtgdt}|||d<t|tjr#|jd|k(jsJy|jd|k(sJy#1swYyxYw)NrGr)rrrrrrr) r rLr(rVr3r4r.r/r0r~r2r)rr[rjr9s r:4test_scalar_setitem_series_with_nested_value_length1r]s  !C%$# C/* sH- ]] H  (#(K Q  ( f -CKQ%$ e#((***wwqzU""" ( (s 2 C66C?ctdddg}t|d<t|jd<tj|jdttj|jdttdddg}t|jd<tj|jdty)NOrurw)rrr)r r2r3r4rJ)s1r7s r:+test_object_dtype_series_set_series_elementrxs c#s ,BhBsG(BFF3K266#;1266#;1 c#s ,BBGGAJ2771:vx0rl).r=r1rrrnumpyr(r. pandas.errorsrpandas.core.dtypes.commonrrrpandasrMrr r r r r rpandas._testing_testingr3pandas.tests.indexing.commonr!pandas.tests.indexing.test_floatsrrrCrrrrr>r@rrrrrrrrrrlr:rs"   ' /5 m -m -`d#d#N@4@4FP7P7f-"-: bTlFfYd5kBFKL "  "% fq!fhbhh1v.  C!Q0H I. fq!fhbhh1v.  C!Q0H I # # fseXRXXse_kekk#u.E F''$ fseXRXXse_kekk#u.E F##0 1rl