eh >ddlmZddlZddlZddlZddlZddlmZm Z m Z m Z m Z m Z mZmZmZmZddlmZddlmcmZej2ddgdZGddZd Zd Zej<j?d d d gddge ddgd zgdZ y)) permutationsN) IndexInterval IntervalIndex Timedelta Timestamp date_rangeinterval_rangeisnanotnatimedelta_rangefoo)paramsc|jSN)param)requests i/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/pandas/tests/indexes/interval/test_interval.pynamers ==cP eZdZejddgddgZdidZdidZdZe jjdgd e j d d d dd ddde jg edde jeddde jj!dej$gdgdZdZdZdZdZe jjdeddd !ed"d#d$d%&eed'd(d)!eed*d+d!gd,Zd-Zd.Zd/Ze jjded0d12ed0d1d34ed5d12gd67d8Z e jjded9d:2ed5d:2gd;Z!e jjdd?g7d@Z#e jjd: eii6||v%%% #=.A e.34[HT5& )bff D  88Cv. ##BJJu$5x@7 , s L'-2L-breaks) rrrr45iiiziiir1?gffffff@ez 2017-01-01z 2017-01-04s)unitzmismatched result unit)reason)marks)1ns2ms3s4min5h6DcBtj||}|j}td|D}t j |||j dtj}|j}td|D}t j ||y)Nrc34K|]}|jywr)length.0ivs r z0TestIntervalIndex.test_length..es3r3src3NK|]}t|r |jn|ywr)r r`ras rrdz0TestIntervalIndex.test_length..ksHBeBiR7Hs#%) rr r`rr;r<insertr(r+)r"rrMrFresultrHs r test_lengthzTestIntervalIndex.test_lengthTs~))&@3U33 fh/ Q'H%HH fh/rc|j|}|jdusJ|j}tjt |t }tj|||j}tjt |t }tj|||j|}|jdusJ|j}tjddgdgt |dz zz}tj|||j}tjddgdgt |dz zz}tj||y)NrFr/Tr) r#hasnansr r(zerosr8boolr;rDr onesr-rBr"rrFrgrHs rtest_with_nansz TestIntervalIndex.test_with_nansns,!!!0}}%%%88CJd3 ##FH5773u:T2 ##FH5**&*9}}$$$88UDMUGs5zA~,FFG ##FH588T5MTFc%j1n,EEF ##FH5rc|j|}|j}|j|sJ|jd}|j|sJ|j|jusJy)NrT)deep)r#copyequalsr=)r"rrHrgs r test_copyzTestIntervalIndex.test_copysj$$F$3}}X&&&D)}}X&&&{{(--///rc|j|}t|d}tj|jj |jj dtj|j j |j j dttj|d}tj|jj |jj dtj|j j |j j dy)NrF)rrsame) check_samerr) r#rr;rDr=valuesrr(rB)r"rrFrgs rtest_ensure_copied_dataz)TestIntervalIndex.test_ensure_copied_datas!!!0u51 ## JJ  v{{11f  ## KK   3 3 rxxU; ## JJ  v{{11f  ## KK   3 3 rctjddtj}tj||}|j |j d}tj||y)Nrrr/rr) r(r*r>rr r#deleter;r<)r"rrMrHrgs r test_deletezTestIntervalIndex.test_deletesX1b1 ,,VFC""&"188; fh/rdatar&neither)periodsrg333333?r%@both)rfreqr20170101 r=z1 dayr5c|d}t|g}|j|}|jd|}tj|||j|}|jt ||}tj|||ddj|j|dd}|jd|}tj|||jdd}|j tjdd}tj||d}tjt|5|jjdddddd}hd|jhz D]}d |d |jd }t|j|j| } |jd| }|j tjd| }tj||tjt |5|jjd| dddtt"j$g|j } t"j$dt&j(fD]P} |ddj| j|dd}|jd| }tj||R|jj*j,d vr|j tjdt&j.}d}tjt|5|jjdt&j.ddd|jdt&j.}tj||y#1swY[xYw#1swYTxYw#1swYZxYw)Nrr2rrz=can only insert Interval objects and NA into an IntervalArraymatchz+'value.closed' is 'left', expected 'right'.>rr=rr~z'value.closed' is 'z ', expected 'z'.r)mM)rappendrfr;r<r8astyperCpytestraises TypeError_datarrr=r ValueErrorr(r+pdNAr0kindNaT) r"r}itemidx_itemrHrgresmsgrbad_itemna_idxnas r test_insertzTestIntervalIndex.test_insertsAw $(??4(Q% fh/;;x(SY- fh/8??8,33DH=Q% fh/kk!U#;;v&--a7 c8,M ]]9C 0 ( JJ  a ' (<:dkk]J /F'x}T[[MLC 4::fEH++a*C{{6*11!X>H  ! !#x 0z5 / !!!X. / /  /x <664' 4BBQxv.55d12h?H[[B'F  ! !&( 3 4 99??  z 1{{6*11!RVVzTestIntervalIndex._c!''lr)idsctj|}|j|}tj|j}t j ||t |d|d}|j|}t |dj|dj}||k(sJ|j|}t|j}t j |||j|d}|dj}||k(sJ|jt|}t|j}t j ||yNrr) rr _maybe_convert_i8asi8r;r<r_valuerlist)r"rMrFrgrHintervals rtest_maybe_convert_i8z'TestIntervalIndex.test_maybe_convert_i8Xs5))&1((/ ,,V[[9 fh/F1Ivay1((2F1I,,fQi.>.>?!!!((0% fh/((3!9##!!!((f6% fh/rz 2018-01-01r4ctj|}|jtjgdzj d}t tjgdztj}|j|}tj|||jd|d}|jdt|dj}|j|}tj||y)Nr2nsr/r)rr _constructorrras_unitrr(r+r@rr;r<rffloatr)r"rMrF to_convertrHrgs rtest_maybe_convert_i8_natz+TestIntervalIndex.test_maybe_convert_i8_nats ))&1(("&&A6>>tD "&&ARZZ8((4 fh/&&q&)4 ??1eF1I,<,<&=>((4 fh/rmake_keyc|SrrMs rrzTestIntervalIndex.srlambdarcftjd|}tj|}||}|j |}|j j }tjtjtjd|}t||} tj|| y)Nr4r/)iuf) r(r*rr rr0rr>uint64r@rr;r<) r"rany_real_numpy_dtyperMrFkeyrgrexpected_dtyperHs rtest_maybe_convert_i8_numericz/TestIntervalIndex.test_maybe_convert_i8_numerics1$89))&1v((-||  !xxbiibjjI$ON3 fh/rc&t|d|dSrrrs rrzTestIntervalIndex.8F1Ivay9rc |dSNrrrs rrzTestIntervalIndex. 6!9r)rrscalarctjd|}tj|}||}|j |}||usJy)Nr4r/)r(r*rr r)r"rrrMrFrrgs r'test_maybe_convert_i8_numeric_identicalz9TestIntervalIndex.test_maybe_convert_i8_numeric_identicalsM1$89))&1v((-}}rzbreaks1, breaks2c,t|jSrrrs rrzTestIntervalIndex.rrc&t|d|dSrrrs rrzTestIntervalIndex.rrc|Srrrs rrzTestIntervalIndex.s6rc |dSrrrs rrzTestIntervalIndex.rr)rrrrrc"tj|}||}d|jd|j}tj|}t j t|5|j|dddy#1swYyxYw)Nz)Cannot index an IntervalIndex of subtype z with values of dtype r) rr r0reescaperrrr)r"breaks1breaks2rrFrrs rtest_maybe_convert_i8_errorsz.TestIntervalIndex.test_maybe_convert_i8_errorss2))'2w8 G&}}o / iin ]]:S 1 )  # #C ( ) ) )s *BBcjtjddgddg}tjddgd}|j d}t j |||j d}t j ||tjddgd}|j d }t j |||j d}t j ||tjtd 5|j tdddddy#1swYyxYw) NrrrFrlr/r2Tr1z contains not implemented for twor) rr'r(rBcontainsr;rDrrNotImplementedErrorr)r"rrHactuals rtest_contains_methodz&TestIntervalIndex.test_contains_methods  % %q!fq!f 588UEN&9A ##FH5A ##FH588T5M8C ##FH5A ##FH5]] 'I  ' JJx1~ & ' ' 's D))D2ctjddg|}tjddtjg|}|j }t j ||tjddtjgddtjg|}|j }t j ||y) N)g?)rg@rrrrrr)rrr(r+dropnar;r<r')r"rrHiirgs r test_dropnazTestIntervalIndex.test_dropnas ,,j*-EfU  & &'? O fh/  & &1bff~1bff~f U fh/rctjddg|}gd}|j|}tjgdd}t j ||d|vsJy) Nrrr)r1rQr)rrrintpr/rQ)rr get_indexerr(rBr;rD)r"rrFtargetrrHs rtest_non_contiguousz%TestIntervalIndex.test_non_contiguouss\))66*:6J ""6*88Jf5 ##FH5%rc|j|}tjdgdgt|dz zz}|j |dd}t j |||j |dg}t j ||tjtjdd|}tjdgt|dz zdgz}|j |}t j |||j |j}t j ||dD]}|j|}tj||k(t|}|j |}t j |||j |j}t j ||y) NrTFrrrr&)rr=rr~) r#r(rBr8isinr;rDrr r*tolistrepeat)r"rrFrHrgother other_closeds r test_isinzTestIntervalIndex.test_isins!!!088TFeWE Q%??@E"1I& ##FH5U1XJ' ##FH5))"))B*;FK88TFc%j1n5?@E" ##FH5ELLN+ ##FH5@ :L%%\%:EyyInvalid comparison between dtype=interval\[int64, right\] and rrzLengths must match to comparer2)rrFr(rBr;rDrr rxr=joinrrrr*r)r"rrHrs rtest_comparisonz!TestIntervalIndex.test_comparisons!Q$**,88UDM* ##FH5#s#djj088UDM* ##FH5hsC00 ##FH5tzz)88T4L) ##FH5tzz) ##FH5tzz) ##FH5djj(88UEN+ ##FH5djj( ##FH5}88FKK ##FH5tzz000 ##FBHHdD\,BC""djj0 ##FBHHdD\,BCtzz000 ##FBHHdD\,BCtzz000 ##FBHHeU^,DEdjj/// ##FBHHeU^,DE""TZZ/ ##FBHHeU^,DEq ##FBHHeU^,DEtzz. ##FBHHeU^,DEhhLQ   ]]9C 0  JJN  ]]9C 0  JJ!O  ]]9C 0 & JJ1 % &. ]]:S 1 & JJ1 % & &     & & & &s0U4 V#V #V4U=V  VV!c Nttjtdd|tdd|g}t j tjddgtjddg|}|j |sJd}tjt|5t j tjddgtjgd|dddtjt|tjgdy#1swY>xYw) NrrrrzMmissing values must be missing in the same location both left and right sidesrr)TFF)rr(r+rrr'rsrrrrBr;rDr )r"rridx2rs rtest_missing_valuesz%TestIntervalIndex.test_missing_values_s VVXa62HQ&4Q R (("&&!Q"&&!QPVWzz$  ]]:S 1   % %A 3F   ##DIrxx8L/MN   s ?DD$c|j|}|j}tj|||jd}tj||dddt t ddt jt ddg}|j}t t ddt ddt jg}tj|||jdd }t t jt ddt ddg}tj||y) NrF) ascendingrrrrfirst)r na_position)r# sort_valuesr;r<rrr(r+rns rtest_sort_valuesz"TestIntervalIndex.test_sort_valuesqs!!!0""$ fe,""U"3 feDbDk2x1~rvvx1~FG""$ (1a.(1a."&&!IJ fh/""U"H "&&(1a.(1a.!IJ fh/rrNcPtd|}t|d}tj|}td|}t|d}t j |j |td||vsJtd||vsJtd||vsJttd|td |}ttd |td|}||vsJ||vsJ|jtd|jrJ|jtd|jsJ|jtd|jsJtd |}t|d d }|j|} tjgdd}t j| |td|}t|d d}|j|} tjgdd}t j| |y)Nz 2000-01-01rr&)startrz2000-01-01T12:00r7z 2000-01-01T12z 2000-01-02z 2000-01-03z 1999-12-31z1999-12-31T12:00r612h)rrr)rrrrrrrrr/z2000-01-08T18:006h)r6r6r%r%r%r%r)rr rr r;r<r?rranyrr(rBrD) r"rrdatesrFrHiv_trueiv_falserrs r test_datetimezTestIntervalIndex.test_datetimes,2.3))%0,4E15 eii2"-U:::R0==="-U::: lr *Ilr,J  lr *Ilr,J %u$$$>>)LR"@AEEGGG~~iB?@DDFFF~~i <=AACCC,4%?""6*8836B ##FH5,4%>""6*882&A ##FH5rctjddgddg|}tjddgddg|}|j|}tjgdgd|}tj|||j||g}tjgdgd |}tj||hd |hz D]|}tjddgddg|}|j|}|j t j|j t }tj||~y) Nrrrrr2)rrrr)rrrr2)rrrrrr)rrrrrr2>rr=rr~)rr'rr;r<rrC)r"rindex1index2rgrHrindex_other_closeds r test_appendzTestIntervalIndex.test_appends&**Aq6Aq6&I**Aq6Aq6&Iv& ,,\RR))&@##u,,,X&'&0Xx4H3II))&@##u,,,INaQ151u9$eq1uo&=>QQ))&@##t+++X&'&0Xx4H3II))&@##t+++INaQ151u9$eq1uo&=>QQ))&@%%V#!!!X&'&0Xx4H3II))&@%%!!! Q P))!Qw?%%KSRRsG GG r;rctj|}|j}tt j |}t j||yr)rr to_tuplesrcomasarray_tuplesafer;r<)r"r;rrgrHs rtest_to_tuplesz TestIntervalIndex.test_to_tuplessD(''/..v67 fh/rna_tupleTFc`tj|}|j|}tt j |dd}|dd}t j|||d}|r7t|tsJt|dk(sJtd|DsJyt|sJy)N)rDrrc32K|]}t|ywr)r )rbrs rrdz6TestIntervalIndex.test_to_tuples_na..Ws21tAw2s) rrr@rrArBr;r< isinstancetupler8allr )r"r;rDrrgexpected_notna result_notna result_nas rtest_to_tuples_naz#TestIntervalIndex.test_to_tuples_na3s.''/1s44VCR[ABcr{  lN;2J i/ //y>Q& &&2 22 22 ? "?rctjddd}tjddd}tj||j}d}||k(sJy)Nrr3i8r/rr4@)r(r*rr'nbytes)r"r=rrgrHs r test_nbyteszTestIntervalIndex.test_nbytes[sQyyAT* !Qd+**47>>!!!r new_closed)r=rrr~ctdd||}|j|}tdd||}tj||y)Nrr4)rr)r set_closedr;r<)r"rrrSrFrgrHs rtest_set_closedz!TestIntervalIndex.test_set_closeddsDq!F>!!*-!!QzE fh/r bad_closed)rr&LEFTTFctdd}d|}tjt|5|j |dddy#1swYyxYw)Nrr4zinvalid option for 'closed': r)r rrrrU)r"rWrFrs rtest_set_closed_errorsz(TestIntervalIndex.test_set_closed_errorslsMq!$-j\: ]]:S 1 )   Z ( ) ) )s AActttdtd}t|g}|jrJy)Nz2017-01-01 00:00:00z2018-01-01 00:00:00)rrr _is_all_dates)r" year_2017year_2017_indexs rtest_is_all_datesz#TestIntervalIndex.test_is_all_datests= + ,i8M.N ( 4"00000r)r)=__name__ __module__ __qualname__rr'rFr#r-rLrmark parametrizer(infr rxfailr to_timedeltarhrortryr|r rrrrrrr rrrrr rrrrrrrr r rrr!r&r+rr>rAr!rCrMrRrVrZr_rrrrrs %M % %q!fq!f 5EC 'AR [[ @ffWdCac3RVV D |\ 2 FLL<C@kk''/G'H  BOOD E   0  06.0 ,0  [[ 1b ; 3F C 9Z0"V L 9W-q I  7070r&4E=N1 [[ z1 - z1 > Ha 0 #00< [[ L! ,ohPQ.RS 0  0 [[ % v  0  0 [[  % % 9 $ 4 [[:q1:q\B!4     #  [[  % % 9 ! $    E  )  . )'. 0 :6>&@O$0( [[TD,#78'69'6R4,r{s"      e}%&] 1] 1@2 QFS#J