U
|
®ý°dÌ9 ã 1 @ sÞ d dl m Z mZ d dlZd dlZd dlmZmZ d dlm Z d dl
|
m Z d dl m Z d dlmZmZmZmZmZmZ d dlmZ d dlmZmZ d d lm Z d d
|
l m Z m Z ej ed d fed d dfed ddfed ddfe !d d¡dfed ddfedddfgddd Z"d d eD d d d D d d!d"g d#d e#d d$D d%d eD Z$ej% &d&e$¡ej% &d'd$d(g¡d)d* Z'd+d, Z(d-d. Z)d/d0 Z*d1d2 Z+d3d4 Z,d5d6 Z-d7d8 Z.d9d: Z/d;d< Z0d=d> Z1d?d@ Z2ej% &dAe#d d$¡dBdC Z3ej% &dDdEdF dGdF g¡dHdI Z4ej% &dJdKdLdMg¡dNdO Z5ej% &dPe6dQdRdSdTgdUdVdWdXgdYdZd[d\gd]dSd^d_gdSd`dadbgdcdddedfgdg 7¡ ¡dhdi Z8ej% &djdkdlgdmdngdodpgg¡ej% &d&ddqdrdsdtdudvg¡dwdx Z9dydz Z:ej% &d{d|d}d~dddgdfd|d}d~ddddddd
dg dfdddddddddddg dfddddddddddddddddddddddddgdfg¡d d¡ Z;d¢d£ Z<d¤d¥ Z=ej% &d¦e >d§¡e ?d§¡e @d§¡e Ad§¡g¡d¨d© ZBej%jCe dªd«d¬d ZDd®d¯ ZEd°d± ZFej% &d²d§d³g¡d´dµ ZGd¶d· ZHej% &d&ddg¡d¸d¹ ZIej% &d&d ddg¡dºd» ZJej% &d¼e d½dF g¡ej% &d&d¾d¿dÀdÁdÂdÃdÄdÅdÆdÇdÈdÉdÊdËdÌdÍdÎdÏdÐdÑdÒdÓdÔdÕdÖd×dØdÙdÚdÛdÜdÝdÞdßdàdádâdãdädådædçdèdédêg-¡dëdì ZKdídî ZLdïdð ZMdñdò ZNdS )óé )ÚdatetimeÚ timedeltaN)ÚDAYSÚMONTHS)Ú _get_offset)ÚINVALID_FREQ_ERR_MSG)Úis_platform_windows)Ú DatetimeIndexÚIndexÚSeriesÚ TimestampÚ
|
date_rangeÚ period_range)Ú DatetimeArrayÚTimedeltaArray)Ú to_datetime)Ú frequenciesÚoffsetsé ÚD)ÚhoursÚH)ÚminutesÚT)ÚsecondsÚSÚnsÚN)Ú microsecondsÚUiè ÚL)Úparamsc C s | j S ©N)Úparam)Úrequest© r% úfd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/tseries/frequencies/test_inference.pyÚbase_delta_code_pair& s r' c C s g | ]}d | qS )úQ-r% )Ú.0Úmonthr% r% r& Ú
|
<listcomp>6 s r+ c C s$ g | ] }t D ]}| d | q qS )ú-)r )r) Zannualr* r% r% r& r+ 7 s )ÚAZBAÚMZBMZBMSc C s$ g | ] }t D ]}d | | q qS )zWOM-)r )r) ÚcountÚdayr% r% r& r+ 9 s é c C s g | ]}d | qS )zW-r% )r) r0 r% r% r& r+ : s ÚfreqÚperiodsé c C s | ¡ }td| |d}t|j}| d¡s@t |¡|jkstnPt |¡}|dkoZ|jdk}|dkol|jdk}|dko~|jd k}|s|s|std S )
|
Nú1/1/2000©r3 r2 r( úQ-DEC)ÚQr7 zQ-SEPzQ-JUNzQ-MARúQ-NOV)r9 zQ-AUGzQ-MAYzQ-FEBúQ-OCT)r: zQ-JULzQ-APRzQ-JAN) Úupperr r ÚvaluesÚ
|
startswithr Ú
|
infer_freqZfreqstrÚAssertionError)r3 r2 ÚgenÚindexZinf_freqZ is_dec_rangeZ is_nov_rangeZ is_oct_ranger% r% r& Útest_infer_freq_range> s
|
|
|
rB c C s: t dddd} d}tjt|d t | ¡ W 5 Q R X d S )Nz1/1/1990é r. )Ústartr3 r2 ú6Check the `freq` attribute instead of using infer_freq©Úmatch)r ÚpytestÚraisesÚ TypeErrorr r> ©rA Úmsgr% r% r& Útest_raise_if_period_index` s rM c C s8 t ddg} d}tjt|d t | ¡ W 5 Q R X d S )Nú
|
12/31/1998ú1/3/1999z(Need at least 3 dates to infer frequencyrF )r rH rI Ú
|
ValueErrorr r> rK r% r% r& Útest_raise_if_too_fewh s rQ c C s$ t dddg} t | ¡dks td S )Nz
|
01/01/1999ú1/4/1999z1/5/1999ÚB©r r r> r? ©rA r% r% r& Útest_business_dailyp s rV c C s$ t dddg} t | ¡d ks td S )NrN rO rR rT rU r% r% r& Ú test_business_daily_look_alikeu s rW c C s$ t dddg} t | ¡dks td S ©Nr5 z1/2/2000z1/3/2000r rT rU r% r% r& Útest_day_corner} s rY c C s$ t dddg} t | ¡dks td S rX )r r r> r? )Údatesr% r% r& Útest_non_datetime_index s r[ c C s$ t dddg} t | ¡d ks td S )Nz
|
2014-03-31z
|
2014-06-30z
|
2015-03-30rT rU r% r% r& Ú test_fifth_week_of_month_infer s r\ c C s& t ddddg} t | ¡dks"td S )Nz
|
2013-08-27z
|
2013-10-01z
|
2013-10-29z
|
2013-11-26zWOM-4TUErT rU r% r% r& Útest_week_of_month_fake s r] c C s. d} t jt| d tddd W 5 Q R X d S )NzVOf the four parameters: start, end, periods, and freq, exactly three must be specifiedrF ú
|
2014-01-01zWOM-5MON©r2 )rH rI rP r ©rL r% r% r& Útest_fifth_week_of_month s ÿra c C s t dddg} | jdks td S )Nú 1/31/2000z 2/29/2000z 3/31/2000r. ©r Ú inferred_freqr? ©Úrngr% r% r& Útest_monthly_ambiguous£ s rg c C s t dddg} | jdks td S )Nrb ú 1/31/2001ú 1/31/2002zA-JANrc re r% r% r& Útest_annual_ambiguous¨ s rj r/ c sh t t ¡ | \}}|| t fddtdD }|dkrN|d| n|}t |¡|ksdtd S )Nc s g | ]} | qS r% r% ©r) Új©ÚbÚincr% r& r+ ³ s z)test_infer_freq_delta.<locals>.<listcomp>é r Úd)r r Únowr Úranger r> r? )r' r/ Ú
|
base_deltaÚcoderA Zexp_freqr% rm r& Útest_infer_freq_delta s rv Ú constructorc s* t d g fddtdD S )Nr4 c s g | ]} | qS r% r% rk ©Údeltarr r% r& r+ ½ s ú <lambda>.<locals>.<listcomp>rp ©r rs ©rr ry r% rx r& Ú<lambda>¼ s $ÿr} c s* t fddtdD d g S )Nc s g | ]} | qS r% r% rk rx r% r& r+ À s rz rp r4 r{ r| r% rx r& r} ¿ s $ÿc C s4 t t ¡ }| \}}|||}t |¡d ks0td S r" )r r rr r r> r? )r' rw rn rt Ú_rA r% r% r& Útest_infer_freq_custom¹ s
|
r z freq,expected)r8 r7 )r9 r9 )r: r: c C s8 t dd| d}t|jddd t¡}|j|ks4td S )NZ1959Q2Z2009Q3r_ r Úe)Úhow)r r
|
Z to_timestampÚastypeÚobjectrd r? )r2 Úexpectedrf r% r% r& Útest_infer_freq_indexÌ s r
zexpected,datesz
|
2009-01-01z
|
2010-01-01ú
|
2011-01-01z
|
2012-01-01z
|
2009-01-31z
|
2009-04-30z
|
2009-07-31z
|
2009-10-31z
|
2010-11-30z
|
2010-12-31z
|
2011-01-31z
|
2011-02-28z
|
2010-12-25z
|
2011-01-08z
|
2011-01-15z
|
2011-01-02z
|
2011-01-03z
|
2011-01-04z2011-12-31 22:00z2011-12-31 23:00z2012-01-01 00:00z2012-01-01 01:00)zAS-JANr: r. zW-SATr r c C s" | }t ||d}|j|ks td S )N)Útzrc )Útz_naive_fixturer rZ r Úidxr% r% r& Útest_infer_freq_tzÖ s r Ú date_pairz
|
2013-11-02z 2013-11-5z
|
2014-03-08z
|
2014-03-11r^ z
|
2014-01-03Ú3HZ10TZ3601SZ3600001LZ 3600000001UZ3600000000001Nc C s. | }t |d |d ||d}|j|ks*td S )Nr r )r2 r )r rd r? )r r r2 r r r% r% r& Ú test_infer_freq_tz_transitionï s r c C s& t dddd d¡} | jd ks"td S )Nz
|
2013-11-03r1 r r6 zAmerica/Chicago)r Z tz_localizerd r? rU r% r% r& Ú$test_infer_freq_tz_transition_custom s ÿr z data,expectedz2014-07-01 09:00z2014-07-01 10:00z2014-07-01 11:00z2014-07-01 12:00z2014-07-01 13:00z2014-07-01 14:00z2014-07-01 15:00z2014-07-01 16:00z2014-07-02 09:00z2014-07-02 10:00z2014-07-02 11:00ZBHz2014-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:00z2014-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:00c C s t | }|j|kstd S r" rc )Údatar r r% r% r& Ú test_infer_freq_business_hour s Pr c C s. t dddg} | d d d
} | jdks*td S )Nrb rh ri éÿÿÿÿz-1A-JANrc re r% r% r& Útest_not_monotonic\ s r c C s2 t dddg} | ¡ }t |¡}|| jks.td S )Nrb rh ri )r Z to_pydatetimer r> rd r? )rf ÚvalsÚresultr% r% r& Útest_non_datetime_index2c s
|
r r é
|
c C s6 d ddg¡}tjt|d t | ¡ W 5 Q R X d S )Nú|z(cannot infer freq from a non-convertiblerE rF )ÚjoinrH rI rJ r r> )r rL r% r% r& Útest_invalid_index_typesk s þÿr zsee gh-10822: Windows issue)Úreasonc C s2 d} t jt| d t t d¡¡ W 5 Q R X d S )Nú Unknown datetime string formatrF r )rH rI rP r r> ÚtmZmakeStringIndexr` r% r% r& Ú test_invalid_index_types_unicode s r c C s4 ddddg} t | ¡}t t| ¡}||ks0td S )Nz2004-01z2004-02z2004-03z2004-04)r r> r
|
r? )r r r r% r% r& Ú test_string_datetime_like_compat s
|
r c C s( t tdd} t | ¡}|dks$td S )NÚ20130101Z20130110r ©r r r r> r? )ÚsÚinferredr% r% r& Ú test_series s
|
r£ Úendg $@c C s: d}t t | ¡}tjt|d t |¡ W 5 Q R X d S )Nú:cannot infer freq from a non-convertible dtype on a SeriesrF )r ÚnpÚarangerH rI rJ r r> )r¤ rL r¡ r% r% r& Útest_series_invalid_type s r¨ c C s4 d} t jt| d t tddg¡ W 5 Q R X d S )Nr rF ZfooÚbar)rH rI rP r r> r r` r% r% r& Ú test_series_inconvertible_string§ s rª c C s> d}t tdd| d}tjt|d t |¡ W 5 Q R X d S )Nr¥ Z2013r r6 rF )r r rH rI rJ r r> )r2 rL r¡ r% r% r& Útest_series_period_index¯ s r« c C s, t tdd| d}t |¡}|| ks(td S )Nr r r6 r )r2 r¡ r¢ r% r% r& Útest_series_datetime_index» s
|
r¬ Ú offset_funcc C s t dd| dS )Nr r1 r6 )r r_ r% r% r& r} Æ ó ZWEEKDAYZEOMzW@MONzW@TUEzW@WEDzW@THUzW@FRIzW@SATzW@SUNzQ@JANzQ@FEBzQ@MARzA@JANzA@FEBzA@MARzA@APRzA@MAYzA@JUNzA@JULzA@AUGzA@SEPzA@OCTzA@NOVzA@DECzY@JANzWOM@1MONzWOM@2MONzWOM@3MONzWOM@4MONzWOM@1TUEzWOM@2TUEzWOM@3TUEzWOM@4TUEzWOM@1WEDzWOM@2WEDzWOM@3WEDzWOM@4WEDzWOM@1THUzWOM@2THUzWOM@3THUzWOM@4THUzWOM@1FRIzWOM@2FRIzWOM@3FRIzWOM@4FRIc C s& t jttd | | W 5 Q R X d S )NrF )rH rI rP r )r r2 r% r% r& Útest_legacy_offset_warnings s :r¯ c C s4 t d} t d}| t ¡ ks t|t ¡ ks0td S )NÚmsZMS)r r ZMillir? Z
|
MonthBegin)ÚleftÚrightr% r% r& Útest_ms_vs_capital_ms s r³ c C sr t d¡ t j¡ d¡} tj| | jd}t |¡}|dks>t
|
| d¡}t j||jd}t |¡}|dksnt
|
d S )Nr zM8[s])Údtyper zm8[ms]r ) r¦ r§ r Úint64Úviewr Z _simple_newr´ r r> r? r )ZarrÚdtaÚresZarr2ZtdaZres2r% r% r& Útest_infer_freq_non_nano s
|
|
|
r¹ c C s: | }t ddd|d}|j d¡}t |¡}|dks6td S )Nz
|
2016-01-01im rS )r3 r2 r r¡ )r Ú_dataZas_unitr r> r? )Ztz_aware_fixturer Zdtir· r¸ r% r% r& Ú test_infer_freq_non_nano_tzaware s
|
|
r» )Or r Únumpyr¦ rH Z pandas._libs.tslibs.ccalendarr r Zpandas._libs.tslibs.offsetsr Zpandas._libs.tslibs.periodr Z pandas.compatr Zpandasr r
|
r r r r Zpandas._testingZ_testingr Zpandas.core.arraysr r Zpandas.core.tools.datetimesr Zpandas.tseriesr r ZfixtureZ timedelta64r' rs ZfreqsÚmarkZ parametrizerB rM rQ rV rW rY r[ r\ r] ra rg rj rv r r
ÚlistÚitemsr r r r r r Z makeIntIndexZmakeFloatIndexZmakePeriodIndexZmakeRangeIndexr Zskipifr r r£ r¨ rª r« r¬ r¯ r³ r¹ r» r% r% r% r& Ú<module> sà
|
ùÿ
|
ÿþý üÿ
|
üþ
|
ÿ
|
|
|
|
|
|
üúÿþ
|
ýþ ÿ ú÷ õ òõ òèåÓþN
|
üþ
|
|
|
|
|
|
|
þþÓþ2
|
