ttj< ddlmZddlZddlZddlZddlZddlmZm Z m Z m Z m Z m Z mZmZmZmZddlmZddlmcmZGddZdZdZdZejd d d gd d ge ddgd zgdZdS)) permutationsN) IndexInterval IntervalIndex Timedelta Timestamp date_rangeinterval_rangeisnanotnatimedelta_rangec eZdZejddgddgZdidZdidZdZe j dgd e j d d d dd ddde j g edddedddejgdgdZdZdZdZdZe j deddded d!d"d#$eed%d&d'eed(d)dgd*Zd+Zd,Zd-Ze j ded.d/0ed.d/d12ed3d/0gd45d6Ze j ded7d80ed3d8d9:gd;Ze j dd?g5d@Z!e j dz5TestIntervalIndex.test_properties..2s9   e T5& ) )   rrrr cng|]1\}}t|rt||n tj2Sr,)r rr!r$r.s rr1z5TestIntervalIndex.test_properties..HsK   e.34[[ DHT5& ) ) )bf   r)rlensizeshapetmassert_index_equalr0rr!r#int64rmidfloat64rziprarrayobjectassert_numpy_array_equalasarrayr&r$)rrindexivsexpected expected_leftexpected_right expected_mids ` rtest_propertiesz!TestIntervalIndex.test_properties&s!!!005zzRzR{e#### ej% "BH0M0M0M*N*NOOO ek51b1Q1Q1Q+R+RSSS eiryd"*/U/U/U)V)VWWW|v%%%%    "599eArll;;   8Cv... #BJu$5$5x@@@**&*995zzRzR{e####q"&!Q1aAqABB &*$s*  ej-888 ek>::: ei666|v%%%%    "=.AA   8Cv... #BJu$5$5x@@@@@rbreaks) rrrr45iiiziiir*?gffffff@ez 2017-01-01z 2017-01-04ns)units)1ns2ms3s4min5h6DcFtj||}|j}td|D}t j|||dtj}|j}td|D}t j||dS)Nrc3$K|] }|jV dSN)lengthr/ivs r z0TestIntervalIndex.test_length..]s$33r333333rrc3FK|]}t|r|jn|VdSr_)r r`ras rrcz0TestIntervalIndex.test_length..cs3HHBeBii7RHHHHHHr) rrr`rr<r=insertr!r$)rrrMrFresultrHs r test_lengthzTestIntervalIndex.test_lengthOs)&@@@33U33333 fh/// Q''HH%HHHHH fh/////rc&||}|jdusJ|}tjt |t }tj||| }tj t |t }tj||| |}|jdusJ|}tj ddgdgt |dz zz}tj||| }tj ddgdgt |dz zz}tj||dS)NrFr(Tr) rhasnansr r!zerosr9boolr<rDr onesr&rBrrrFrfrHs rtest_with_nansz TestIntervalIndex.test_with_nansfs`!!!00}%%%%8CJJd333 #FH55573u::T222 #FH555**&*99}$$$$8UDMUGs5zzA~,FFGG #FH5558T5MTFc%jj1n,EEFF #FH55555rc||}|}||sJ|d}||sJ|j|jusJdS)NrT)deep)rcopyequalsr0)rrrHrfs r test_copyzTestIntervalIndex.test_copy}s$$F$33}}X&&&&&D))}}X&&&&&{(-//////rc||}t|d}tj|jj|jjdtj|jj|jjdttj|d}tj|jj|jjdtj|jj|jjddS)NrF)rqsame) check_samerq) rrr<rDr0valuesrr!rB)rrrFrfs rtest_ensure_copied_dataz)TestIntervalIndex.test_ensure_copied_datas!!!00u5111 # J v{1f     # K  3    rxU;;; # J v{1f     # K  3      rctjddtj}tj||}||d}tj||dS)Nrrr(rr) r!r#r>rrrdeleter<r=)rrrMrHrfs r test_deletezTestIntervalIndex.test_deletesl1b111 ,VFCCC""&"1188;; fh/////rdatarneither)periodsrg333333?r@both)r~freqr20170101 r0z1 dayr5c"|d}t|g}||}|d|}tj||||}|t ||}tj|||dd||dd}|d|}tj|||dd}|tdd}tj||d}tj t|5|j dddddn #1swxYwYd}hd|j hz D]}d |d |j d }t|j|j| } |d| }|td| }tj||tj t |5|j d| dddn #1swxYwYtt"jg|j } t"jdt&jfD]e} |dd| |dd}|d| }tj||f|jjjd vr|tdt&j}d}tj t|5|j dt&jdddn #1swxYwY|dt&j}tj||dS)Nrr2rfooz=can only insert Interval objects and NA into an IntervalArraymatchz+'value.closed' is 'left', expected 'right'.>rr0rr}z'value.closed' is 'z ', expected 'z'.r)mM)rappendrer<r=r9astyperCpytestraises TypeError_datarrr0r ValueErrorr!r$pdNAr)kindNaT) rr|itemidx_itemrHrfresmsgrbad_itemna_idxnas r test_insertzTestIntervalIndex.test_insertsVAw $((??4((Q%% fh///;;x((SYY-- fh///8??8,,33DH==Q%% fh///kk!U##;;v&&--a77 c8,,,M ]9C 0 0 0 ( ( J  a ' ' ' ( ( ( ( ( ( ( ( ( ( ( ( ( ( (<:::dk]J / /FLLLT[LLLC 4:fEEEH++a**C{{6**11!X>>H  !#x 0 0 0z555 / / !!!X... / / / / / / / / / / / / / / /x <<<64' 4 4BBQBxv..55d122h??H[[B''F  !&( 3 3 3 3 9? z 1 1{{6**11!RV<zTestIntervalIndex.Wc!'llr)idsctj|}||}tj|j}t j||t |d|d}||}t |dj|dj}||ksJ||}t|j}t j||||d}|dj}||ksJ|t|}t|j}t j||dSNrr) rr_maybe_convert_i8asi8r<r=r_valuerlist)rrMrFrfrHintervals rtest_maybe_convert_i8z'TestIntervalIndex.test_maybe_convert_i8PsX)&11((// ,V[99 fh///F1Ivay11((22F1I,fQi.>??!!!!((00%% fh///((33!9#!!!!((f66%% fh/////rz 2018-01-01r4us)r~rUc(tj|}|tjgdzd}t tjgdztj }| |}tj ||| d|d}| dt|dj}| |}tj ||dS)Nr2rr(r)rr _constructorrras_unitrr!r$r@rr<r=refloatr)rrMrF to_convertrHrfs rtest_maybe_convert_i8_natz+TestIntervalIndex.test_maybe_convert_i8_natws)&11(("&A66>>tDD "&ARZ888((44 fh///&&q&)44 ??1eF1I,<&=&=>>((44 fh/////rmake_keyc|Sr_r,rMs rrzTestIntervalIndex.srlambdarcRtjd|}tj|}||}||}|jj}tjtjtj d|}t||} tj || dS)Nr4r()iuf) r!r#rrrr)rr>uint64r@rr<r=) rrany_real_numpy_dtyperMrFkeyrfrexpected_dtyperHs rtest_maybe_convert_i8_numericz/TestIntervalIndex.test_maybe_convert_i8_numerics1$8999)&11hv((--| !xbibjII$ON333 fh/////rc:t|d|dSrr-rs rrzTestIntervalIndex.8F1Ivay99rc|dSNrr,rs rrzTestIntervalIndex. 6!9r)rrscalarctjd|}tj|}||}||}||usJdS)Nr4r()r!r#rrr)rrrrMrFrrfs r'test_maybe_convert_i8_numeric_identicalz9TestIntervalIndex.test_maybe_convert_i8_numeric_identicals^1$8999)&11hv((--}}}}}}rzbreaks1, breaks2c*t|jSr_rrs rrzTestIntervalIndex.rrc:t|d|dSrr-rs rrzTestIntervalIndex.rrc|Sr_r,rs rrzTestIntervalIndex.s6rc|dSrr,rs rrzTestIntervalIndex.rr)rrrrrc"tj|}||}d|jd|j}tj|}t jt|5||ddddS#1swxYwYdS)Nz)Cannot index an IntervalIndex of subtype z with values of dtype r) rrr)reescaperrrr)rbreaks1breaks2rrFrrs rtest_maybe_convert_i8_errorsz.TestIntervalIndex.test_maybe_convert_i8_errorss2)'22hw /  / /&} / / inn ]:S 1 1 1 ) )  # #C ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s!BB Bctjddgddg}tjddgd}|d}t j|||d}t j||tjddgd}|d }t j|||d}t j||tjtd 5|tddddddS#1swxYwYdS) NrrrFrkr(r2Tr*z contains not implemented for twor) rr r!rBcontainsr<rDrrNotImplementedErrorr)rrrHactuals rtest_contains_methodz&TestIntervalIndex.test_contains_methodss  %q!fq!f 5 58UEN&999A #FH555A #FH5558T5M888C #FH555A #FH555] 'I    ' ' JJx1~~ & & & ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 's $D>>EEctjddg|}tjddtjg|}|}t j||tjddtjgddtjg|}|}t j||dS) N)g?)rg@rrrrrr)rrr!r$dropnar<r=r )rrrHiirfs r test_dropnazTestIntervalIndex.test_dropnas ,j*-EfUUU  &'? O O O fh///  &1bf~1bf~f U U U fh/////rctjddg|}gd}||}tjgdd}t j||d|vsJdS) Nrrr)r*rQr)rrrintpr(rQ)rr get_indexerr!rBr<rD)rrrFtargetrrHs rtest_non_contiguousz%TestIntervalIndex.test_non_contiguoussz)66*:6JJJ ""6**8JJJf555 #FH555%rc||}tjdgdgt|dz zz}||dd}t j||||dg}t j||tjtj dd|}tjdgt|dz zdgz}||}t j|||| }t j||dD]}||}tj ||kt|}||}t j|||| }t j||dS) NrTFrrrr)rr0rr}) rr!rBr9isinr<rDrrr#tolistrepeat)rrrFrHrfother other_closeds r test_isinzTestIntervalIndex.test_isins!!!008TFeWE Q%??@@E"1"I&& #FH555U1XJ'' #FH555)")B*;*;FKKK8TFc%jj1n5?@@E"" #FH555ELLNN++ #FH555@ : :L%%\%::EyInvalid comparison between dtype=interval\[int64, right\] and rrzLengths must match to comparer2)rrFr!rBr<rDrrrwr0joinrrrr#r)rrrHrs rtest_comparisonz!TestIntervalIndex.test_comparisonsE!Q$*,8UDM** #FH555#s##dj08UDM** #FH555hsC000 #FH555tz)8T4L)) #FH555tz) #FH555tz) #FH555dj(8UEN++ #FH555dj( #FH555}8FKKK #FH555tz00 #FBHdD\,B,BCCC"dj0 #FBHdD\,B,BCCCtz00 #FBHdD\,B,BCCCtz00 #FBHeU^,D,DEEEdj// #FBHeU^,D,DEEE"TZ/ #FBHeU^,D,DEEEq #FBHeU^,D,DEEEtz. #FBHeU^,D,DEEEhhLQ    ]9C 0 0 0   JNN                ]9C 0 0 0   J!OO                ]9C 0 0 0 & & J1 % % & & & & & & & & & & & & & & &. ]:S 1 1 1 & & J1 % % & & & & & & & & & & & & & & & & & &sH) PPP' P??QQ%RRR7S""S&)S&c Tttjtdd|tdd|g}t jtjddgtjddg|}||sJd}tjt|5t jtjddgtj gd|dddn #1swxYwYtj t|tj gddS) NrrrrzMmissing values must be missing in the same location both left and right sidesrr)TFF)rr!r$rrr rrrrrrBr<rDr )rrridx2rs rtest_missing_valuesz%TestIntervalIndex.test_missing_valuesZs\ VXa6222HQ&4Q4Q4Q R  ("&!Q"&!QPVWWWzz$  ]:S 1 1 1    %A 3 3F                    #DIIrx8L8L8L/M/MNNNNNs"9C''C+.C+c||}|}tj|||d}tj||dddt t ddt jt ddg}|}t t ddt ddt jg}tj|||dd }t t jt ddt ddg}tj||dS) NrF) ascendingrrrrfirst)r  na_position)r sort_valuesr<r=rrr!r$rms rtest_sort_valuesz"TestIntervalIndex.test_sort_valuesls8!!!00""$$ fe,,,""U"33 feDDbDk222x1~~rvx1~~FGG""$$ (1a..(1a.."&!IJJ fh///""U"HH "&(1a..(1a..!IJJ fh/////rrNctd|}t|d}tj|}td|}t|d}t j|j|td||vsJtd||vsJtd||vsJttd|td |}ttd |td|}||vsJ||vsJ|td| rJ|td| sJ|td| sJtd |}t|d d }| |} tj gdd}t j | |td|}t|d d}| |} tj gdd}t j | |dS)Nz 2000-01-01rr)startr~z2000-01-01T12:00r7z 2000-01-01T12z 2000-01-02z 2000-01-03z 1999-12-31z1999-12-31T12:00r612h)rr~r)rrrrrrrrr(z2000-01-08T18:006h)r6r6rrrrr)rr rrr<r=r?rranyrr!rBrD) rrrdatesrFrHiv_trueiv_falserrs r test_datetimezTestIntervalIndex.test_datetimes,2...333)%00,444E1555 ei222"---U::::R000===="---U:::: lr * * *Ilr,J,J,J   lr * * *Ilr,J,J,J  %u$$$$>>)LR"@"@"@AAEEGGGGG~~iB???@@DDFFFFF~~i <<<==AACCCCC,444%???""6**83336BBB #FH555,444%>>>""6**8222&AAA #FH55555rctjddgddg|}tjddgddg|}||}tjgdgd|}tj|||||g}tjgdgd |}tj||hd |hz D]}tjddgddg|}||}|t |t }tj||dS) Nrrrrr2)rrrr)rrrr2)rrrrrr)rrrrrr2>rr0rr})rr rr<r=rrC)rrindex1index2rfrHrindex_other_closeds r test_appendzTestIntervalIndex.test_appends*Aq6Aq6&III*Aq6Aq6&IIIv&& ,\\\<<.s4RRR151u9$eq1uo&=>RRRr)rrr3rFc6g|]}|zz|dzzzfS)rr,r(s rr1z9TestIntervalIndex.test_is_overlapping..4QQQ151u9$eq1uo&=>QQQrr2Tc6g|]}|zz|dzzzfSr'r,r(s rr1z9TestIntervalIndex.test_is_overlapping..r,rr)rrN#(-r327<AFKPU)rNr.r/r0r1r2r3r3r4r5r6r7r8r9r:r;Zr)rris_overlappingrr ) rrr*na_valuertuplesrFrfrHabs `` rtest_is_overlappingz%TestIntervalIndex.test_is_overlappingsSRRRR RRR)&@@@#u,,,,X&FF(H1EF)&@@@#u,,,,RQQQQaQQQ)&@@@#t++++X&FF(H1EF)&@@@#t++++RQQQQaQQQ)&@@@%V#!!!!X&FF(H1EF)&@@@%!!!! Q P P P P P)!Qw???%rr?rct|}tj|}|}t t j|}tj||dSr_) rrr to_tuplesrcomasarray_tuplesafer<r=)rr?rrfrHs rtest_to_tuplesz TestIntervalIndex.test_to_tupless] f'//.v6677 fh/////rna_tupleTFctj|}||}tt j|dd}|dd}t j|||d}|rIt|tsJt|dksJtd|DsJdSt|sJdS)N)rHrrc34K|]}t|VdSr_)r )r/rs rrcz6TestIntervalIndex.test_to_tuples_na..Ss(221tAww222222r) rrrDrrErFr<r= isinstancetupler9allr )rr?rHrrfexpected_notna result_notna result_nas rtest_to_tuples_naz#TestIntervalIndex.test_to_tuples_na+s6'//11s4VCRC[AABBcrc{  lN;;;2J  #i// / //y>>Q&&&&22 22222 2 22 2 2 ?? " "? " "rctjddd}tjddd}tj||j}d}||ksJdS)Nrr3i8r(rr4@)r!r#rr nbytes)rr0rrfrHs r test_nbyteszTestIntervalIndex.test_nbytesWs_yAT*** !Qd+++*477>!!!!!!rnamerctdd||}||}tdd||}tj||dS)Nrr4)rrW)r set_closedr<r=)rrWrrrFrfrHs rtest_set_closedz!TestIntervalIndex.test_set_closed`s\q!F>>>!!,//!!Q|$GGG fh/////r bad_closed)rrLEFTTFctdd}d|}tjt|5||ddddS#1swxYwYdS)Nrr4zinvalid option for 'closed': r)r rrrrY)rr[rFrs rtest_set_closed_errorsz(TestIntervalIndex.test_set_closed_errorshsq!$$:j:: ]:S 1 1 1 ) )   Z ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )sAAActtdtd}t|g}|jrJdS)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_datespsN + , ,i8M.N.N  ( 44"000000r)r);__name__ __module__ __qualname__rr rFrr&rLrmark parametrizer!infr r to_timedeltargrnrsrxr{r rrrrrrr rrrrrrrrrrrrrr rrrr$r$rrBrArrGrQrVrZr^rcr,rrrrsK %M %q!fq!f 5 5ECCCC    'A'A'AR [ @ @ @fWdCac3RV D J|\ = = = J|\ < < < BODDD E E     0 0   0666.000   ,000  [ N1b ; ; ; N3F C C C N99Z00"V L L L N99W--q I I I  707070r&&&4E=E=E=N111 [ Jz1 - - - Jz1 > > > OHa 0 0 0 # "000< [ J|Q / / / OHad ; ; ;  0 0 0 [  % v  0 0  0 [  % 9 9 $ $ 4 3 3 [  :q111 :q\BBB!444      # "   [  % 9 9 ! ! $ $    E D D   ) )    . )'''. 0 0 0   :::6>&>&>&@OOO$000( [TD,#788'6'698'6R444,<<r()rrr2rzmatching signednessr)r!rBrrrrr )r0rs r-test_from_arrays_mismatched_signedness_raisesrlys 8IIIW - - -D HYYYh / / /E y(= > > >//!$...//////////////////s A//A36A3cbtjddgddg}t|}d|vsJdS)Nrrrr)rr dir)rFrfs rtest_dirros>  %q!fq!f 5 5E ZZF       rcpttddtddg}|||}tjddg|j}t j|||j||}t j||dS)Nrrrr() rr searchsortedr!rBr)r<rDr) listlike_boxrwrfrHs r,test_searchsorted_different_argument_classesrss HQNNHQNN; < rs""""""                           ^ 1^ 1^ 1^ 1^ 1^ 1^ 1^ 1B///222 QFS#J