U mdB @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) TestAsUnitcCs4tdd}|d|kst|d}|j|jdks:t|jtjjksLt|d}|j|jksft|j|jksvt|d}|j|jdkst|jtjjkst|d}|j|jkst|j|jkst|d}|j|jd kst|jtj jkst|d}|j|jkst|j|jks0tdS) Ndaysnsusms@Bsʚ;) r as_unitAssertionError_value_cresor NPY_FR_usvalue NPY_FR_msNPY_FR_s)selftdresrtr$e/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/pandas/tests/scalar/timedelta/test_timedelta.py test_as_units(       zTestAsUnit.test_as_unitc Csnd}t|tjj}d}tjt|d|dW5QRX|d}|j |dksXt |j tj jksjt dS)N&o>Cannot cast 106752 days 00:00:00 to unit='ns' without overflowmatchrrr) r _from_value_and_resorrrpytestraisesr rrrrr)r rr!msgr"r$r$r%test_as_unit_overflows:s z!TestAsUnit.test_as_unit_overflowsc Csvtdd}|d}tdd}||ks*t|jtjjkst|d}|j|jdksZt|j|jksjtdSNrPg@)r]nanrrrrr<r$r$r%test_truediv_numericsz TestNonNano.test_truediv_numericc Cs||dkstd||dks$tt|j}d}tjt|d||W5QRX||}|dkslttdtj j }tdtj j }||}|dkst||}|dkstdS) NrrbrPr(r)rircrd)r r!rSr.r"rerfrgr$r$r%test_floordiv_timedeltalikes    z'TestNonNano.test_floordiv_timedeltalikecCs|tjtkst|d}|j|jdks.t|j|jks>t|d}|j|jdksZt|j|jksjt|ttjtkst|td}|j|jdkst|j|jkst|td}|j|jdkst|j|jkstdSri)r]rjrrrrarrayr<r$r$r%test_floordiv_numericsz!TestNonNano.test_floordiv_numericc CsXtddd}||}|j|jks(t|j|jdks|j|kst|jj|jks t|jjttjjks:tdSrC)maxrrrr]iinfoint64rrr$r$r%test_max)szTestNonNano.test_maxcCs4td|j}|j}||ks t|j|jks0tdSrp)r r+r resolutionr)r r!r5rgr$r$r%test_resolution.s zTestNonNano.test_resolutionN)r>r?r@r,ZfixturerFrIrJr!rLrNrOrRrTrWrarhrkrlrnrorsrwryr$r$r$r%rAfs.      !rAcCstjttjdksttjjtjjks,ttj tt t j j ksHttj jtjjks\ttj tdksnttj jtjjkstdSrp)r rqrrrrrZ NPY_FR_nsrrtr]rurvrxr$r$r$r%'test_timedelta_class_min_max_resolution5s rzc@seZdZddZddZdS)TestTimedeltaUnaryOpsc Cstddd}d}tjt|d |W5QRXtjt|d|W5QRXd}tjt|d|W5QRXdS)N drIzbad operand type for unary ~r)z0ufunc 'invert' not supported for the input types)r r,r- TypeErrorrUrZ)r r!r.Zumsgr$r$r% test_invertCs z!TestTimedeltaUnaryOps.test_invertcCstddd}| tdddks"t| tdks4t| tdddksJtt||ksZtt| |ksltt| tdkstdS)Nr|r}r~z-10dZ10d)r rrMrrr$r$r%test_unary_opsRs z$TestTimedeltaUnaryOps.test_unary_opsN)r>r?r@rrr$r$r$r%r{Bsr{c@seZdZejddddgddZddZd d Zd d Z d dZ ddZ ejddddDdddDdddDdddDdddDdddDddd Dejd!e j eejgd"d#Zejd$d%d&d'gd(d)Zd*d+Zd,d-Zd.d/Zejd0d1ed2ed3fd4ed5ed6fd7ed8ed9fd:ed;ed<fd=ed>ed?fd@edAedBfdCedDedEfdFedGedHfdIedJedKfdLedMedNfg dOdPZdQdRZdSdTZejjdUdVdWeejedXej dYdZejd[ej!ej"ej#gd\d]Z$ejd$d^d_d`dagdbdcZ%dddeZ&dfdgZ'dhdiZ(djdkZ)dldmZ*dndoZ+dpdqZ,drdsZ-dtduZ.dvS)wTestTimedeltaszunit, value, expected)rgsh|#@i')rgr#@i)rgh#@lc( cCs<t||d}|j|ksttt||}|j|ks8tdS)Nr~)r rrstr)r rIrr5rgr$r$r%&test_rounding_on_int_unit_constructionas z5TestTimedeltas.test_rounding_on_int_unit_constructioncCs<td}d}t||ttj}t|s8tdS)Nz1 days, 10:11:12.100123456g  @)r tmZassert_almost_equalr9r]rjisnanr)r rngZexptr$r$r%test_total_seconds_scalarps  z(TestTimedeltas.test_total_seconds_scalarcCstdddtdfD]}|}|t|ks0t||kst|j dksNt|jdks^t|j dksnt|jdks~t|jdkst||j||j ||j ||j||j ||j||jt dj}|jdkst|j dkst|j dkst|jdkst|j dks.t|jdks>t|jdksNtdS)NcSst|tstdSrC)r\intr)rr$r$r%checksz)TestTimedeltas.test_fields..checkz1 days, 10:11:12ri@rz('Timedelta' object has no attribute '{}'r:r)minutesr2z-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 rrsecondsr1 nanosecondsr,r-AttributeErrorformatr:rr2r rMrrr8)r rrr.r!tupr$r$r% test_fieldsst             zTestTimedeltas.test_fieldscCs tddd}td|kstdS)Nrrr~Z P0DT0H0M1S)r r r)r r5r$r$r%test_iso_conversions z"TestTimedeltas.test_iso_conversioncCs`td}|jjdkst|dtks.ttd}|jjdksJt|dtks\tdS)NZnatMrvrj)r r[r`kindrastyper)r rgr$r$r%test_nat_converterss   z"TestTimedeltas.test_nat_convertersz unit, np_unitcCsg|] }|dfqS)Wr$.0rr$r$r% szTestTimedeltas.)rwcCsg|] }|dfqS)Dr$rr$r$r%rs)rr}rdayZDaysZDaycCsg|] }|dfqS)mr$rr$r$r%rs) rminuterqrtZMinuteZMinZMinutesTcCsg|] }|dfqS)rr$rr$r$r%rs)rrsecsecondSZSecondsZSecZSecondcCsg|] }|dfqS)rr$rr$r$r%rs) rr2Z millisecondZmilliZmillislZMSZ MillisecondsZ MillisecondZMilliZMillisLcCsg|] }|dfqS)rr$rr$r$r%rs) rr1 microsecondmicromicrosuZUSZ MicrosecondsZ MicrosecondZMicroZMicrosUcCsg|] }|dfqS)rr$rr$r$r%r%s) rrZ nanosecondZnanonanosnZNSZ NanosecondsZ NanosecondZNanoZNanosNwrappercs.tfddtdDdd}t|tdd}t||t|tdd}t||fddtdD}t||}t||t||}t||tt d d }tdd}||kst tdd}||kst td }||kst td }||ks*t dS) Ncsg|]}t|qSr$)r]r^)ri)np_unitr$r%r<sz3TestTimedeltas.test_unit_parser..m8[ns])r`r~csg|]}|qSr$r$)rxr~r$r%rEsrPztimedelta64[ns]2) r r]Zarangetolistr rangerZassert_index_equalr r^rr)r rIrrr5rgZstr_reprr$)rrIr%test_unit_parsers,W          zTestTimedeltas.test_unit_parserrIYyrc Csxd}tjt|dtd|W5QRXtjt|dtd|W5QRXtjt|dtddg|W5QRXdS)Nz/Units 'M', 'Y', and 'y' are no longer supportedr)r|rrP)r,r-r3r r )r rIr.r$r$r%test_unit_m_y_raisesWsz#TestTimedeltas.test_unit_m_y_raisescCstdtddksttdtddks0ttdddtddksLttdddtddkshttdddtddksttdddtddksttdddtdd kstdS) Nrrr|r~rrrr}r)r r]r^rr r$r$r%test_numeric_conversionscsz'TestTimedeltas.test_numeric_conversionscCspttddtdddks$tttddtdddksHtttddtdddksltdS) Nrrrrr0rrr)r rr]r^rrrr$r$r%test_timedelta_conversionsms  z)TestTimedeltas.test_timedelta_conversionsc Csntd}||kstd}tjt|d|dW5QRXtjt|d|jddW5QRXdS)NZ10m7sz$dtype and copy arguments are ignoredr)rYT)copy)r rZr[rr,r-r3)r r!r.r$r$r%test_to_numpy_aliasvsz"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}||}||ks&t||}||kstd}dD],\}}tjt|d||W5QRXq dS)Nr))rz, is a non-fixed frequency)rz# is a non-fixed frequency)ZfoobarzInvalid frequency: foobarr))r r,r-r3r)r rrr.r$r$r%test_round_invalids z!TestTimedeltas.test_round_invalidc Cstjd}tjtddtd}||ks0ttjd}tjtd}||ksVttjtddtjdW5QRXd d d g}tjt|dtjdW5QRXdS) Nrrrii2zvalue too larger)|z)Python int too large to convert to C longzint too big to convert) r rqceilrrtfloorr,r- OverflowErrorjoin)r rgr5r.r$r$r% test_round_implementation_boundss    z/TestTimedeltas.test_round_implementation_boundszFailing on buildsF)reasonstrictr)Z min_valueZ max_value)rJmethodcCst|}t|}||d|ks$t||d}d}t||j|ksJt|j|dks\t||d}d}t||j|kst|j|dkst||d}d}t||j|kst|j|dkst||d }d }t||j|kst|j|dkst||d }d }t||j|ks.t|j|dksBt||d }d}t||j|ksjt|j|dks~tdS)NrrrrrrrrrqlXp7hl qa rl"R:)r]rvr rrMr)r rJrr!r"rr$r$r%test_round_sanitys6       z TestTimedeltas.test_round_sanityrrrrcCstd|}|d}|tdks(t|j|jks8t|d}|tdksRt|j|jksbt|d}|tdks|t|j|jkstdS)Nrrqrz1 days 02:34:00)r rrrrrr)r rIr!r"r$r$r%test_round_non_nanos   z"TestTimedeltas.test_round_non_nanocCs,tddd}t|tstt|ts(tdS)Nr|r}r~)r r\rrrrr$r$r% test_identitys zTestTimedeltas.test_identityc Cszdd}tdtddks ttdtddks8ttdtddksPttd tddkshttd td dksttd td dksttd td dksttdtddksttdtddksttdtddksttdtddksttdtddks,ttdtddksFttdtddks`ttdtddkszttdtddksttdtdd ksttdtddksttdtd dksttd!td"dksttd#td$dksttd%|td&d'ks6ttd(|td&d' ksVttd)|td&d'kstttd*|tdd'ksttd+|tdd'ksttd,|td d'ksttd-|tdd'ksttd.|tdd'ks ttd/|tdd' ks*td0}tjt|d1td2W5QRXd3}tjt|d1td4W5QRXdS)5NcSs |dSNrrvr$r$r%convsz9TestTimedeltas.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^rr,r-r3)r rr.r$r$r%test_short_format_converterssJ  z+TestTimedeltas.test_short_format_convertersc CsRdd}tdd}td||ks(ttd||ks.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 rr,r-r3)r rZd1r.r$r$r%test_full_format_converters7s*  "  z*TestTimedeltas.test_full_format_converterscCs"td}t|}||kstdS)Nz1 days 10:11:12.0123456)r rZround_trip_pickler)r rZv_pr$r$r% test_pickleUs zTestTimedeltas.test_picklecCstdd}tdd}t|t|ks(t|di}||dks@tddtdD}tdd |Dshttdd }t|t|kstdS) NrrrrPcSs g|]}tddt|dqS)rrr)r )rrr$r$r%rcsz?TestTimedeltas.test_timedelta_hash_equality..css"|]}t|t|kVqdSrC)hashrU)rr!r$r$r% dsz>TestTimedeltas.test_timedelta_hash_equality..r)r rrrrallrU)r rr!r}ZtdsZns_tdr$r$r%test_timedelta_hash_equalityZs   z+TestTimedeltas.test_timedelta_hash_equalityc Csttj}ttj}|jtdks&t|jtjks6t|tddtksLtd}t j t |d|tddW5QRXt j t |d|tddW5QRXt|jdd}|tkstd}t j t |dt|jddW5QRXd}t j t |dt|jddW5QRXdS)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 rqrtrrrri8maxrr,r-rr )r Zmin_tdZmax_tdr.r!r$r$r%test_implementation_limitsjs$   z)TestTimedeltas.test_implementation_limitscCstddksttddks(ttddks@0gz-2Sgz5.324SgK@g#B ;)r r9rrr$r$r%test_total_seconds_precisions z+TestTimedeltas.test_total_seconds_precisioncCstddjdksttdddjdks*ttdddjdks@ttddd jd ksVttddd jd kslttddd jdksttdddjdkstdS)Nrrrrr:r)rrr)rrr)rr2r)rr1r)rrr)r Zresolution_stringrrr$r$r%test_resolution_stringsz%TestTimedeltas.test_resolution_stringcCs@tddd}|j}|tddks$ttj}|tddksr?r@r,mark parametrizerrrrrrr]rmlistpdIndexrrrrrr rrrZxfailrstZintegersrrrrrrrrrrrrrrrrrr$r$r$r%r`s  J   .?S       $ 2  rzvalue, expectedZ10STz-10Sr|rr~FrcCst||kstdSrC)boolr)rr5r$r$r%test_truthinesssrcCsZtddd}|jd}||j7}|d9}||j7}|d9}||j7}|j}||ksVtdS)Nl?2\BT,rr~r7rr)r rrr1rrr)r!rgr5r$r$r%"test_timedelta_attribute_precisions     r)$__doc__datetimerZ hypothesisrrrnumpyr]r,Z pandas._libsrZpandas._libs.tslibsrrZpandas._libs.tslibs.dtypesrZ pandas.errorsr Zpandasrr r r Zpandas._testingZ_testingrr rArzr{rrrrrr$r$r$r%sD     JP K