U
|
®ý°d ( ã @ sb d dl m Z mZmZ d dlZd dlZd dlmZmZm Z m
|
Z
|
m Z m Z d dl mZ G dd dZdS )é )ÚdatetimeÚ timedeltaÚtimezoneN)ÚOutOfBoundsDatetimeÚOutOfBoundsTimedeltaÚ TimedeltaÚ TimestampÚoffsetsÚ to_offsetc @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z ej dedde dd¡eddg¡dd Zej d e d d ¡d fe d d ¡d fe d d!¡d"fe d d#¡d$fg¡d%d& Zej d'ed(ed(d)d*g¡ej d+d,e d,¡ejd,d-gejd.ejdd/gejd.g¡d0d1 Z ej d2d3d4g¡d5d6 Z ej d2d3d4g¡d7d8 Z d9d: Z d;S )<ÚTestTimestampArithmeticc C sX t d}tdd }t d}|| |ks,t|| |ks<tt d}|| |ksTtd S )Nú2000/1/1ÚDéd z
|
2000/04/10z
|
1999/09/23)r r
|
ÚAssertionError)ÚselfÚstampZoffset_no_overflowÚexpected© r údd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/scalar/timestamp/test_arithmetic.pyÚtest_overflow_offset s z,TestTimestampArithmetic.test_overflow_offsetc C s t d d¡}dt d¡ }d}d}tjt|d || W 5 Q R X tjt|d || W 5 Q R X tjt|d || W 5 Q R X t d d¡}td d
|
}d }tjt|d || W 5 Q R X tjt|d || W 5 Q R X tjt|d || W 5 Q R X d S ) Nz2017-01-13 00:00:00ÚnsiÔÄ3é zathe add operation between \<-?\d+ \* Days\> and \d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2} will overflowzECannot cast -?20169940 days \+?00:00:00 to unit='ns' without overflow©Úmatchr r l d( zHCannot cast -?10000000000 days \+?00:00:00 to unit='ns' without overflow) r Úas_unitr ZDayÚpytestÚraisesr Ú OverflowErrorr
|
)r r Zoffset_overflowÚmsgZlmsg2Zlmsg3r r r Útest_overflow_offset_raises% s* ÿ ÿz3TestTimestampArithmetic.test_overflow_offset_raisesc C sb d}t d d¡}t d d¡}tjt|d || W 5 Q R X || ¡ | ¡ | ks^td S )NzResult is too largez2101-01-01 00:00:00r z1688-01-01 00:00:00r )r r r r r Ú to_pydatetimer )r r ÚaÚbr r r Ú test_overflow_timestamp_raisesM s z6TestTimestampArithmetic.test_overflow_timestamp_raisesc C s( t d}|td }|j|jks$td S )Nl { Ô|%*x r )r r Z
|
nanosecondr )r ÚvalÚresultr r r Útest_delta_preserve_nanosY s z1TestTimestampArithmetic.test_delta_preserve_nanosc C s t d}td|d}|| }|| |ks,t| ¡ | |ks@t|d kr^| ¡ | |kstn*d}tjt|d | ¡ | W 5 Q R X d S )Nl @~z
|
2021-01-01©Útzz;Cannot subtract tz-naive and tz-aware datetime-like objectsr )r r r r Z to_datetime64r r Ú TypeError)r Ztz_naive_fixtureÚtdÚtsÚotherr r r r Útest_rsub_dtscalars^ s z+TestTimestampArithmetic.test_rsub_dtscalarsc C sD t ddd}tt ddd}|| jdks.t|| jdks@td S )NéÝ é
|
é é r éÿÿÿÿ)r r Údaysr )r Údtr+ r r r Útest_timestamp_sub_datetimem s z3TestTimestampArithmetic.test_timestamp_sub_datetimec C sF t d}tddddtjd}|| }t|ts2t|tdksBtd S )Nz2020-10-22T22:00:00+00:00iä r/ é ©Útzinfoz0 days)r r r ÚutcÚ
|
isinstancer r ©r Út1Út2r% r r r Ú test_subtract_tzaware_datetimes s
|
z6TestTimestampArithmetic.test_subtract_tzaware_datetimec C sF t d d¡}t d d¡}|| }t|ts2t|tdksBtd S )NÚ20130101z
|
US/EasternÚCETú0 days 06:00:00)r Ú tz_localizer: r r r; r r r Ú/test_subtract_timestamp_from_different_timezone| s
|
zGTestTimestampArithmetic.test_subtract_timestamp_from_different_timezonec C sx t dddttddd}td d¡}|| }t|ts>t|tdksNt|| }t|tsdt|td ksttd S )
|
Nr. r éûÿÿÿ©Úhoursr7 r? r@ rA z-1 days +18:00:00)r r r r rB r: r r r; r r r Ú5test_subtracting_involving_datetime_with_different_tz
s zMTestTimestampArithmetic.test_subtracting_involving_datetime_with_different_tzc C sN t d}| d¡}| |¡td }|| }t|ts:t|tdksJtd S )Nr? ÚUTCz0 days 05:00:00)r rB Z
|
tz_convertr r: r )r Ztz_aware_fixtureZt_rawZt_UTCZt_diffr% r r r Ú$test_subtracting_different_timezones s
|
z<TestTimestampArithmetic.test_subtracting_different_timezonesc C sÚ t ddd}tdd}t|}d}tjt|d |d W 5 Q R X tjt|d |d W 5 Q R X t|| tksztt|| tkstt|| tks¢tt
|
dd¡}t|| tksÂtt|| tksÖtd S ) NiÞ é é r )Úsecondsz Addition/subtraction of integersr r ) r r r r r r) Útyper r ÚnpÚ timedelta64)r r4 r* r+ r Ztd64r r r Ú test_addition_subtraction_types s
|
z7TestTimestampArithmetic.test_addition_subtraction_typesr* rJ rE Úhc C s |}|| || kst d S )N)r )r r* Z fixed_now_tsr+ r r r Útest_radd_tdscalar¶ s z*TestTimestampArithmetic.test_radd_tdscalarzother,expected_differencei
ÿÿÿr iÚIÚusi þÿÚmsi@+«øc C sR t ¡ }t| d¡}|| }|j|j }||ks6tt|}|| |ksNtd S )Nr )r Úutcnowr r Ú_valuer )r r, Zexpected_differenceÚnowr+ r% ZvaldiffÚts2r r r Ú#test_timestamp_add_timedelta64_unit¾ s
|
z;TestTimestampArithmetic.test_timestamp_add_timedelta64_unitr+ z
|
1776-07-04rH r' r, r é ©ZdtyperK c C s d}t jt|d || W 5 Q R X t jt|d || W 5 Q R X t jt|d || W 5 Q R X d}t jt|d || W 5 Q R X d S )Nz3Addition/subtraction of integers and integer-arraysr zunsupported operand type)r r r) )r r+ r, r r r r Útest_add_int_with_freqÑ s z.TestTimestampArithmetic.test_add_int_with_freqÚshape)é )rZ rJ c sò t d d¡ t d¡ d¡ |¡} | } fddtdD }tjdd |D dd |¡}t ||¡ | }t ||¡ | } fd
|
dtdD }tjd d |D dd |¡}t ||¡ d }t
|
j t |d | W 5 Q R X d S )Nú2020-04-04 15:45r r^ úm8[h]c s g | ]} t |d qS ©rE ©r ©Ú.0Ún©r+ r r Ú
|
<listcomp>÷ s zATestTimestampArithmetic.test_addsub_m8ndarray.<locals>.<listcomp>c S s g | ]
|
}|j qS r ©Zasm8©rd Úxr r r rg ø s zM8[ns]r[ c s g | ]} t |d qS ra rb rc rf r r rg ÿ s c S s g | ]
|
}|j qS r rh ri r r r rg s úDunsupported operand type\(s\) for -: 'numpy.ndarray' and 'Timestamp'r ) r r rN ÚarangeÚastypeÚreshapeÚrangeÚarrayÚtmÚassert_numpy_array_equalr r r) ©r r] r, r% Z ex_stampsr r r rf r Útest_addsub_m8ndarrayï s z-TestTimestampArithmetic.test_addsub_m8ndarrayc sÔ t ddd t d¡ d¡ |¡} | } fddtdD }t |¡ |¡}t ||¡ | }t ||¡ | } fddtdD }t |¡ |¡}t ||¡ d }t j
|
t |d
|
| W 5 Q R X d S ) Nr_ z
|
US/Pacificr' r^ r` c s g | ]} t |d qS ra rb rc rf r r rg s zITestTimestampArithmetic.test_addsub_m8ndarray_tzaware.<locals>.<listcomp>c s g | ]} t |d qS ra rb rc rf r r rg s rk r ) r rN rl rm rn ro rp rq rr r r r) rs r rf r Ú test_addsub_m8ndarray_tzaware s z5TestTimestampArithmetic.test_addsub_m8ndarray_tzawarec C sD t ddd}t||d}t||d}|| }td}||ks@td S )Niå r r' r )r r r r )r Z utc_fixtureZ utc_fixture2r4 Zts1rX r% r r r r Ú#test_subtract_different_utc_objects s z;TestTimestampArithmetic.test_subtract_different_utc_objectsN) Ú__name__Ú
|
__module__Ú __qualname__r r r# r& r- r5 r> rC rG rI rP r ÚmarkZ parametrizer rN rO r rR rY r Úint64rp Úint32Zuint64r\ rt ru rv r r r r r sZ (
|
ÿ
|
üþ
|
|
|
þþüþ
|
|
r )r r r ÚnumpyrN r Zpandas._libs.tslibsr r r r r r
|
Zpandas._testingZ_testingrq r r r r r Ú<module> s
|
|
