gL i <dZddlmZddlZddlmZmZmZmZddlZddl m Z ddl Z ddl Z ddlmZddlmZddlmZdd lmZddlmcmZddlZdd lmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.ddl/m0Z1dd l2m3Z3dd l4m5Z5dd l6m7Z7e jpjsde jte jtdgde jte jtggdge jtdgde jtgddgggdZ;GddZ<GddZ=GddZ>GddZ?GddZ@Gdd ZAGd!d"ZBGd#d$ZCGd%d&ZDGd'd(ZEGd)d*ZFe jpjsd+gd,d-d.gdd/gfgd0d-d.gdd/gfgd0d-d1gdd2gfgd3ZGd4ZHd5ZId6ZJd7ZKd8ZLd9ZMd:ZNd;ZOd<ZPd=ZQd>ZRe jpjsd?d@e jd@e jd@gdAZUe jpjsdBe*dCdDfe*dEdFGdHfgdIZVdJZWe jpjdKe jpjdLdMZYdNZZGdOdPZ[y)Qz$ test label based indexing with loc ) namedtupleN)datedatetimetime timedelta)gettz)using_string_dtypeindex)np_version_gt2) IndexingError) CategoricalCategoricalDtypeCategoricalIndex DataFrame DatetimeIndexIndex IndexSlice MultiIndexPeriod PeriodIndexSeries SparseDtype Timedelta Timestamp date_rangetimedelta_range to_datetime to_timedelta) is_scalar)_one_ellipsis_message)"check_indexing_smoketest_or_raisesz series, new_series, expected_serba)FTTFTctd|i}||jdddf<td|i}tj|j |tj|j |yNA)rloctmassert_frame_equalisnanotna)series new_series expected_serdfexpecteds d/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/pandas/tests/indexing/test_loc.pytest_not_change_nan_locr32s` C= !BBFF1c6N#|,-H"'')X."((*xi0c eZdZdZej j dddgdZej j dddgdZej j ddgd d gdd gd gd gd d gd dgd gd dgdgd d gdggej j dddgdZ ej j dgdgdggdgdggej j dddgdZ ej j dgddgd ggdgddggdgddggddgdggej j dddgdZ ej j ddd gej j dddgd!Z d"Z ej j dddgd#Zej j d$edd%gd&d eged'd(dgdeged)d*d+gdeged)d*d+gdeged)d,d)d+gdeggej j dddgd-Zd.Zd/Zej j d0edgged1g2ed d d1fedgd13fedged1g4d1dgfedgged1g4d1edgd13fgd5Zy )6TestLocctgddgt}|jdJtgddgd}|r%tj|jdsJy|jdJy)N)12Nr$columnsdtype)r$str)robjectr(npisnan)selfusing_infer_stringr0s r2"test_none_values_on_string_columnsz*TestLoc.test_none_values_on_string_columnsCsi '#f Evvf~%%% '#e D 88BFF6N+ ++66&>) ))r4kindr-framecR|j|d}t|ddty)N_labelsr(r=failsgetfixturevaluer"KeyErrorrBrErequestobjs r2test_loc_getitem_intzTestLoc.test_loc_getitem_intNs)%%g&67*3qIr4cR|j|d}t|ddty)N_emptyr(crIrKrNs r2test_loc_getitem_labelzTestLoc.test_loc_getitem_labelTs(%%fo6*3s(Kr4zkey, typs, axesf)intsuintslabelsmixedtsNfloats)rWrXrZrYr[rcf|D],}|j|d|}t|d||t.yN_r(axesrJrKrBkeytypsrbrErOtyprPs r2#test_loc_getitem_label_out_of_rangez+TestLoc.test_loc_getitem_label_out_of_rangeZs@ C))TF!C5/:C .UCd(  r4z key, typsrr=)rWrXr\)ri@r'cd|D]+}|j|d|}t|d|t-y)Nr`r(rIrK)rBrdrerErOrfrPs r2test_loc_getitem_label_listz#TestLoc.test_loc_getitem_label_listns= PC))TF!C5/:C .sE3h O Pr4empty)rr= )ri))riro)ri)r=multicf|D],}|j|d|}t|d||t.yr_rKrcs r2(test_loc_getitem_label_list_with_missingz0TestLoc.test_loc_getitem_label_list_with_missing|s@ C))TF!C5/:C .UCd(  r4rerWrXc\|j|d|}t|dgddty)Nr`r()r](rirarK)rBrerErOrPs r2!test_loc_getitem_label_list_failsz)TestLoc.test_loc_getitem_label_list_failss2%%av&67*  1H r4cyNrBs r2!test_loc_getitem_label_array_likez)TestLoc.test_loc_getitem_label_array_likes r4cZ|j|d}gd}t|d|ty)NrSTFTFr(rI)rLr" IndexError)rBrErOrPr#s r2test_loc_getitem_boolzTestLoc.test_loc_getitem_bools+%%fo6 &*3q Kr4zslc, typs, axes, failsro)rYrZrmr[r\2013010220130104r=rZrrc^|D](}|j|d|}t|d|||*yr_)rLr") rBslcrerbrJrErOrfrPs r2test_loc_getitem_label_slicez$TestLoc.test_loc_getitem_label_slicesD, C))TF!C5/:C .   r4ctddgddgddggddggd}tjd|jd <tddgd dgd dggddggd}t j ||y) Nr]r$col1col2)rnririr;r r=)rirrri)rr@aranger(r)r*rBr0r1s r2 test_setitem_from_duplicate_axisz(TestLoc.test_setitem_from_duplicate_axiss #Yc S#J /V$  IIaLy#YC1c( +ff5EZ  b(+r4c tgddtjdgtdtj tdgd}tdgtdgd }|j |j|d jd d gf<tgdgd tdtdtdgd}tj||y) Nrir=roString 1String 2z2019-06-11 11:00:00z2019-06-11 12:00:00)channelr'B)dataString 3r'rr'r)rrr) rr@nanrpdNaTvaluesr(r+r)r*)rBr0df2r1s r2test_column_types_consistentz$TestLoc.test_column_types_consistents $ "&&*534FF34  "|95J+K*LM .1ZZr#w||~Sz)*$9343434   b(+r4z obj, key, expFr;namer c|j|}t|ttfrt j ||y||k(sJyr|)r( isinstancerrr) assert_equal)rBrPrdexpress r2#test_loc_getitem_single_boolean_argz+TestLoc.test_loc_getitem_single_boolean_args:ggcl cIv. / OOC %#: :r4)__name__ __module__ __qualname__rDpytestmark parametrizerQrUrgrlrvrzrrslice TypeErrorrMrrrrrrrrr}r4r2r6r6Bs * [[Vh%89J:J  [[Vh%89L:L  [[  [[A3%w81e8$s'  QCueW~ .s ; uE5'N 3UFA3UZ,d?Z-d@Z.ej jOd8dAZ/ej j dBdCge0dDdCdDej6dCggej j dEdFgej6dFggdGZ1ej j dHd)dIggdJfdKe2gdLdMdNejfejfggdLggdJOfdPdQdRgfdKdSge2dPdIejfejfggdTdUdVejfejfgggdJOfdPgdWfej6gdXej8*e2dPdIejfggdXdUdVejfgggdWOfe0dPdMdDgdYfgdXgdZge2dPdIejfejfggd[gd\ggdJOfe0dPdMdDgd]fej6gdXgdZgej8*e2dPdIejfejfggd^gd_ggdJOfe0dDdDdDdCdQgfe2dKdSgd`dagdbdcggdCdQgOe2gddgdegdfggdWOfgdgZ4dhZ5diZ6djZ7dkZ8dlZ9ej jtej j dmdndogdpdqggdrZ;dsZdvZ?dwZ@dxZAdyZBdzZCd{ZDd|ZEd}ZFd~ZGdZHdZIdZJej j dgdej j dej8ejeLgdZMdZNdZOdZPej j dBddgdZQej j deRej6eSeTgdZUdZVej j ddeSdPgfdeSdIgfgdZWdZXej j dd)gdfe0dNdSgdfdMdUggdfgdZYdZZdZ[dZ\ej j dddggdZ]dZ^ej j ddCdCgdCe0dDfgdZ_dZ`dZadZbdZcdZdej j dddgdZedZfdZgdZhdZidZjdZkyD)TestLocBaseIndependentcttdd}|jtjdgdd}|j dd}t j||y)N20212022r z2021/6/1r)rrr(r@arrayilocr)r*rBr0resultr1s r2test_loc_npstrz%TestLocBaseIndependent.test_loc_npstrsS Z7 8*.q1347734= fh/r4zmsg, keyz'Period\('2019', 'Y-DEC'\), 'foo', 'bar'foobarz&Period\('2019', 'Y-DEC'\), 'y1', 'bar'y1z&Period\('2019', 'Y-DEC'\), 'foo', 'z1'z1z;Period\('2018', 'Y-DEC'\), Period\('2016', 'Y-DEC'\), 'bar'z&Period\('2018', 'Y-DEC'\), 'foo', 'y1'z;Period\('2017', 'Y-DEC'\), 'foo', Period\('2015', 'Y-DEC'\)z&Period\('2017', 'Y-DEC'\), 'z1', 'bar'c :ttdddgtdtddgtddtd ggd gd d jgd }tjt |5|j |dddy#1swYyxYw)a( parse_datetime_string_with_reso return parameter if type not matched. PeriodIndex.get_loc takes returned value from parse_datetime_string_with_reso as a tuple. If first argument is Period and a tuple has 3 items, process go on not raise exception rx1x2rrrrrrrrnr]rx)r'rCV1V2r'rrmatchN)rr set_indexrraisesrMr()rBmsgrdr0s r2;test_contains_raise_error_if_period_index_is_in_multi_indexzRTestLocBaseIndependent.test_contains_raise_error_if_period_index_is_in_multi_indexs6TlD$/TlF4L$7TlD&,7"    )O $ ]]83 /  FF3K   s 8BBctddgi}tjtd5|jdddfdddy#1swYyxYw)Nr$riuאr)rrrrMr(rBr0s r2$test_loc_getitem_missing_unicode_keyz;TestLocBaseIndependent.test_loc_getitem_missing_unicode_key5sE aSz " ]]88 4 FF1h;    s AAc Dttjjdjdt dDcgc] }d|dz c}}|j d}|j dddfj d }t j||ycc}w) Nr=)r]rsr]ABCDErsr )r'rrr')rr@random default_rngranger(r)assert_series_equal)rBxr0r1rs r2test_loc_getitem_dupsz,TestLocBaseIndependent.test_loc_getitem_dups:s II ! !! $ + +G 4+095a71q5>5 66&>1!!#& vx0 6sB c tddddtdgggddg}tddddtdggdd}|jd }t j |||j d}t j ||y) Nrir=rr20130101r$r$r$r$r$rr rr)rrrrr)rr(rBr0r1rs r2test_loc_getitem_dups2z-TestLocBaseIndependent.test_loc_getitem_dups2EsE5)J"7 8 9-#   5%:!6 7+   vx0 vx0r4c ttdtdtjdddzdztjdddzdzd j d }d d d gf}|j }|j |xxdzcc<tj|j |d|j |zd}|j }|j |xxdzcc<|j |d|j |zk(sJdd d gf}|j }|j |xxdzcc<tj|j |d|j |zy)Nrtttiaaadersfloat64r<gq= ףp?r=g(\տ)merrbar2rrrr@)rrt) rlistr@rrcopyr(r)rr*)rBdf_origindexerr0s r2test_loc_setitem_dupsz,TestLocBaseIndependent.test_loc_setitem_dupsYsW7mG}yy)4t;a? !95=A    )D/   FO \\^ w3 rvvggkk'6J0JK \\^ w3vvg# G(<"<<<<  FO \\^ w3 bffWosW[[5I/IJr4c.tgdtgddd}|ddk(}|j|dfdz}||j|df<tgdtgd dd}tj||tgdgddd }|ddk(}|j|df}tj t d 5||j|df<dddtgdgddd }tj||y#1swY4xYw) N)rriri)dr,uint32rr$r#r$rir#)ri-uint64item of incompatible dtyper)rrr(r)r*assert_produces_warning FutureWarning)rBdf1ixnewb1r1rnewb2s r2test_loc_setitem_slicez-TestLocBaseIndependent.test_loc_setitem_slice|s if_H.UVW X]C 1$ C&"I J  c8,io>hO X]C   ' ' !=  % %CGGBG  %9?C8T c8,  % %s D  Dc ltdgdgdgdgd}gd}|jdd|fjd|jdd|f<tdgtjdgd tjdgd tjdgd d}t j ||y) Nr'333333?g)idr$r#rTr$r#rTfloat32rr)rr(astyper@rr)r*)rBr0colsr1s r2test_loc_setitem_dtypez-TestLocBaseIndependent.test_loc_setitem_dtypes se3%se4&I J&&D/00;q$weXXse95XXse95XXtfI6    b(+r4c8ttdgd}tjtd5|ddgdddttd}tjtd5|ddgdddy#1swYIxYw#1swYyxYw) Nrorr not in indexrr$dr)rrrrrM)rBss r2$test_getitem_label_list_with_missingz;TestLocBaseIndependent.test_getitem_label_list_with_missings 58? 3]]8> :  sCjM  58  ]]8> :  q!fI       sB3BB Br TFrctgd}dt|dt|}tjt|5|j |dddy#1swYyxYw)Nrz Boolean index has wrong length: z instead of r)rlenrrrr()rBr rrs r2test_loc_getitem_bool_diff_lenz5TestLocBaseIndependent.test_loc_getitem_bool_diff_lensV 9 0U LQQ ]]:S 1  EE%L   s AA%cyr|r}r~s r2test_loc_getitem_int_slicez1TestLocBaseIndependent.test_loc_getitem_int_slices r4cDttjjdjdgdgd}dtjt d}t jt|5|jd dgd dgfdddy#1swYyxYw) Nr=roror)erVgr r;z$\"None of \[Index\(\[1, 2\], dtype=''\)\] are in the \[index\]\"rri) rr@rrr<intrrrMr(rBr0rs r2test_loc_to_failz'TestLocBaseIndependent.test_loc_to_fails  II ! !! $ + +F 3!# 4BHHSM?C" " ]]83 / # FFAq6Aq6> " # # #s 7BBc*tt}d|jd<d|jd<tjt d5|jdddddt jtd }tjt |5|jdd gdddd }tjt |5|jtd gtdddd |jd<tjt d5|jdd gdddd|d<dt jtd }tjt |5|jd gddd|d=tjt |5d|jd g<dddy#1swYsxYw#1swY.xYw#1swYxYw#1swYxYw#1swYsxYw#1swYyxYw)Nrrir=r$z^-1$rz&\"None of \[Index\(\[-1, -2\], dtype='rzE\"None of \[Index\(\['4'\], dtype='object'\)\] are in the \[index\]\"4ror z"\"None of \[Index\(\[-2\], dtype='r) rr?r(rrrMr@r<rrrBrrs r2test_loc_to_fail2z(TestLocBaseIndependent.test_loc_to_fail2s  ac ]]87 3  EE"I 6bhhsm_E" " ]]83 /  EE2r(O W ]]83 / . EE%V, - .b ]]8> :  EE2r(O #1"((3-A" " ]]83 /  EE2$K  cF ]]83 / AEE2$K  9     . .       sH G G G%#G1:G=1H  GG"%G.1G:=H Hcptdgdggddgdg}dtjtd}t j t | 5|jd gddfdddt j t | 5|jd gdddy#1swY>xYw#1swYyxYw) Nr$r#rir=valuerz!\"None of \[Index\(\[3\], dtype='rrro)rr@r<rrrrMr(rs r2test_loc_to_fail3z(TestLocBaseIndependent.test_loc_to_fail3su~aVgY G1#@" " ]]83 /  FFA36N ]]83 /  FFA3K      sB B, B),B5ctgd}|jdgdtjtd}t j ttj|5|jdgdddt j td5|jddgdddy#1swY?xYw#1swYyxYw)Nrr=z"None of [Index([3], dtype='z')] are in the [index]rror ) rr(r@r<rrrrMreescaper#s r2test_loc_getitem_list_with_failz6TestLocBaseIndependent.test_loc_getitem_list_with_fails 9  qc -bhhsm_ :  EE1#J ]]8> :  EE1a&M       s)B8C8CC cttjjdjdgd}|jj d}|j tj|}|j |}tj|||j |j}tj|||j tj|d}tj||y) Nr=)rsrnsize)alpha_0alpha_1alpha_2beta_0beta_1r c d|vS)Nalphar}rs r2z7TestLocBaseIndependent.test_loc_index..s glr4booleanr) rr@rrr mapr(rr)r*rr)rBr0maskr1rs r2test_loc_indexz%TestLocBaseIndependent.test_loc_indexs II ! !! $ + + + 9G  xx||2366"((4.) fh/ $ fh/Y78 fh/r4c,ttjjdjdgdgd}|jddddfj ddddf}|j ddgk(jsJ|jddgk(jsJttdgd gd j d}ttdd gd d gd }tj|||jtk(sJy)Nr=rrrrr'rrDrr'rrrrirr$r#r)rr@rrr(rr;allr rrr)rr<r?rs r2test_loc_generalz'TestLocBaseIndependent.test_loc_general(s  II ! !! $ + +F 3(& 3s7 #((1a03*,11333 c *//111)J"7!8sCDII!L9Z0!4S#JQO vx0||v%%%r4cvttddttdtj dS)N 2000-01-01 2000-01-5rsrrval)rrrrr@int64r~s r2frame_for_consistencyz,TestLocBaseIndependent.frame_for_consistency:s0"<=eAhbhh7   r4rFrr)rrrrrc|ttdtdtjttdtjd}|j }t jtd5||jdddf<dddt j||y#1swY xYw) Nrrsr r<rrEincompatible dtyperr rrrr@rGrr)rrr(r*)rBrHrFr1r0s r2test_loc_setitem_consistencyz3TestLocBaseIndependent.test_loc_setitem_consistencyCsqaAeAhbhh7   # ' ' )  ' ' =Q R $ #BFF1f9  $ b(+ $ $s ?B22B;c^ttdtdttdtjd}|j }t jtd5d|jdddf<dddt j||y#1swY xYw) Nrrsr rrErKrrrLrBrHr1r0s r2(test_loc_setitem_consistency_dt64_to_strz?TestLocBaseIndependent.test_loc_setitem_consistency_dt64_to_strUsuE!H5eAhbhh7   # ' ' )  ' ' =Q R & %BFF1f9  & b(+ & & 0B##B,c^ttdtdttdtjd}|j }t jtd5d|jdddf<dddt j||y#1swY xYw) N?rsr rrErKrrrLrOs r2*test_loc_setitem_consistency_dt64_to_floatzATestLocBaseIndependent.test_loc_setitem_consistency_dt64_to_floatdss%(3eAhbhh7   # ' ' )  ' ' =Q R $ #BFF1f9  $ b(+ $ $rQc$tdttdgi}tjt d5d|j dddf<dddtdtdgi}tj||y#1swY7xYw)Nr20180101rKrstring)rrrr)rrr(r*rs r2'test_loc_setitem_consistency_single_rowz>TestLocBaseIndependent.test_loc_setitem_consistency_single_rowrs~ *(='> ?@ A  ' ' =Q R ) (BFF1f9  )ffhZ&89: b(+ ) )s BBc~tddg}tddg}tjd5d|jdddf<dddtj||tddg}d|d<|dj t j|d<tj||y#1swYnxYw)Nryrri)rr)rr(r*r r@rGrBr1r0s r2"test_loc_setitem_consistency_emptyz9TestLocBaseIndependent.test_loc_setitem_consistency_empty{sc3Z0 Sz *  ' ' - BFF1c6N  b(+Sz *3  ,,RXX6  b(+  s B33B<zTODO(infer_string)reasonc>dgdzgdgdg}tj|gd}gdgdg}tj|d d g}d d d dtjggdd dddtjggdg}t |||}t j }|rtjtd}|5t|jdddf|jdddf<ddd|5t|jdddf|jdddf<ddd|ry|j}|jdddf|jdddfz |jdddf<tjtd5|jdddft!dz |jdddf<dddt#gd|j$d} tj&|d| y#1swYxYw#1swYxYw#1swYVxYw)NRegion_1rr)Site_1raSite_2rb)lBP[l _Zlq=.Zl<.Z)RegionSite RespondentIDnames) RespondentrhrhOtherCatri) Something StartDateEndDatezYes/No SomethingElseLevel_0Level_1r'z5/25/2015 10:59z5/25/2015 11:22Yes)r'z5/21/2015 9:40z5/21/2015 9:52rprpz5/20/2015 8:27z5/20/2015 8:41)r'z5/20/2015 8:33z5/20/2015 9:09rpNor Invalid valuer)rhrk)rhrl)rhDurationrKlXp7)g7@(@g,@gB@r)r from_arraysr@rr contextlib nullcontextrrrrr( infer_objectsr)rrrrr r) rBrClevelsmiclevelsr rr0ctxr1s r2-test_loc_setitem_consistency_slice_column_lenzDTestLocBaseIndependent.test_loc_setitem_consistency_slice_column_lens) L1  4 <   # #F2T U O L %%gi5KL#%6rvv F C "$4eRVV D B   vR 6$$& -- AC  5@q5566BFF111 2  3>q3344BFF1// 0        FF1// 0ffQ334 5 q,,- ' ' =Q R *46FF--5.)5*BFF100 1 *  $BHH;U  r"<=xH?    & * *s$9.G:1.H 1H:HHHunit)YMr?hmrmsusctdtjdtjdgi}|jdddfjj d|d|jdd|f<|jdddfjj d|d|d<t |jdddf|}tj|jdd|f|y)N timestampz2017-02-11 12:41:29z1991-11-07 04:22:37z datetime64[]r1r) rr@ datetime64r(rr rr)r)rBr~r0r1s r2test_loc_assign_non_ns_datetimez6TestLocBaseIndependent.test_loc_assign_non_ns_datetimesMM"78MM"78  &&K077>>TFRS?TUq$w;/66== D6QR>ST:"&&J/d; rvvag9r4ctjdgdi}t|ddd|d<|jdddf|jdddf<|jdd gdf|jdd gdf<tgd gd gd gd ggd}ddg}||j t||<t j ||y)Nr)CO!gs~!g4 8A{?rT)r~cachedate_dt date_dt_cpr=ro)r2017-01-24 13:26:12.711r)r2017-01-24 13:52:05.110r)r2018-10-22 13:44:05.888r)r2018-10-23 08:08:41.025r)rrrr)r from_dictrr(applyr)r*)rBr0r1r;s r2test_loc_modify_datetimez/TestLocBaseIndependent.test_loc_modify_datetimes  Q R $BvJTF9 "$&&I"6q,')vvq!fi.?'@1v|#$UUUU   6 l+$W-33K@ b(+r4z.ignore:Setting a value on a view:FutureWarninghas_refctgddgt}|r|dd}tjgdd|jgddf<t gdgdt }td|i}t j||y) Nrorsrrr'r r;r<rrGrrrrorsr=rorirJ)rfloatr@rr(rr)r*)rBrr0viewserr1s r2#test_loc_setitem_frame_with_reindexz:TestLocBaseIndependent.test_loc_setitem_frame_with_reindexskYU C a5D!#)7!Cy#~Yiu=c3Z( b(+r4c(tgdddgt}d|d<tjgdd|jgd df<t gd gdd }td|j ti}d|d<tj||y) Nrr'rrrWrrGrrrrJ) rrr@rr(rr r)r*)rBr0rr1s r2)test_loc_setitem_frame_with_reindex_mixedz@TestLocBaseIndependent.test_loc_setitem_frame_with_reindex_mixeds{ Yc % H3!#)7!Cy#~Yiw?c3::e#456   b(+r4ctgdddgt}d|d<tjgdd|jt dd d df<tgd dd gd }t j||y)Nrr'rrrWrGrrorr )rjrrSrr )rrr@rr(rr)r*rs r2*test_loc_setitem_frame_with_inverted_slicezATestLocBaseIndependent.test_loc_setitem_frame_with_inverted_slicesf Yc % H3')xx 'IuQ2#$?B)T b(+r4ctdDcgc]}dt|z}}tjdd}tdDcgc]}dt|z}}tjdd}t t |j |}t|}tj|d<|r|dd} ||j|df<tj|d<||j|df<t||tj } t||} t| | d tddg j|} tj|| ycc}wcc}w) Nrs@rGrrrr r'rrJrr)rr>r@rrsetunionrrr(rrrreindexr)r*) rBrikeys1val1keys2val2r r0rseraserbr1s r2test_loc_setitem_empty_framez3TestLocBaseIndependent.test_loc_setitem_empty_frames9(-Qx0!s1v00yy'*',Qx0!s1v00yy'*SZ%%e,- U #&&3 a5D!ucz&&3!uczd%rzz:d%(4d3UC:=NOWWX  b(+/11s E  E%cttjjdj dt dt d}|r|dd}|j d}d|jd<|jd}|dk(sJ|j d}|dk(sJd |jddd d f<|jddd d f}|j ddddf}tj||y) Nr=r=abcdABCDrrrri)r$r'rrr?) rr@rrstandard_normalrrr(r)r*)rBrr0rrr1s r2test_loc_setitem_framez-TestLocBaseIndependent.test_loc_setitem_frame8s II ! !! $ 4 4V <v,L  a5Dx!{{{{q#c'z66!SW*%AB fh/r4c tgdtjd}|j|j|j |jkDdf<tgdtjd}t j||y)Nrrr)rr@rr'r(rr)r*rs r2/test_loc_setitem_frame_nan_int_coercion_invalidzFTestLocBaseIndependent.test_loc_setitem_frame_nan_int_coercion_invalidQs]YRVV4 5#%44rttbdd{C 9266:; b(+r4cLtddgddgddgd}|jdddgf}tddgtddgt td }t j ||td dgd dgddgd}d d g|jdddgf<t j||y) Nrir=rorrr$r#)rir=r$rrr r<rrsrp)rr(rrr?r)rr*rs r2#test_loc_setitem_frame_mixed_labelsz:TestLocBaseIndependent.test_loc_setitem_frame_mixed_labelsYsAq6q!fC:> ?Aq6 " F%Af5V!  vx0!QQF#sDEFq1a&y b(+r4c tgdttdtjd}|j dd}|j dd|_||j ddtgdtgd tjd}tj||ttd d ttd tjd }ttdtdtdtdtdgtgdtjd }|j dd}|j dd |_||j ddtj||y)Nrrbazrorrrir=r)rrr)rir=r=rCrDrsrE200001012000010220000103)rrirrir=rr) rrrr@rGr(r r)r*rr)rBwarn_copy_on_writer0rhsr1s r2 test_loc_setitem_frame_multiplesz7TestLocBaseIndependent.test_loc_setitem_frame_multipleshsC'fU1XRXX.N O ffQqkHHQqM q 'fYbhh.O P  b(+"<=eAhbhh7   j)j)j)j)j) oRXX>   ffQqkHHQqM q  b(+r4rr'Nr&Zctddgddggddgjdti}||jd|f<|jd }t |r|d k(sJy) Nrir=rorrr'rrr)rr'r)rr r?r(r )rBrr&r0rs r2"test_loc_setitem_with_scalar_indexz9TestLocBaseIndependent.test_loc_setitem_with_scalar_indexseAA'#s < C CS&M R"q'z Vs]22]r4zindex,box,expectedr=r>rq)rqrqrqrqrorrrrirr?r)rorrrqrrsrprrqr )rrr?rn )rorqrr)rsrnrr)rr'r?)rrrrqr)rrprnrrrnrr)rqr=r)rrrrn)rrprc~tddgddgddggddg }||j|<tj||y) Nrir=rorrrsrpr'rrrr(r)r*)rBr boxr1r0s r2 test_loc_setitem_missing_columnsz7TestLocBaseIndependent.test_loc_setitem_missing_columnssEjAAA/#s Du  b(+r4cFtdtdjdtjgi}|j }|j dg}tj|j ||j dg}tj|j |y)NrrUTCrri) rr tz_localizerrdtypesrr)rrs r2test_loc_coercionz(TestLocBaseIndependent.test_loc_coercionsy :!6!B!B5!I266 RS T99! v}}h7! v}}h7r4c &tdtdddtdddgi}|j}|jdg}t j |j||jdg}t j |j|y)Nrriir=r)rrrrr)rrs r2test_loc_coercion2z)TestLocBaseIndependent.test_loc_coercion2sw $1!5xa7K LM N99! v}}h7! v}}h7r4ctddgdgdzzi}|j}|jdd}tj|j||jdd}tj|j|y)Ntextz some wordsrrr=ro)rrrr)rrs r2test_loc_coercion3z)TestLocBaseIndependent.test_loc_coercion3sm $! ;< =991 v}}h7 v}}h7r4c tdjdtdjdg}t|tddg}tt}|d||d<|d ||d<t j ||y) N*rirrr rrri)rrrrr?r)r)rB indexer_slvalsr1rs r2test_setitem_new_key_tzz.TestLocBaseIndependent.test_setitem_new_key_tzs O ' ' .   ( ( / $eUEN&;<6"!%a 3!%a 3 sH-r4c<tgdgddgd}d}tjt|5|jdddddd }tjt|5|jd ddddd}tjt|5|jdd dddtgdgddgdj d }|jdd}tgd gddgd}t j|||jd d}t j|||jdd }tgdgddgd}t j||y#1swY6xYw#1swYxYw#1swYxYw)Nrir=rorrrsrp)rorrrsrprqrr)rrirrir=ror z5'Cannot get left slice bound for non-unique label: 1'rriz5'Cannot get left slice bound for non-unique label: 0'rr=axis)r=rrrsrp)rrrprqr)ririr=ro)r=rrrs)rrrprq)ririr=)rrrrMr( sort_indexr)r*)rBr0rrr1s r2test_loc_non_uniquez*TestLocBaseIndependent.test_loc_non_uniquesd $+= >FX F ]]83 /  FF12J E ]]83 /  FF12J E ]]83 /  FF1QK $+= >FX *!*  DTX   FF2q5M2772A2q5>DTX   FF2JAQU 56s D  D%ctdgdi}|jdd}||usJ|dd|usJ|jddddf|usJtj|dj|djsJt j |5gd|jdddf<ddd|r|dgdk(jsJ|ddk(jsJttjjdjd}|s|r|d|jdddfusJ|d|jdddfusJtgd}|jdd}||usJ|dd|usJt j |5gd |dd ddd|rt|dd gdk(sJyt|dd gd k(sJy#1swY+xYw#1swYDxYw) Nr$r)rrrrrrrrr=rnrrrrrro) rr(r@ shares_memory_valuesr)assert_cow_warningr@rrrr)rBusing_copy_on_writer original_df sliced_dfr0original_series sliced_seriess r2&test_identity_slice_returns_new_objectz=TestLocBaseIndependent.test_identity_slice_returns_new_objectjs  i 01 OOA&  +++1~[000q!t$K777 C 0 8 8)C.:P:PQQQ " "#5 6 0&/KOOAsF # 0 cNi/446 66cNa',,. ..ryy,,Q/??H I "4a5q!t , ,,a5BFF1a4L( ((!!34'++A. O333q!888  " "#5 6 ,"+OBQ  , }Ra(I56 66}Ra(I56 663 0 0( , ,sG G GG#c|s1tjjd}|j|t t t dt dddg}|j}|jdddf}|dz }tj|||j}|j|jdf}|dz }tj||y)Nz!accidental fix reverted - GH37497r]ror$r#rr=) rrxfail applymarkerrziprrr(r)r*r )rBrOrrrrZqzs r2test_loc_copy_vs_viewz,TestLocBaseIndependent.test_loc_copy_vs_views#;;$$,O$PD    % c%(E!H-Sz B FFH EE!S&M Q a# FFH EE!''3,  Q a#r4ctjdj}tddg|dz |g}|j|dz }|j d}||k(sJ|j|dz g}|j dg}t j|||j|dz |g}t j||y)Nrrir=r r)r@iinfomaxrr(rr)r)rBumaxrrr1s r2test_loc_uint64z&TestLocBaseIndependent.test_loc_uint64sxx!%%aVD1Hd#34"88A;!!!$($88QC= vx0$(D)* vs+r4chtjdj}tddg|dz |g}t j t d5|jddddt j t d5|jdgdddy#1swY>xYw#1swYyxYw)Nrrir=r z-1rr )r@rrrrrrMr()rBrrs r2!test_loc_uint64_disallow_negativez8TestLocBaseIndependent.test_loc_uint64_disallow_negativesxx!%%aVD1Hd#34 ]]84 0  GGBK ]]84 0  GGRDM       sBB(B%(B1c gd}t|tjtjgt |zt d}tddg}||j dddf<tj||yNrrrrZrrZr) rr@rrrr?r(r)r*rBrr1r0s r2*test_loc_setitem_empty_append_expands_rowszATestLocBaseIndependent.test_loc_setitem_empty_append_expands_rowssfRXXrvvhT&:&I J  Sz *q#v b(+r4c Hgd}t|tjtjgt |zt d}tddg}|dj tj|d<||jdddf<tj||yr) rr@rrrr?r rGr(r)r*rs r26test_loc_setitem_empty_append_expands_rows_mixed_dtypezMTestLocBaseIndependent.test_loc_setitem_empty_append_expands_rows_mixed_dtypesRXXrvvhT&:&I J Sz *S'..*3q#v b(+r4ctdgtjgd}tddgt}|jd|jd<t j ||y)NrSrrrZr:)rr)rr@rrr(r)r*r[s r2*test_loc_setitem_empty_append_single_valuezATestLocBaseIndependent.test_loc_setitem_empty_append_single_valuesOC5x89 Sz 7!f-v b(+r4cddg}tddg}|jjtj|_dtj t d}tjt|5||jd dgdf<dddd }tjt|5||jd ddf<dddy#1swYCxYw#1swYyxYw) Nrir=rrZrz"None of \[Index\(\[0, 1\], dtype='z'\)\] are in the \[index\]rr)setting an array element with a sequence.) rr r r@rGr<rrrrMr( ValueError)rBrr0rs r2$test_loc_setitem_empty_append_raisesz;TestLocBaseIndependent.test_loc_setitem_empty_append_raisess1v Sz *88??288,1"((3-A$ $ ]]83 / '"&BFFAq63;  ': ]]:S 1 $#BFF1Q38  $ $  ' ' $ $s9C2CCC$ctddgddgg}|jtjd}t ddgd}t j ||y)Nrir=rorrrr)rr(r@rrr)r)rBr0rrs r2test_indexing_zerodim_np_arrayz5TestLocBaseIndependent.test_indexing_zerodim_np_arraysO AA' ( $ Aq6 " vq)r4cptddg}|jtjd}|dk(sJy)Nrir=r)rr(r@r)rBrrs r2%test_series_indexing_zerodim_np_arrayzr@rrrr)r*rrrrr(r r r?)rBrCr`col_datarr1rs r23test_loc_setitem_str_to_small_float_conversion_typezJTestLocBaseIndependent.test_loc_setitem_str_to_small_float_conversion_typesiMRRSHUqC --a0779EABUU8cU3Xu5 fh/ y@ MCK0Laq0L 6<<,- M M@H,H!U1X,HFJJv||S( ) C5FMMfUH  ! !&( 3*22AuQx2s XuEB fh/'V1M M M-I 3s0AFF$F6FF% F*FF"ctddd}|jdk(|jdk(z}ttj j djt|df| }tj||}|jtdd}|j|}tj|||jd d}|jtdd}|d d}|jjd|_|jjd|_tj||y) N1/1/2000z1/5/20005minfreqrrxr=ror z1/4/2000r )rhourminuterr@rrrrr)get_objr(rrr _with_freq) rBframe_or_seriesrngr:rPrrchunkr1s r2test_loc_getitem_time_objectz3TestLocBaseIndependent.test_loc_getitem_time_object(sZf=A #**"23 II ! !! $ 4 4c#h] C3 jjo.a%ggdm $ $42;'"#;||..t4 !2248 )r4 spmatrix_t) coo_matrix csc_matrix csr_matrixr<ctjd}t||}d\}}|tj||||}t j j|}td|}|j|j} |j|} tj| | |j|jj} tj|t!|d} tj| | y)N scipy.sparse)rsrqrr=r) fill_value)r importorskipgetattrr@eyersparse from_spmatrixrr(rtoarrayr)assert_numpy_array_equalrfullr) rBr9r< sp_sparserowsr spmatrixr0itr_idxrr1s r2$test_loc_getitem_range_from_spmatrixz;TestLocBaseIndependent.test_loc_getitem_range_from_spmatrix>s''7 Y 3  dbffT4u=UK    + +H 54.''##%g. ##FH5''..774Uq!AB ##FH5r4c tdtjddgtdi}|jddg}t j ||y)Nr'rrGrri)rrrrr(r)r*rs r2,test_loc_getitem_listlike_all_retains_sparsezCTestLocBaseIndependent.test_loc_getitem_listlike_all_retains_sparseWsF RXXq!fK4HIJ KA fb)r4ctjd}tjj |j d}|j td}tgdgdgtdd}tj|||j tdj td }tgdgtdd}tj||y) Nr>rsr=rSrrrr)rrSrrrrrrri) rr@rrCrDrBr(rrr)r*)rBrHr0rr1s r2test_loc_getitem_sparse_framez4TestLocBaseIndependent.test_loc_getitem_sparse_frame\s''7    + +IMM!,< =a! &(A Bi-  fh/a!%%eAh/ & '{9c/J  fh/r4c~tgdtdd}|jtd}tddgtdd}t j |||jtdjtd}tddgtdd}t j ||y)NrPrrrr=rSro)rrr(rr)r)rBrrr1s r2test_loc_getitem_sparse_seriesz5TestLocBaseIndependent.test_loc_getitem_sparse_seriesns ,K 34O PuQx3*K 3,GH vx0uQx$$U1X.3*K 3,GH vx0r4r(rctgdgdgtt}t||d}t gdttd}t j ||y)N)rSr?)rrrrr)r<r)rrrrArr)r)rBrr0rr1s r2!test_getitem_single_row_sparse_dfz8TestLocBaseIndependent.test_getitem_single_row_sparse_dfzsN9UAS TW%a(/U1C!L vx0r4key_typec|gd}|jdd|f}|jddgdf}tj||y)Nr)r(r)r*)rB float_framerWidxrr1s r2test_loc_getitem_iterablez0TestLocBaseIndependent.test_loc_getitem_iterablesA'C(??1o#56 fh/r4c,ttjjdj d}t ddd|_|jtddddf}|jddddf}tj||y)Nr=rr-0srnrstartperiodsr0) rr@rrnormalrr r(rr)r*rs r2#test_loc_getitem_timedelta_0secondsz:TestLocBaseIndependent.test_loc_getitem_timedelta_0secondssw ryy,,Q/66G6D E"rD66)D/+Q./q! fh/r4 val,expectedc~tddgddg}|j|}||_tj||y)Nrir=rdrer )rr(rr)r)rBrFr1r0rs r2test_loc_getitem_uint64_scalarz5TestLocBaseIndependent.test_loc_getitem_uint64_scalars= 1vi%7 8  vx0r4c|}tgdtgd|}|j}d|jd<d|jd<t j ||y)Nr)r?rirr zoorir=)rrrr(rr)r)rBfloat_numpy_dtyper<rr1s r2+test_loc_setitem_int_label_with_float_indexzBTestLocBaseIndependent.test_loc_setitem_int_label_with_float_indexsP!_E+U,KL88:   a sH-r4zindexer, expected) r]rir=rorrrsrprqrr) rrir=ror]r]r]r]rr) rrir=r]rrr]rprqrrcttdd}tdtdid|}d|j|j|df<t||dgd}t j ||y) Nrnrr~rrGr<r r]r)rrrr(r r)r*)rBrr1tdir0s r2/test_loc_setitem_listlike_with_timedelta64indexzFTestLocBaseIndependent.test_loc_setitem_listlike_with_timedelta64indexsq593/ U2Y'wc B)+rxx #%& E   h+r4c|tgdtdd}tgdtdd}tjtd5t ddgd dg |j d d d f<t ddgd dg |j d d df<dddtj||y#1swY xYw)Nriririririaaaaar)rir#r#ririaabbarrr#r$ categoriesrir=ro)rrr)rrrr(r*)rBr0rs r28test_loc_setitem_categorical_values_partial_column_slicezOTestLocBaseIndependent.test_loc_setitem_categorical_values_partial_column_slices_4=A B1W FG  ' ' !=  N +C:3*MBFF1Q38 *C:3*MBFF1Q38   N b#&  N Ns AB22B;cTtdgdgd}t|dgd}||jdddf<|d}t||jdj |st nd}tj||||d<tj|dt|d y) Nr$r)AlphaNumericrzrrvrr>r) rrr(rr r r?r)r)rBrCr0rwrr1s r2'test_loc_setitem_single_row_categoricalz>TestLocBaseIndependent.test_loc_setitem_single_row_categoricals #A37 8 GI (q'zG*BHH7CJJ,F%  vx0!7  r'{F:G,LMr4ctdtdgdzi}tjd|jdddf<td|jdk(sJtd|jdk(sJt j td 5td d d |jd <dddtdj|jd k(sJy#1swY5xYw)NrTz 2010-10-01roz 2008-08-08rri)rrT)rirTrKrirs)r=rTz 2005-05-05) rrr@rr(r)rrrrs r2"test_loc_setitem_datetime_coercionz9TestLocBaseIndependent.test_loc_setitem_datetime_coercions i 56:; <==6qsCx&"&&.888&"&&.888  ' ' =Q R .!$1-BFF6N .&++-??? . .s C%%C.idxervarc>|}tddd|}td|dg}t|dgtj}t j |dk(rt ndd 5||jdd|f<dddt j||y#1swY xYw) Nz 2015-07-12ror)r_r`r0tzrrrrrKr) rrr@rr)rrr(r*)rBrtz_naive_fixturerrZr1rs r2!test_loc_setitem_datetimeindex_tzz8TestLocBaseIndependent.test_loc_setitem_datetimeindex_tzs|QSRHSeW=ugRZZH  ' '"e^M=Q  ,$,FJJq%x  , fh/  , ,s BBcvtddd}ttjj dj t |df|}tdd d }ttd d d td d d }gd }gd }|j}d |j|<|j|}|j|j} d | jdd|r| jt} tj|| |j}d |j|<|j||j|<tj|||j}d |j|<|j|}|j|j} d | jdd|r| jt} tj|| |j}d |j|<|j||j|<tj||y)Nz 2012-01-01z 2012-01-0530minr/r=rsr rr )Hx) JKLz{|)rrr@rrrrrrrr(r rr)r*r) rBusing_array_managerr r0akeybkeyaindsbindsrr1s r2test_loc_setitem_time_keyz0TestLocBaseIndependent.test_loc_setitem_time_keys<GD  II ! !! $ 4 4c%j!_ EU B1~T"a^T"a^4"F 4D!66$<$$& Q u-H fh/ 4775> 4 fb) 4D!66$<$$& Q u-H fh/ 4775> 4 fb)r4rdctjgd}tgdgdg|}|j}t j dd|j dd|f<tgdgd g|}tj|||jd }t j dd|j dd|f<|jd }tj||y) N))r'rr)r3)r'r9rrrrsrprr=r=rGr)rr=r)rrsrrir) r from_tuplesrrr@zerosr(r)r*r)rBrdrzr0rPr1s r2,test_loc_setitem_unsorted_multiindex_columnszCTestLocBaseIndependent.test_loc_setitem_unsorted_multiindex_columns"s # #$F G  9-r :ggi((693i3R@ c8, ]]] "&8q#v&&A&. b(+r4ctgd|}d|jd<tgd|}tj||y)Nrrrrr)rrr=rorr(r)r)rBany_int_numpy_dtyper-r1s r2test_loc_setitem_uint_dropz1TestLocBaseIndependent.test_loc_setitem_uint_drop1s; )<= 1 )+>? vx0r4cbtdtjddgz}tjdd|jgd<tdtjddgz}t tjdd|jgd<t j ||y)Nrnrr]r)rr@ timedelta64r(rr)r)rBrr1s r2test_loc_setitem_td64_non_nanoz5TestLocBaseIndependent.test_loc_setitem_td64_non_nano9sR2>>"c2334^^B4 "r3 7889"+BNN2s,C"D Y sH-r4ctjjdjd}t t dd}d}t jt|5||jt d<dddt jt|5||jdddddy#1swY=xYw#1swYyxYw)Nr=rrrr r) r@rrrrrrrr!r()rBrrrs r2 test_loc_setitem_2d_to_1d_raisesz7TestLocBaseIndependent.test_loc_setitem_2d_to_1d_raisesAsyy$$Q'77?U1XY/9 ]]:S 1 % $CGGE!H  %]]:S 1 CGGAJ   % %  s"B7C7CC ctjdd}tgdgdgdg|gd}d }|jd }t j ||y) Nrro)r_r`rrrrrri)rir')rinterval_rangerr(r)assert_almost_equal)rBr r0r1rs r2test_loc_getitem_interval_indexz6TestLocBaseIndependent.test_loc_getitem_interval_indexMsQ!!15   9 -UO ! vx0r4ctjddd}tgdgdgdg|gd }tjdd d d }td d g|d}|jd}t j ||y)Nrroboth)r_r`closedrrrrrr=ri)r_r`r0rrrr'rrir')rrrrr(r)r)rBr r0 index_expr1rs r2 test_loc_getitem_interval_index2z7TestLocBaseIndependent.test_loc_getitem_interval_index2Xsu!!1VD   9 -UO %%AqqP 1a& < vx0r4tplririr=ctddgdd}t|}|j|g}t|gdd}t|}tj||y)Nrrr'Fr tupleize_colsr )rrr(r)r*)rBrrZr0rr1s r2+test_loc_getitem_index_single_double_tupleszBTestLocBaseIndependent.test_loc_getitem_index_single_double_tuplesds` 6N  S !SE593' fh/r4ctdddg}|dd}|dd}t||gdd }td d g|d dg}|j|ddd }|dk(sJy)N IndexTyper$r#rrrbofcomposite_indexFrr)rorrr'rrri)rrrr()rBridx1idx2r r0rs r2!test_loc_getitem_index_namedtuplez8TestLocBaseIndependent.test_loc_getitem_index_namedtuplessz{S#J7 &&tTl):%P 'usCj I %/05{{r4cttjjdj dgdgd}d|d<tj |j |jddddf<ttj dtj dtj g|stndj}tj|dj|y) Nr=rsro)r$r#rTrrrrquxr>r) rr@rrrrr(r rr?rr)r)rBrCr0r1s r2$test_loc_setitem_single_column_mixedz;TestLocBaseIndependent.test_loc_setitem_single_column_mixed}s  II ! !! $ 4 4V <+)  5 ')vvrxx!}e#$ VVUBFFE266 2 2&  &  r%y//:r4cttjjdjdt d}tj |d<t jtd5d|jd <ddd|j}ttjd gd ztjd gzgd }t j||y#1swYixYw)Nr=)rxroABCreventrrr)rnrrror?)r'rrrr )rr@rrtuplerr)rrr(rrr<rrs r2test_loc_setitem_cast2z-TestLocBaseIndependent.test_loc_setitem_cast2sryy,,Q/66w?u Vff7  ' ' !=  (#(BFF;  ( XXi !A %();(< <*  vx0 ( (s 1C))C2ctdtjdtji}d|jd<|j j tjtjk(sJtjd|_|j j tjtjk(sJy)Nonerpr)rirrq)rr@rint8r(rrr<rs r2test_loc_setitem_cast3z-TestLocBaseIndependent.test_loc_setitem_cast3s ryy"'':; <xyy}} 1111yy}} 1111r4c|tdgd}gd}|jdk(rdgdgdgg}||jtd <|gd |j}t j ||y) Nrs)rorrrirr=r )rrnrr=rrnrro)rrirnrr)rndimr(r r)r)rBr5rPrr1s r2test_loc_setitem_range_keyz1TestLocBaseIndependent.test_loc_setitem_range_keyseeAho> 88q=cB4"&F"a"#4CIIF X&r4ctgdgdd}tddgddg|jdddf<tgdgdd}tj||y) Nrsrrr=rirvr$)rir=r=ririrrr(r)r*rs r2.test_loc_setitem_numpy_frame_categorical_valuezETestLocBaseIndependent.test_loc_setitem_numpy_frame_categorical_valuesW _3LM N&1v1a&AqsCx?9RST b(+r4)lrrrrrrrrrrrrrrr rrrrr$r'r+r;rAfixturerHr@rrGrMrPrTrXr\r r r}rrfilterwarningsrrrrrrrrrrrrrrrrrrarm_slowrrrrrrrrrrr"r$r&r(r+r8rcomplexrLrNrQrSrViterrrr[rbrgrlrqrxr|r~rrrrrrrrrrrrrrrr}r4r2rrsi0 [[ 7&,u9U V 6tU8S T 6ud8S TNvd|U3 7ud8S TNufTl3 7tU8S T $%$* 11(!KF-0,&  [[We}6P&QRS  #%N  0*&$ ^^   [[ HBHHQbhh '/)RS ,  , , ,,," [[)+4HI;IJ;Iz [[V%OP:Q:",4 [[ PQ [[Yu 6 ,7R , ,, [[ PQ [[Yu 6,7R,6 [[ PQ [[Yu 607R0., , [[ PQ#,R#,J [[SE5sD18288SE?C [[Wuhbhhuo&>? 3@ 3 [[Q-.!Aq"&&"& <H0 S#JABFFBFF+\Aq"&&"&&;QR0 O$"((3BFF^YArvv? q!T"O4L)BFFBFF+\?K0 q!T"O4)\2"((CBFFBFF+\?K0 tT4(3*5Aq6Ar7RH5SzJ K8/ Q/ 2f,g2f, 8 8 8 . 0D [[ [[\S#J8H+IJ 0K 0D& &)7V$(,"  , ,,$"* 100*, [[\+UV [[WrxxW&EF6GW6.* 0$ 1 [[Y8191 [[Z$&%)HI0J0 0 [[)VQC[1E61#;3GH11 . [[/ 0 1a[< =V5 6   , , 'N&@ [[Wuug&67 08 0&*P [[US3%#uT{1C$DE ,F ,1.  1 1 [[UT6N3 04 0 ;1 2 ',r4rceZdZejej ejgdZejdZ dZ ejjddZ dZy) TestLocWithEllipsis)paramsc|jSr|)param)rBrOs r2rzTestLocWithEllipsis.indexers}}r4c:|}|tur|j}|Sr|)rto_frame)rBseries_with_simple_indexr5rPs r2rPzTestLocWithEllipsis.objs& & (,,.C r4cF||d}tj||y)N.)r)r)rBrPrrs r2test_loc_iloc_getitem_ellipsisz2TestLocWithEllipsis.test_loc_iloc_getitem_ellipsissc" $r43ignore:PeriodDtype\[B\] is deprecated:FutureWarningc|}|tjust|dk(rdn|jd}|tjur|jj dk(ry|tjurOt |jtr5d}tjt|5||d|gf}dddn>t|dk7r0||d|gf}|||g}tj|||tjurdn |j}|j} || d|gf}|| dd|gf}tj||y#1swYnxYw)Nrr8z2MultiIndex does not support indexing with Ellipsisr.)r)rrr r( inferred_typerrrrNotImplementedErrorrrrr*) rBrrrPrdrrr1key2r0s r2&test_loc_iloc_getitem_leading_ellipsesz:TestLocWithEllipsis.test_loc_iloc_getitem_leading_ellipsessB&rww&#c(a-aciil bff !8!8I!E  bff CIIz!BFC2#> 2 cC5j1 2 2X]S\#u*-Fs|SE*H  " "68 4rww&qCHH \\^S4&[)2;q4&y) fh/ 2 2s )E$$E-cj|tjust|dk(rdn|jd}t j t t5||ddddt j t t5||d|gdfdddt j t t5||dd|fdddt j t d5|||ddfdddy#1swYxYw#1swY~xYw#1swYTxYw#1swYyxYw)Nrr)...Too many indexers)r)rrr rrr r!)rBrPrrds r20test_loc_iloc_getitem_ellipses_only_one_ellipsiszDTestLocWithEllipsis.test_loc_iloc_getitem_ellipses_only_one_ellipsissrww&#c(a-aciil ]]=0E F # CL " #]]=0E F * CLseS ) *]]=0E F ( CLc3 ' ( ]]=0C D ( CLc3 ' ( ( # # * * ( (  ( (s0 DD;D-D)DDD&)D2N)rrrrrr)r(rrrPrrrrrr}r4r2rrssV^^BFFBGG,-. ^^ % [[ VW0X00(r4rc reZdZejj dddggdgdgfddggdgdgfddgddgfgdgdgfddgddgfgdgd gfddgddgfgdgd gfddgddgfgdgdgfd dgddgfgd gd gfgejj d ddgdZdZdZdZ dZ dZ dZ dZ dZdZdZejj dej$ej(dej*fdZdZejj dd d!gd"Zd#Zd$Zy)%TestLocWithMultiIndexzkeys, expectedr#r$)r#r#r$r$)rir=rir=)r$r$r#r#rir=)r=rir=rirT)rTr$r$)rir=ridimr r;c|gdgdgi}ttjdjddfi|}t j |}|dk(r4|j |ddf}tj|j|y|dk(r4|j dd|f}tj|j|yy)N)rTr$r$r#r#)ririr=rir=rsr r;) rr@rreshaperrur(r)assert_index_equalr r;)rBrkeysr1kwargsr0 exp_indexrs r2'test_loc_getitem_multilevel_index_orderz=TestLocWithMultiIndex.test_loc_getitem_multilevel_index_orders 1?CD ryy},,Q2 =f =**84 '>&&q/C  ! !#))Y 7 I &&D/C  ! !#++y 9r4c|}|jd}|djd}|jj|jjddk(sJ|jj|jjddk(sJ|jd}|djd}|jj|jjdk(sJ|jj|jjdk(sJy)Nr'ri)rr=r=)r(r rgr)rB/multiindex_year_month_day_dataframe_random_dataymdrresult2s r2test_loc_preserve_namesz-TestLocWithMultiIndex.test_loc_preserve_namess=c(,,t$||!!SYY__QR%8888}}""ciiooab&9999!c(,,w'||  CIIOOA$6666}}!!SYY__Q%7777r4c tjdgddgg}td|}|jg}|dd}t j |||j|j dd}t j ||y)Nrrir )r from_productrr(r)rr)rBrzrrr1res2s r2.test_loc_getitem_multiindex_nonunique_len_zerozDTestLocWithMultiIndex.test_loc_getitem_multiindex_nonunique_len_zerosx  $ $qcAq6] 3Qb!ggbkr7 sH-wwsxx!}% tX.r4cttdgtjdgdgg}|jd}|Jtjdgddgg}tdgt |zt |}|jd}|Jtdgt |zt |}|jd}|dk(sJy) NLevel1Level2)rr Level2_aLevel2_bro)rr ri)rrrur(rrr?)rBrrmidxs r20test_loc_getitem_access_none_value_in_multiindexzFTestLocWithMultiIndex.test_loc_getitem_access_none_value_in_multiindex+sdVZ33hZ(4LMN-.~~&& Z4L'MNdVc$i'vTB/0~~aS3t9_F$?/0{{r4ctjtgdgdddg}tgd|}d|jd d tgd |}t j ||y) N)rrrrrrrr)rtworrrrrrfirstsecondrf)ririririririririr r)rr)rr)ririrrrrriri)rrr rr(r)r)rBr rr1s r2!test_loc_setitem_multiindex_slicez7TestLocWithMultiIndex.test_loc_setitem_multiindex_slice<sh&& HH H%  0>47 >.1:%H vx0r4ctddd}ttd|}|jt dddt ddd}t j ||y)NrC10minrr0r`ilrii4)rrrr(rr)r)rBtimesrrs r27test_loc_getitem_slice_datetime_objs_with_datetimeindexzMTestLocWithMultiIndex.test_loc_getitem_slice_datetime_objs_with_datetimeindexNsRtest_loc_multiindex_levels_contain_values_not_in_index_anymorezTTestLocWithMultiIndex.test_loc_multiindex_levels_contain_values_not_in_index_anymoresn-.tF|\6R S ]]8+B C 8 FF2c7X%q( ) - -seQh 7 8 8 8s ,A//A8cJtjtjtjg}ddg}tj||g}t ddg|}|j dddf}t dgtjg}tj||y)Nrrrrir ) r@rrrrurr(r)r)rBlev1lev2rzrrr1s r2'test_loc_multiindex_null_slice_na_levelz=TestLocWithMultiIndex.test_loc_multiindex_null_slice_na_levelsxx()u~  # #T4L 1aV2&E"1#bffX. vx0r4c8tjtdtdddgggd}tt d|}|j d ddddf}|j jd dd }tj|j |y) Nabxyrir=)rHrInumrfrr r$rrr) rrrrrr(r droplevelr)r)rBrzr loc_resultr1s r2test_loc_drops_levelz*TestLocWithMultiIndex.test_loc_drops_levels $ $ $ZdaV ,4G U1XR(WWS!QY' 99&&q)"1- j..9r4)rrrrrrrrrr rrrr#r'r*r.r@rrNArr3r:rBrFrMr}r4r2rrs [[3Z. = >3Z. = >Cj1a& !$8,#G HCj1a& !$8,#G HCj1a& !$8,#G HCj1a& !$8,#G HCj1a& !OY#? @   [[UWi$89 ::  : 8 /"1$, 0 0D11.* [[TBFFBEE4#@A 6B 602 [[Z"b2838 1 :r4rcpeZdZdZdZdZdZdZejjdddd d ggd  d Z dZ dZ dZdZejj!dejjdddgdZejjdgddZdZy)TestLocSetitemWithExpansionc0d}|j5|jtd|tdt |id}||j |<dddtdt |dzid}t j|y#1swY;xYw)N2 _SIZE_CUTOFFrrGrri)contextsetattrlibindexrrr(r)r*)rB monkeypatch size_cutoffrr1s r2/test_loc_setitem_with_expansion_large_dataframezKTestLocSetitemWithExpansion.test_loc_setitem_with_expansion_large_dataframes  " 2   .+ FU;%78HF&1FJJ{ # 2c5q#9:'J fh/  2 2s ?B  Bctt}d|jd<tj|tdgdgd|jd<tj|tddgddgy)Nrrir rorr?r(r)rrBrs r2test_loc_setitem_empty_seriesz9TestLocSetitemWithExpansion.test_loc_setitem_empty_seriessb6"  sFA3qc$:;  sFAq6!Q$@Ar4ctt}d|jd<tj|tdgdgd|jd<tj|tddgddgy)NrrSrir rjror[r\s r2#test_loc_setitem_empty_series_floatz?TestLocSetitemWithExpansion.test_loc_setitem_empty_series_floatsb6"  sFC5$<=  sFC:aV$DEr4c tt}d|jd<tj|tdgt dgd|jd<tj|tddgt ddgd|jd<tj|tgdt gd y) Nrrirr rorrr)rirorr)rrro)rr?r(r)rrr\s r2%test_loc_setitem_empty_series_str_idxzATestLocSetitemWithExpansion.test_loc_setitem_empty_series_str_idx s6" sFA3eUGn$EF sFAq6u~9N$OP  sF9EBS$sar4c"|jSr|) to_datetime64r6s r2r7z$TestLocSetitemWithExpansion.% aoo'r4c"|jSr|) to_pydatetimer6s r2r7z$TestLocSetitemWithExpansion.&rnr4c,tj|Sr|)r@rr6s r2r7z$TestLocSetitemWithExpansion.'sbmmA&r4)rBrmrpz np.datetime64)idsctd}td}t}d|j||df<d|j||df<|r<|jjj dk(sJ|jj}nd}tddd git ||g| t dg }tj||y) Nz20130101 09:00:00z20130101 10:00:00rrrrdatetime64[ns]gY@gi@rr) rrr(r r<rErr)r*)rBrjrCdt1dt2r0 exp_dtyper1s r2#test_loc_setitem_datetime_keys_castz?TestLocSetitemWithExpansion.test_loc_setitem_datetime_keys_cast!s+,+, [#&tCy% #&tCy% 88>>&&#- --I(I UEN #c )45'N  b(+r4ctddgi}tdg||jdddf<tdgtdg|d}tj||y)Nr'rir#)r9rrr)rBr9rr1s r21test_loc_setitem_categorical_column_retains_dtypezMTestLocSetitemWithExpansion.test_loc_setitem_categorical_column_retains_dtypeBsWC!:&(#@ 1c6A3[#-PQR fh/r4c~tdd}tdd}tdd}t||dd}||vsJt|d g }d |j|d f<ttj gt |zd gz|jt|gd gt }tj||y) Nz2017-10-29 00:00:00+0200z Europe/Madridrz2017-10-29 03:00:00+0100z2016-10-10 03:00:00leftr) inclusiver0r&rrr) rrrr(r@rrappendrr?r)r*)rBr_endr[rZrr1s r20test_loc_setitem_with_expansion_and_existing_dstzLTestLocSetitemWithExpansion.test_loc_setitem_with_expansion_and_existing_dstIs4I2G , AvC@}}wi8"$ 2w; VVHs3x 2$ &**]B401I   fh/r4cttddgdg}ddg|d<|jdjj d|_||j dk(jdjjd }|j}||j|j dk(df<t|d jd|jd gd }tj|j ||j|j dk(dftd z}||j|j dk(df<tj|j|j dk(df|y)Nz2015-03-30 20:12:32z2015-03-12 00:11:11r)rr;newoldnew_colrrcr)rirr1s)rrrr rrr tz_convertrr(rr)rr)rBr0vrr1s r2test_setitem_with_expansionz7TestLocSetitemWithExpansion.test_setitem_with_expansion[sK 35JKLH 9 ,,v&,,88? rzzU" # - -f 5 ; ; F F| Tggi01 u$f,-1Q4??51266)3DEFS sxx2 FF2::&. /)D/ A./rzzU"F*+ rvvbjjE&96&ABAFr4c.t}d|jd<d|jd<d|jdtjf<|j}t ddtjgtj }tj||y)Nrirr=)ririrorr) rr(r@infr;rrr)rrs r20test_loc_setitem_with_expansion_inf_upcast_emptyzLTestLocSetitemWithExpansion.test_loc_setitem_with_expansion_inf_upcast_emptyqsl [t t q"&&y!Qrzz: fh/r4z"ignore:indexing past lexsort depthrTFcht|stjd|jd}t|}t j |j tj}t||dg}d}||vsJ|jt||}t|tr |dd|k(s J|d|k(sJt j |dzj tj}t||dg} |j} |r| dd} || j|df<tj | | |jd} |r| dd} || j|<| dj tj} tj"| | |j} |r| dd} || j|df<tt%|tj&gztj&g|zt)|gzd| } tj | | y) NzNot relevant for empty Indexr=rrkapowr rirrir )rrskiprepeatr@rr rGrinsertrrrrr(r)r*rrrr) rBr rNarrorigrdrexp_datar1r0rrs r2/test_loc_setitem_with_expansion_nonunique_indexzKTestLocSetitemWithExpansion.test_loc_setitem_with_expansion_nonunique_index|s5z KK6 7 Q Jiil!!"((+EA37%LLUS1 eZ (R=#s* **R=C' ''99QU#**2::6XYDYY[ a5DsAv b(+iik!n q6D A;%%bhh/ sH-YY[ a5DsAvS RVVH$"&&Aq )B C  b(+r4r<)Int32Int64UInt32UInt64Float32Float64cttgd|}td|i}t|j}||j|jdf<t j ||dt|j}|j |j|jdf<t j ||dy)NrArrr Fr)rrr r(r)r*r)rBr<rr0rs r26test_loc_setitem_with_expansion_preserves_nullable_intzRTestLocSetitemWithExpansion.test_loc_setitem_with_expansion_preserves_nullable_ints \/ } %*'* 288V#$ fbEB*'*{{ 288V#$ fbEBr4ctdtdi}tddd}||jgddf<|j t t jt jg|j }ttd|d }|jd|jk(sJtj||y) Nr'rs 2016-01-01rorcr`rrhrrr) rrrr(rrrrr<rr)r*)rBr0rFbexr1s r2#test_loc_setitem_ea_not_full_columnz?TestLocSetitemWithExpansion.test_loc_setitem_ea_not_full_columns U1X 'q\B!$y#~jj'7syyIJ58#67s#syy000 b(+r4N)rrrrYr]r_rarirrrrxrzrrrrrrrr}r4r2rPrPs0BF W1 [[  ' ' &  H , ,.00$G, 0 [[ DE [[Yu 61,7F1,f [[M C C ,r4rPc$eZdZdZdZdZdZy)TestLocCallablectgdtdgdd}|jd}tj||j|j dkD|jdddf}tj||j|j dk(ddf|jdd f}tj||j|j dkDgd f|jd d f}tj||j|j dkDd f|jddf}tj||j|j dkDdd gf|jddf}tj||j|j dk(dd gf|jddf}||jdk(sJy)Nrir=rorrr@rc |jdkDSNr=r'r6s r2r7zATestLocCallable.test_frame_loc_getitem_callable..qssQwr4r=c |jdk(S)Nr#)rr6s r2r7zATestLocCallable.test_frame_loc_getitem_callable..sqssczr4r#c |jdkDSrrr6s r2r7zATestLocCallable.test_frame_loc_getitem_callable..rr4c |jdk(SNrrr6s r2r7zATestLocCallable.test_frame_loc_getitem_callable..s!))s2Br4)FTFc |jdkDSrrr6s r2r7zATestLocCallable.test_frame_loc_getitem_callable..rr4cyrr}r6s r2r7zATestLocCallable.test_frame_loc_getitem_callable..r4rc |jdkDSrrr6s r2r7zATestLocCallable.test_frame_loc_getitem_callable..rr4c ddgSNr'rr}r6s r2r7zATestLocCallable.test_frame_loc_getitem_callable..s 3*r4r'c |jdk(Srrr6s r2r7zATestLocCallable.test_frame_loc_getitem_callable..sqssaxr4c ddgSrr}r6s r2r7zATestLocCallable.test_frame_loc_getitem_callable..s C:r4cy)Nrir}r6s r2r7zATestLocCallable.test_frame_loc_getitem_callable..rr4cyr&r}r6s r2r7zATestLocCallable.test_frame_loc_getitem_callable..rr4r)rrr(r)r*r'rrrBr0rs r2test_frame_loc_getitem_callablez/TestLocCallable.test_frame_loc_getitem_callables \V <P Qff&' c266"$$(#34ff)1,- c266"$$#+q.#9:ff&(BBC c266"$$(4H*H#IJff& 56 sBFF244!8S=$9:ff&(<<= c266"$$(S#J*>#?@ff')==> c266"$$!)c3Z*?#@Aff[-/0bffVn$$$r4ctgdtdgdd}|jdddgf}tj||j|j dkDddgf|jddgd f}tj||jddgddgf|jdd f}tj ||jdddgfy) Nrr@rc |jdkDSrrr6s r2r7zITestLocCallable.test_frame_loc_getitem_callable_mixture..rr4r'rr=roc ddgSrr}r6s r2r7zITestLocCallable.test_frame_loc_getitem_callable_mixture..s Szr4c ddgSrr}r6s r2r7zITestLocCallable.test_frame_loc_getitem_callable_mixture..s 3*r4)rrr(r)r*r'rrs r2'test_frame_loc_getitem_callable_mixturez7TestLocCallable.test_frame_loc_getitem_callable_mixtures \V <P Qff&c 23 c266"$$(S#J*>#?@ffaV112 c2661a&3**<#=>ffQ,,- sBFF1sCj=$9:r4ctgdtddtd}|jd}tj||jddg|jd ddf}tj||jddgddf|jd d f}tj ||jddgd f|jd df}tj||jddgd gf|jddgdf}tj ||jddgd f|jddgdf}tj||jddgd gf|jdd f}tj ||jddgd f|jdd gf}tj||jddgd gfy)Nrr@Xrrr c ddgSNr'rr}r6s r2r7zHTestLocCallable.test_frame_loc_getitem_callable_labels.. Szr4r'rc ddgSrr}r6s r2r7zHTestLocCallable.test_frame_loc_getitem_callable_labels.. rr4c ddgSrr}r6s r2r7zHTestLocCallable.test_frame_loc_getitem_callable_labels.. rr4cyNrr}r6s r2r7zHTestLocCallable.test_frame_loc_getitem_callable_labels.. rr4rc ddgSrr}r6s r2r7zHTestLocCallable.test_frame_loc_getitem_callable_labels.. rr4cdgSrr}r6s r2r7zHTestLocCallable.test_frame_loc_getitem_callable_labels.. scUr4cyrr}r6s r2r7zHTestLocCallable.test_frame_loc_getitem_callable_labels.. rr4cdgSrr}r6s r2r7zHTestLocCallable.test_frame_loc_getitem_callable_labels.. sC5r4c ddgSrr}r6s r2r7zHTestLocCallable.test_frame_loc_getitem_callable_labels.. rr4c ddgSrr}r6s r2r7zHTestLocCallable.test_frame_loc_getitem_callable_labels.. rr4)rrr(r)r*rrs r2&test_frame_loc_getitem_callable_labelsz6TestLocCallable.test_frame_loc_getitem_callable_labelss \V =T&\ Rff)* c2663*#56ff)1,- c2663*a-#89ff)=89 sBFFC:s?$;<ff)?:; c2663*se*;#<=ffc3Z./ sBFFC:s?$;<ffc3Z01 c2663*se*;#<=ff)3./ sBFFC:s?$;<ff)C501 c2663*se*;#<=r4ctgdttdtdtd}|j }d|j d<|j }d|j d d g<t j|||j }d |j d ddf<|j }d |j d d gddf<t j|||j }d |j ddf<|j }d |j d d gdf<t j|||j }ddg|j ddf<|j }ddg|j d d gdgf<t j|||j }tjd dg|j d d gdf<|j }tjd dg|j d d gdf<t j|||j }d|j d d gdf<|j }d|j d d gdgf<t j|||j }d|j ddf<|j }d|j d d gdf<t j|||j }d|j ddgf<|j }d|j d d gdgf<t j||y)Nrr@rrrr ic ddgSrr}r6s r2r7zATestLocCallable.test_frame_loc_setitem_callable..! 3*r4r'rr]c ddgSrr}r6s r2r7zATestLocCallable.test_frame_loc_setitem_callable..' rr4r c ddgSrr}r6s r2r7zATestLocCallable.test_frame_loc_setitem_callable..- rr4cyrr}r6s r2r7zATestLocCallable.test_frame_loc_setitem_callable..- rr4rrsrnc ddgSrr}r6s r2r7zATestLocCallable.test_frame_loc_setitem_callable..3 rr4cdgSrr}r6s r2r7zATestLocCallable.test_frame_loc_setitem_callable..3 sr4r!cyrr}r6s r2r7zATestLocCallable.test_frame_loc_setitem_callable..: rr4cdgSrr}r6s r2r7zATestLocCallable.test_frame_loc_setitem_callable..@ sser4c ddgSrr}r6s r2r7zATestLocCallable.test_frame_loc_setitem_callable..F rr4c ddgSrr}r6s r2r7zATestLocCallable.test_frame_loc_setitem_callable..L rr4) rrrr?rr(r)r*r@r)rBr0rrs r2test_frame_loc_setitem_callablez/TestLocCallable.test_frame_loc_setitem_callable s VDL%G Hv,  ggi(+$%ggi!c  c3'ggi+-$a'(ggi!#c A  c3'ggi79$m34ggi#%c C  c3'ggi:;R$o56ggi&'Wc SE!" c3'ggi-/XXr2h-?c M)*ggi#%88RH#5c C  c3'ggi/1c O+,ggi%'c SE!" c3'ggi-/$c)*ggi#%c C  c3'ggi/1$se+,ggi%'c SE!" c3'r4N)rrrrrrrr}r4r2rrs%8 ;><7(r4rcleZdZdZdZdZdZejjdddgdZ y) TestPartialStringSlicingctgdgddtgd}tddgdd gdtd d g}|jd }tj||y) Nrr)rr) 2020-08-01z 2020-07-02 2020-08-05r r$rTrirorrz2020-08)rrr(r)r*rs r25test_loc_getitem_partial_string_slicing_datetimeindexzNTestPartialStringSlicing.test_loc_getitem_partial_string_slicing_datetimeindexS sf $i 8HI 3Z!Q 0|\:;  " fh/r4ctjddd}|j}|jdd}|jdd}t j ||y)N 2017-01-01 2018-01-01rr_rr0z2017-12r )r period_range to_seriesr(rr)r)rBpirrr1s r28test_loc_getitem_partial_string_slicing_with_periodindexzQTestPartialStringSlicing.test_loc_getitem_partial_string_slicing_with_periodindex` sN __<\ Llln)$88CR= vx0r4ctddd}|j}|jdd}|jdd}t j ||y)N1 day2 days1hrz1 daysr )rrr(rr)r)rBrrrr1s r2;test_loc_getitem_partial_string_slicing_with_timedeltaindexzTTestPartialStringSlicing.test_loc_getitem_partial_string_slicing_with_timedeltaindexh sJ 7t Dlln(#88CR= vx0r4ctdtdittdd}|jd}|jd}t j ||y)Nrrodaysrnr r0 days)rrrrr(r)r)rBr0r1sliceds r2#test_loc_getitem_str_timedeltaindexzl+ $   # C!Gl+=>$    k2W )! )r4) rrrrrrrrrrrr}r4r2rrR sB 0111 [[]T3R,ST!*U!*r4rc HeZdZdZdZej jdejddde dd d gd Z d Z d Z ej jdejdddeddgejddejddeddgdejddeddggdZej jdddgdZej jdddgd Zd!Zy)"TestLabelSlicingc *ttjdddDcgc]}tdd|ddc} }ttjd dd Dcgc]}tdd|ddc} }||fD]}|jtddddd}t j |||jdtddd d}t j |||jtddddtddd d}t j |||jtdddd d}|jdd}t j |||jdd}t j || ycc}wcc}w)Nrrrririrnrr r )rir=r=rorrrrrz 20010101 11)rr@rrr(r)r*r)rBr df_uniquedf_dupsr0rr1s r2(test_loc_getitem_slicing_datetimes_framez9TestLabelSlicing.test_loc_getitem_slicing_datetimes_frame sw IIc +9EFA8D!QB/F  IIc +9HIA8D!QB/I  g& 4BVVHT1a467F  ! !&" -VV6htQ267F  ! !&" -VVHT1a4xaB7OPF  ! !&" -VVHT1a467Fwwqr{H  ! !&( 3VVMN+F  ! !&( 3 4G Js F F ctdddd}tgd|}tdd }td d }|j||}td d g|d d }t j ||||}d }||k(sJy)Nz2017-10-29 01:30:00z Europe/Berlinrsz30 min)rr`r0)rrir=rorrr z2017-10-29 02:30:00+02:00r|z2017-10-29 02:00:00+01:00r=rorr)rrrr(r)r)rBrZseries2t_1t_2rr1s r2'test_loc_getitem_label_slice_across_dstz8TestLabelSlicing.test_loc_getitem_label_slice_across_dst s !oqx 43H3HS%1a&Aa1 vx0!!!r4r rrrrrrrc|j}|jd|d}|jdd}tj||y)Nr!r )rr(rr)r)rBr rrr1s r2-test_loc_getitem_label_slice_period_timedeltaz>TestLabelSlicing.test_loc_getitem_label_slice_period_timedelta sCoo59%88CR= vx0r4c8gd}ttjjdjd|}|jdd}t |dk(sJ|jdd}t |dk(sJ|jdd}t |d k(sJy) N)gx"p|@g\Va|@g|@r=ror=r g3333c|@g|@g3333|@g|@ro)rr@rrr(r)rBr r0s1s r2%test_loc_getitem_slice_floats_inexactz6TestLabelSlicing.test_loc_getitem_slice_floats_inexact s: ryy,,Q/66v>e L VVGG $2w!|| VVGG $2w!|| VVGG $2w!||r4c|}ttjjdjdtjdd|}t |j dddk(sJt |j ddd k(sJtjdd|}d |d<||_t |j dddk(sJt |j ddd k(sJy) Nr=rnr]rr rtrg)@rqgffffff(@)rr@rrrrr(r )rBrkr<rrZs r2'test_loc_getitem_float_slice_floatindexz8TestLabelSlicing.test_loc_getitem_float_slice_floatindex s! II ! !! $ + +B /ryyRu7U 37745>"a'''37745>"a'''iiBe,A 37745>"a'''37745>"a'''r4zstart,stop, expected_slicernsNrrir?rprqrrrscttdtdd}|jt ||}|j |}t j||y)Nrrz10 days)rrrr(rrr)r)rBr_stopexpected_slicerrr1s r2$test_loc_getitem_slice_label_td64objz5TestLabelSlicing.test_loc_getitem_slice_label_td64obj sNU2Y) DEud+,88N+ vx0r4r_20182020c|gdtdtdtdg}tjtd5|j|ddddy#1swYyxYw) Nr201620192017r z,Value based partial slicing on non-monotonicrr)rrrrMr()rBr5r_rPs r2)test_loc_getitem_slice_unordered_dt_indexz:TestLabelSlicing.test_loc_getitem_slice_unordered_dt_index sa V$i&769JK ]] J  " GGE& ! " " "s AA(r&rUc|td|dddg}|j|d}|td|dddg}tj||y)Nrrrr=thirdr )rr(r)r)rBr5r&rPrr1s r21test_loc_getitem_slice_labels_int_in_object_indexzBTestLabelSlicing.test_loc_getitem_slice_labels_int_in_object_index sTeAhugq'.JKw'"58E7Aw3OP )r4ctdddddg}tddggdgtddgt}tj|j ddddf|y) Nrir=ro)testrir=rr r)rr r;)rrr?r)r*r(rs r2*test_loc_getitem_slice_columns_mixed_dtypez;TestLabelSlicing.test_loc_getitem_slice_columns_mixed_dtype s[ aA.qc :a&!eQF&.I  bffQUmX6r4)rrrrr rrrrrrr rrr@rrrrr r#r}r4r2rr sF4:"" [[ BOO,Ls K 'xd C 11  ( [[$ R^^At $dE!RL 9 R^^As #^R^^As%;U1a[ I >2>>!S)5A; 7 11 [[Wvv&67"8" [[Wq#h/*0*7r4rceZdZejj dddgdZejj dddgdZdZy)TestLocBooleanLabelsAndSlices bool_valueTFc|d}|jdk7r>||d}tjt|5|j|dddyy#1swYyxYw)Nz7: boolean label can not be used without a boolean indexr8r?rJr)rrrrMr()rBr r5r&messagerPs r2'test_loc_bool_incompatible_index_raiseszETestLocBooleanLabelsAndSlices.test_loc_bool_incompatible_index_raises! sd  L WX   ) +!X>Cxw7 $ # $ $ , $ $s AAcP|tddgdd}|j|y)NTFr8rr?rJ)rr()rBr5r&rPs r2test_loc_bool_should_not_raisez s U2Y' (b 4gk2c7a<C1= 0 + - j-8 4JD$WWYF$&FJJtSy ! "5937 H  ! !(F 3 4r4rNrctjgd}tdd|}tdtjdi|j d}|j }|j|ddf|j|ddf<tj|||j }|j||j|<tj||y)Nr20010101rrrr$r r) r@rrrrr rr(r)r*)rBrr:rZr0rs r2+test_loc_setitem_mask_with_datetimeindex_tzz>TestLocBooleanMask.test_loc_setitem_mask_with_datetimeindex_tzT sxx23Q26 RYYq\*# 6 = =i H ffT1Wo 47 fb)66$< 4 fb)r4c ttjdjddt dt ddd}|j }|jd gtjtjgz|d <|jd k}|j|j|j|d f<tj||y) Nr!ror=ABr-r)r`r0rrrri)rr@rrrrrr rrr'r(r)r*)rBr0r1r:s r22test_loc_setitem_mask_and_label_with_datetimeindexzETestLocBooleanMask.test_loc_setitem_mask_and_label_with_datetimeindexe s IIcN " "1a (JZ>  779!*+rvvrvv.>> ttaxFF4L..tSy b(+r4cdtd}td}tdt||gi}|j}t|g}|djdj }||j ddg<|djdj }||k(sJtj||y)Nrl ,GsO1colriTF) rrrrr_valuer(r)r*)rBtd1td2r0df_copyrr1rs r2'test_loc_setitem_mask_td64_series_valuez:TestLocBooleanMask.test_loc_setitem_mask_td64_series_valuet sl)* vsCj12 3'')cUme9>>!$++ #e}E"))6!!! b'*r4c|j}|ddkD}d|j|df<|jj}d||jdf<t||j|j }t j||y)Nr'rrrir)rr(rrr r;r)r*)rBrYr1r:rs r2#test_loc_setitem_boolean_and_columnz6TestLocBooleanMask.test_loc_setitem_boolean_and_column s}##%3!#%& c "%%'!"t{{A~V8>>8CSCST k84r4c,tgdgdd}tdgdigd|jgddgf<tgd gdd}tj||tgdgdd}t gd gd|jgddgf<tj||tgdgdd}t gd gd|jgddf<tj||tgdgdd}|j }|d}tj |5t gd gd|jgd<ddd|rtj||ytj||y#1swY9xYw) Nrrrr$r)r=rirr )FFT)rir=rnr)rr(r)r*rrr)rBrrr0r1rrs r2)test_loc_setitem_ndframe_values_alignmentz? b(+YY7 8.4\.S#cU*+ b(+YY7 8,2779;=c7<<>rxxrxx03%78 b(+r4ctdgdgdt}tdgdgdt}|dj|jt j dgtj dgf<tj||y)NrrZrrr#Tr$rGrs r2test_loc_indexer_length_onez.TestLocBooleanMask.test_loc_indexer_length_one sl cU#/v >C5u5VD: sqr4cLtddtjg}tt d|}|j ||}t j|||||}t j|||jj ||}t j||j|dd}|dd}d}tjt|5|j ||dddtjt|5|||dddtjt|5|jj ||dddy#1swYxYw#1swY]xYw#1swYyxYw) Nr'rror r rir r) rr@rrrr(r)rrr*rrrM)rBrcirrser2ci2rs r28test_loc_getitem_list_of_labels_categoricalindex_with_nazHTestLocListlike.test_loc_getitem_list_of_labels_categoricalindex_with_na sXsC0 1U1XR(R! vs+SW vs+##CG, fclln53Bxf ]]83 /  HHSX  ]]83 /  SN ]]83 / * MMO  C ) * *      * *s$-F& F$FF FF#ctjgdd}tgdtddd}t j t d 5|j|dddy#1swYyxYw) N) 2001-01-04z 2001-01-02rYz 2001-01-14rr)r=rsrrz 2001-01-01r?rrrr r)r@rrrrrrMr()rBrdrs r21test_loc_getitem_series_label_list_missing_valueszATestLocListlike.test_loc_getitem_series_label_list_missing_values sZhh DL ]J|#q$QR ]]8> :  GGCL   s A((A1cttjddgtjddg}tjt d5|j tjddgdddy#1swYyxYw)NlPF AHlUBeNlP2$$nr rr rlNZ4FG)rr@rrrrMr(r\s r29test_loc_getitem_series_label_list_missing_integer_valueszITestLocListlike.test_loc_getitem_series_label_list_missing_integer_values sw((,.?@A-/ABC ]]8> : E GGBHH.0ABC D E E Es %BB  to_periodTFctdddd}|r|jd}tddg|d }tdtdg}|r|Dcgc]}|jd}}|j|}tddg|d }|s |j j d|_tj||d tdtdtdg}|r|Dcgc]}|jd}}tgd t|d|j d }|j|}tj||d tdtdtdg}|r|Dcgc]}|jd}}tjtd5|j|dddycc}wcc}wcc}w#1swYyxYw)N 2011-01-01z 2011-01-02r?rZ)r0r皙?皙?rrT)r)rbrbra)rr< 2011-01-03r r)rr^rrr(r r4r)rrr<rrrM)rBr^rZrrrrrs r2.test_loc_getitem_listlike_of_datetimelike_keysz>TestLocListlike.test_loc_getitem_listlike_of_datetimelike_keys s|#EJ --$Cc3Zs5,'<)@A .23AKK$3D3c3Zs5 ,,T2CI vsTB l # l # l #  .23AKK$3D3 5E#KRU  vsTB l # l # l #  .23AKK$3D3 ]]8> :  GGDM  9444  sG(G5G *GGctddgddgddgggddd g }|jdd}d |j_|jt d d gd }t j||y)Nrir=rrrsrqr)cobraviper sidewinder max_speedshieldrrrfrgr)rrr rr(rr)r*rs r2test_loc_named_indexz$TestLocListlike.test_loc_named_index st VaVaV $2 (+  772A;#w0u=> fh/r4N)rrrrrrr@asarrayrrWrZr]rdrkr}r4r2rQrQ sk [[U["**d$CD*E*6E [[[4-8'9'R 0r4rQz%columns, column_key, expected_columns)rirmrri)rmrAllrnr=cttjjdjd|t d}|j dd|f}|j gd|f}tj||dy)Nr=rrrrTr) rr@rrrrr(r)r*)r; column_keyexpected_columnsr0r1rs r2*test_loc_getitem_label_list_integer_labelsrr+ sn  a ''/U  Bwwq**+H VVOZ/ 0F&(dCr4ctjjdjd}t |}tj |j dddf<tj|tjtj gdzjddf}t |gd}tj||t |}tj |j dddf<tj||y)Nr=)rrrrirri)rrSrrjrir) r@rrrrrr(hstackrrr)r*) rand_datarr2r1s r2test_loc_setitem_float_intindexrv> s %%a(88@I y !FFJJq#vIIy"((BFF8a<*@*H*HA*NOPM0IJH&(+ y !FFJJq#v&(+r4cdDcgc] }dD]}||f}}}ttjdtdt j |}|j ddd }ttjd tdt j gd }tj||ycc}}w) N)r)rqrrrn)rnr ABCDEFGHIJrrirrxrrrr)rz)rxrn)rrqr{) rr@rrrrr(r)r*)yrrr r0rr1s r2test_loc_axis_1_slicer}L s* BM BqRG BG BD B  L!&&t, B VVV^Ii 0F L!&&'TUH &(+ CsCc tdtdtjtdtdg}|j }|j ddgf|j ddgf<t j||y)Nr$r=rrr)rrrrrr(r)r*)r1rs r2!test_loc_set_dataframe_multiindexr] so 58Z%<%r $Slicing a positional slice with .locrr=ro)rrrrrr()rs r2%test_loc_with_positional_slice_raisesro sF q!5 6C y(N O s A  Actdd}ttjj djd|}|d}d}||fD]h}t j t| 5|jd ddddt j td 5d |jd ddddjt j t| 5|jd dd fdddt j td 5d|jd dd f<dddy#1swYxYw#1swYxYw#1swYYxYw#1swYyxYw) Nrror`r=r r rzOcannot do slice indexing on DatetimeIndex with these indexers \[1\] of type intrrir) rrr@rrrrrr()dtir0rrrPs r2#test_loc_slice_disallows_positionalrw s> \1 -C 299((+226:# FB Q%C & Cy ]]9C 0  GGAaL ]]9,R S CGGAaL    y , qsAv y(N OqsAv    s00D,#D8EE,D5 8E E Ec .ttjjdj dgdt ddd}|j gd j}|j}tj|j}d }tjt| 5|j|dddtjt| 5|d j|dddy#1swY@xYw#1swYyxYw) Nr=rr2012rrsrr)rr=r=roz/None of \[TimedeltaIndex.* are in the \[index\]rr$)rr@rrrrrrr rTimedeltaIndexasi8rrrMr()r0rrprs r2'test_loc_datetimelike_mismatched_dtypesr s  a 008c15 B   # # %B ((C  CHH %C TestLocSeries.test_loc_getitem_setitem_integer_slice_keyerrorso s II ! !! $ 4 4R 8U1bRS_@U  XXZ" " "'')))XXZ" " "**..000!A''!B-;;}- vx0 w1XXd58ntE!QO'<< = ]]86 2  FF1RL  ]]86 2 BFF1RL      s G=G"G"G+ct|jdd}|j|}tj||ddy)Nrn)rr r(r)r)rBrrZrs r2test_loc_getitem_iteratorz'TestLocSeries.test_loc_getitem_iterator s@=&&s+,""3' v}Sb'9:r4c||jkD}|j}d|j|<|}d||<tj||y)Nr)rrr(r)r)rBrr:rr1s r2test_loc_setitem_booleanz&TestLocSeries.test_loc_setitem_boolean sO}3355##% 4  vx0r4ct|jgd}d|j|<d}tjt |5d|j|dgz<dddy#1swYyxYw)N)rsrrrsz\['foo'\] not in indexrr)rr r(rrrM)rBrrrs r2test_loc_setitem_cornerz%TestLocSeries.test_loc_setitem_corner seM'' 34"# $' ]]83 / 201M  deWn - 2 2 2s A$$A-cf|jgd}|j}|j}d||<d|j|<tj|||j}|j}d||d|dd|j|d|dtj||y)N)rsrnrrr=)r rr(r)r)rBrindicesrrs r2test_basic_setitem_with_labelsz,TestLocSeries.test_basic_setitem_with_labels s!'' 4  ! ! #""$7  r3'  ! ! #""$&'71: #+, WQZ( r3'r4c  ttjjdj dt t ddd}gd}tjgd}|j}|j}d||<d|j|<tj|||j}|j}d||<d|j|<tj||gd}tjgd}d}tjt| 5d||<dddtjt| 5d||<dddy#1swY3xYw#1swYyxYw) Nr=rnrr]r )rrrrprrrrsrpz\[5\] not in indexr)rr@rrrrrrrr(r)rrrrM Exception) rBrrarr_indsrr inds_notfoundarr_inds_notfoundrs r2!test_loc_setitem_listlike_of_intsz/TestLocSeries.test_loc_setitem_listlike_of_ints s@ II ! !! $ 4 4R 8U1bRS_@U 88I& XXZhhjD   r3' XXZhhjH  r3'$ HH\2# ]]83 / #!"C  # ]]9C 0 '%&C! " ' ' # # ' 's5E-E9-E69Fcjttdddgd}|j}tdd}||jd <|jd }||k(sJ|j}||j d <|j d }||k(sJ|j}||d <|d }||k(sJy) Nr`rorrrr rcr|r$r)rrrrr(r)rBrrr1rs r2test_loc_setitem_dt64tz_valuesz,TestLocSeries.test_loc_setitem_dt64tz_values s |Q< @! XXZ\l;s !!! XXZ !!! XXZ3C!!!r4array_fnr.rc\|dg|z}t|ddddgtdt}tdtdt}||jd<t j ||tdtdt}||j d<t j ||y)NrabcderJr$)rrr?r(r)rr)rBr.rrr1rs r2#test_loc_iloc_setitem_with_listlikez1TestLocSeries.test_loc_iloc_setitem_with_listlike ssTz"31a+4=OQd7m6:  sH-Qd7m6:  sH-r4rr'Nc\ttjgdgd}d}tjt |5|j |ddfdddtjt |5d|j |ddf<dddy#1swYAxYw#1swYyxYw)N))r'0)r'r8)rr)r5r=r\rrri)rrrrrr r()rBrrrs r2+test_loc_series_getitem_too_many_dimensionsz9TestLocSeries.test_loc_series_getitem_too_many_dimensions s(()MN " ]]= 4 GGGQJ  ]]= 4 $"#CGGGQJ  $ $   $ $sB9B"B"B+c|jgd}|j}d|j|<|j}d|jgd<t j ||d|jddd|ddt j |||jddg\}}d|j||d|ddt j ||d|j|<d|j|<||dk(sJ||dk(sJy)Nrrsrnrrprerr)r rr(rr)r)rBrrrr1rrs r2test_loc_setitemzTestLocSeries.test_loc_setitem s""9-##% 4 %%'#$ i  vx0 Ab2 vx0$$aW-B 2b2 vx0!" " ! "R A%%%R A%%%r4r<r?rWctddgddgd|}ddd|jd ddf<tddgddgd|}tj||y) NrdefghijklrrnewAnewBrr)rBr<r0r1s r2test_loc_assign_dict_to_rowz)TestLocSeries.test_loc_assign_dict_to_row s]eU^5%.A O#&1q!t FE?&%IQVW b(+r4ctddg}tdddd|jd <tdddd|jd <ttdddggddgd g }tj||y) Nrr&rrprrnr)rr&rir)rrr(r)r*r&s r2,test_loc_setitem_dict_timedelta_multiple_setz:TestLocSeries.test_loc_setitem_dict_timedelta_multiple_set& sFG#45!*13!7%H 1 !*13!7%H 1 $e , -7HQRPS  fh/r4c2tddgdg}ddg|jdddgf<tttjtjgd dtjgdtjgd ddg }t j ||y) Nrir=r$rrprqr#rTr?rrr )rr(rr@rr)r*rs r23test_loc_set_multiple_items_in_multiple_new_columnszATestLocSeries.test_loc_set_multiple_items_in_multiple_new_columns1 s aVcU 3!"Aq3*}RVVRVV,H=[[  a&   b(+r4cd}tjt|5tjddd}t t d|}|jdd k(sJ dddy#1swYyxYw) Nz#Period with BDay freq is deprecatedr2000r]rr^r z 2000-01-14r)r)rrrrrrr()rBrr r-s r2 test_getitem_loc_str_periodindexz.TestLocSeries.test_getitem_loc_str_periodindexA se3  ' ' S A 1OO&"3GEE"IU3F::l+q0 00 1 1 1s AA++A4cttd}t|dzd}tdt j t |tji|}|j|}td|jdjtjitt j|jdd}tj||y) NrirrrFr stable)rE)rrrrr@rrintpr(argsortr rrr)r*)rBrrr r0rr1s r2test_loc_nonunique_masked_indexz-TestLocSeries.test_loc_nonunique_masked_indexI s59ocDj0 ryyU277CDE R EMMxM077@ A ,,T2'B  fh/r4)!rrrrrrrrrrrrrrrrr@rrrrrrrrrrrNrOrrrrr}r4r2rrJ s= [[^nj-IJ(K( >& :; 12 ('8", [[Z"((BHHdE)JK [[V\2.3L.  [[YCF);c5;=O(PQ $R $&4 [[Wx&:;,<,''0(0, 1 0r4r)\__doc__ collectionsrrvrrrrr) dateutil.tzrnumpyr@rpandas._configr pandas._libsr rVpandas.compat.numpyr pandas.errorsr pandas.util._test_decoratorsutil_test_decoratorsrNpandasrrrrrrrrrrrrrrrrrrrpandas._testing_testingr)pandas.api.typesr pandas.core.indexingr!pandas.tests.indexing.commonr"rrrr3r6rrrrPrrrr%r/rQrrrvr}rrrrrrrrrrrrrrrrrrr}r4r2rs*"   -*.'))(&6K& &&"&&# bffbff 57JK &&#bff t}511@@Fq,q,h-:(:(zs:s:la,a,H((DG*G*TA7A7H  :E,E,Pc0c0L+ dD\Aq62 tTlQF3 tUmaV4DD ,,",8,1 $<*(3  # "FG(H(& < "23