U
|
®ý°d»E ã @ s¶ d dl mZ d dlmZ d dlZd dlZd dlmZ d dl m
|
Z
|
d dl m Z m Z mZmZ dd Zd d
|
Zd d Zd d Zdd Zdd Zej dedddddddd ¡ ¡ej d ejejej ej ej ej g¡d d Z ej d d d!d"d#d$d%d&d'd(d)d*g ¡d+d, Z!d-d. Z"ej d/d0d1d2d3d4d5g¡d6d7 Z#ej d/d8d9d:d;d<g¡d=d> Z$ej d?d@e dAdBdCdDdDdEdFfdGe dHdBdCdDdDdEdFfdIe dJdEdKdLdMfdNe dOdPfdQe dDdRfdSe dTdUfdVe dTdWfdXe dTdYdZdZdMfd[e dTd\fd]e d d\fd^e d_d d`fdae dTdbfdce dTdTddfdee d_dbfdfe dgd\fdhe dTdbfdie dHdCdjfdke dHdldjfdme dAdldjfg¡dndo Z%ej dpdqdrdsdtdudvdwg¡dxdy Z&ej dze d{dPd|fe dTdTdTdTdTdTdTdTd}d~fe dTdRe dTdP dfe dTdRe dTdP dfe dTdRdLe ddP dfg¡dd Z'd
d Z(dd Z)dd Z*ej de dde+dffedde+dffedddCgde+dffg¡dd Z,ej ddd dD ¡dd Z-dd Z.dd Z/dd Z0dS )é )Ú timedelta)ÚproductN)ÚOutOfBoundsTimedelta)ÚNpyDatetimeUnit)ÚNaTÚ TimedeltaÚoffsetsÚ to_timedeltac C sP t jtdd tddd W 5 Q R X t jtdd tddd W 5 Q R X d S )Nzwithout overflow©Úmatchì ÈNÎZoÚW©Úunitg ÈNgmÁ«C)ÚpytestÚraisesr r © r r úfd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/scalar/timedelta/test_constructors.pyÚ'test_construct_with_weeks_unit_overflow s r c C s~ t dd¡} tjtdd t| dd W 5 Q R X tjtdd t| dd W 5 Q R X tjtdd t| W 5 Q R X d S )Nl éJ.m´ Úhz123456789000000000 hoursr
|
Zpsr Úns)ÚnpÚ timedelta64r r r r )Úobjr r r Ú"test_construct_from_td64_with_unit s r c C s´ t dd¡} t| }|j| d¡ks(t|jtjj ks:tt dd¡}t|}|jtj
|
j ks`t||kslt|j dksztt dd¡}t|}| ¡ dkst|jtj j ks°td S ) Ni90 ÚmsÚi8iÒ ÚDr Úusl J)£)r r r Ú_valueÚviewÚAssertionErrorÚ_cresor Ú NPY_FR_msÚvalueÚNPY_FR_sÚdaysÚ total_secondsÚ NPY_FR_us)r ÚtdÚobj2Útd2Zobj3Ztd3r r r Ú test_from_td64_retain_resolution* s r, c C s: t ddd} t| }| ¡ | ks$t|jtjjks6td S )Né é )r& Úminutes)r r Zto_pytimedeltar! r" r r( r$ )r) Úresultr r r Ú test_from_pytimedelta_us_reso@ s r1 c C sÖ t ¡ } t| jtjjks tt ¡ } t| jtj jks<tt
|
¡ } t| jtj jksZtt ¡ } t| jtj jksxtt ¡ } t| jtj jkstt ¡ } t| jtj jks´tt ¡ } t| jtj jksÒtd S ©N)r ZNanor r" r Z NPY_FR_nsr$ r! ZMicror( ZMillir# ÚSecondr% ZMinuteÚHourZDay)Ztickr r r Útest_from_tick_resoH s r5 c C s t dd¡ d¡ d¡} tdddj| ks.ttdddj| ksDttdj| ksVttdd j| ksjttdd j| ks~t| t dd
|
¡ d¡ d¡7 } td j| ks¬ttddd j| ksÂttdd dj| ksØttdddj| ksîttdjdkstdttdksttdjdks,tdttdksBttdtddksZttdtddksrttdtdd ksttdtdd ks¦ttd tddks¾ttd tddksÖttd tddksîttd tddksttd tdd ks ttd!tdd"ks8ttd#td$d%ksPttd&tdd"kshttd'tdd"ksttd(tdd%ksttd)tdd%ks°ttd*tdd%ksÈttd+tdd,ksàttd-tdd,ksøttd.tdd,ksttd/tdd0ks(ttd1tdd0ks@ttd2tdd0ksXttd3tdd0kspttd4td5ksttd6td7ksttd8td7ks²ttd9td7ksÈttd:tddd;ksâttd<tddd;ksüttd=tddddd>ksttd?tddddd> ks:ttd?tddddd> ksZttd@tdddddAdB ks|ttdCtddddDdAdB kstdE}t j
|
t |dF tdG W 5 Q R X t j
|
t |dF tdH W 5 Q R X dI}t j
|
t |dF tdJ W 5 Q R X dK}t j
|
t |dF t W 5 Q R X dL}t j
|
t |dF tdM W 5 Q R X dN}t j
|
t |dF tddO W 5 Q R X t dd
|
¡ d¡ d¡t dPdQ¡ d¡ d¡ } tdRd
|
dj| ksÆtt t dS¡tdSdksättt dS¡tdSdksttt dS¡tdSd%ks ttdT} tdT}|| ks>tt t dS¡tdUksZtdL}t j
|
t |dF tdV W 5 Q R X d S )WNé
|
r úm8[ns]r Údr g $@z10 days©r& Úsz10 days 00:00:10©r& Úsecondsi' )r& Ú millisecondsi )r& Ú microsecondsl >P- z0 days 22:58:59.999850l @BÍ]»? z1 days 10:11:12.001z0:00:00r )Úhoursz00:00:00z-1:00:00é z -01:00:00z1 hz1 hourz1 hrz1 hoursz-1 hoursz1 m©r/ z1.5 méZ ©r< ú1 minutez 1 minutesz1 sz1 secondz 1 secondsz1 ms©r= z1 milliz 1 millisecondz1 us©r> u 1 µsz1 microsz 1 microsecondz1.5 microsecondz00:00:00.000001500z1 nsz00:00:00.000000001z1 nanoz 1 nanosecondz10 days 1 hour©r& r? z 10 days 1 hz10 days 1 h 1m 1s©r& r? r/ r< z-10 days 1 h 1m 1sz-10 days 1 h 1m 1s 3usr. )r& r? r/ r< r> z-10 days 1 h 1.5m 1s 3usé ú'only leading negative signs are allowedr
|
z-10 days -1 h 1.5m 1s 3usz10 days -1 h 1.5m 1s 3uszno units specifiedz3.1415z cannot construct a Timedeltaz unit abbreviation w/o a numberZfoozMcannot construct a Timedelta from the passed arguments, allowed keywords are )Údayiô r g %@é Z1Hz0 days, 02:00:00zfoo bar)r r Úastyper r r r! Ústrr r r Ú
|
ValueErrorr r r4 r3 )ÚexpectedÚmsgr0 r r r Útest_construction` sâ ÿ ÿ ÿ ÿ ÿÿ ÿþþ rR Úitemr r: r r Úmr r )r& r< r> r= r/ r? ÚweeksÚnpdtypec C s@ |\}}t d|¡ d¡ d¡}tf || dij|ks<td S )Nr@ r7 r )r r rM r r r r! )rV rS ZpykwargZnpkwargrP r r r Ú#test_td_construction_with_np_dtypesÝ s rW ÚvalZ1sz-1sZ1usz-1usz1 dayz-1 dayz-23:59:59.999999z-1 days +23:59:59.999999z-1nsZ1nsz-23:59:59.999999999c C s^ t | }t |j|kstt t||ks.tt |jdd|ksFtt | ¡ |ksZtd S )NÚall)Úformat)r r r! rN Z
|
_repr_base)rX r) r r r Útest_td_from_repr_roundtripõ s
|
r[ c C s t djd } d}tjt|d t | W 5 Q R X d}tjt|d t ddd W 5 Q R X t td d
|
}|jtjj ks~t
|
|j d kst
|
d S ) NZ1dayiÔÄ3zCCannot cast 1742682816000000000000 from ns to 'ns' without overflowr
|
z2Cannot cast 139993 from D to 'ns' without overflowiÙ" r r i÷ r9 ) r r r r r r r" r r( r$ r! r& )r$ rQ r) r r r Ú test_overflow_on_construction s r\ z val, unit)i´ ÚM)i; r )i ¡ r )i ' r )iô) rT )l }K r: c C sþ t | |¡}| d¡ d¡dk s$tt|}|dkrB|j|ksBt|jjdksRtd}tj t
|
|d | d¡ W 5 Q R X t|d |d kst|d
|
9 }| d¡ d¡dks²tt|}d }tj t
|
|d | d¡ W 5 Q R X t|d |d ksútd S ) Nr7 r r r] zm8[s]z:Cannot cast 1067\d\d days .* to unit='ns' without overflowr
|
r r@ éÿÿÿÿz;Cannot cast -1067\d\d days .* to unit='ns' without overflow) r r rM r r! r Úasm8Zdtyper r r Úas_unit)rX r Útd64r) rQ r+ r r r Ú&test_construction_out_of_bounds_td64ns! s" rb )l 0
|
à r] )l ~ßL{7r )l F\ r )l àhH r )l ýrT c C sN t | |¡}tjtt|d t| W 5 Q R X t|d |d ksJtd S )Nr
|
i Ê;)r r r r r rN r r! )rX r ra r r r Ú%test_construction_out_of_bounds_td64sM s rc zfmt,expzP6DT0H50M3.010010012Sé é2 r. r6 é )r& r/ r< r= r> Ú nanosecondszP-6DT0H50M3.010010012SiúÿÿÿZ P4DT12H30M5Sr- é é rH zP0DT0H0M0.000000123Sé{ ©rg zP0DT0H0M0.00001SrF zP0DT0H0M0.001Sr@ rE Z
|
P0DT0H1M0SrA Z P1DT25H61M61Sé é= ZPT1SrC ZPT0SZP1WT0Sé r; ZP1Dr9 ZP1DT1HrG ZP1WZPT300Si, ZP1DT0H0M00000000000SzPT-6H3M)r? r/ z-PT6H3Méýÿÿÿz -PT-6H+3Mc C s t | |kstd S r2 ©r r! )ÚfmtÚexpr r r Útest_iso_constructor` s -rs rq Z PPPPPPPPPPPPZPDTHMSZP0DT999H999M999SzP1DT0H0M0.0000000000000Sz P1DT0H0M0.SÚPz-Pc C s0 d| }t jt|d t| W 5 Q R X d S )Nz#Invalid ISO 8601 Duration format - r
|
©r r rO r )rq rQ r r r Útest_iso_constructor_raises s
|
rv zconstructed_td, conversionéd Z100ns)r& r? r/ rU r< r= r> rg l )D^ã_ Z1us1nsZ999nséþÿÿÿZ990nsc C s | t |kstd S r2 rp )Zconstructed_tdÚ
|
conversionr r r Ú"test_td_constructor_on_nanoseconds¢ s rz c C s, d} t jt| d tdd W 5 Q R X d S )Nz1Invalid type <class 'str'>. Must be int or float.r
|
Úabcrk )r r Ú TypeErrorr )rQ r r r Ú test_td_constructor_value_error½ s r} c C s( t t dd¡} t | }|| ks$td S )Nr@ r: )r r r r! )rP r0 r r r Ú#test_timedelta_constructor_identityà s r~ c C s8 t dd} d}tjt|d t | dd W 5 Q R X d S )Nr@ r9 zRCannot pass both a Timedelta input and timedelta keyword arguments, got \['days'\]r
|
rL )r r r rO )r) rQ r r r Ú(test_timedelta_pass_td_and_kwargs_raisesÊ s
|
|
ÿr z%constructor, value, unit, expectationZ10szunit must not be specifiedÚ1rL c C s2 |\}}t j||d | ||d}W 5 Q R X d S )Nr
|
r )r r )Ú constructorr$ r Z expectationrr r Ú_r r r Útest_string_with_unitÕ s r r$ c C s( g | ] }t d |dD ]}d |¡qqS )z+-, )ÚrepeatÚ )r Újoin)Ú.0Z
|
repetitionÚelementsr r r Ú
|
<listcomp>å s þr )r@ rL c C s6 | dkr dnd}t jt|d t| W 5 Q R X d S )Nz--zsymbols w/o a numberrJ r
|
ru )r$ rQ r r r Útest_string_without_numbersã s ÿýr c C s t dd¡} t| tks td S )Nr r )r r r r r! )Znatr r r Útest_timedelta_new_npnatö s r c C s* G dd dt } | d}t|| s&td S )Nc @ s e Zd ZdS )z2test_subclass_respected.<locals>.MyCustomTimedeltaN)Ú__name__Ú
|
__module__Ú __qualname__r r r r ÚMyCustomTimedeltaþ s r rD )r Ú
|
isinstancer! )r r) r r r Útest_subclass_respectedü s r c C st t ddd d¡j} | dks td}t dd d¡d }tjt|d |j W 5 Q R X |j d ¡} | d
|
ksptd S ) Nr6 r r r: l ,Z6G zCannot convert Timedelta to nanoseconds without overflow. Use `.asm8.view\('i8'\)` to cast represent Timedelta in its own unit \(here, s\).$iè r
|
r l `®;P ) r r` r$ r! r r Ú OverflowErrorr_ r )r0 rQ r) r r r Útest_non_nano_value s ÿ r )1Údatetimer Ú itertoolsr Únumpyr r Zpandas._libs.tslibsr Zpandas._libs.tslibs.dtypesr Zpandasr r r r r r r, r1 r5 rR ÚmarkZ parametrizeÚlistÚitemsÚint64Úint32Úint16Úfloat64Úfloat32Úfloat16rW r[ r\ rb rc rs rv rz r} r~ r rO r r r r r r r r r Ú<module> sH }ùÿþ ÿ õþ
|
|
úþ
|
!ûþ
|
|
úþ úþ Ùþ,
|
ùþ
|
ø
|
õ ïþ
|
ýþ
|
þþ
|
|
