gL i)ddlmZddlZddlZddlZddlmZddlm Z ddl m Z m Z GddZGddZGd d Zy) ) timedeltaN) Timedelta) DatetimeArrayTimedeltaArrayceZdZejgddZejdZdZdZejjde jdZ d Zd Zd Zd Zejjd ej&ddej&ddgdZdZdZdZdZdZdZdZdZdZdZy) TestNonNano)smsus)paramsc|jSN)param)selfrequests i/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/pandas/tests/arrays/test_timedeltas.pyunitzTestNonNano.units }}ctjdtjjd|d}t j ||j S)Ndtypem8[])nparangeint64viewr _simple_newr)rrarrs rtdazTestNonNano.tdas@ii*//#dV1 >))#SYY??rctjdtjjd|d}t j ||j }|j |j k(sJ|dj|k(sJy)Nrrrrr)rrrrrrrr)rrr r!s r test_non_nanozTestNonNano.test_non_nanoskii*//#dV1 >((CII>yyCII%%%1v{{d"""rc2tjtd5|jddddt j |}tjtd5|jddddy#1swYTxYw#1swYyxYw)NzSupported unitsmatchD)pytestraises ValueErroras_unitpdIndex)rr!tdis rtest_as_unit_raiseszTestNonNano.test_as_unit_raises sv ]]:-> ?  KK  hhsm ]]:-> ?  KK        sB&B B  Bfieldc|jjd}tj||j}t ||}t ||}t j||yNm8[ns]r)_ndarrayastyperrrgetattrtmassert_numpy_array_equal)rr!r0as_nanotda_nanoresultexpecteds r test_fieldszTestNonNano.test_fields)sT,,%%h/!--gW]]Ke$8U+ ##FH5rc|jjd}tj||j}|j }|j }t j||yr2)r4r5rrrto_pytimedeltar7r8)rr!r9r:r;r<s rtest_to_pytimedeltazTestNonNano.test_to_pytimedelta2sX,,%%h/!--gW]]K##%**, ##FH5rc|jjd}tj||j}|j }|j }t j||yr2)r4r5rrr total_secondsr7r8)rrr!r9r:r;r<s rtest_total_secondszTestNonNano.test_total_seconds:sX,,%%h/!--gW]]K""$))+ ##FH5rctdj}tjtdgjd}||k(sJy)Nz2 minr)rrBr,array)rr<r;s r"test_timedelta_array_total_secondsz.TestNonNano.test_timedelta_array_total_secondsBsGW%3359W-./==?B!!!rc@tjdgjd}tjdgjd}||z jt j ddz }||z j jj}||k(sJy)Nz2145-11-02 06:00:00datetime64[ns]z2145-11-02 07:06:00r )r,Seriesr5valuesr timedelta64dtrB)r start_timeend_timer<r;s rtest_total_seconds_nanosecondsz*TestNonNano.test_total_seconds_nanosecondsIsYY 567>>?OP 99345<<=MNz)11BNN1c4JJZ'++99;BB!!!rnatNaTnsr cL||z}t|tsJ|j|jk(sJ|jj sJ||z}t|tsJ|j|jk(sJ|jj sJyr) isinstancer_cresoisnaall)rrQr!r;s r test_add_nat_datetimelike_scalarz,TestNonNano.test_add_nat_datetimelike_scalarQss&-000}} ***{{}  """s&-000}} ***{{}  """rc|tjz}t|tsJ|j|jk(sJ|j j sJtj|z}t|tsJ|j|jk(sJ|j j sJyr)r,rRrUrrVrWrX)rr!r;s rtest_add_pdnatzTestNonNano.test_add_pdnat_srvv&.111}} ***{{}  """#&.111}} ***{{}  """rc0tjd|jd}|jd|z}||z}tj||||z}tj|||t dz }|j |jz}tj||jjdj|j}||z}tj||||z}tj||y)Nz 2016-01-01)tzrSrIrUTC)r, Timestampr+r7assert_extension_array_equalrr4asm8rrr tz_localize tz_convertr])rr!tz_naive_fixturetsr<res exp_valuesr;s rtest_add_datetimelike_scalarz(TestNonNano.test_add_datetimelike_scalarls \\,+; < D DT J;;t$r)Bh ''X63h ''X6 il\\BGG+  % %j 8H8H I [  Z   r ''9c ''9rcd}||z}tj|j|z|j}t j |||j |j k(sJyNrrrr4rr7r`rVrr!otherr;r<s rtest_mul_scalarzTestNonNano.test_mul_scalarUu!--cllU.B#))T ''9}} ***rctjt|}||z}tj|j |z|j }tj|||j|jk(sJyNr rrlenrrr4rr7r`rVrms rtest_mul_listlikezTestNonNano.test_mul_listlikeb #c(#u!--cllU.B#))T ''9}} ***rc.tjt|}||jtz}t j |j|z|j}tj|||j|jk(sJyrr) rrrtr5objectrrr4rr7r`rVrms rtest_mul_listlike_objectz$TestNonNano.test_mul_listlike_objectsl #c(#u||F++!--cllU.B#))T ''9}} ***rcd}||z }tj|j|z |j}t j |||j |j k(sJyrjrlrms rtest_div_numeric_scalarz#TestNonNano.test_div_numeric_scalarrprctd}||z }|jtjddz }t j ||y)NrIsecondsr )rr4rrLr7r8rms rtest_div_td_scalarzTestNonNano.test_div_td_scalars>!$u<<"..C"88 ##FH5rctjt|}||z }tj|j |z |j }tj|||j|jk(sJyrrrsrms rtest_div_numeric_arrayz"TestNonNano.test_div_numeric_arrayrvrc|j|jdz}||z }|j|z }tj||y)N)r4r7r8rms rtest_div_td_arrayzTestNonNano.test_div_td_arrays@ s||B//u<<%' ##FH5rc|jd}|dz}||z}tj||||z}tj|||dz}||z }tj||||z }tj||y)Nr3rkr)r5r7r`)rr!r:r<rfs rtest_add_timedeltaarraylikez'TestNonNano.test_add_timedeltaarraylikes::h'a<n ''X6Hn ''X6a<Hn ''X6n ''X6rN)__name__ __module__ __qualname__r(fixturerr!r#r/mark parametrizer _field_opsr=r@rCrFrPr datetime64rYr[rhroruryr{rrrrrrrrsV^^,-. ^^@@# [[Wn&?&?@6A666"" [[  eT*MBMM%,FG # # #:0++++6 +6 7rrceZdZejj deejejddgdZ dZ ejj de de dje djgd Zejj d dejdd ej d ej$d dej&dddzdzdzej&ddzdzdzj)dej$d j+dg ejj dddgdZy)TestTimedeltaArrayruint32uint64ctjtdtdgd}tj|tj k7r6t jtd5|j|dddy|j|}|jjd}tj||y#1swYyxYw)N1h2hr3rzDo obj.astype\('int64'\)r%i8)r_from_sequencerrrrr(r) TypeErrorr5r4rr7r8)rrr r;r<s rtest_astype_intz"TestTimedeltaArray.test_astype_ints++ t_io .h  88E?bhh &y0KL " 5! " E"<<$$T* ##FH5  " s -C  Cctjdddj}td|d<|jJy)Nrrkhperiodsfreqr)r,timedelta_range_datarr)ras rtest_setitem_clears_freqz+TestTimedeltaArray.test_setitem_clears_freqs7   tQS 9 ? ?!vv~~robjrIr}c~tjddd}|j}||d<|dtdk(sJy)N2 DaysrrrrIr})r,rrr)rrr.r s rtest_setitem_objectsz'TestTimedeltaArray.test_setitem_objectssA  13?iiA1v1----rrng?rRz 2021-01-01invalid rrʚ;rHr'indexTFc\tjdddzdzdz}tj|dj}|rtj |}d j d d g}tjt| 5|j|dddy#1swYyxYw) Nrrrrrrr')r|z0searchsorted requires compatible dtype or scalarzrs@  r7r7jA$A$H1;1;r