U
|
ý°dûd ã @ s d dl mZm Z mZmZ d dlZd dlZd dlZd dlm Z m
|
Z
|
m Z m Z m Z mZ d dlmZ d dlmZ e ddde ddd ZZG dd d ZG d
|
d d ZG d d d ZG dd dZG dd dZG dd dZG dd dZ G dd dZ G dd dZ dS )é )ÚdateÚdatetimeÚtimeÚ timedeltaN)Ú DatetimeIndexÚIndexÚ TimestampÚ bdate_rangeÚ
|
date_rangeÚnotna)Ú to_offsetiÙ é éÚ c @ sX e Zd Zdd Zdd Zej dddg¡dd Zej dddg¡d
|
d Z d d Z
|
dS )Ú TestGetItemc C sD t ddd}t ddd}t||ddd}|dd
j|jks@td S ) Nz2013-07-01 00:00:00zAmerica/Los_Angeles©Útzz2013-07-02 00:00:00ÚHZ
|
timebucket©ÚfreqÚnamer )r r
|
r ÚAssertionError)ÚselfÚstÚetZdr© r úcd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/datetimes/test_indexing.pyÚ test_getitem_slice_keeps_name s z)TestGetItem.test_getitem_slice_keeps_namec C sT t ddddd}t dddddd}||fD ]"}|d }|td|jd ksNt|dd
|
}t dd d|jdd}t ||¡ |j|jkst|dd d
}t ddd|jdd}t ||¡ |j|jksÈt|ddd
}t ddd|jdd}t ||¡ |j|jkst|dd d
}td ddddgd|jdd}t ||¡ |j|jks*tq*d S ) Nú
|
2011-01-01ú
|
2011-01-31ÚDÚidxr ú
|
Asia/Tokyo©r r r r r é ú
|
2011-01-05é
|
é z
|
2011-01-09Ú2Diìÿÿÿéûÿÿÿé z
|
2011-01-12z
|
2011-01-24Z3Dé éÿÿÿÿú
|
2011-01-04ú
|
2011-01-03ú
|
2011-01-02z-1D)r
|
r r r ÚtmÚassert_index_equalr r ©r Zidx1Zidx2r ÚresultÚexpectedr r r Ú test_getitem# sd ÿ ÿ ÿ ÿ ü zTestGetItem.test_getitemr ÚBÚCc C sâ t tt|d}|d d
}t| tj¡d d
|d}t ||¡ |j |j ksRt
|
|j |j ksbt
|
|d d d
}|j t |d kst
|
|dddddg }t |dks¨t
|
t |ts¶t
|
|j d ksÄt
|
|d |t d¡ ksÞt
|
d S )N©r r# r* r) r& r r )r ÚSTARTÚENDr ÚviewÚnpZndarrayr/ r0 r r r ÚlenÚ
|
isinstanceÚint_)r r ÚrngZsmallerÚexpZslicedZ fancy_indexedr r r Útest_dti_business_getitemL s z%TestGetItem.test_dti_business_getitemc C s< t tt|d}tjtdd |d d
d f W 5 Q R X d S )Nr7 zMulti-dimensional indexing©Úmatch)r r8 r9 ÚpytestÚraisesÚ
|
ValueError)r r r? r r r Ú/test_dti_business_getitem_matplotlib_hackaround` s z;TestGetItem.test_dti_business_getitem_matplotlib_hackaroundc C sv t dddd}|dddg }|d }|d }|d }|td
|
ksDt|td ksTt|td ksdt|jd ksrtd S ) Nz1/1/2005z 12/1/2005ÚM)ÚstartÚendr r r) r# r r& z 2/28/2005z 4/30/2005z 6/30/2005)r
|
r r r )r ÚdtiZdti2Zv1Zv2Zv3r r r Útest_getitem_int_listg s z!TestGetItem.test_getitem_int_listN) Ú__name__Ú
|
__module__Ú __qualname__r r4 rD ÚmarkÚ parametrizerA rG rL r r r r r s )
|
|
r c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) Ú TestWherec C s` t ddddd}dddg}t|d |d |d gd dd
|
}| ||d d d
¡}t ||¡ d S ) NÚ20130101r) r r )Úperiodsr r TFr r r r+ )r
|
r Úwherer/ r0 )r rK Úcondr3 r2 r r r Ú test_where_doesnt_retain_freqx s
|
|
z'TestWhere.test_where_doesnt_retain_freqc C sÐ t dddd}tjtjfD ]&}|jt||d}|}t ||¡ q| ¡ }t
|
tjtjg|dd
¡ }| t||¡}t ||¡ | ¡ }t
|
tjtjg|dd
¡ }| t||j ¡}t ||¡ d S )NrS r) ú
|
US/Eastern©rT r )Úotherr& ) r
|
r; ÚnanÚpdÚNaTrU r r/ r0 Úcopyr ÚtolistZ_values)r ÚiÚarrr2 r3 Úi2r r r Útest_where_other s zTestWhere.test_where_otherc C sì t dddd}|dd
¡ }ttjtjg| }t|}| ||j¡}ttjjtjjg| t d}t
|
||¡ | d ¡}| ||¡}t|d |d g|dd
¡ t d}t
|
||¡ | d ¡ d ¡}| ||¡}t|d |d g| t d}t
|
||¡ |j d
|
¡} | || ¡}t| d | d g| t d}t|d tjsHtt
|
||¡ | ||j¡}ttjjtjjg| t d}t|d tstt
|
||¡ tjd d }
|
| ||
|
¡}t|
|
|
|
g| t d}|d |
|
ksÜtt
|
||¡ d S ) NrS r) rX rY r& ©Údtyper r r ztimedelta64[ns]r* )Údays)r
|
r_ r r\ r] r rU ÚvaluesZasm8Úobjectr/ r0 Ú tz_localizeZ to_periodZasi8r: r= r; Ú timedelta64r Ú_valueÚintÚ Timedelta) r rK Útailrb Úmaskr2 r3 ZnaiveÚpiZtdaÚtdr r r Útest_where_invalid_dtypes s:
|
( z#TestWhere.test_where_invalid_dtypesc C st |}t dd|d}t dddg¡}t dd¡}t|d ||d gtd
|
}|d |ksXt| ||¡}t ||¡ d S ) Nz
|
2013-01-01r) rY TFr] Únsr r& rd r )
|
r
|
r; Úarrayrj r rh r rU r/ r0 )r Útz_aware_fixturer rK rV Ztdnatr3 r2 r r r Útest_where_mismatched_nat½ s z#TestWhere.test_where_mismatched_natc C sv t dddd}| t|¡}|}t ||¡ | ¡ }ttjtjg|dd
¡ }| t|¡}|}t ||¡ d S )NrS r) rX rY r& )
|
r
|
rU r r/ r0 r^ r r\ r] r_ )r r` r2 r3 rb r r r Ú test_where_tzÉ s zTestWhere.test_where_tzN)rM rN rO rW rc rr rv rw r r r r rR w s
|
* rR c @ sP e Zd Zdd Zdd Zdd Zej ddd d
|
g¡d d Z d d Z
|
dd Z dS )ÚTestTakec C sR t tjtdtdg}| dddg¡}t |d |d |d g}t ||¡ d S )NrS Z20130102r+ r r )r r\ r] r Útaker/ r0 )r Úindexr2 r3 r r r Ú test_take_nan_first_datetime× s z%TestTake.test_take_nan_first_datetimec C s t ddddd}t dddddd}||fD ]n}| dg¡}|td|jd ksRt| dd
|
d g¡}t dd d|jdd}t ||¡ |j|jkst| dd d g¡}t ddd|jdd}t ||¡ |j|jksÒt| dd d
|
g¡}t ddd|jdd}t ||¡ |j|jkst| dd dg¡}tdd dgd |jdd}t ||¡ |jd ksXt| dd dg¡}tdd dgd |jdd}t ||¡ |jd ks*tq*d S )Nr r r r r r! r" r r r r& r- r* r$ r' é z
|
2011-01-08r. z-3Dr) r# r, z
|
2011-01-06éýÿÿÿz
|
2011-01-29) r
|
ry r r r r/ r0 r r r1 r r r Ú test_takeÝ sv ÿ ÿ ÿ ÿ ü ü zTestTake.test_takec C s¬ t ddddd}dddd d
|
d d d g}d}tjt|d |j|dd W 5 Q R X d}tjt|d |j||d W 5 Q R X d}tjt|d |j|dd W 5 Q R X d S )Nr r r r r r é r# é r% é é r) z1take\(\) got an unexpected keyword argument 'foo'rB r& )Úfooz$the 'out' parameter is not supported)Úoutz%the 'mode' parameter is not supportedZclip)Úmode)r
|
rD rE Ú TypeErrorry rF )r r ÚindicesÚmsgr r r Útest_take_invalid_kwargs s z!TestTake.test_take_invalid_kwargsr NrX r! c C sÒ t ddddt ddddt ddddt ddddg}tddd |d
|
d }t|d d
|
|d }| d dddg¡}|d dddg }||fD ]L}t ||¡ t|tst|jd ks¬t|j |j ks¼t|j
|
|j
|
kstqd S )Nr r é r é é z2010-01-01 09:00z2010-02-01 09:00r r )rI rJ r r r )r r r r# r é é ) r r
|
r ry r/ r0 r= r r r r )r r Údatesr r3 Ztaken1Ztaken2Ztakenr r r Ú
|
test_take2" s* üû zTestTake.test_take2c C sZ t dddgdd}| t dddg¡¡}t dddgdd}t ||¡ |jt dddg¡d d
|
}t ddd gdd}t ||¡ |jt dddg¡d d d }t dddgdd}t ||¡ d}tjt|d |jt dddg¡d d
|
W 5 Q R X tjt|d |jt dddg¡d d
|
W 5 Q R X d}tjt |d | t ddg¡¡ W 5 Q R X d S )Nr ú
|
2011-02-01ú
|
2011-03-01Úxxx)r r r r+ T©Ú
|
fill_valuer] F©Z
|
allow_fillr úJWhen allow_fill=True and fill_value is not None, all indices must be >= -1rB éþÿÿÿr( ú out of bounds©
|
r ry r; rt r/ r0 rD rE rF Ú
|
IndexError©r r r2 r3 r r r r Útest_take_fill_value> s& ÿ$$z TestTake.test_take_fill_valuec C sb t dddgddd}| t ddd g¡¡}t dddgddd}t ||¡ |jt ddd g¡d
|
d }t ddd gddd}t ||¡ |jt ddd g¡d d
|
d}t dddgddd}t ||¡ d}tjt|d |jt dddg¡d
|
d W 5 Q R X tjt|d |jt dddg¡d
|
d W 5 Q R X d}tjt |d | t ddg¡¡ W 5 Q R X d S )Nr r r r rX )r r r r r+ Tr r] Fr r rB r r( r r r r r r Ú"test_take_fill_value_with_timezone\ sF ÿ ÿ ÿ ÿ ÿ$$z+TestTake.test_take_fill_value_with_timezone) rM rN rO r{ r~ r rD rP rQ r r r r r r r rx Ö s 4
|
rx c @ sx e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Ze j
|
d e d¡e d¡edg¡dd Zdd Zdd ZdS )Ú
|
TestGetLocc C s@ t ddd}|d d¡}| |¡}|dks0t||ks<td S )Nú
|
2000-01-01r) ©rT r Úms)r
|
Úas_unitÚget_locr )r r ÚkeyÚlocr r r Ú test_get_loc_key_unit_mismatch s
|
|
z)TestGetLoc.test_get_loc_key_unit_mismatchc C sh t dddj d¡}t|}|d d¡t d¡ }tjt dd |
|
|¡ W 5 Q R X ||ksdt d S )
|
Nr r) r¡ zM8[s]r rs r z,Timestamp\('2000-01-01 00:00:00.000000001'\)rB ) r
|
Ú_dataÚastyper r£ r\ rm rD rE ÚKeyErrorr¤ r )r ÚdtarK r¥ r r r Ú+test_get_loc_key_unit_mismatch_not_castable s ÿz6TestGetLoc.test_get_loc_key_unit_mismatch_not_castablec C sf t dddd}| td¡}t dg¡}tj||dd | tdd¡}t g ¡}tj||dd d S ) Nr é r ©rT r r F)Z check_dtypeé )r
|
r¤ r r; rt r/ Úassert_numpy_array_equal©r r r2 r3 r r r Útest_get_loc_time_obj s
|
z TestGetLoc.test_get_loc_time_objc C s ddl m} |tjddgtjd }tddd}|jd |jd
|
|j }d }|D ]ª}t d |d d}t
|
j tj |¡|d}tj|||tjd} |j |¡}
|
t |
|
| ¡ t || |j| ¡ | ¡ | ¡ } } | | d9 < | j| d9 < t | | ¡ qRd S )Nr )Ú _SIZE_CUTOFFiÿÿÿéd rd r é r¯ i é< iQ z
|
2014-11-26ÚSr® )rz iöÿÿÿ)Zpandas._libs.indexr³ r; rt Úint64r ÚhourÚminuteÚsecondr
|
r\ ZSeriesÚrandomZrandnZarangeÚintprz r¤ r/ r° Zassert_series_equalZilocr^ ) r r³ rs r¥ rI ÚstepÚnr ÚtsZlocsr2 ÚleftÚrightr r r Útest_get_loc_time_obj2¢ s z!TestGetLoc.test_get_loc_time_obj2c C sD t dddd}ttjg}| |¡}tjg tjd}t ||¡ d S )Nr é+ iH7 )rº r» Ú microsecondrd )
|
r r r\ r] r¤ r; rt r½ r/ r° )r ÚticrK r¦ r3 r r r Útest_get_loc_time_natº s
|
|
z TestGetLoc.test_get_loc_time_natc C s t ddg}| tj¡dks t| d ¡dks2t| tj¡dksFt| tj¡dksZt| t d¡¡dksrtt j
|
t dd | t d¡¡ W 5 Q R X d S )Nú1/3/2000r] r rB ) r r¤ r\ r] r r; r[ ZNAZ
|
datetime64rD rE rª rj ©r rz r r r Útest_get_loc_natÄ s zTestGetLoc.test_get_loc_natr¥ r r c C s8 t ddd}d}tjt|d | |¡ W 5 Q R X d S )Nz
|
1970-01-01r% r¡ z,Cannot index DatetimeIndex with [Tt]imedeltarB )r
|
rD rE r r¤ )r r¥ rK r r r r Ú"test_get_loc_timedelta_invalid_keyÔ s z-TestGetLoc.test_get_loc_timedelta_invalid_keyc C s2 t dg}tjtdd | d¡ W 5 Q R X d S )NrÈ Ú2000rB ú1/1/2000)r rD rE rª r¤ rÉ r r r Ú!test_get_loc_reasonable_key_errorÜ s
|
z,TestGetLoc.test_get_loc_reasonable_key_errorc C s. t dd}| d¡}tdd}||ks*td S )NrÍ z1/1/2010Z2009iØ iE )r
|
r¤ Úslicer ©r r? r2 r3 r r r Útest_get_loc_year_strâ s
|
|
|
z TestGetLoc.test_get_loc_year_strN)rM rN rO r§ r¬ r² rà rÇ rÊ rD rP rQ r\ rm r rË rÎ rÑ r r r r r s
|
"
|
r c
|
@ sR e Zd Zdd Zej ddddgdddgdddgdddgddd gg¡d
|
d Zd S ) Ú TestContainsc C s, t ddddd}t||g}||ks(td S )NiÛ r r# é r¯ )r r r )r ÚdÚixr r r Ú!test_dti_contains_with_duplicatesë s z.TestContains.test_dti_contains_with_duplicatesÚvalsr r r+ Ú2015Z2016Z2014c C s t |}|d |kstd S )Nr )r r )r r× r r r r Útest_contains_nonuniqueð s z$TestContains.test_contains_nonuniqueN)rM rN rO rÖ rD rP rQ rÙ r r r r rÒ ê s ûþ
|
rÒ c
|
@ sÆ e Zd Zdd Zdd Zej deddde dge d eddd
|
gg¡d d Z
|
ej d eddde d gddgfe d edddgddgfedddedddgddgfg¡dd Z dd Z dS )ÚTestGetIndexerc C s: t ddd}| | dd ¡¡}| |¡}t ||¡ d S )NrÍ rÓ r¡ c S s | ¡ S )N)r )Úxr r r Ú<lambda> ó z;TestGetIndexer.test_get_indexer_date_objs.<locals>.<lambda>)r
|
Ú get_indexerÚmapr/ r° rÐ r r r Útest_get_indexer_date_objs s
|
z)TestGetIndexer.test_get_indexer_date_objsc C s t ddd}tjdddgtjd}t | |¡|¡ |d t dd d
|
g¡ }t | |d ¡tjd ddgtjd¡ t | |d ¡tjdddgtjd¡ t | |d¡tjdddgtjd¡ t |j|dt d¡dtjdd dgtjd¡ t d¡t d¡t d¡
|
¡ g}t |j|ddd |D dtjdd dgtjd¡ t d¡
|
¡ t d¡
|
¡ dg}d}t j t |d |j|d|d W 5 Q R X t j t dd |j|dg ddd W 5 Q R X d S )Nr r) r¡ r r r& rd z-1 hourz12 hoursz 1 day 1 hourÚpadr+ ZbackfillZnearestz1 hour)Ú tolerancec S s g | ]}t |¡qS r )r; rj )Ú.0rÛ r r r Ú
|
<listcomp>" s z3TestGetIndexer.test_get_indexer.<locals>.<listcomp>z2 hourr z*Could not convert 'foo' to NumPy timedeltarB zabbreviation w/o a number)Úmethodrâ )r
|
r; rt r½ r/ r° rÞ r\ Z to_timedeltarm Zto_timedelta64rD rE rF )r r r@ ÚtargetZtol_rawZtol_badr r r r Útest_get_indexer sR
|
ÿ
|
ÿ
|
ÿþ ý ÿü ýz TestGetIndexer.test_get_indexerræ éä r ú
|
2020-01-02ú
|
2020-01-01r& c C sB t tdtdg}| |¡}tjddgtjd}t ||¡ d S )Nrê ré r r rd ©r r rÞ r; rt r½ r/ r° )r ræ rg r2 r3 r r r Ú test_get_indexer_mixed_dtypes1 s
|
z,TestGetIndexer.test_get_indexer_mixed_dtypesztarget, positionsi' r+ r c C s> t tdtdg}| |¡}tj|tjd}t ||¡ d S )Nrê ré rd rë )r ræ Z positionsrg r2 r3 r r r Ú#test_get_indexer_out_of_bounds_date? s
|
z2TestGetIndexer.test_get_indexer_out_of_bounds_datec C sL t dddd}|dddg }d}tjt|d |j|d
|
d W 5 Q R X d S ) NrÍ z3/1/2000r5 r7 r r r& z0index must be monotonic increasing or decreasingrB rá )rå )r
|
rD rE rF rÞ )r r? Zrng2r r r r Ú*test_get_indexer_pad_requires_monotonicityN s
|
z9TestGetIndexer.test_get_indexer_pad_requires_monotonicityN) rM rN rO rà rç rD rP rQ r r rì rí rî r r r r rÚ s$ )þþ
|
ýþ
|
rÚ c @ s e Zd Zdd Zdd ZdS )ÚTestMaybeCastSliceBoundc C sR t dddd}| dd¡}td}||ks.t| dd¡}td }||ksNtd S )
|
NZ1Hr rØ )r rT rJ z
|
2015-01-02r z 2015-01-02 23:59:59.999999999rÁ z2015-01-02 00:00:00)r
|
Ú_maybe_cast_slice_boundr r )r Z empty_idxr r@ rÁ r r r Ú"test_maybe_cast_slice_bounds_emptyZ s z:TestMaybeCastSliceBound.test_maybe_cast_slice_bounds_emptyc C s0 t ddg}| dd¡}td}||ks,td S )NZ2017z
|
2017-01-01rÁ )r rð r r r± r r r Ú)test_maybe_cast_slice_duplicate_monotonicf s zATestMaybeCastSliceBound.test_maybe_cast_slice_duplicate_monotonicN)rM rN rO rñ rò r r r r rï Y s rï c @ s e Zd Zej deeeg¡ej dddg¡dd Z ej deeg¡ej ddd g¡ej d
|
d d g¡d d Z
|
ej deeg¡dd Z dS )ÚTestGetSliceBoundsÚboxzside, expected)rÁ r* )r r# c C sr |}t dd |¡}|dddd}|d k rTtjtdd |j||d W 5 Q R X n|j||d }||ksntd S )
|
Nú
|
2000-01-03ú
|
2000-02-11iÐ r r| ©ÚyearÚmonthÚdayúCannot compare tz-naiverB ©Úside©r ri rD rE r Zget_slice_boundr ) r rô rý r3 ru r rz r¥ r2 r r r Ú%test_get_slice_bounds_datetime_withino s z8TestGetSliceBounds.test_get_slice_bounds_datetime_withinrý rÁ r zyear, expected)iÏ r )rè r¯ c
|
C sr |}t dd |¡}||ddd}|d k rTtjtdd |j||d W 5 Q R X n|j||d} | |ksntd S ) Nrõ rö r r| r÷ rû rB rü rþ )
|
r rô rý rø r3 ru r rz r¥ r2 r r r Ú&test_get_slice_bounds_datetime_outside s z9TestGetSliceBounds.test_get_slice_bounds_datetime_outsidec C s |}t ddg |¡}|ddd}|d k rZtjtdd | ||ddd¡ W 5 Q R X n$| ||ddd¡}d}||ks~td S ) Nz
|
2010-01-01z
|
2010-01-03r r rû rB r& )r r )r ri rD rE r Z
|
slice_locsr )r rô ru r rz r¥ r2 r3 r r r Útest_slice_datetime_locs s z+TestGetSliceBounds.test_slice_datetime_locsN) rM rN rO rD rP rQ r r r rÿ r r r r r r ró n s ró c @ s0 e Zd Zdd Zej ddddg¡dd Zd S )
|
ÚTestIndexerBetweenTimec C sP t dddd}d}tjt|d& | tddd dtddd d
|
¡ W 5 Q R X d S ) NrÍ ú1/5/2000Ú5minr7 zGCannot convert arg \[datetime\.datetime\(2010, 1, 2, 1, 0\)\] to a timerB r r r& r# )r
|
rD rE rF Úindexer_between_timer )r r? r r r r Útest_indexer_between_time¦ s z0TestIndexerBetweenTime.test_indexer_between_timeÚunitÚusr¢ Úsc C sÒ t dddd}|jj}| d| d¡}t|jj||jd}t|}|j|jksVtt dd }t d
|
d }|
|
||¡} |
|
||¡}
|
t | |
|
¡ t dd d d }t d
|
d d d}|
|
||¡} |
|
||¡}
|
t | |
|
¡ d S )NrÍ r r r7 zM8[ú]rd r é r& é r in² iÒ ) r
|
r¨ Z_ndarrayr© ÚtypeZ _simple_newre r r r r r/ r° ) r r r? Zarr_nanora r« rK rÆ Ztocr2 r3 r r r Ú"test_indexer_between_time_non_nano s
|
|
z9TestIndexerBetweenTime.test_indexer_between_time_non_nanoN)rM rN rO r rD rP rQ r r r r r r ¥ s r ) r r r r Únumpyr; rD Zpandasr\ r r r r r
|
r Zpandas._testingZ_testingr/ Zpandas.tseries.frequenciesr r8 r9 r rR rx r rÒ rÚ rï ró r r r r r Ú<module> s" \_ -hY7
|
