U
|
ý°dgS ã @ sÆ d dl m Z d dlZd dlZd dlZd dlm mZ d dl Z
|
d dl m Z m Z m Z mZmZmZ d dlmZ d dlmZmZmZ e ddde ddd ZZG dd d ZG d
|
d d ZG d d d Z dS )é )ÚdatetimeN)Ú DataFrameÚ DatetimeIndexÚIndexÚSeriesÚ bdate_rangeÚ
|
date_range)Ú BMonthEndÚMinuteÚMonthEndiÙ é iÚ c @ s* e Zd ZddddddgZdd Zej d ej e
|
e g¡d
|
d Z ej d e¡d d Z dd Zdd Zdd Zdd Zdd Zdd Zdd Zd d Zd d Zej d ddddg¡d!d" Zej d#d$d%g¡d&d' Zd(d) Zej d e¡d*d+ Zd,d- Zd.d/ Z ej d dddg¡d0d1 Z d2d3 Z dS )4ÚTestDatetimeIndexSetOpsNÚUTCz
|
Asia/Tokyoú
|
US/Easternzdateutil/Asia/Singaporezdateutil/US/Pacificc C s@ t d¡}|d d
}|dd
}|j||d}t ||¡ d S ©Né
|
é ©Úsort)ÚtmÚ makeDateIndexÚunionÚassert_index_equal)Úselfr Ú
|
everythingÚfirstÚsecondr © r úad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/datetimes/test_setops.pyÚ test_union2( s
|
|
z#TestDatetimeIndexSetOps.test_union2Úboxc C sX t d¡}|d d
}|dd
}|j||d}||j}|j||d}t ||¡ d S r )r r r Úvaluesr ) r r r r r r ÚexpectedÚcaseÚresultr r r Ú test_union3/ s
|
|
z#TestDatetimeIndexSetOps.test_union3Útzc C s@ t ddd|d}t ddd|d}t ddd|d}tt|t| }t ddd|d}t ddd|d}t ddd|d} tt|t|d d
}
|
t ddd|d} tg |d
|
} t ddd|d} | }||||f||| |
|
f| | | |ffD ]X\}}}}|j||d }t ||¡ |j||d }|d kr.t ||¡ qât ||¡ qâd S ) Nú1/1/2000ÚDr ©ÚfreqÚperiodsr& ú1/6/2000r ú1/4/2000é é ©r& r )r r Úlistr r r )r r& r Úrng1Úother1Ú expected1Zexpected1_notsortedÚrng2Úother2Ú expected2Zexpected2_notsortedÚrng3Úother3Ú expected3Zexpected3_notsortedÚrngÚotherÚexpZ exp_notsortedZ result_unionr r r Ú
|
test_union; s,
|
|
|
ý
|
z"TestDatetimeIndexSetOps.test_unionc C sn t dddg}t | ¡ dd}|j||d}t ||¡ |d d
j||d}t ||¡ |j|jksjtd S )Nz
|
2000-01-03z
|
2000-01-01z
|
2000-01-02Zinfer©r* r r )r Ú sort_valuesr r r r* ÚAssertionError)r r ÚidxZorderedr$ r r r Útest_union_coverageZ s z+TestDatetimeIndexSetOps.test_union_coveragec C sp t dddd}t dddd}|j||d}t|t|dd
}|d krXtt|}nt|}t ||¡ d S )Nú1/1/2012é Z3H©r+ r* Z4Hr r )r r r1 r Úsortedr r )r r Zrng_aZrng_br$ r= r r r Útest_union_bug_1730d s z+TestDatetimeIndexSetOps.test_union_bug_1730c C sV t dg}t dddg}|j||d}t ddddg}|d krF| ¡ }t ||¡ d S )Nz2012-05-11 15:19:49.695000z2012-05-29 13:04:21.322000z2012-05-11 15:27:24.873000z2012-05-11 15:31:05.350000r )r r r@ r r )r r ÚleftÚrightr$ r= r r r Útest_union_bug_1745p s$
|
ýÿüÿz+TestDatetimeIndexSetOps.test_union_bug_1745c C sp ddl m} tdd}||dd }|j||d}t|t| }|d krXtt|}nt|}t ||¡ d S )Nr ©Ú
|
DateOffsetz
|
2013-01-01z
|
2013-02-01é )Úminutesr ) ÚpandasrM r r r1 r rG r r )r r rM rI rJ r$ r= r r r Útest_union_bug_4564 s
|
z+TestDatetimeIndexSetOps.test_union_bug_4564c C s@ t ddd}d |j_|j||d}t ||¡ |jd ks<td S )NZ20150101r ©r+ r )r Ú_datar* r r r rA )r r r" r$ r r r Útest_union_freq_both_none s
|
z1TestDatetimeIndexSetOps.test_union_freq_both_nonec C sn t ddd}|ddddg }|dddg }|jd ks8t|jd ksFt| |¡}t ||¡ |jd ksjtd S )
|
Nz
|
2016-01-01r rR r r r/ rE é r( )r r* rA r r r )r ÚdtirI rJ r$ r r r Útest_union_freq_infer s
|
z-TestDatetimeIndexSetOps.test_union_freq_inferc C sv t dddd}ttj t|¡|}t dddd}ttj t|¡|}t||d}t dddd}t |j |¡ d S )Nz1/1/1999rD ZMSr? z1/1/1980z 12/1/2001)Ús1Ús2)
|
r r ÚnpÚrandomZrandnÚlenr r r Úindex)r r2 rX r5 rY Zdfr= r r r Útest_union_dataframe_index s z2TestDatetimeIndexSetOps.test_union_dataframe_indexc C sF t tjdddtjd}tdddd}|j||d |j||d d S )
|
Nr é rU ©Zdtypez2012-01-03 00:00:00r r( )Ústartr+ r* r )r rZ ZarangeÚint64r r )r r Úi1Úi2r r r Ú test_union_with_DatetimeIndex¸ s z5TestDatetimeIndexSetOps.test_union_with_DatetimeIndexc s¢ t d¡}|dd
| ¡}t | ¡s0t fddtjttfD }|D ] }| |¡}t | ¡sNtqNt dddg}| |¡}t g t
|
d}t ||¡ d S ) Nr r c s g | ]}| j qS r )r! )Ú.0Úklass©r r r Ú
|
<listcomp>È s z>TestDatetimeIndexSetOps.test_intersection2.<locals>.<listcomp>ÚaÚbÚcr` ) r r Ú intersectionZ equalContentsrA rZ Úarrayr r1 r Úobjectr )r r Z intersectZcasesr# r$ Úthirdr" r rh r Útest_intersection2Á s
|
|
|
|
z*TestDatetimeIndexSetOps.test_intersection2c C s® t ddddd}t ddddd}t ddddd}t ddddd}t dddd d}t d d
|
ddd}tg ddd} ||f||f|| ffD ].\}
|
} | |
|
¡} t | | ¡ | j| jkstqtd d d dg|dd}td d ddg|dd}td d g|dd}td d ddg|dd}td d g|d d}t d d
|
d|dd}tg |dd} | jd ksJt||f||f|| ffD ]H\}
|
} |j|
|
|d} |d kr| ¡ } t | | ¡ | j| jks`tq`d S )Nú6/1/2000z 6/30/2000r( rB )r* Únamez 5/15/2000z 6/20/2000r< z7/1/2000z 7/31/2000z
|
2011-01-05z
|
2011-01-04z
|
2011-01-02z
|
2011-01-03)r& rs z
|
2011-02-02z
|
2011-02-03)r* r& rs r )r r rm r r r* rA r@ ) r r& r Úbaser5 r7 r8 r: Zrng4Z expected4r; r" r$ r r r Útest_intersectionÒ sV ý
|
|
ÿ
|
ÿ
|
ýý
|
z)TestDatetimeIndexSetOps.test_intersectionr* ÚTÚBc C s |}t dd||d}|dd
|¡}t|dks6t|j|jksFt| |dd
¡}t|dksht|j|jksxt|dk}|d d
|dd
¡}t ||d d
¡ |rÂ|j|jksÂt|dd
|d d
¡}t ||d d
¡ |r|j|jkstd S )Nrr z 6/15/2000)r* r& r rv r/ éýÿÿÿ)r rm r\ rA r* r r )r Ztz_aware_fixturer* r& r; r$ Z
|
check_freqr r r Útest_intersection_empty s" z/TestDatetimeIndexSetOps.test_intersection_emptyc C sF ddl m} tdddd}||dd }| |¡}t|dksBtd S ) Nr rL rD rE Z12HrF r )Úhours)rP rM r rm r\ rA )r rM Zindex_1Zindex_2r$ r r r Útest_intersection_bug_1708, s
|
|
z2TestDatetimeIndexSetOps.test_intersection_bug_1708c C sæ dddddg}t ||d}tddd |d
|
}t ||d}t ||d}tddd |d
|
}t |d d
|d} t ||d}
|
t g |d} t ||d} |||f||| f|
|
| | ffD ]:\} }}| ||¡}|d krÔt|rÔ| ¡ }t ||¡ q¦d S ) Nz1/2/2000z1/3/2000r' r- z1/5/2000r0 r, r( r r) r/ )r r Ú
|
differencer\ r@ r r )r r& r Z rng_datesr2 r3 r4 r5 r6 r7 r8 r9 r: r; r< r" Z result_diffr r r Útest_difference5 s$ ý z'TestDatetimeIndexSetOps.test_differencec C s t dddd}t dddd}tddgd d}| ||¡}t ||¡ t d||¡ t dddd}| ||¡}tddgd d}t ||¡ t d||¡ d S ) NZ20160920Z20160925r( r? Z20160921Z20160924r* Z20160922)r r r| r r Zassert_attr_equal)r r r] r< r" Zidx_diffr r r Útest_difference_freqP s z,TestDatetimeIndexSetOps.test_difference_freqc C sH t dtddddd}t dtddddd}t| ||¡dksDtd S ) NzQ-JANiÍ é é éd )r* ra r+ éb rU )r r r\ r| rA )r r Zdti1Zdti2r r r Útest_datetimeindex_diff` s z/TestDatetimeIndexSetOps.test_datetimeindex_diffc C sâ t ddd|d}|d d
|dd
¡}|j|jks:t|j|jksJt|j|jksZt|d d
|dd
¡}|j|jkst|j|jkst|j|jks¤t|d d
|dd
¡}|j|jksÎt|jd ksÜt|j|jksìt|d d
|d d
|
¡}|j|jkst|jd ks(t|j|jks:ttt |d d
|
d d }|d d
|¡}|jd kstt|j|jkst|j|jkst|d d
|¡}|jd ksºt|j|jksÌt|j|jksÞtd S )Nr' z1/1/2002rB )rs r& é2 r é é< é éK r( r< ©rs )
|
r r rs rA r* r& rm Zfreqstrr r1 )r r& r; r$ Znofreqr r r Útest_setops_preserve_freqe s4 z1TestDatetimeIndexSetOps.test_setops_preserve_freqc C sR t ddddddgdd}|d d d
|d
|
d d
¡}|d d
}t ||¡ d S ) Nz
|
2018-12-31z
|
2019-03-31z
|
2019-06-30z
|
2019-09-30z
|
2019-12-31z
|
2020-03-31zQ-DECr? rU r r )r rm r r )r rV r$ r" r r r Ú&test_intersection_non_tick_no_fastpath s ú÷ z>TestDatetimeIndexSetOps.test_intersection_non_tick_no_fastpath) Ú__name__Ú
|
__module__Ú __qualname__r& r ÚpytestÚmarkÚ parametrizerZ rn r r1 r% r> rC rH rK rQ rT rW r^ re rq ru ry r{ r} r~ r r r r r r r r sH ú
|
|
|
|
|
ÿ
|
;
|
|
|
"r c @ sd e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Ze j
|
d d Z e j dddg¡dd ZdS )ÚTestBusinessDatetimeIndexc C s" t tt}|d d
}|dd
}|j||d}t|ts>t|d d
}|dd
}|j||d}t|tsrt|d d
}|dd
}|j||d}t|ts¦t|d krÆt |j||d|¡ n*tt
|
|t
|
| }t |j||d|¡ t ttt d}|j||d}t|ts td S )Nr r r r? ) r ÚSTARTÚENDr Ú
|
isinstancer rA r r r r1 r r )r r r; rI rJ Ú the_unionr" r r r r> s(
|
z$TestBusinessDatetimeIndex.test_unionc C s¾ t ddt d}|dd
}|d d
}|j||d}|d krLt ||¡ n0tt|dd
t|d d
}t ||¡ |dd
}|dd
}|j||d}|dd
}t ||¡ d S ) Nr' r rF r r r rN é# )r r
|
r r r r r1 )r r r; r2 r5 r r" r r r Útest_union_not_cacheable¿ s $ z2TestBusinessDatetimeIndex.test_union_not_cacheablec C s¶ t ddt d}|dd
}|d d
}| |¡}|dd
}t ||¡ t|tsXt|j|jksht| | t¡¡}t ||¡ |d d
|dd
¡}tg }t ||¡ d S )Nr' r rF r r )
|
r r
|
rm r r r r rA r* Úview)r r; r2 r5 Zthe_intr" r r r ru Ð s
|
z+TestBusinessDatetimeIndex.test_intersectionc C s> t dd}t dd}| |¡}t ||¡ |j|jks:td S )Nú
|
11/30/2011ú
|
12/31/2011ú
|
12/10/2011ú
|
12/20/2011©r rm r r r* rA ©r rj rk r$ r r r Útest_intersection_bugâ s
|
|
|
|
z/TestBusinessDatetimeIndex.test_intersection_bugc C s: t d¡t d¡g}t|dd}| |¡}t ||¡ d S )Nz
|
2020-01-01z
|
2020-02-01rj r )ÚpdÚ Timestampr rm r r )r r! rB Úresr r r Útest_intersection_listê s
|
z0TestBusinessDatetimeIndex.test_intersection_listc
|
C sz ddl m} |d}tddd}tddd}tddd}tddd}t|||t d}t|||t d} |j| |d d S )
|
Nr )Útimezoner éÛ r r/ r ©ra Úendr& r* r )Úpytzr¥ r r r r )
|
r r r¥ r& Ú early_startÚ early_endÚ
|
late_startÚlate_endÚearly_drÚlate_drr r r Ú test_month_range_union_tz_pytzò s z8TestBusinessDatetimeIndex.test_month_range_union_tz_pytzc
|
C sz ddl m} |d}tddd}tddd}tddd}tddd}t|||t d}t|||t d} |j| |d d S )
|
Nr )Údateutil_gettzr r¦ r r/ r r§ r )Z pandas._libs.tslibs.timezonesr± r r r r )
|
r r r± r& rª r« r¬ r r® r¯ r r r Ú"test_month_range_union_tz_dateutil s z<TestBusinessDatetimeIndex.test_month_range_union_tz_dateutilr FNc C sV t t d¡t d¡t d¡g}|j||d}t t d¡t d¡g}t ||¡ d S )Nz
|
2019-12-13z
|
2019-12-12r )r r¡ r¢ rm r r )r r Úidx1r$ r" r r r Ú test_intersection_duplicates s ýÿz6TestBusinessDatetimeIndex.test_intersection_duplicates)r r r r> r ru r r¤ r° ÚtdZskip_if_windowsr² r r r r´ r r r r r s $
|
r c @ s@ e Zd Zdd Zdd Zej dddde d ¡g¡d
|
d Z
|
dS ) ÚTestCustomDatetimeIndexc C sö t ttdd}|d d
}|dd
}|j||d}t|tsBt|d d
}|dd
}| ||¡}t|tstt|d d
}|dd
}|j||d}t|ts¨t|d krÆt |j||d|¡ t
|
ttt d}|j||d}t|tsòtd S )NÚCr? r r r ) r r r r r r rA r r r r r )r r r; rI rJ r r r r r> # s$ z"TestCustomDatetimeIndex.test_unionc C sF t dddd}t dddd}| |¡}t ||¡ |j|jksBtd S )Nr r r· r? r r r r r r r r D s
|
|
z-TestCustomDatetimeIndex.test_intersection_bugr& Nr z Europe/BerliniÄÿÿÿc C sJ t ddd|d}t ddd|d}| |¡}t ddd|d}t ||¡ d S )Nz
|
2020-03-27r r( )r+ r* r& z
|
2020-03-30rU )r rm r r )r r& r³ Zidx2r$ r" r r r Ú test_intersection_dst_transitionL s
|
|
z8TestCustomDatetimeIndex.test_intersection_dst_transition) r r r r> r r r r r© Z FixedOffsetr¸ r r r r r¶ " s ! ÿr¶ ) r ÚnumpyrZ r r© Z pandas.util._test_decoratorsÚutilZ_test_decoratorsrµ rP r¡ r r r r r r Zpandas._testingZ_testingr Zpandas.tseries.offsetsr r
|
r r r r r r¶ r r r r Ú<module> s
|
