j< ddlmZddlZddlZddlZddlZddlmZm Z m Z m Z m Z m Z mZmZmZmZddlmZddlmcmZGddZdZdZdZej:j=d d d gd d ge ddgd zgdZy)) permutationsN) IndexInterval IntervalIndex Timedelta Timestamp date_rangeinterval_rangeisnanotnatimedelta_rangeceZdZejddgddgZdidZdidZdZe jjdgd e j d d d dd ddde jg edddedddej gdgdZdZdZdZdZe jjdeddded d!d"d#$eed%d&d'eed(d)dgd*Zd+Zd,Zd-Ze jjded.d/0ed.d/d12ed3d/0gd45d6Ze jjded7d80ed3d8d9:gd;Ze jjdd?g5d@Z!e jjd: eii6||v%%% #=.A e.34[HT5& )bff D  88Cv. ##BJJu$5x@7 , s L'-2L-breaks) rrrr-5iiiziiir*?gffffff@ez 2017-01-01z 2017-01-04ns)units)1ns2ms3s4min5h6DcBtj||}|j}td|D}t j |||j dtj}|j}td|D}t j ||y)Nrc34K|]}|jywN)length.0ivs r z0TestIntervalIndex.test_length..]s3r3src3NK|]}t|r |jn|ywrX)r rYrZs rr]z0TestIntervalIndex.test_length..csHBeBiR7Hs#%) rrrYrr4r5insertr!r$)rrrFr?resultrAs r test_lengthzTestIntervalIndex.test_lengthOs~))&@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) rhasnansr r!zerosr1boolr4r=r onesr&r;rrr?r`rAs rtest_with_nansz TestIntervalIndex.test_with_nansfs,!!!0}}%%%88CJd3 ##FH5773u:T2 ##FH5**&*9}}$$$88UDMUGs5zA~,FFG ##FH588T5MTFc%j1n,EEF ##FH5rc|j|}|j}|j|sJ|jd}|j|sJ|j|jusJy)NrT)deep)rcopyequalsr6)rrrAr`s r test_copyzTestIntervalIndex.test_copy}sj$$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)rksame) check_samerk) rrr4r=r6valuesrr!r;)rrr?r`s 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#r7rrrdeleter4r5)rrrFrAr`s r test_deletezTestIntervalIndex.test_deletesX1b1 ,,VFC""&"188; fh/rdatarneither)periodsrg333333?r@both)rxfreqr20170101 r6z1 dayr.c|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)Nrr+rfooz=can only insert Interval objects and NA into an IntervalArraymatchz+'value.closed' is 'left', expected 'right'.>rzr6rrwz'value.closed' is 'z ', expected 'z'.r)mM)rappendr_r4r5r1astyper<pytestraises TypeError_datarrr6r ValueErrorr!r$pdNAr)kindNaT) rrvitemidx_itemrAr`resmsgrbad_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.Wc!''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_i8asi8r4r5r_valuerlist)rrFr?r`rAintervals rtest_maybe_convert_i8z'TestIntervalIndex.test_maybe_convert_i8Ps5))&1((/ ,,V[[9 fh/F1Ivay1((2F1I,,fQi.>.>?!!!((0% fh/((3!9##!!!((f6% fh/rz 2018-01-01r-us)rxrNctj|}|jtjgdzj d}t tjgdztj}|j|}tj|||jd|d}|jdt|dj}|j|}tj||y)Nr+rr(r)rr _constructorrras_unitrr!r$r9rr4r5r_floatr)rrFr? to_convertrAr`s rtest_maybe_convert_i8_natz+TestIntervalIndex.test_maybe_convert_i8_natws))&1(("&&A6>>tD "&&ARZZ8((4 fh/&&q&)4 ??1eF1I,<,<&=>((4 fh/rmake_keyc|SrXrFs rrzTestIntervalIndex.srlambdarcftjd|}tj|}||}|j |}|j j }tjtjtjd|}t||} tj|| y)Nr-r()iuf) r!r#rrrr)rr7uint64r9rr4r5) rrany_real_numpy_dtyperFr?keyr`rexpected_dtyperAs 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)Nr-r()r!r#rrr)rrrrFr?rr`s r'test_maybe_convert_i8_numeric_identicalz9TestIntervalIndex.test_maybe_convert_i8_numeric_identicalsM1$89))&1v((-}}rzbreaks1, breaks2c,t|jSrXrrs rrzTestIntervalIndex.rrc&t|d|dSrrrs rrzTestIntervalIndex.rrc|SrXrrs 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) rrr)reescaperrrr)rbreaks1breaks2rr?rrs 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) NrrrFrer(r+Tr*z contains not implemented for twor) rr r!r;containsr4r=rrNotImplementedErrorr)rrrAactuals 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$dropnar4r5r )rrrAiir`s r test_dropnazTestIntervalIndex.test_dropnas ,,j*-EfU  & &'? O fh/  & &1bff~1bff~f U fh/rctjddg|}gd}|j|}tjgdd}t j ||d|vsJy) Nrrr)r*rJry)rrrintpr(rJ)rr get_indexerr!r;r4r=)rrr?targetrrAs 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)rr6rzrw) rr!r;r1isinr4r=rrr#tolistrepeat)rrr?rAr`other 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 comparer+)rr?r!r;r4r=rrrqr6joinrrrr#r)rrrArs 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 rlrrrr;r4r=r )rrridx2rs rtest_missing_valuesz%TestIntervalIndex.test_missing_valuesZs 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_valuesr4r5rrr!r$rgs rtest_sort_valuesz"TestIntervalIndex.test_sort_valuesls!!!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)startrxz2000-01-01T12:00r0z 2000-01-01T12z 2000-01-02z 2000-01-03z 1999-12-31z1999-12-31T12:00r/12h)rrxr{)rrrrrrrrr(z2000-01-08T18:006h)r/r/rrrrr)rr rrr4r5r8rranyrr!r;r=) rrrdatesr?rAiv_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) Nrrrrr+)rrrr)rrrr+)rrrrrr)rrrrrr+>rzr6rrw)rr rr4r5rr<)rrindex1index2r`rArindex_other_closeds r test_appendzTestIntervalIndex.test_appends&**Aq6Aq6&I**Aq6Aq6&Iv& ,,\RR))&@##u,,,X&FF(H1EF))&@##u,,,INaQ151u9$eq1uo&=>QQ))&@##t+++X&FF(H1EF))&@##t+++INaQ151u9$eq1uo&=>QQ))&@%%V#!!!X&FF(H1EF))&@%%!!! Q P))!Qw?%%KSRRsF:F? Gr5rct|}tj|}|j}t t j |}tj||yrX) rrr to_tuplesrcomasarray_tuplesafer4r5)rr5rr`rAs rtest_to_tuplesz TestIntervalIndex.test_to_tuplessM f''/..v67 fh/rna_tupleTFc`tj|}|j|}tt j |dd}|dd}t j|||d}|r7t|tsJt|dk(sJtd|DsJyt|sJy)N)r>rrc32K|]}t|ywrX)r )r[rs rr]z6TestIntervalIndex.test_to_tuples_na..Ss21tAw2s) rrr:rr;r<r4r5 isinstancetupler1allr )rr5r>rr`expected_notna result_notna result_nas rtest_to_tuples_naz#TestIntervalIndex.test_to_tuples_na+s6''/1s44VCR[ABcr{  lN;2J i/ //y>Q& &&2 22 22 ? "?rctjddd}tjddd}tj||j}d}||k(sJy)Nrr,i8r(rr-@)r!r#rr nbytes)rr6rr`rAs r test_nbyteszTestIntervalIndex.test_nbytesWsQyyAT* !Qd+**47>>!!!rnamerctdd||}|j|}tdd||}tj||y)Nrr-)rrM)r set_closedr4r5)rrMrrr?r`rAs rtest_set_closedz!TestIntervalIndex.test_set_closed`sDq!F>!!,/!!Q|$G fh/r bad_closed)rrLEFTTFctdd}d|}tjt|5|j |dddy#1swYyxYw)Nrr-zinvalid option for 'closed': r)r rrrrO)rrQr?rs rtest_set_closed_errorsz(TestIntervalIndex.test_set_closed_errorshsMq!$-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_datesps= + ,i8M.N ( 4"00000r)r);__name__ __module__ __qualname__rr r?rr&rErmark parametrizer!infr r to_timedeltararhrmrrrur rrrrrrr rrrrrrrrrrrrrrr rrr r$rr8r:rr=rGrLrPrTrYrrrrrsl %M % %q!fq!f 5EC 'AR [[ @ffWdCac3RVV D |\ = |\ < BOOD E    0  06.0 ,0  [[ 1b ; 3F C 9Z0"V L 9W-q I  7070r&4E=N1 [[ z1 - z1 > Ha 0 #00< [[ |Q / Had ;  0 0 [[ % v  0  0 [[  % % 9 $ 4 [[:q1:q\B!4     #  [[  % % 9 ! $    E  )  . )'. 0 :6>&@O$0( [[TD,#78'69'6R4,/!!$.///s A..A7cZtjddgddg}t|}d|vsJy)Nrrrr)rr dir)r?r`s rtest_dirres3  % %q!fq!f 5E ZF   rcTttddtddg}|j||}tjddg|j }t j|||jj||}t j||y)Nrrrr() rr searchsortedr!r;r)r4r=r) listlike_boxrqr`rAs r,test_searchsorted_different_argument_classesris HQNHQN; rus"      ^ 1^ 1B/2 QFS#J