U
|
®ý°dâF ã @ sÞ d Z ddlmZmZmZmZ ddlZddlZddlmZm Z ddl
|
Z
|
ddl Z ddl m Z mZ ddlmZ ddlmZ ddlmZ ddlm mZ dd lmZmZ zdd
|
l m Z W n e k
|
rÊ dZ Y nX G d d d Z dS ) z.
|
Tests for Timestamp timezone-related methods
|
é )ÚdateÚdatetimeÚ timedeltaÚtimezoneN)ÚgettzÚtzoffset)ÚAmbiguousTimeErrorÚNonExistentTimeError)Ú timezones)ÚNpyDatetimeUnit)ÚOutOfBoundsDatetime)ÚNaTÚ Timestamp)ÚZoneInfoc @ sh e Zd Zdd Zej dddddg¡dd Zd
|
d Zej d d dddg¡dd Z dd Z
|
dd Z ej dddddg¡d d Z d d Z ej d e d!¡ed!d!d"g¡d#d$ Zej d%d&d'd(d)gd&d'd*d+gd&d'd,ed-d.gd&d'd/ed0d.gd1d2d3d)gd1d2d4d+gd1d2d5ed-d.gd1d2d6ed0d.gg¡ej d7d8d9g¡ej dddddg¡d:d; Zej d<d0d-g¡d=d> Zej dddddg¡d?d@ Zej dddddg¡dAdB Zej dddddg¡dCdD Zej dEd!d"g¡dFdG ZejdHdI ZdJdK Z dLdM Z dNdO Z ej d e d!¡ed!d!d"g¡dPdQ Z ej d e d!¡ed!d!d"g¡dRdS Z dTdU Z!dVS )WÚTestTimestampTZOperationsc C sÔ dt j d¡ dt j }t j d¡}|jt jjks8t| d¡ tjt |d t j d¡ W 5 Q R X dt j
|
d¡ dt j
|
}t j
|
d¡}|jt j
|
jk s t| d¡ tjt |d t j
|
d¡ W 5 Q R X d S )Nz Converting z%Y-%m-%d %H:%M:%Sz underflows past ú
|
US/Pacificú
|
Asia/Tokyo©Úmatchz overflows past ) r ÚminÚstrftimeÚ tz_localizeÚ_valueÚAssertionErrorÚ
|
tz_convertÚpytestÚraisesr Úmax)ÚselfÚmsgZpacZtokyo© r úcd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/scalar/timestamp/test_timezones.pyÚ%test_tz_localize_pushes_out_of_bounds, s ÿ
|
ÿ
|
z?TestTimestampTZOperations.test_tz_localize_pushes_out_of_boundsÚunitÚnsÚusÚmsÚsc C s: t d |¡}t ddd}t ddd}d}tjtj|d | d¡ W 5 Q R X tjtj|d | d¡ W 5 Q R X td k rÆz td}W n tk
|
r Y n(X tjtj|d | |¡ W 5 Q R X |jdd d
|
}||ksàt |j
|
t t d | j ksüt |jdd d
|
}||kst |j
|
t t d | j ks6t d S ) Nz2015-11-01 01:00:03z2015-11-01 01:00:03-0500z
|
US/Central©Útzz2015-11-01 01:00:03-0600z.Cannot infer dst time from 2015-11-01 01:00:03r zdateutil/US/CentralT©Z ambiguousÚNPY_FR_F)r Úas_unitr r Úpytzr r r ÚKeyErrorr Ú_cresoÚgetattrr Úvalue)r r# ÚtsZ expected0Z expected1r r) Úresultr r r! Ú test_tz_localize_ambiguous_boolD s* z9TestTimestampTZOperations.test_tz_localize_ambiguous_boolc C sÊ t d}|jddd}|jddd}|j|j dks8tt d¡}tjt|d |jdd d W 5 Q R X d
|
}tjt |d t d dd d ¡ W 5 Q R X d}tjt |d t d
|
d ¡ W 5 Q R X d S )Nz2014-11-02 01:00ú
|
US/EasternTr* Fi zK'ambiguous' parameter must be one of: True, False, 'NaT', 'raise' (default)r ZinferzBCannot localize tz-aware Timestamp, use tz_convert for conversionsz
|
2011-01-01r( r z>Cannot convert tz-naive Timestamp, use tz_localize to localize) r r r r ÚreÚescaper r Ú
|
ValueErrorÚ TypeErrorr )r r2 Zts_dstZ ts_no_dstr r r r! Útest_tz_localize_ambiguouse s ÿ z4TestTimestampTZOperations.test_tz_localize_ambiguousz stamp, tz)z2015-03-08 02:00r5 )z2015-03-08 02:30r )z2015-03-29 02:00ú Europe/Paris)z2015-03-29 02:30zEurope/Belgradec C sn t |}tjt|d | |¡ W 5 Q R X tjt|d |j|dd W 5 Q R X |j|ddtksjtd S )Nr Úraise©Z nonexistentr )r r r r r r r )r Ústampr) r2 r r r! Ú test_tz_localize_nonexistent{ s z6TestTimestampTZOperations.test_tz_localize_nonexistentc C s8 t d}d}tjt|d |jddd W 5 Q R X d S )Nz2015-11-1 01:00z/Cannot infer dst time from 2015-11-01 01:00:00,r r r< r* )r r r r r )r r2 r r r r! Ú test_tz_localize_ambiguous_raise s z:TestTimestampTZOperations.test_tz_localize_ambiguous_raisec C s< |}t d}d}tjt|d |j|dd W 5 Q R X d S )Nz2015-03-29 02:00:00úoThe nonexistent argument must be one of 'raise', 'NaT', 'shift_forward', 'shift_backward' or a timedelta objectr Úfoor= )r r r r8 r )r Úwarsawr) r2 r r r r! Ú(test_tz_localize_nonexistent_invalid_arg s ÿzBTestTimestampTZOperations.test_tz_localize_nonexistent_invalid_argr> z2014-02-01 09:00z2014-07-08 09:00z2014-11-01 17:00z2014-11-05 00:00c C sz |}t |}| |¡}|t ||dks*td}tjt|d | |¡ W 5 Q R X | d ¡}||ksht|jd ksvtd S )Nr( z"Cannot localize tz-aware Timestampr )r r r r r r9 Útzinfo)r r> Útz_aware_fixturer) r2 Z localizedr Úresetr r r! Útest_tz_localize_roundtrip s
|
|
|
z4TestTimestampTZOperations.test_tz_localize_roundtripc C sô t d}d}d}|j|dd}|j|dd}|j|jks<t|jdksJt| ¡ ¡ dks^t| ¡ ¡ dksrtt|t|kst|j|dd}|j|dd}|j|jks²t|jd ksÀtt|t|ksÔt| ¡ ¡ | ¡ ¡ ksðtd S )
|
Nz2013-10-27 01:00:00z Europe/Londonzdateutil/Europe/LondonFr* ielRÚGMTTiWlR)r r r r Ú to_pydatetimeÚtznameÚstr)r ÚnaiveZ pytz_zoneZ dateutil_zoneZ result_pytzZresult_dateutilr r r! Ú!test_tz_localize_ambiguous_compat· s&
|
|
ÿÿz;TestTimestampTZOperations.test_tz_localize_ambiguous_compatr) r5 zdateutil/US/Easternc C s> t d}| |¡}t d|d}|j|jks.t||ks:td S )Nú3/11/2012 04:00r( )r r Úhourr ©r r) r> r3 Úexpectedr r r! Útest_timestamp_tz_localizeÖ s
|
|
|
z4TestTimestampTZOperations.test_timestamp_tz_localizezstart_ts, tz, end_ts, shiftú2015-03-29 02:20:00z Europe/Warsawz2015-03-29 03:00:00Zforwardz 2015-03-29 01:59:59.999999999Zbackwardz2015-03-29 03:20:00é ©Úhoursz2015-03-29 01:20:00éÿÿÿÿz2018-03-11 02:33:00r z2018-03-11 03:00:00z 2018-03-11 01:59:59.999999999z2018-03-11 03:33:00z2018-03-11 01:33:00Útz_typeÚ z dateutil/c C sÞ || }t |trd| }t| |¡}|j||d}t| |¡} |dkrb|| jddks¾tn\|dkr| j| jd }
|
|| j|
|
ddks¾tn,|d kr²|| jdddks¾tn || ks¾t|jt t
|
d
|
| j ksÚtd S ) NZshift_r= r% r )Ú
|
nanosecondr& iè )Ú microsecondr[ r' r+ ) Ú
|
isinstancerL r r, r Úreplacer r\ r/ r0 r r1 ) r Ústart_tsr) Zend_tsÚshiftrY r# r2 r3 rR Zmicrosr r r! Ú,test_timestamp_tz_localize_nonexistent_shiftç s 1
|
zFTestTimestampTZOperations.test_timestamp_tz_localize_nonexistent_shiftÚoffsetc C sB |}t d}d}tjt|d |j|t|dd W 5 Q R X d S )NrT z<The provided timedelta will relocalize on a nonexistent timer )Úsecondsr= )r r r r8 r r )r rb rC r) r2 r r r r! Ú4test_timestamp_tz_localize_nonexistent_shift_invalid* s
|
zNTestTimestampTZOperations.test_timestamp_tz_localize_nonexistent_shift_invalidc C s0 |}t d |¡}|j|dd}|tks,td S )NrT r r= )r r, r r r )r rC r# r) r2 r3 r r r! Ú*test_timestamp_tz_localize_nonexistent_NaT3 s zDTestTimestampTZOperations.test_timestamp_tz_localize_nonexistent_NaTc C sp |}t d |¡}d}tjtj|d |j|dd W 5 Q R X d}tjt|d |j|dd W 5 Q R X d S )NrT r r< r= rA rB )r r, r r r- r r r8 )r rC r# r) r2 r r r r! Ú,test_timestamp_tz_localize_nonexistent_raise; s ÿzFTestTimestampTZOperations.test_timestamp_tz_localize_nonexistent_raisec C s^ |}t |dd}| |¡}| d ¡}|t |ks4t|jd ksBt|| d¡ d ¡ksZtd S )NÚUTCr( )r r r rE r )r r> rF r) r2 Z convertedrG r r r! Útest_tz_convert_roundtripM s
|
|
|
z3TestTimestampTZOperations.test_tz_convert_roundtripÚtzstrc C s> t ddd}| |¡}| |¡}||ks,tt|t s:td S )Nz3/11/2012 22:00rg r( )r r Ú
|
astimezoner r] )r ri ZutcdaterR r3 r r r! Útest_astimezonea s
|
|
|
z)TestTimestampTZOperations.test_astimezonec C sX t dt d¡d}|| tj ¡ ¡ks*tt dt d¡d}|| tj ¡ ¡ksTtd S )Nz2001-01-05 11:56z dateutil/UTCr( )r r
|
Ú maybe_get_tzr Údateutilr) Ztzutcr )r r2 r r r! Ú#test_tz_convert_utc_with_system_utck s z=TestTimestampTZOperations.test_tz_convert_utc_with_system_utcc C sJ t ddd}|jtjks t|jdks*tt d d¡}|jdksFtd S )Nú3/11/2012 05:00Úutcr( é )r rE r rp r rP r )r Z utc_stampr r r! Ú!test_timestamp_constructor_tz_utcz s
|
z;TestTimestampTZOperations.test_timestamp_constructor_tz_utcc C s2 t d d}td|d}t| ¡ }||ks.td S )Ni rO r( )r r rJ r )r rE rR r3 r r r! Ú#test_timestamp_to_datetime_tzoffset s
|
z=TestTimestampTZOperations.test_timestamp_to_datetime_tzoffsetc
|
C sº dD ]V}t d|d}t d |¡}||ks.td}tjtj|d t d|d W 5 Q R X qt ddd}t d d¡}||kstd }tjtj|d t d dd W 5 Q R X t d
|
}| d ¡ d ¡}t d d d}||ksÞtt ddd}t ddd}||kstt ddd}t ddd}||ks*td }tjtj|d t d dd W 5 Q R X t ddd}t | d¡j
|
}| d ¡ d¡}||kstt ddd}t ddd}||ks¶td S )N)zEurope/Brusselsz Europe/Praguez2015-10-25 01:00r( z.Cannot infer dst time from 2015-10-25 02:00:00r z2015-10-25 02:00z2017-03-26 01:00r; z2017-03-26 02:00z2015-11-18 10:00:00rg z Asia/Kolkataz2015-11-18 15:30:00+0530z2017-03-26 00:00z2017-03-26 00:00:00+0100z2017-03-26 01:00:00+0100z2017-03-26 02:00:00+0100r$ z2017-03-26 03:00z2017-03-26 03:00:00+0200) r r r r r r- r r r r, r )r r) r3 rR r rM r r r! Ú,test_timestamp_constructor_near_dst_boundary sB zFTestTimestampTZOperations.test_timestamp_constructor_near_dst_boundaryc C s@ t tddd|d}t d|d}|j|jks0t||ks<td S )NiÜ é é r( z 3/11/2012)r r rP r )r r) r3 rR r r r! Ú)test_timestamp_constructed_by_date_and_tzº s zCTestTimestampTZOperations.test_timestamp_constructed_by_date_and_tzc C s6 t d|d}|tdd }t d|d}||ks2td S )Nz3/10/2012 22:00r( é rV ro )r r r rQ r r r! Ú3test_timestamp_add_timedelta_push_over_dst_boundaryÍ s zMTestTimestampTZOperations.test_timestamp_add_timedelta_push_over_dst_boundaryc C sL t |¡}td|d}tdddddd|d }| ¡ }| ¡ }||ksHtd S )
|
Nz2018-06-04 10:20:30r( iâ rx é é
|
é é )rP ÚminuteÚsecondrE )r
|
rl r r Útimetzr )r Ztz_naive_fixturer) r> Ú _datetimer3 rR r r r! Ú1test_timestamp_timetz_equivalent_with_datetime_tzã s
|
zKTestTimestampTZOperations.test_timestamp_timetz_equivalent_with_datetime_tzN)"Ú__name__Ú
|
__module__Ú __qualname__r" r ÚmarkZ parametrizer4 r: r? r@ rD rH rN r- r r rS r ra rd re rf rh rk ÚtdZskip_if_windowsrn rr rs rt rw ry r r r r r! r ( sä
|
üþ
|
|
üþ
|
üþ
|
|
üüü
|
üüüßþ+
|
|
|
üþ
|
|
|
2üþ
|
|
üþ
|
r ) Ú__doc__r r r r r6 rm Z dateutil.tzr r r r- Zpytz.exceptionsr r Zpandas._libs.tslibsr
|
Zpandas._libs.tslibs.dtypesr Z pandas.errorsr Z pandas.util._test_decoratorsÚutilZ_test_decoratorsr Zpandasr r Zzoneinfor Ú ImportErrorr r r r r! Ú<module> s"
|
|
