U
|
ý°dÞ± ã @ sR d Z ddlmZmZmZ ddlZddlZddlZddlmZ ddl m
|
Z
|
ddl m Z m Z mZmZmZ ddlmZ ddlm mZ ddlZddlmZmZmZm Z m Z m Z ddl m Z! dd l"m#Z$ ed
|
d d ed d d Z%Z&d d Z'G dd dZ(G dd dZ)G dd dZ*G dd dZ+G dd dZ,G dd dZ-dd Z.G d d d Z/dS ) zP
|
test date_range, bdate_range construction from the convenience range functions
|
é )ÚdatetimeÚtimeÚ timedeltaN)Útimezone)Ú timezones)ÚBDayÚCDayÚ
|
DateOffsetÚMonthEndÚprefix_mapping)ÚOutOfBoundsDatetime)Ú DatetimeIndexÚ TimedeltaÚ TimestampÚ bdate_rangeÚ
|
date_rangeÚoffsets)Ú_generate_rangeéÙ é éÚ c C sÄ | |d k}||d k}|dkr2|r2|dd
}n|dkrL|rL|dd
}nt|dkrj|rj|rj|dd
}nV|dkr|r|dd
}n<|dkr|r|dd
}n"|dkr´|dd
}n |dd
}|S ) z8Helper to get expected range from a both inclusive ranger éÿÿÿÿÚleftNÚrightr ÚneitherÚboth© )Zbegin_to_matchZ end_to_matchÚ
|
both_rangeZinclusive_endpointsZ
|
left_matchZ right_matchÚexpected_ranger r úed:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/datetimes/test_date_range.pyÚ_get_expected_range* s r c @ sB e Zd Zdd Zdd Zdd Zejdd Zd d
|
Z d d Z
|
d S )ÚTestTimestampEquivDateRangec C s2 t dddd}|d }tddd}||ks.td S ©NÚ20090415Ú20090519ú
|
US/Eastern©Útzr ©r r ÚAssertionError©ÚselfÚrngÚstampÚtsr r r Ú test_date_range_timestamp_equivL s z;TestTimestampEquivDateRange.test_date_range_timestamp_equivc C s2 t dddd}|d }tddd}||ks.td S )Nr# r$ údateutil/US/Easternr& r r( r* r r r Ú(test_date_range_timestamp_equiv_dateutilS s zDTestTimestampEquivDateRange.test_date_range_timestamp_equiv_dateutilc C s> t ddt d¡d}|d }tdt d¡d}||ks:td S r" )r Úpytzr r r) r* r r r Ú-test_date_range_timestamp_equiv_explicit_pytzZ s zITestTimestampEquivDateRange.test_date_range_timestamp_equiv_explicit_pytzc C sF ddl m} tdd|dd}|d }td|dd}||ksBtd S )Nr )Údateutil_gettzr# r$ r% r& )Ú pandas._libs.tslibs.timezonesr4 r r r) )r+ Zgettzr, r- r. r r r Ú1test_date_range_timestamp_equiv_explicit_dateutila s
|
zMTestTimestampEquivDateRange.test_date_range_timestamp_equiv_explicit_dateutilc C s6 t ddd}t|dddd }t|}||ks2td S )NéÞ é é r ÚD©ÚperiodsÚfreqr )r r r r) )r+ Zdatetime_instanceÚtimestamp_instancer. r r r Ú6test_date_range_timestamp_equiv_from_datetime_instancek s zRTestTimestampEquivDateRange.test_date_range_timestamp_equiv_from_datetime_instancec C s* t ddddd }td}||ks&td S )Nz
|
2014-03-05r r: r; r r( )r+ r> r. r r r Ú2test_date_range_timestamp_equiv_preserve_frequencyt s zNTestTimestampEquivDateRange.test_date_range_timestamp_equiv_preserve_frequencyN) Ú__name__Ú
|
__module__Ú __qualname__r/ r1 r3 ÚtdZskip_if_windowsr6 r? r@ r r r r r! F s
|
r! c @ sâ e Zd Zej ddddddddg¡d d
|
Zd d Zd d Zdd Z dd Z
|
dd Z ejj dd Z dd Zdd Zej ddd g¡d d Zej dd d g¡d!d" Zej dd#d$g¡d%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zej d3d4d5d6ged7d8d8ed7d8d9d6ged4ed5d6ged4d6d:ed5d6d:d6ged4d6d:ed5d6d:d;gg¡d<d= Zd>d? Z d@dA Z dBdC Z dDdE Z dFdG Z ej dHdI¡dJdK Z!dLdM Z"dNdO Z#dPdQ Z$dRdS Z%dTdU Z&ej dVeedWd9dXd6d:eedWd9dYd6d:geedZd[d8d6d:eedZd[dXd6d:gg¡d\d] Z'd^d_ Z(ej dd`dadbdcddd g¡dedf Z)ej dd`dadbdcddd g¡dgdh Z*ej dd`dadbdcddd g¡didj Z+dkdl Z,dmdn Z-dodp Z.dqdr Z/dsdt Z0dudv Z1dwdx Z2dydz Z3d;S ){ÚTestDateRangesr= ÚNÚUÚLÚTÚSÚHr: c sÈ t d| tdt d |d}t fddtddD |d }t ||¡ td |d}tg |d }t ||¡ t |d}t g|d }t ||¡ d S )
|
NÚ1z
|
1970-01-01r9 ©ÚstartÚendr= c s g | ]}| qS r r )Ú.0Ún©rD r. r r Ú
|
<listcomp> s z8TestDateRanges.test_date_range_edges.<locals>.<listcomp>r é ©r= )r r r r ÚrangeÚtmÚassert_index_equal)r+ r= ÚidxÚexpr rR r Útest_date_range_edges| s4
|
ýþ
|
ý ýz$TestDateRanges.test_date_range_edgesc C s6 t d}tjtdd ttjd|d W 5 Q R X d S )Nr úCannot generate range with©Úmatché ©rO r< r= )r ÚpytestÚraisesr r r Úmin)r+ r= r r r Ú)test_date_range_near_implementation_bound s z8TestDateRanges.test_date_range_near_implementation_boundc C s\ d}t jt|d tdtjdd W 5 Q R X t jt|d ttjddd W 5 Q R X d S )Nz$Neither `start` nor `end` can be NaTr] ú
|
2016-01-01r: rM )ra rb Ú
|
ValueErrorr ÚpdZNaT©r+ Úmsgr r r Útest_date_range_nat¦ s
|
z"TestDateRanges.test_date_range_natc C sx t d ¡ tdddd}W 5 Q R X |d tdks8tt|dksHtd}tjt|d tdd d
|
d W 5 Q R X d S ) Nú
|
1677-09-22iÿA r: ©rN r< r= r r\ r] z
|
1969-05-04i Âë Z30000Dr; ) rW Zassert_produces_warningr r r) Úlenra rb r )r+ Údtiri r r r Ú'test_date_range_multiplication_overflow® s z6TestDateRanges.test_date_range_multiplication_overflowc C sZ t dddd}t |d t|dd}| |¡s2tt |d t|dd}| |¡sVtd S ) Nrk z
|
2262-04-11r: rM r rl r r` )r rm Úequalsr) )r+ rn Údti2Zdti3r r r Ú*test_date_range_unsigned_overflow_handling½ s
|
z9TestDateRanges.test_date_range_unsigned_overflow_handlingc C sX d}t jt|d tdddd W 5 Q R X t jt|d tdddd W 5 Q R X d S ) Nr\ r] ú
|
1970-02-01i ' rK rl ú
|
1969-11-14r` ©ra rb r r rh r r r Ú.test_date_range_int64_overflow_non_recoverableÉ s
|
z=TestDateRanges.test_date_range_int64_overflow_non_recoverablec C s¼ t d}t d}t||dd}|d |ks.t|d |ks>tt|t|dd}t ||¡ t d}t d }t||dd}|d |kst|d |kstt|t|dd
|
}t ||¡ d S ) Nz
|
2262-02-23rt z-1HrM r r r` rs z
|
1677-10-22rl )r r r) rm rW rX ) r+ rN rO Úexpectedrn Zstart2Zend2Ú expected2rq r r r Ú>test_date_range_int64_overflow_stride_endpoint_different_signsÔ s zMTestDateRanges.test_date_range_int64_overflow_stride_endpoint_different_signsc C sX d}t jt|d tdddd W 5 Q R X t jt|d tdddd W 5 Q R X d S ) NzCannot generate ranger] re i r: r; z
|
1763-10-12r` ru rh r r r Ú test_date_range_out_of_boundsì s
|
z,TestDateRanges.test_date_range_out_of_boundsc C s" t dddd}t|dks td S )Nz1/1/2000 00:00z1/1/2000 00:18Z5minrU r9 ©r rm r) ©r+ r, r r r Útest_date_range_gen_errorô s z(TestDateRanges.test_date_range_gen_errorZASZYSc C s4 t dd|d}tdddddg|d}t ||¡ d S ) Nú1/1/2013ú7/1/2017rU ú
|
2013-01-01z
|
2014-01-01z
|
2015-01-01re ú
|
2017-01-01©r r rW rX ©r+ r= r, rZ r r r Útest_begin_year_aliasø s þz$TestDateRanges.test_begin_year_aliasÚAÚYc C s2 t dd|d}tddddg|d}t ||¡ d S )Nr~ r rU ú
|
2013-12-31ú
|
2014-12-31ú
|
2015-12-31z
|
2016-12-31r r r r r Útest_end_year_alias s
|
ÿz"TestDateRanges.test_end_year_aliasZBAZBYc C s2 t dd|d}tddddg|d}t ||¡ d S )Nr~ r rU r r r z
|
2016-12-30r r r r r Ú test_business_end_year_alias s
|
ÿz+TestDateRanges.test_business_end_year_aliasc C sx t dddd}tdddgdd}t ||¡ |jdks:tt dd dd}tdd
|
d gd d}t ||¡ |jd ksttd S ) Nz
|
2011-12-31z-2Ar8 )r= r< z
|
2009-12-31z
|
2007-12-31rU z
|
2011-01-31z-2Mz
|
2010-11-30z
|
2010-09-30)r r rW rX r= r) )r+ r, rZ r r r Ú test_date_range_negative_freq s z,TestDateRanges.test_date_range_negative_freqc C s* t dddd}td}|d |ks&td S )Nú1/1/2000é
|
ZBMSr; z
|
2000-01-03r r( )r+ r, Zex_firstr r r Útest_date_range_bms_bug s z&TestDateRanges.test_date_range_bms_bugc s t ¡ d}t|ddd}td t fddt|D d}t ||¡ td |dd
|
d}td d }|D ]}| ¡ |ksnt qnd S ) Né2 FZ2D)r< Ú normalizer= r_ c s g | ]}| qS r r ©rP Úi©ÚoffsetZsnapr r rS . s z<TestDateRanges.test_date_range_normalize.<locals>.<listcomp>rU z1/1/2000 08:15ÚBé é )
|
r Útodayr r r rV rW rX r r) )r+ rQ r, ÚvaluesZthe_timeÚvalr r r Útest_date_range_normalize' s
|
z(TestDateRanges.test_date_range_normalizec C sF t ddtjddddd}|d td ks.t|d td
|
ksBtd S ) Nr r_ r r8 Znearest)Z startingMonthÚweekdayZ variationrl r z
|
2013-01-31z
|
2014-01-30)r r ZFY5253r r) ©r+ Údrr r r Útest_date_range_fy52527 s ýz%TestDateRanges.test_date_range_fy5252c C sV t dddddd}t dddddd}d}tjt|d t||d
|
d d W 5 Q R X d S ) NéÛ r rT r8 é( r é úVOf the four parameters: start, end, periods, and freq, exactly three must be specifiedr] r Úsr; )r ra rb rf r ©r+ rN rO ri r r r Ú#test_date_range_ambiguous_arguments@ s ÿz2TestDateRanges.test_date_range_ambiguous_argumentsc C sr t dddd}tdddgd d}t ||¡ t d d
|
d dd }ttd d dtdd dtdd dg}t ||¡ d S )Nz
|
2018-04-24z
|
2018-04-27r8 ©r< z2018-04-24 00:00:00z2018-04-25 12:00:00z2018-04-27 00:00:00rU z2018-04-01 01:00:00z2018-04-01 04:00:00zAustralia/Sydney)r' r< z2018-04-01 01:00:00+1100r& z2018-04-01 02:00:00+1000z2018-04-01 04:00:00+1000)r r rW rX r ©r+ Úresultrw r r r Ú#test_date_range_convenience_periodsL s& þ ü
|
|
|
ýÿz2TestDateRanges.test_date_range_convenience_periodszstart,end,result_tzÚ20180101Z20180103r% iâ r r8 r& Nc C s0 t ||d|d}t ddddd}t ||¡ d S )Nr8 ©r< r' r¬ r: r% ©r< r= r' ©r rW rX )r+ rN rO Z result_tzrª rw r r r Ú test_date_range_linspacing_tzf s z,TestDateRanges.test_date_range_linspacing_tzc C sº t ddddddddgd d
|
}tddd d
|
}t ||¡ t dd gd d
|
}tdd d d
|
}t ||¡ t ddddddddd d d dddddddddddddgd d
|
}tddd d
|
}t ||¡ d S )Nz2014-07-04 09:00z2014-07-04 10:00z2014-07-04 11:00z2014-07-04 12:00z2014-07-04 13:00z2014-07-04 14:00z2014-07-04 15:00z2014-07-04 16:00ZBHrU z2014-07-07 09:00z2014-07-07 10:00z2014-07-07 11:00z2014-07-07 12:00z2014-07-07 13:00z2014-07-07 14:00z2014-07-07 15:00z2014-07-07 16:00z2014-07-08 09:00z2014-07-08 10:00z2014-07-08 11:00z2014-07-08 12:00z2014-07-08 13:00z2014-07-08 14:00z2014-07-08 15:00z2014-07-08 16:00)r r rW rX )r+ rY r, r r r Ú test_date_range_businesshour~ s^ ø
|
õ èå z+TestDateRanges.test_date_range_businesshourc C s: d}d}t ||dd}t ||tddd}t ||¡ d S )Nz
|
2020-01-01z
|
2020-01-11Ú3DrU r8 )Údays)r r rW rX )r+ rN rO Úrng1Úrng2r r r Útest_date_range_timedelta³ s
|
z(TestDateRanges.test_date_range_timedeltac C s d}t jt|d tdd W 5 Q R X t jt|d tdd W 5 Q R X t jt|d tdd W 5 Q R X t jt|d tddd W 5 Q R X t jt|d tddd
|
W 5 Q R X t jt|d tddd W 5 Q R X t jt|d t W 5 Q R X d S ) Nr¤ r] r )rN )rO r r¨ rK )rN r= )rO r= r; ©ra rb rf r rh r r r Útest_range_misspecifiedº s ÿz&TestDateRanges.test_range_misspecifiedc C s& t tdddd}t|dks"td S )Nz1960-04-01 00:00:00éL zQS-JANr; )r r rm r) )r+ rª r r r Útest_compat_replaceÖ s z"TestDateRanges.test_compat_replacec C sL t jdd}d}tjt|d$ ttdddtddd|d W 5 Q R X d S ) NrT )Úminutez4Offset <DateOffset: minute=5> did not increment dater] r¡ é é rU )r r ra rb rf r r )r+ r ri r r r Útest_catch_infinite_loopÜ s z'TestDateRanges.test_catch_infinite_loopr< )r r_ c C s" t d|dd}t||ks td S )NZ20110101zWOM-1MONrl r{ )r+ r< Úresr r r Ú test_wom_lenã s zTestDateRanges.test_wom_lenc C sb t djddd}t djddd}t ddd||g}t|dd }td
|
d ddd }t ||¡ d S ) Nz2010-11-07 01:00:00z
|
US/PacificT)Z ambiguousFz2010-11-07 00:00:00r& rK rU z 2010-11-7r8 )rN r< r= r' )r Z tz_localizer r rW rX )r+ Zpre_dstZpst_dstZ expect_datarw rª r r r Útest_construct_over_dsté s ÿ ÿ
|
ý z&TestDateRanges.test_construct_over_dstc C s< t dddd}ttdtdtdgdd}t ||¡ d S )Nz2013-01-01 00:00:00+09:00z2013/01/01 02:00:00+09:00rK rU z2013-01-01 01:00:00+09:00z2013-01-01 02:00:00+09:00)r r r rW rX r© r r r Ú5test_construct_with_different_start_end_string_formatú s ÿýúzDTestDateRanges.test_construct_with_different_start_end_string_formatc C s4 d}t jt|d tddtdd W 5 Q R X d S )Nz.Offset <0 \* MonthEnds> did not increment dater] r z1/1/2001r rU )ra rb rf r r
|
rh r r r Ú test_error_with_zero_monthends s z-TestDateRanges.test_error_with_zero_monthendsc sT t dd tdd d}tdddt fdd td
|
D d}t ||¡ d S ) Nr8 )Úmonthsú2011-1-1z 2012-1-31rU r¡ r c s g | ]}| qS r r r ©r rN r r rS s z1TestDateRanges.test_range_bug.<locals>.<listcomp>rT )r r r r rV rW rX r© r rÆ r Útest_range_bug s
|
|
z TestDateRanges.test_range_bugc C sê t d}| tddd¡}| tddd¡}t|dd}|jj|jksJt|d |ksZt|d |ksjtt|dd}|jj|jkst|d |kst|d |ks¨tt||d }|jj|jksÆt|d |ksÖt|d |ksætd S )
|
Nr% r¡ r r8 ©rN r< r r_ ©rO r< ©rN rO )r Zlocalizer r r' Úzoner) ©r+ r' rN rO r r r r Útest_range_tz_pytz s z!TestDateRanges.test_range_tz_pytzz
|
start, endr7 é r½ iÝ r¼ c C sÞ t ||dd}|d |ks t|d |ks.tt |jdk¡sBtt ||ddd}|d |ksbt|d |ksrtt |jdk¡stt |jd d|jd dddd}|d |ks¶t|d |ksÆtt |jdk¡sÚtd S )Nr: rU r r r% ©r= r' ©Útzinfo)r r) ÚnpÚallÚhourÚreplace)r+ rN rO r r r r Ú test_range_tz_dst_straddle_pytz, s"
|
|
üz.TestDateRanges.test_range_tz_dst_straddle_pytzc sþ ddl m fdd}tddd|dd}tddd |dd}t|d d
|
}|j|dks^t|d |ksnt|d |ks~tt|d d }|j|dkst|d |ks¬t|d |ks¼tt||d }|j|dksÚt|d |ksêt|d |ksútd S )Nr ©Ú maybe_get_tzc s d| S )Nz dateutil/r )Úxr× r r Ú<lambda>T ó z7TestDateRanges.test_range_tz_dateutil.<locals>.<lambda>r¡ r r% rÐ r8 rÈ r_ rÉ rÊ )r5 rØ r r r' r) rÌ r r× r Útest_range_tz_dateutilN s z%TestDateRanges.test_range_tz_dateutilZ1Dr² Z2MZ7WZ3Hc C sV t ddd}t ddd}t||||d}t||d|d}t||||}t ||¡ d S )Nr¡ r r7 ©Ú inclusiver= r )r r r rW rX ©r+ r= Úinclusive_endpoints_fixtureÚbeginrO Ú result_ranger r r r r Útest_range_closedh s ÿ ÿz TestDateRanges.test_range_closedc C sV t ddd}t ddd}t||||d}t||d|d}t||||}t ||¡ d S )Nú2011/1/1r% r& ú2014/1/1rÝ r ©r r r rW rX rß r r r Ú)test_range_closed_with_tz_aware_start_endw s ÿüz8TestDateRanges.test_range_closed_with_tz_aware_start_endc
|
C sj t d}t d}t ddd}t ddd}t||||dd}t||d|dd}t||||} t | |¡ d S )Nrä rå r% r& )rÞ r= r' r ræ )
|
r+ r= rà rá rO ZbegintzZendtzrâ r r r r r Ú1test_range_with_tz_closed_with_tz_aware_start_end s2 û ÿüz@TestDateRanges.test_range_with_tz_closed_with_tz_aware_start_endc
|
C sØ t ddd|d}t ddd|d}t ddd|d}t ddd|d}|}|}|}|dkrb|dd
}n6|d krx|d d
|
}n |d kr|dd
}|d d
|
}|dd
|
} t ||¡ t ||¡ t ||¡ t || ¡ d S ) Nz
|
2015-09-12z
|
2015-12-01zQS-MAR©r= rÞ z
|
2015-09-01z
|
2015-09-11r r r r r r¯ )
|
r+ rà Zright_boundaryZ left_boundaryZ both_boundaryZneither_boundaryZexpected_rightZ expected_leftZ expected_bothZexpected_neitherr r r Útest_range_closed_boundary¨ sN üüüü z)TestDateRanges.test_range_closed_boundaryc C sB t dddd}|d tdddks&t|d tdd
|
dks>td S ) NZ2014Z2015ÚMrU r r7 r é r r½ )r r r) r r r r Útest_years_onlyÖ s z TestDateRanges.test_years_onlyc C s` t dddd}t dddd}tddgd d
|
d d }td d gd d
|
d d }t ||¡ t ||¡ d S )Nz2005-01-12 10:00z2005-01-12 16:00Z345minrU z2005-01-13 10:00z2005-01-13 16:00z2005-01-12 10:00:00z2005-01-12 15:45:00údatetime64[ns]Z345T©Údtyper= r' z2005-01-13 10:00:00z2005-01-13 15:45:00r )r+ Zresult_1Zresult_2Z
|
expected_1Z
|
expected_2r r r Ú test_freq_divides_end_in_nanosÜ s üü z-TestDateRanges.test_freq_divides_end_in_nanosc C sB t ddtddd}t|dks$t|d tddd d
|
ks>td S ) Nz2010-09-01 05:00:00r rÎ )Úhoursr; r r r£ r rT )r r rm r) r r| r r r Útest_cached_range_bugï s z$TestDateRanges.test_cached_range_bugc C s. t ddd}t|ddd}t|dks*td S )Nú20130220 10:00r% r& r_ r )r r rm r) )r+ rN rª r r r Ú test_timezone_comparison_bugô s z+TestDateRanges.test_timezone_comparison_bugc C s< t ddd}d}tjt|d t|ddd W 5 Q R X d S ) Nrô r% r& z0Inferred time zone not equal to passed time zoner] r_ z Europe/Berlinr )r ra rb r) r )r+ rN ri r r r Ú test_timezone_comparison_assertú s z.TestDateRanges.test_timezone_comparison_assertc C s> |}t ddd|d}t ddd|dd d d
}t ||¡ d S )Nz
|
2011-06-01z
|
2011-01-01z-1MS)rN rO r= r' Z1MS)rO rN r= r' r r¯ )r+ Ztz_aware_fixturer' rª rw r r r Ú1test_negative_non_tick_frequency_descending_dates s ÿz@TestDateRanges.test_negative_non_tick_frequency_descending_datesc C sb d}d}t ||d|d}t ||ddd}|dkr>|dd
}n|dkrR|d d
}t ||¡ d S ) Nz
|
2021-09-02r: )rN rO r= rÞ r r r r )r r r r¯ )r+ rà rN rO rª r rw r r r Ú test_range_where_start_equal_end s ÿ z/TestDateRanges.test_range_where_start_equal_end)4rA rB rC ra ÚmarkÚ parametrizer[ rd rj ro rr rv Zslowry rz r} r r r r r r r r§ r« r r r° r± r¶ r¸ rº r¾ rÀ rÁ r rà rÇ rÍ rÖ rÜ rã rç rè rê rí rñ ró rõ rö r÷ rø r r r r rE { s
|
"
|
|
|
|
|
|
ý
|
|
ý÷þ
|
5
|
þþûþ
|
|
|
|
. rE c @ sn e Zd ZdZdd Zej dddg¡dd Zej dddg¡d d
|
Z d d Z
|
ej dddg¡d d Z dS )ÚTestDateRangeTZz#Tests for date_range with timezonesc C s t ddddd}|j d S )Nz
|
2012-01-01z
|
2012-01-10r: ZHongkongrÏ )r rÔ r r r r Útest_hongkong_tz_convert s z(TestDateRangeTZ.test_hongkong_tz_convertÚtzstrr% r0 c C sZ t ddddd}|jdk ¡ s"tt dd|d }|j}tjdgd d
|
d }t ||¡ d S ) Nz03/06/2012 00:00éÈ zW-FRIr% r® r z
|
2012-11-02r r Úint32©rð )r rÔ rÓ r) rg ÚIndexrW rX )r+ rý r rª rw r r r Ú#test_date_range_span_dst_transition$ s z3TestDateRangeTZ.test_date_range_span_dst_transitionc C s6 t |¡}tdd|d}tdd|d}t ||¡ d S )Nr r r )r rØ r rW rX )r+ rý r' rª rw r r r Ú%test_date_range_timezone_str_argument2 s
|
z5TestDateRangeTZ.test_date_range_timezone_str_argumentc C sv ddl m} |}tdddddd|d}tdddddd|d}t||d }||jksVtt ||g¡}||jksrtd S )
|
Nr )Úfixed_off_no_nameiÜ r8 r¼ rT rÐ rÎ rÊ )Z-pandas.tests.indexes.datetimes.test_timezonesr r r r' r) rg r )r+ r ÚoffrN rO r, rY r r r Ú'test_date_range_with_fixedoffset_noname: s z7TestDateRangeTZ.test_date_range_with_fixedoffset_nonamec C s> t d|d}|jdksttddd|d}||d ks:td S ) Nz3/11/2012 05:00r& rT z3/11/2012 04:00r rK r® r )r rÔ r) r )r+ rý r- r, r r r Útest_date_range_with_tzF s z'TestDateRangeTZ.test_date_range_with_tzN) rA rB rC Ú__doc__rü ra rù rú r r r r r r r r rû s
|
|
rû c @ s² e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd \Z Z
|
d\Z Z e j dee e dee
|
fee ee
|
e dfee e dee
|
e dfee e dee
|
e dfg¡dd ZdS )ÚTestGenRangeGenerationc C s> t tttd t dd}t tttd ddd}||ks:td S )NÚns©r< r Úunitr )ÚlistÚgenerate_rangeÚSTARTÚENDr r) ©r+ r´ rµ r r r Ú test_generateQ s z$TestGenRangeGeneration.test_generatec C s> t tttd t dd}t tttd ddd}||ks:td S )Nr
|
r ÚC)r r r r r r) r r r r Útest_generate_cdayV s z)TestGenRangeGeneration.test_generate_cdayc C sH t ttdddd dt dd}tdddtdddg}||ksDtd S )Nr r8 é r_ r
|
©rN rO r< r r é ©r r r r r) ©r+ r, rw r r r Útest_1[ s
|
ûÿ z TestGenRangeGeneration.test_1c C sZ t ttdddtdddd t dd}tdddtdddtdddg}||ksVtd S )NéØ r r8 r
|
r r_ r r r r r Útest_2h s
|
|
ûÿ "z TestGenRangeGeneration.test_2c C s< t ttdddtdddd t dd}g }||ks8td S )Nr r rT rÎ r
|
r r r r r r Útest_3u s
|
|
ûÿ z TestGenRangeGeneration.test_3c
|
C s| t dddd}t dddd}ddd d
|
g}d d d dddddddg
|
}t|ddd d}t|ddd d}t ||¡ t ||¡ d S )Nz2015-04-15 00:00:03z2016-04-22 00:00:00ÚQrM z2015-06-22 00:00:04ÚWz2015-06-30 00:00:03z2015-09-30 00:00:03z2015-12-31 00:00:03z2016-03-31 00:00:03z2015-04-19 00:00:03z2015-04-26 00:00:03z2015-05-03 00:00:03z2015-05-10 00:00:03z2015-05-17 00:00:03z2015-05-24 00:00:03z2015-05-31 00:00:03z2015-06-07 00:00:03z2015-06-14 00:00:03z2015-06-21 00:00:03rî zQ-DECrï zW-SUNr )r+ Zresult1Zresult2Zexpected1_listZexpected2_listZ expected1rx r r r Ú test_precision_finer_than_offset sP ÿ ÿüö ÿ ÿ z7TestGenRangeGeneration.test_precision_finer_than_offset)r r )r% z Europe/Londonz start,endr& c C sV d}t jt|d t|| W 5 Q R X t jt|d t||t d W 5 Q R X d S )Nz>Start and end cannot both be tz-aware with different timezonesr] rU )ra rb Ú TypeErrorr r r¦ r r r Ú test_mismatching_tz_raises_err¨ s
|
z5TestGenRangeGeneration.test_mismatching_tz_raises_errN)rA rB rC r r r r r r Zdt1Zdt2Ztz1Ztz2ra rù rú r r" r r r r r P s" #üþ r c @ sZ e Zd Zdd Zdd Zdd Zdd Zej d d
|
d d d g¡dd Z
|
dd Z dd Z dS )ÚTestBusinessDateRangec C s² t ttt d t tdt d t tdt d d}tjt|d tddd W 5 Q R X tjt|d t ddd W 5 Q R X d
|
}tjt|d t ttd d d W 5 Q R X d S ) NrU é r; r` z periods must be a number, got Br] rÅ ú2012-1-1r z>freq must be specified for bdate_range; use date_range insteadr )r r r r ra rb r! r rh r r r Útest_constructor» s z&TestBusinessDateRange.test_constructorc C sZ t ddd}t|dd}|dt }t|dks6t|d |ksFt|d |ksVtd S ) Nr rT é r$ rÉ é r r )r r r rm r) ©r+ rO r Z firstDater r r Ú test_miscË s z TestBusinessDateRange.test_miscc C s d}d}t jt|d t| W 5 Q R X t jt|d t|dd W 5 Q R X t jt|d t|dd W 5 Q R X t jt|d t|| W 5 Q R X d S )Nz
|
2007/100/1z;Unknown datetime string format, unable to parse: 2007/100/1r] r rÈ rÉ )ra rb rf r r )r+ Zbadly_formed_dateri r r r Útest_date_parse_failureÔ s z-TestBusinessDateRange.test_date_parse_failurec C sB t dd}t dd}|jjt ks&t| |¡}t|ts>td S )Nú 12/5/2011ú 12/2/2011)r Ú_datar= r r) ÚunionÚ
|
isinstancer ©r+ r´ rµ rª r r r Útest_daterange_bug_456ä s
|
|
|
|
z,TestBusinessDateRange.test_daterange_bug_456rÞ r r r r c C s> d}d}t ||d|d}d}d}t ||dd}t ||¡ d S ) Nz
|
2018-07-21z
|
2018-07-29r ré z
|
2018-07-23z
|
2018-07-27r: rU r¯ )r+ rÞ rN rO rª Z
|
bday_startZbday_endrw r r r Ú test_bdays_and_open_boundariesí s z4TestBusinessDateRange.test_bdays_and_open_boundariesc C s> t j d¡ ¡ }t|d ddd}t|gdd}t ||¡ d S )Nr: r r r` rU )r ÚmaxÚfloorÚ to_pydatetimer r rW rX )r+ rN r, rw r r r Útest_bday_near_overflowú s z-TestBusinessDateRange.test_bday_near_overflowc C s@ d}t j d¡ ¡ }tjt|d t|ddd W 5 Q R X d S )Nz"Out of bounds nanosecond timestampr: r] r_ r r; )r r4 r5 r6 ra rb r r )r+ ri rN r r r Útest_bday_overflow_error s z.TestBusinessDateRange.test_bday_overflow_errorN) rA rB rC r& r* r+ r2 ra rù rú r3 r7 r8 r r r r r# º s
|
r# c @ s¶ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z e
|
j ddd e D ¡dd Ze
|
j ddddg¡dd Ze
|
j ddd d gfd d d gfd d d d gfg¡d d! Zd"S )#ÚTestCustomDateRangec C s t ttt d t tdt d t tdt d d}tjt|d tddd W 5 Q R X tjt|d t ddd W 5 Q R X d S )
|
NrU r$ r; r` z periods must be a number, got Cr] rÅ r% r )r r r r ra rb r! r rh r r r r&
|
s z$TestCustomDateRange.test_constructorc C s\ t ddd}t|ddd}|dt }t|dks8t|d |ksHt|d |ksXtd S )
|
Nr rT r' r$ r r` r( r r )r r r rm r) r) r r r r* s z TestCustomDateRange.test_miscc C sJ t dddd}t dddd}|jjt ks.t| |¡}t|tsFtd S )Nr, r rU r- )r r. r= r r) r/ r0 r r1 r r r r2 s
|
|
z*TestCustomDateRange.test_daterange_bug_456c C s@ t dddd}tdddgdd}t ||¡ |j|jks<td S )Nú
|
2013-05-01r8 r r; ú
|
2013-05-02ú
|
2013-05-03rU )r r rW rX r= r) r© r r r Útest_cdaterange( s z#TestCustomDateRange.test_cdaterangec C sp t ddddd}tdddg|jd}t ||¡ |j|jks@td }tjt|d
|
t dddd W 5 Q R X d S ) Nr: r8 r úSun Mon Tue Wed Thu)r< r= Úweekmaskr; ú
|
2013-05-05rU ú[a custom frequency string is required when holidays or weekmask are passed, got frequency Br] )r< r? © r r r= rW rX r) ra rb rf ©r+ rª rw ri r r r Útest_cdaterange_weekmask. s ÿ ÿ ÿz,TestCustomDateRange.test_cdaterange_weekmaskc C st t ddddgd}tdddg|jd}t ||¡ |j|jksBtd }tjt|d
|
t dddgd W 5 Q R X d S ) Nr: r8 r )r< r= Úholidaysr; r< ú
|
2013-05-06rU rA r] )r< rE rB rC r r r Útest_cdaterange_holidays@ s ÿ ÿz,TestCustomDateRange.test_cdaterange_holidaysc C sx t dddddgd}tdddg|jd }t ||¡ |j|jksDtd
|
}tjt|d t ddddgd W 5 Q R X d S ) Nr: r8 r r> )r< r= r? rE r; r@ rF rU rA r] )r< r? rE rB rC r r r Ú%test_cdaterange_weekmask_and_holidaysP s, û ÿ ÿüz9TestCustomDateRange.test_cdaterange_weekmask_and_holidaysr= c C s g | ]}| d ¡r|qS )r )Ú
|
startswith)rP r= r r r rS l s
|
z TestCustomDateRange.<listcomp>c C sR t tt|ddgd |d }d| }tjt|d t tt|d W 5 Q R X d S )Nz Mon Wed Friz
|
2009-03-14)r= r? rE ZFOOz!invalid custom frequency string: r] rU )r r r ra rb rf )r+ r= Zbad_freqri r r r Útest_all_custom_freqk s ÿ
|
z(TestCustomDateRange.test_all_custom_freqÚ start_end)z2018-01-01T00:00:01.000Zz2018-01-03T00:00:01.000Z)z2018-01-01T00:00:00.010Zz2018-01-03T00:00:00.010Z)z2001-01-01T00:00:00.010Zz2001-01-03T00:00:00.010Zc C s2 |\}}t ||ddd}t|g}t ||¡ d S )Nr_ r )rN rO r< rÞ r )r+ rK rN rO rª rw r r r Ú&test_range_with_millisecond_resolutiony s
|
|
z:TestCustomDateRange.test_range_with_millisecond_resolutionzstart,period,expectedz2022-07-23 00:00:00+02:00r z2022-07-25 00:00:00+02:00z2022-07-22 00:00:00+02:00r_ c C s& t ||dd}t|}t ||¡ d S )Nr rl r )r+ rN Zperiodrw rª r r r Ú.test_range_with_timezone_and_custombusinessday s zBTestCustomDateRange.test_range_with_timezone_and_custombusinessdayN)rA rB rC r& r* r2 r= rD rG rH ra rù rú r rJ rL rM r r r r r9 s> ÿ
|
ýþ
|
|
|
ýýþ r9 c C sB t jddgd} tdd| d}tddd d
|
g| d }t ||¡ d S ) Nz15:00z
|
2020-11-26)rN rE z2020-11-25 15:00r9 rl z2020-11-25 15:00:00z2020-11-25 16:00:00z2020-11-27 15:00:00z2020-11-27 16:00:00rU )r ZCustomBusinessHourr r rW rX )r= rª rw r r r Ú$test_date_range_with_custom_holidays s üù rN c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) ÚTestDateRangeNonNanoc C s2 d}t jt|d tddddd W 5 Q R X d S )Nz+'unit' must be one of 's', 'ms', 'us', 'ns'r] re z
|
2016-03-04r8 Úh©r< r r· rh r r r Ú test_date_range_reso_validation¬ s z4TestDateRangeNonNano.test_date_range_reso_validationc C s2 d}t jt|d tddddd W 5 Q R X d S )Nz)Use a lower freq or a higher unit insteadr] re z
|
2016-01-02r
|
Úms©r= r r· rh r r r Ú%test_date_range_freq_higher_than_reso± s z:TestDateRangeNonNano.test_date_range_freq_higher_than_resoc C sÄ t ddddd}tjddtjd}t| d¡dd }t ||¡ t ddd
|
d
|
d}tjd d tjd}t| d ¡d
|
d }t ||¡ t ddddd}tjddtjd}t| d¡dd }t ||¡ d S )Nre z2016-01-01 00:00:01rS rT l \÷tGl é_÷tGr zM8[ms]rU Úusl `¦gé ) l A"Ågé ) úM8[us]z2016-01-01 00:00:00.001r
|
l ^cQ)! l AB|cQ)! zM8[ns])r rÒ ÚarangeÚint64r ÚviewrW rX )r+ rn r, rw r r r Ú!test_date_range_freq_matches_reso¸ s ÿz6TestDateRangeNonNano.test_date_range_freq_matches_resoc C s t d}t d}tjtdd t||ddd W 5 Q R X t||dd d}tj| d ¡j| d ¡jgtj d
|
}t
|
| d ¡}t ||¡ d S ) Nz2022-10-19 11:50:44.719781z2022-10-19 11:50:47.066458z Cannot losslessly convert unitsr] r8 r¥ rQ r_ rV r rW )r ra rb rf r rÒ ÚarrayZas_unitÚ_valuerY r rZ rW rX )r+ rN rO rn r, rw r r r Ú)test_date_range_freq_lower_than_endpointsË s ÿz>TestDateRangeNonNano.test_date_range_freq_lower_than_endpointsc C s t d¡}t d¡}t||ddd}|jdks2t|jdks@tt | d¡ d¡|d d¡ d¡d ¡ d¡}t
|
| ¡ |¡ d S )
|
Nz
|
1066-10-14z
|
2305-07-13r: r¥ rT zM8[s]Úi8r iQ ) rÒ Z
|
datetime64r r= r) rð rX ZastyperZ rW Zassert_numpy_array_equalZto_numpy)r+ rN rO rn rZ r r r Útest_date_range_non_nanoÜ s
|
|
ýüz-TestDateRangeNonNano.test_date_range_non_nanoN)rA rB rC rR rU r[ r^ r` r r r r rO « s
|
rO )0r r r r ÚnumpyrÒ ra r2 r Zpandas._libs.tslibsr Zpandas._libs.tslibs.offsetsr r r r
|
r Z pandas.errorsr Z pandas.util._test_decoratorsÚutilZ_test_decoratorsrD Zpandasrg r r r r r r Zpandas._testingZ_testingrW Z pandas.core.arrays.datetimesr r r r r r! rE rû r r# r9 rN rO r r r r Ú<module> s: 5 $6jO
|
