U
|
®ý°dR ã @ sÞ d dl m Z d dlmZ d dlmZmZ d dlZd dl Z d dl
|
Z
|
d dl
|
m Z d dl m Z d dlmZmZmZmZmZmZmZ d dlmZ d d lmZ d dlm m Z d dl m Z G d
|
d d Z!e j" #d d d g¡dd Z$dS )é )Údatetime)Úgettz)ÚgivenÚ
|
strategiesN)Úutc)Úlib)ÚNaTÚOutOfBoundsDatetimeÚ TimedeltaÚ TimestampÚ
|
conversionÚiNaTÚ to_offset)ÚNpyDatetimeUnit)ÚINVALID_FREQ_ERR_MSGc @ sD e Zd Zdd Zej ddddddd d
|
g¡d d Zd d Zdd Z dd Z
|
dd Z dd Z ej dddddd d d d g¡d d! Z ej d"d#d$d%d&d'd(g¡ej d)d*d+d,g¡d-d. Zej d/d0d1d2d3g¡d4d5 Zej d/d0d1d2d3g¡d6d7 Zej d8d*d,d+g¡ej d/d0d1d2d3g¡d9d: Zej d;d*d<d=gd,d<d=gd+d>d?gg¡ej d/d0d1d2d3g¡d@dA Zej dBdCdDdEdFdGdHg¡ej dIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[g¡d\d] Zd^d_ Zejjd`dadbee edc ej¡ddej d8e j e j e j g¡dedf Z dgdh Z!didj Z"dkdl Z#dmdn Z$dodp Z%dqdr Z&dsdt Z'dudv Z(dwdx Z)e*j+dydz Z,ej d{e- .d|¡d}d~ fe/d|dd~ fg¡dd Z0ej d/d0d1d2d3g¡dd Z1ej dd
dcg¡ej dddg¡ej d/d0d1d2d3g¡dd Z2ej dddg¡ej d/d0d1d2d3g¡dd Z3dd Z4e*j+dd Z5dS )ÚTestTimestampUnaryOpsc C s4 t d}d}tjt|d | d¡ W 5 Q R X d S )Nú
|
2016-01-01z Division by zero in rounding©ÚmatchZ0ns)r ÚpytestÚraisesÚ
|
ValueErrorÚround)ÚselfÚtsÚmsg© r úcd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/scalar/timestamp/test_unary_ops.pyÚ!test_round_divison_by_zero_raises s z7TestTimestampUnaryOps.test_round_divison_by_zero_raisesztimestamp, freq, expected)ú20130101 09:10:11ÚDÚ20130101)z20130101 19:10:11r Ú20130102)z20130201 12:00:00r Z20130202)z20130104 12:00:00r Z20130105)ú2000-01-05 05:09:15.13r z2000-01-05 00:00:00)r# ÚHz2000-01-05 05:00:00)r# ÚSz2000-01-05 05:09:15c C s* t |}| |¡}t |}||ks&td S ©N©r r ÚAssertionError)r Ú timestampÚfreqÚexpectedÚdtÚresultr r r Útest_round_frequencies( s
|
z,TestTimestampUnaryOps.test_round_frequenciesc C sT t ddd}| d¡}t ddd}||ks.tt ddd}| d¡}||ksPtd S )Nr ú
|
US/Eastern©Útzr r! Úsr' ©r r, r- r+ r r r Útest_round_tzaware: s
|
|
z(TestTimestampUnaryOps.test_round_tzawarec C s* t d}| d¡}t d}||ks&td S )Nz20130104 12:32:00Z30Minz20130104 12:30:00r' r3 r r r Útest_round_30minD s
|
z&TestTimestampUnaryOps.test_round_30minc C s t d d¡}t d}||ks"tt d d¡}t d}||ksDtt d}dD ]}|| |¡ksPtqPt d d¡}t d }||kstd S )
|
Nz2016-10-17 12:00:00.0015Úmsz2016-10-17 12:00:00.002000z2016-10-17 12:00:00.00149z2016-10-17 12:00:00.001000)ÚusÚnsú 2016-10-17 12:00:00.001501031Ú10nsz 2016-10-17 12:00:00.001501030r' )r r- r+ r r* r r r Útest_round_subsecondK s z*TestTimestampUnaryOps.test_round_subsecondc C s( t d¡ td d¡ W 5 Q R X d S )NFr9 Z1010ns)ÚtmZassert_produces_warningr r )r r r r Útest_round_nonstandard_freq] s z1TestTimestampUnaryOps.test_round_nonstandard_freqc C s0 t d}tjttd | d¡ W 5 Q R X d S )Nr# r Úfoo)r r r r r r )r Zstampr r r Útest_round_invalid_arga s z,TestTimestampUnaryOps.test_round_invalid_argz#test_input, rounder, freq, expected)ú2117-01-01 00:00:45ÚfloorÚ15sr@ )r@ ÚceilrB r@ )z 2117-01-01 00:00:45.000000012rA r: z 2117-01-01 00:00:45.000000010)z 1823-01-01 00:00:01.000000012rC r: z 1823-01-01 00:00:01.000000020)ú1823-01-01 00:00:01rA Ú1srD )rD rC rE rD )r rA rE r )r rC rE r c C sH t |}t||}||}|tkr0|tksDtnt |}||ksDtd S r& )r Úgetattrr r( )r Ú
|
test_inputÚrounderr* r+ r, Úfuncr- r r r Útest_ceil_floor_edgef s
|
z*TestTimestampUnaryOps.test_ceil_floor_edgeztest_input, freq, expected)ú2018-01-01 00:02:06Ú2srK )ú2018-01-01 00:02:00Z2TrM )ú2018-01-01 00:04:00Z4TrN )ú2018-01-01 00:15:00Z15TrO )ú2018-01-01 00:20:00Z20TrP )ú2018-01-01 03:00:00Z3HrQ rH rC rA r c C s2 t |}t |}t||}||}||ks.td S r& )r rF r( )r rG r* r+ rH r, rI r- r r r Útest_round_minute_freq s
|
|
z,TestTimestampUnaryOps.test_round_minute_freqÚunitr8 r7 r6 r2 c C s@ t d |¡}| d¡}t d}||ks,t|j|jks<td S )Nr r r" )r Úas_unitrC r( Ú_creso©r rS r, r- r+ r r r Ú test_ceil s
|
|
z TestTimestampUnaryOps.test_ceilc C s@ t d |¡}| d¡}t d}||ks,t|j|jks<td S )Nr r r! )r rT rA r( rU rV r r r Ú
|
test_floor¦ s
|
|
z TestTimestampUnaryOps.test_floorÚmethodc C sö t ddd d¡}| |¡}t||ddd}||ks:t|jttd| jksVtt||dd d}t d
|
dd d¡}||kst|jttd| jks¢tt||dd d}|tksÀtd }t j
|
t j |d t||ddd W 5 Q R X d S )Nz2017-10-29 00:00:00ÚUTCr0 z Europe/Madridr$ T©Z ambiguousÚNPY_FR_Fz2017-10-29 01:00:00r zCannot infer dst timer Úraise) r Ú
|
tz_convertrT rF r( rU r Úvaluer r r ÚpytzZAmbiguousTimeError)r rY rS r r- r+ r r r r Ú test_round_dst_border_ambiguous® s
|
ÿ z5TestTimestampUnaryOps.test_round_dst_border_ambiguouszmethod, ts_str, freqz2018-03-11 01:59:00-0600Z5minz2018-03-11 03:01:00-0500Z2Hc C s¬ t |dd |¡}t|||dd}t ddd}||ks<t|jttd| jksXtt|||dd}|tksvtd}tj t
|
j |d t|||d
|
d W 5 Q R X d S ) NúAmerica/Chicagor0 Z shift_forward)Z nonexistentz2018-03-11 03:00:00r\ r z2018-03-11 02:00:00r r] ) r rT rF r( rU r r_ r r r r` ZNonExistentTimeError) r rY Zts_strr* rS r r- r+ r r r r Ú!test_round_dst_border_nonexistentÊ s z7TestTimestampUnaryOps.test_round_dst_border_nonexistentr) z2018-01-01 0:0:0.124999360z2018-01-01 0:0:0.125000367z2018-01-01 0:0:0.125500z2018-01-01 0:0:0.126500z2018-01-01 12:00:00z2019-01-01 12:00:00r* Z2nsZ3nsZ4nsZ5nsZ6nsZ7nsZ250nsZ500nsZ750nsZ1usZ19usZ250usZ500usZ750usrE rL Z3sZ1Dc C sD t | d¡}t|j}| |¡}|j| dks@td| dd|j|j kr\|k sfn td| |¡}|j| dkstd| dd|j|j krª|k s´n td| |¡}|j| dksÜtd| dt |j|j |d ksütd
|
|d dkr@t |j|j |d kr@|j| d dks@td d S ) Nr8 r z floor not a z multiplez floor errorz ceil not a z
|
ceil errorz round not a é z round errorzround half to even error)
|
r rT r ÚnanosrA Ú_valuer( rC r Úabs)r r) r* r, rS r- r r r Útest_round_int64å s %
|
|
&
|
&
|
(z&TestTimestampUnaryOps.test_round_int64c C s® t j d¡}t dddddd}||ks*tt j d¡}t jtd }||ksPttjt d d
|
t j d¡ W 5 Q R X d
|
d d g¡}tjt |d
|
t j d¡ W 5 Q R X d S )Nr2 i é é r é é, iÿ×ò2zvalue too larger ú|z)Python int too large to convert to C longzint too big to convert) r ÚminrC r( ÚmaxrA r
|
r r Ú OverflowErrorÚjoin)r r- r+ r r r r Ú test_round_implementation_bounds s ÿz6TestTimestampUnaryOps.test_round_implementation_boundszFailing on buildsF)ÚreasonÚstricté )Úvalc sØ t |¡}t|} fdd} |d|ks0t |d}d}t || j¡|k sVt|j| dksht|||| |d}d}t || j¡|k st|j| dks¬t|||| |d }d
|
}t || j¡|k sÞt|j| dksðt|||| |d }d }t || j¡|k s$t|j| dks8t|||| |d }d}t || j¡|k slt|j| dkst|||| |d}d}t || j¡|k s´t|j| dksÈt|||| d S )Nc s^ t jkr,t | | j¡}||d ksZtn. t jkrD| |ksZtn t jkrZ| |ksZtd S )Nrd )r r Únprg rf r( rA rC )Úresr re Zdiff©rY r r Úchecker< s
|
|
|
z8TestTimestampUnaryOps.test_round_sanity.<locals>.checkerr8 r7 iè r r6 i@B r2 i Ê;rn l Xp7 Úhl qa r l "R: )rw Úint64r r( rg rf )r rv rY r rz rx re r ry r Útest_round_sanity3 sD
|
|
|
|
|
|
|
z'TestTimestampUnaryOps.test_round_sanityc C s| t d d¡}d}tjt|d |jdd W 5 Q R X | d¡}|jdd}|jdks\t|jt t
|
dd¡jksxtd S ) Nr r8 z8Out of bounds nanosecond timestamp: 99999-01-01 00:00:00r i ©Úyearr6 z 99999-01-01) r rT r r r Úreplacer r( rf rw Z
|
datetime64)r r r r- r r r Ú%test_replace_out_of_pydatetime_boundsn s
|
z;TestTimestampUnaryOps.test_replace_out_of_pydatetime_boundsc C sb t dtjjd ¡}| ¡ tdddks*t|jdd}|j |j ksFt| ¡ tdddks^td S )Nl T1)
|
i rk é i$ r~ )
|
r Z_from_value_and_resor Z NPY_FR_usr_ Ú to_pydatetimer r( r rU )r r r- r r r Útest_replace_non_nano{ s ÿ z+TestTimestampUnaryOps.test_replace_non_nanoc C s, t d}|jdd}t d}||ks(td S )Nú2016-01-01 09:00:00r ©Úhourú2016-01-01 00:00:00©r r r( )r r r- r+ r r r Útest_replace_naive
s z(TestTimestampUnaryOps.test_replace_naivec C s8 |}t d|d}|jdd}t d|d}||ks4td S )Nr
r0 r r r r ©r Útz_aware_fixturer1 r r- r+ r r r Útest_replace_aware s
|
z(TestTimestampUnaryOps.test_replace_awarec C s8 |}t d|d}|jdd}t d|d}||ks4td S )Nú 2016-01-01 09:00:00.000000123r0 r r z 2016-01-01 00:00:00.000000123r r r r r Ú test_replace_preserves_nanos s
|
z2TestTimestampUnaryOps.test_replace_preserves_nanosc
|
C sF |}t d|d}|jddddddddd}t d|d}||ksBtd S ) Nr r0 iß rd r é )r ÚmonthÚdayr ÚminuteÚsecondÚ microsecondZ
|
nanosecondz 2015-02-02 00:05:05.000005005r r r r r Útest_replace_multiple s ø
|
z+TestTimestampUnaryOps.test_replace_multiplec C s> |}t d|d}d}tjt|d |jdd W 5 Q R X d S )Nr r0 z.replace\(\) got an unexpected keyword argumentr r )r> )r r r Ú TypeErrorr ©r r r1 r r r r r Útest_replace_invalid_kwarg° s
|
z0TestTimestampUnaryOps.test_replace_invalid_kwargc C s> |}t d|d}d}tjt|d |jdd W 5 Q R X d S )Nr r0 z;value must be an integer, received <class 'float'> for hourr g¹?r )r r r r r r r r r Útest_replace_integer_args¸ s
|
z/TestTimestampUnaryOps.test_replace_integer_argsc C s* t ddd}| d ¡|jd dks&td S )Nz 2013-11-03 01:59:59.999999-0400r/ r0 ©Útzinfo)r Ú tz_localizer r( )r r r r r Ú*test_replace_tzinfo_equiv_tz_localize_noneÀ s z@TestTimestampUnaryOps.test_replace_tzinfo_equiv_tz_localize_nonec C sü t dddd}t d¡j|ddj}|j|d}t|j|d}t d ¡ | ¡ | ¡ ks`t
|
W 5 Q R X ||ksvt
|
|| ¡ kst
|
|j|djd d}t|j|djd d}t d ¡ | ¡ | ¡ ksÒt
|
W 5 Q R X ||ksèt
|
|| ¡ ksøt
|
d S )
|
Nià é é ru ZCETF)Zis_dstr rZ ) r r` ÚtimezoneZlocalizer r r r< Ú set_timezoner) r( r )r r, r Z result_dtZ result_pdr r r Útest_replace_tzinfoÆ s z)TestTimestampUnaryOps.test_replace_tzinfoz tz, normalizer/ c C s | j | ¡S r& )r Ú normalize©Úxr r r Ú<lambda>ã ó z TestTimestampUnaryOps.<lambda>c C s | S r& r r¥ r r r r§ ä r¨ c C sd t d}t ||¡}|||ks$t|jdd}|j|jf|j|jfksLt||}||ks`td S )Nz2017-12-03 16:03:30é )r )r r Zlocalize_pydatetimer( r r r )r r1 r¤ Zts_naiveZts_awareZts2Zts2br r r Útest_replace_across_dstà s
|
z-TestTimestampUnaryOps.test_replace_across_dstc C sV t ddd |¡}|jdd}t ddd}||ks6t|jttd| jksRtd S )Nz 2013-11-3rb r0 r r z2013-11-3 03:00:00r\ )r rT r r( rU rF r r_ )r rS Útr- r+ r r r Útest_replace_dst_borderú s
|
z-TestTimestampUnaryOps.test_replace_dst_borderÚfoldr r1 zdateutil/Europe/Londonz Europe/Londonc C s| t ddddd}t||d |¡}|jd|d}tt dddddj|| d }||ks\t|jttd
|
| j ksxtd S ) Niã é
|
r rd é r0 ru )r r r[ r\ )
|
r r rT r r r( rU rF r r_ )r r r1 rS Údr r- r+ r r r Útest_replace_dst_fold s ÿ z+TestTimestampUnaryOps.test_replace_dst_foldÚargú
|
2013-11-30z2013-11-30 12:00:00c C sV |}t ||d |¡}| ¡ }t d|d}||ks6t|jttd| jksRtd S )Nr0 r³ r\ )r rT r¤ r( rU rF r r_ )r Ztz_naive_fixturer² rS r1 r r- r+ r r r Útest_normalize s z$TestTimestampUnaryOps.test_normalizec C s$ t d ¡ }t d}||ks td S )Nz1969-01-01 09:00:00z1969-01-01 00:00:00)r r¤ r( )r r- r+ r r r Ú test_normalize_pre_epoch_dates s z4TestTimestampUnaryOps.test_normalize_pre_epoch_datesc C s |}|j td}| ¡ | ¡ ks$ttddd}| d¡}| ¡ | ¡ ksNtt d¡" | ¡ }| ¡ | ¡ ksvtW 5 Q R X d S )Nr z 2014-10-11 11:00:01.12345678z
|
US/Centralr0 rZ ) r r r) r( r r^ r< r¢ r )r Z fixed_now_tsr ZutsZtscZutscr, r r r Útest_timestamp& s
|
z$TestTimestampUnaryOps.test_timestampN)6Ú__name__Ú
|
__module__Ú __qualname__r r ÚmarkÚ parametrizer. r4 r5 r; r= r? rJ rR rW rX ra rc rh rr Zxfailr ÚstZintegersr r Zi8maxr r rA rC r} r r r r r r r r r ÚtdZskip_if_windowsr£ r` r¡ r rª r¬ r± r´ rµ r¶ r r r r r s ùþ
|
|
îþ
|
úþ
|
|
|
|
þ ýþ
|
þ úþ îþ ÿ6
|
|
þþ
|
|
r r ru c C s< t d}tddddd| |d}|jdd }|j| ks8td S )
|
Nz Europe/MoscowiÙ r® é rd r¯ )r r r r r r r ru )r )r r r r r( )r r1 r Z ts_replacedr r r Útest_replace_preserves_fold; s r¿ )%r Z dateutil.tzr Z
|
hypothesisr r r¼ Únumpyrw r r` r Z pandas._libsr Zpandas._libs.tslibsr r r
|
r r r r Zpandas._libs.tslibs.dtypesr Zpandas._libs.tslibs.periodr Z pandas.util._test_decoratorsÚutilZ_test_decoratorsr½ Zpandas._testingZ_testingr< r rº r» r¿ r r r r Ú<module> s& $ !
|
