o dB @s<dZddlmZddlmZmZddlZddl Z ddl m Z ddl m Z mZddlmZddlmZddlZdd lmZmZmZddlmZGd d d ZGd d d ZddZGdddZGdddZe j !deddfeddfeddddfeddddfeddddfeddfe dfgddZ"dd Z#dS)!z test the scalar Timedelta ) timedelta)given strategiesN)lib)NaTiNaT)NpyDatetimeUnit)OutOfBoundsTimedelta) TimedeltaTimedeltaIndex to_timedeltac@s,eZdZddZddZddZddZd S) TestAsUnitcCs.tdd}|d|usJ|d}|j|jdksJ|jtjjks&J|d}|j|jks3J|j|jks;J|d}|j|jdksJJ|jtjjksSJ|d}|j|jks`J|j|jkshJ|d}|j|jd kswJ|jtjjksJ|d}|j|jksJ|j|jksJdS) Ndaysnsusms@Bsʚ;) r as_unit_value_cresor NPY_FR_usvalue NPY_FR_msNPY_FR_s)selftdresrtr#`/app/.heroku/python/lib/python3.10/site-packages/pandas/tests/scalar/timedelta/test_timedelta.py test_as_units(       zTestAsUnit.test_as_unitcCsd}t|tjj}d}tjt|d |dWdn1s#wY|d}|j |dks6J|j tj jks?JdS)N&o>Cannot cast 106752 days 00:00:00 to unit='ns' without overflowmatchrrr) r _from_value_and_resorrrpytestraisesr rrrr)rrr msgr!r#r#r$test_as_unit_overflows:s  z!TestAsUnit.test_as_unit_overflowscCstdd}|d}tdd}||ksJ|jtjjksJ|jdks%Jtjt dd|jddd WddS1s?wYdS) Ni microsecondsrr) millisecondszCannot losslessly convert unitsr(F)Zround_ok) r rrrrrrr+r, ValueErrorrr r!expectedr#r#r$test_as_unit_roundingGs    "z TestAsUnit.test_as_unit_roundingcCstddd}|jdksJ|jdksJ|jjdksJ|jdks%J|dks-J|d}|jdks9J|jjdksAJ|jjdksIJ|jdksPJ|jdksWJ|dks_JdS) Nrrri\&Qrl`;Pr) r rrr components_d total_secondshours_hrr r!r#r#r$test_as_unit_non_nanoTs z TestAsUnit.test_as_unit_non_nanoN)__name__ __module__ __qualname__r%r.r5r=r#r#r#r$r s  r c@seZdZejgddddZejddZejddZejd d Zd d Z d dZ ddZ ddZ ddZ ddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)S)* TestNonNano)rrr)paramscCs|jSN)param)rrequestr#r#r$unit_strgszTestNonNano.unit_strcCsd|}tt|jS)NZNPY_FR_)getattrrr)rrFattrr#r#r$unitks  zTestNonNano.unitcCs<d}|tjjkr |}|S|tjjkr|d}|S|d}|S)Nr&rr)rrrr)rrIrrr#r#r$valqs  zTestNonNano.valcCs t||SrC)r r*)rrIrJr#r#r$r }s zTestNonNano.tdcCs:t||}|j|ks J|j|ksJ|jdksJdS)N)r r*rrr)rrIrJr r#r#r$test_from_value_and_resos z$TestNonNano.test_from_value_and_resocCs6t|j|ks J| j|ksJ| j|ksJdSrC)absr)rr rIr#r#r$test_unary_non_nanoszTestNonNano.test_unary_non_nanocCs2||}td|}||ksJ|j|ksJdS)Nr)r r*r)rr rIr!r4r#r#r$test_sub_preserves_resos  z#TestNonNano.test_sub_preserves_resocCs.|d}|j|jdksJ|j|ksJdS)N)rrrr rIr!r#r#r$test_mul_preserves_resosz#TestNonNano.test_mul_preserves_resocCs@tddd}||ks J||ksJ||krJ||ksJdS)Nir)rrIr )rr otherr#r#r$test_cmp_cross_resos    zTestNonNano.test_cmp_cross_resocCs2|}tdd}t|tusJ||ksJdS)NrKr)to_pytimedeltartyper3r#r#r$test_to_pytimedeltas zTestNonNano.test_to_pytimedeltacCs|||jfD]=}t|tjsJ|d|jksJ|tj j kr,|j dks+Jq |tj j kr:|j dks9Jq |tj j krG|j dksGJq dS)Ni8m8[s]zm8[ms]zm8[us])to_timedelta64to_numpyZasm8 isinstancenp timedelta64viewrrrrdtyperrrQr#r#r$test_to_timedelta64s   zTestNonNano.test_to_timedelta64c Cs||dksJd||dksJt|j}d}tjt|d ||Wdn1s/wY||}||}||ksHJtdtjj }tdtj j }||}|dksbJ||}|dkslJdS)Nr@z?Cannot cast 106752 days 00:00:00 to unit='ns' without overflow.r(2gMbP?r r rr+r,r rVr*rrrr) rr rTr-r!r4leftrightresultr#r#r$test_truediv_timedeltalikes      z&TestNonNano.test_truediv_timedeltalikecCsn|tjtus J|d}|j|jdksJ|j|jksJ|d}|j|jdks-J|j|jks5JdSNrPg@)r^nanrrrr<r#r#r$test_truediv_numericsz TestNonNano.test_truediv_numericcCs||dksJd||dksJt|j}d}tjt|d ||Wdn1s/wY||}|dks@Jtdtjj }tdtj j }||}|dksZJ||}|dksdJdS) NrrcrPr'r(rirdre)rr rTr-r!rfrgrhr#r#r$test_floordiv_timedeltalikes     z'TestNonNano.test_floordiv_timedeltalikecCs|tjtus J|d}|j|jdksJ|j|jksJ|d}|j|jdks-J|j|jks5J|ttjtusAJ|td}|j|jdksRJ|j|jksZJ|td}|j|jdkskJ|j|jkssJdSrj)r^rkrrrarrayr<r#r#r$test_floordiv_numericsz!TestNonNano.test_floordiv_numericcCstddd}||}|j|jksJ|j|jdksJ||}|j|jks*J|j|jdks4J||}|j|jks@J|j|jdksJJ||}|j|jksVJ|jd|jks`Jtd}d}tjt|d ||Wdn1s|wYtjt|d ||Wdn1swYtjt|d ||Wdn1swYtjt|d ||WddS1swYdS)Nrrrir'r()r rrrr+r,r )rr rTrhZother2r-r#r#r$test_addsub_mismatched_resos6    "z'TestNonNano.test_addsub_mismatched_resocCs:|j|ksJ|jj|jksJ|jjtjdksJdSNr)minrrrrr r#r#r$test_min$szTestNonNano.test_mincCs>|j|ksJ|jj|jksJ|jjttjjksJdSrC)maxrrr^iinfoint64rsr#r#r$test_max)szTestNonNano.test_maxcCs4td|j}|j}||ksJ|j|jksJdSrq)r r*r resolution)rr r4rhr#r#r$test_resolution.s zTestNonNano.test_resolutionN)r>r?r@r+ZfixturerFrIrJr rLrNrOrRrUrXrbrirlrmrorprtrxrzr#r#r#r$rAfs0      ! rAcCstjttjdks JtjjtjjksJtjtt t j jks$Jtjjtjjks.Jtj tdks7Jtj jtjjksAJdSrq) r rrrrrrZ NPY_FR_nsrrur^rvrwryr#r#r#r$'test_timedelta_class_min_max_resolution5s r{c@seZdZddZddZdS)TestTimedeltaUnaryOpscCstddd}d}tjt|d |Wdn1swYtjt|d |Wdn1s9wYd}tjt|d|WddS1sXwYdS)N drIzbad operand type for unary ~r(z0ufunc 'invert' not supported for the input types)r r+r, TypeErrorrVr[)rr r-Zumsgr#r#r$ test_invertCs   "z!TestTimedeltaUnaryOps.test_invertcCstddd}| tdddksJ| tdksJ| tdddks%Jt||ks-Jt| |ks6Jt| tdksAJdS)Nr}r~rz-10dZ10d)r rMrsr#r#r$test_unary_opsRs z$TestTimedeltaUnaryOps.test_unary_opsN)r>r?r@rrr#r#r#r$r|Bs r|c@seZdZejdgdddZddZddZd d Z d d Z d dZ ejddddDdddDdddDdddDdddDdddDdddDejde j eejgd d!Zejd"gd#d$d%Zd&d'Zd(d)Zd*d+Zejd,d-ed.ed/fd0ed1ed2fd3ed4ed5fd6ed7ed8fd9ed:ed;fdfd?ed@edAfdBedCedDfdEedFedGfdHedIedJfg dKdLZdMdNZdOdPZejjdQdRdSeejedTej dUdVejdWej!ej"ej#gdXdYZ$ejd"gdZd[d\Z%d]d^Z&d_d`Z'dadbZ(dcddZ)dedfZ*dgdhZ+didjZ,dkdlZ-dmdnZ.doS)pTestTimedeltaszunit, value, expected))rgsh|#@i')rgr#@i)rgh#@lc( cCs<t||d}|j|ks Jtt||}|j|ksJdS)Nr)r rstr)rrIrr4rhr#r#r$&test_rounding_on_int_unit_constructionas z5TestTimedeltas.test_rounding_on_int_unit_constructioncCs<td}d}t||ttj}t|sJdS)Nz1 days, 10:11:12.100123456g  @)r tmZassert_almost_equalr9r^rkisnan)rrngZexptr#r#r$test_total_seconds_scalarps  z(TestTimedeltas.test_total_seconds_scalarcCstdddtdfD]H}|}|t|ksJ||ksJt|tr(t|tr*J|t|jdks5J|}|t|jdksDJ||ksJJt|tjsRJq td}||ks_JdS)Nr}r~rz1 days, 10:11:12.012345rz1 days, 10:11:12.012345678)r rVr]rr^r_rr[)rr ZpydtZtd64r#r#r$test_conversionys  zTestTimedeltas.test_conversioncCsdd}td}|jdksJ|jdksJ|jdksJ|jdks$Jd}tjt|dd  |j Wdn1s>wYtjt|d d  |j Wdn1s[wYtjt|d d  |j Wdn1sxwY||j||j||j||jt d }t |t d ksJt|dksJ| t dksJt d j dksJt d jdksJtd}|jdksJ|jdksJ|jdksJ|jdksJd}tjt|dd  |j Wdn1swYtjt|d d  |j Wdn 1swYtjt|d d  |j Wdn 1s9wYtddj}|jdksLJ|j dksTJ|j dks\J|jdksdJ|j dkslJ|jdkstJ|jdks|J||j||j ||j ||j||j ||j||jt dj}|jdksJ|j dksJ|j dksJ|jdksJ|j dksJ|jdksJ|jdksJdS)NcSst|tsJdSrC)r]int)rr#r#r$checksz)TestTimedeltas.test_fields..checkz1 days, 10:11:12ri@rz('Timedelta' object has no attribute '{}'r:r(minutesr1z-1 days, 10:11:12z13:48:48z-1 days +10:11:12z0 days 13:48:48lg4lg4z-1 days, 10:11:12.100123456iir;iz -1 days 1 us)r rsecondsr0 nanosecondsr+r,AttributeErrorformatr:rr1r rMrrr7)rrrr-r tupr#r#r$ test_fieldss             zTestTimedeltas.test_fieldscCs tddd}td|ksJdS)NrrrZ P0DT0H0M1S)r r )rr4r#r#r$test_iso_conversions z"TestTimedeltas.test_iso_conversioncCs`td}|jjdksJ|dtksJtd}|jjdks%J|dtks.JdS)NZnatMrwrk)r r\rakindastyper)rrhr#r#r$test_nat_converterss  z"TestTimedeltas.test_nat_convertersz unit, np_unitcCg|]}|dfqS)Wr#.0rr#r#r$ zTestTimedeltas.)rwcCr)Dr#rr#r#r$rr)rr~rdayZDaysZDaycCr)mr#rr#r#r$r) rminuterrrtZMinuteZMinZMinutesTcCr)rr#rr#r#r$rr)rrsecsecondSZSecondsZSecZSecondcCr)rr#rr#r#r$rr) rr1Z millisecondZmilliZmillislZMSZ MillisecondsZ MillisecondZMilliZMillisLcCr)rr#rr#r#r$rr) rr0 microsecondmicromicrosuZUSZ MicrosecondsZ MicrosecondZMicroZMicrosUcCr)rr#rr#r#r$r%r) rrZ nanosecondZnanonanosnZNSZ NanosecondsZ NanosecondZNanoZNanosNwrappercs*tfddtdDdd}t|tdd}t||t|tdd}t||fddtdD}t||}t||t||}t||tt d d }tdd}||ksmJtdd}||ksyJtd }||ksJtd }||ksJdS) Ncsg|]}t|qSr#)r^r_)ri)np_unitr#r$r<z3TestTimedeltas.test_unit_parser..m8[ns])rarcsg|]}|qSr#r#)rxrr#r$rErrPztimedelta64[ns]2) r r^Zarangetolistr rangerZassert_index_equalr r_r)rrIrrr4rhZstr_reprr#)rrIr$test_unit_parsers,W           zTestTimedeltas.test_unit_parserrI)YyrcCsd}tjt|d td|Wdn1swYtjt|d td|Wdn1s5wYtjt|dtddg|WddS1sTwYdS)Nz/Units 'M', 'Y', and 'y' are no longer supportedr(r}rrP)r+r,r2r r )rrIr-r#r#r$test_unit_m_y_raisesWs  "z#TestTimedeltas.test_unit_m_y_raisescCstdtddks JtdtddksJtdddtddks&Jtdddtddks4JtdddtddksBJtdddtddksPJtdddtdd ks^JdS) Nrrr}rrrrr~r)r r^r_rr#r#r$test_numeric_conversionscs z'TestTimedeltas.test_numeric_conversionscCspttddtdddksJttddtdddks$Jttddtdddks6JdS) Nrrrrr/rrr)r rr^r_rrr#r#r$test_timedelta_conversionsms  (z)TestTimedeltas.test_timedelta_conversionscCstd}||ksJd}tjt|d |dWdn1s'wYtjt|d|jddWddS1sEwYdS)NZ10m7sz$dtype and copy arguments are ignoredr(rZT)copy)r r[r\r+r,r2)rr r-r#r#r$test_to_numpy_aliasvs "z"TestTimedeltas.test_to_numpy_aliasz freq,s1,s2r1 days 02:34:56.789123456-1 days 02:34:56.789123456rz1 days 02:34:56.789123000z-1 days 02:34:56.789123000rz1 days 02:34:56.789000000z-1 days 02:34:56.789000000r1 days 02:34:57z-1 days 02:34:57Z2Sz1 days 02:34:56z-1 days 02:34:56Z5Sz1 days 02:34:55z-1 days 02:34:55r1 days 02:35:00z-1 days 02:35:00Z12Tz1 days 02:36:00z-1 days 02:36:00Hz1 days 03:00:00z-1 days 03:00:00r~z1 daysz-1 dayscCs@td}td}||}||ksJ||}||ksJdS)Nrr)r round)rfreqs1s2t1t2r1r2r#r#r$ test_rounds    zTestTimedeltas.test_roundc CsRtd}dD] \}}tjt|d ||Wdn1s!wYqdS)Nr))rz, is a non-fixed frequency)rz# is a non-fixed frequency)ZfoobarzInvalid frequency: foobarr()r r+r,r2r)rrrr-r#r#r$test_round_invalids  z!TestTimedeltas.test_round_invalidcCstjd}tjtddtd}||ksJtjd}tjtd}||ks+JtjtddtjdWdn1sCwYdd d g}tjt|dtjdWddS1shwYdS) Nrrrii2zvalue too larger(|z)Python int too large to convert to C longzint too big to convert) r rrceilrufloorr+r, OverflowErrorjoin)rrhr4r-r#r#r$ test_round_implementation_boundss    "z/TestTimedeltas.test_round_implementation_boundszFailing on buildsF)reasonstrictr)Z min_valueZ max_value)rJmethodcCsxt|}t|}||d|ksJ||d}d}t||j|ks%J|j|dks.J||d}d}t||j|ksAJ|j|dksJJ||d}d}t||j|ks]J|j|dksfJ||d }d }t||j|ksyJ|j|dksJ||d }d }t||j|ksJ|j|dksJ||d }d}t||j|ksJ|j|dksJdS)NrrrrrrrrrrlXp7hl qa rl"R:)r^rwr rMr)rrJrr r!rr#r#r$test_round_sanitys6       z TestTimedeltas.test_round_sanity)rrrrcCstd|}|d}|tdksJ|j|jksJ|d}|tdks)J|j|jks1J|d}|tdks>J|j|jksFJdS)Nrrrrz1 days 02:34:00)r rrrrr)rrIr r!r#r#r$test_round_non_nanos   z"TestTimedeltas.test_round_non_nanocCs,tddd}t|ts Jt|tsJdS)Nr}r~r)r r]rrsr#r#r$ test_identitys zTestTimedeltas.test_identitycCsdd}tdtddksJtdtddksJtdtddks(Jtd tddks4Jtd td dks@Jtd td dksLJtd td dksXJtdtddksdJtdtddkspJtdtddks|JtdtddksJtdtddksJtdtddksJtdtddksJtdtddksJtdtddksJtdtdd ksJtdtddksJtdtd dksJtd!td"dksJtd#td$dksJtd%|td&d'ksJtd(|td&d' ks!Jtd)|td&d'ks0Jtd*|tdd'ks?Jtd+|tdd'ksNJtd,|td d'ks]Jtd-|tdd'kslJtd.|tdd'ks{Jtd/|tdd' ksJd0}tjt|d1 td2Wdn 1swYd3}tjt|d1 td4WddS1swYdS)5NcS |dSNrrvr#r#r$conv z9TestTimedeltas.test_short_format_converters..conv10r}rZ10ns100dZ100nsZ1000rZ1000nsZ1000NSZ10usi'Z100usiZ1000usrZ1000UsZ1000uSZ1msZ10msiZ100msiZ1000msrz-1sZ1sZ10sld( Z100slh]Z1000slJ)Z1drrz-1dZ1DZ10DZ100DZ1000DZ10000Dz 10000D z - 10000D zinvalid unit abbreviationr(Z1foozunit abbreviation w/o a numberZfoo)r r^r_r+r,r2)rrr-r#r#r$test_short_format_converterssN    $z+TestTimedeltas.test_short_format_converterscCsddd}tdd}td||ksJtd||ksJtd|| ks)Jtd|tdd ks7Jtd |td d ksEJtd |td d ksSJtd |tddksaJtd|| tdd ksrJtd||td d ksJtd||tddksJd}tjt|d tdWddS1swYdS)NcSrrrrr#r#r$r8rz8TestTimedeltas.test_full_format_converters..convrrZ1daysz1days,z- 1days,z00:00:01rz06:00:01iaTz 06:00:01.0z 06:00:01.01iIrz- 1days, 00:00:01z1days, 06:00:01z1days, 06:00:01.01zhave leftover unitsr(z - 1days, 00)r^r_r r+r,r2)rrZd1r-r#r#r$test_full_format_converters7s,   "   "z*TestTimedeltas.test_full_format_converterscCs"td}t|}||ksJdS)Nz1 days 10:11:12.0123456)r rZround_trip_pickle)rrZv_pr#r#r$ test_pickleUs zTestTimedeltas.test_picklecCstdd}tdd}t|t|ksJ|di}||dks JddtdD}tdd |Ds4Jtdd }t|t|ksEJdS) NrrrrPcSs g|] }tddt|dqS)rrrrS)rrr#r#r$rcs z?TestTimedeltas.test_timedelta_hash_equality..css$|] }t|t|kVqdSrC)hashrV)rr r#r#r$ ds"z>TestTimedeltas.test_timedelta_hash_equality..r)r rrrallrV)rrr r~ZtdsZns_tdr#r#r$test_timedelta_hash_equalityZs   z+TestTimedeltas.test_timedelta_hash_equalitycCsnttj}ttj}|jtdksJ|jtjksJ|tddtus&Jd}tj t |d|tddWdn1sAwYtj t |d|tddWdn1s_wYt|jdd}|tusrJd}tj t |dt|jddWdn1swYd}tj t |dt|jddWddS1swYdS)Nrrzint too (large|big) to convertr(rPzACannot cast -9223372036854775809 from ns to 'ns' without overflowz@Cannot cast 9223372036854775808 from ns to 'ns' without overflow) r rrrurrri8maxrr+r,rr )rZmin_tdZmax_tdr-r r#r#r$test_implementation_limitsjs,   "z)TestTimedeltas.test_implementation_limitscCstddks JtddksJtddksJtddks(Jtddd ks4Jdtdd ks@JdS) NZ30Sg>@0gz-2Sgz5.324SgK@g#B ;)r r9rr#r#r$test_total_seconds_precisions z+TestTimedeltas.test_total_seconds_precisioncCstddjdks JtdddjdksJtdddjdks Jtddd jd ks+Jtddd jd ks6Jtddd jdksAJtdddjdksLJdS)Nrrrrr:r)rrr)rrr)rr1r)rr0r)rrr)r Zresolution_stringrr#r#r$test_resolution_stringsz%TestTimedeltas.test_resolution_stringcCs@tddd}|j}|tddksJtj}|tddksJdS)Nrr)r)r ry)rr rhr#r#r$test_resolution_deprecateds z)TestTimedeltas.test_resolution_deprecatedN)/r>r?r@r+mark parametrizerrrrrrr^rnlistpdIndexrrrrrr rrrZxfailrstZintegersrrrrrrrrrrrrrrrrrr#r#r#r$r`s  J   .?S         $ 2 rzvalue, expectedZ10STz-10Sr}rrFrcCst||usJdSrC)bool)rr4r#r#r$test_truthinesssr cCsZtddd}|jd}||j7}|d9}||j7}|d9}||j7}|j}||ks+JdS)Nl?2\BT,rrr6rr)r rrr0rr)r rhr4r#r#r$"test_timedelta_attribute_precisions     r )$__doc__datetimerZ hypothesisrrr numpyr^r+Z pandas._libsrZpandas._libs.tslibsrrZpandas._libs.tslibs.dtypesrZ pandas.errorsr Zpandasrr r r Zpandas._testingZ_testingrr rAr{r|rrrr r r#r#r#r$sF     JP K