U
|
ý°d]A ã @ sb d Z ddlmZ ddlZddlZddlmZmZmZm Z m
|
Z
|
m Z m Z ddl mZ G dd dZdS )z& test partial slicing on Series/Frame é )ÚdatetimeN)Ú DataFrameÚ DatetimeIndexÚIndexÚSeriesÚ TimedeltaÚ TimestampÚ
|
date_rangec @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Ze j
|
d ddddddg¡dd Z dd Z dd Zdd Zd d Zd d Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Ze j
|
d.d/ed/ed/ ¡ g¡e j
|
d0d1ed1ed1 ¡ g¡d2d3 Zd4d5 Z d6S )7Ú TestSlicingc C s^ t tj dd¡tdddd}|jd }|j|jd ks<t|j d }|j|jd ksZtd S )Né
|
é z1/1/2000©Úperiods©Úindexz1/3/2000é )
|
r ÚnpÚrandomZrandnr ÚlocÚnamer ÚAssertionErrorÚT©ÚselfÚdfÚresult© r újd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/datetimes/test_partial_slicing.pyÚ'test_string_index_series_name_converted s
|
|
|
z3TestSlicing.test_string_index_series_name_convertedc C s6 d}t |dddd}tt d¡|d}|dd
d S )Nz
|
2013-01-07Z1dr z
|
US/Eastern)ÚstartÚfreqr Útzr z 2013-01-14 23:44:34.437768-05:00)r r r Úarange)r r Úidxr r r r Ú test_stringified_slice_with_tz s z*TestSlicing.test_stringified_slice_with_tzc C sÌ t dddd}ttd|d}ttd|d |d |d gd}d }|j| }|jdd
}t ||¡ |jd d d
|
j| }|jd d d
|
d d
|
}t ||¡ |j| }|jd d d
}t ||¡ d S ) Nú2015-5-13 23:59:00Úminé ©r r r r é r r z 2015-5-14 00éÿÿÿÿ)r r Úranger ÚilocÚtmÚassert_series_equal) r ÚdtiÚserÚser2Úkeyr ÚexpectedÚresult2Ú expected2r r r Ú.test_return_type_doesnt_depend_on_monotonicity& s "
|
|
z:TestSlicing.test_return_type_doesnt_depend_on_monotonicityc C s t dddd}ttd|d}ttd|d |d |d gd}d }|j| }|dksZt|jd d d
|
j| }|dks|t|j| }|dkstd S ) Nr% r& r' r( r r) r r z2015-5-14 00:00:00r* )r r r+ r r r, )r r/ r0 r1 r2 r r4 r r r Ú:test_return_type_doesnt_depend_on_monotonicity_higher_reso@ s "
|
|
zFTestSlicing.test_return_type_doesnt_depend_on_monotonicity_higher_resoc C sê t ttd}dddddg}t|}||d< t ttd|d}t ||¡ td d
|
d }t d d dd
|
gi|d d d
d}t d d i|dd
d d d
d}|jd }t ||¡ |jd d d
jd }|jd d d
}t ||¡ d S )Né z
|
2018-01-02z
|
2017-02-10z
|
2016-03-10z
|
2015-03-15z
|
2014-03-16Údate)r r9 z20170101 01:00:00r' r ÚAr) r r* r z
|
2017-01-03) r Úlistr+ r r- Úassert_frame_equalr r r, ) r r Z date_listZ
|
date_indexr3 r/ r r4 r5 r r r Ú test_monotone_DTI_indexing_bugW s& û "
|
z*TestSlicing.test_monotone_DTI_indexing_bugc C s t dtddddd}tt t|¡|d}|d }||jjdk }t ||¡ t
|
tj t|d¡|d}|j d }||jjdk }t ||¡ d S ) NÚBéÕ r) éô ©r r r r Z2005r8 )r r r r r" Úlenr Úyearr- r. r r Úrandr r< )r r/ Úsr r3 r r r r Útest_slice_yearw s
|
zTestSlicing.test_slice_yearÚ partial_dtimeZ2019Z2019Q4zDec 2019z
|
2019-12-31z 2019-12-31 23z2019-12-31 23:59c C sD t dddd}ttd|d}|| }|jd d
}t ||¡ d S )Nz 2019-12-31 23:59:55.999999999r rE ©r r r r8 )r r r+ r, r- r. )r rG r/ r0 r r3 r r r Ú#test_slice_end_of_period_resolution s
|
z/TestSlicing.test_slice_end_of_period_resolutionc C st t dtddddd}tt t|¡|d}t|d d ks@tttj t|d
|
¡|d}t|j
|
d d ksptd S ) NÚDiÐ é r) r@ rA r Z2001Q1éZ r8 Z1Q01) r r r r r" rB r r r rD r ©r r/ rE r r r r Útest_slice_quarter s
|
z TestSlicing.test_slice_quarterc C s t dtddddd}tt t|¡|d}t|d dks@tttj t|d ¡|d}t|j
|
d dksptt |d |d
|
¡ d S ) NrJ r? r) r@ rA r z2005-11é r8 z11-2005) r r r r r" rB r r r rD r r- r. rM r r r Útest_slice_month¡ s z TestSlicing.test_slice_monthc C sØ t dtddddd}tt t|¡|d}|dd
}|d d
|
}t ||¡ |dd
}|d d
}t ||¡ |d d
}|d d
|
}t ||¡ |d }||jd ks²t t
|
j t d d |d W 5 Q R X d S )NrJ r? r) r@ rA r z2005-05z2006-02Z20050501Z20060228ú2005-1-1r z^'2004-12-31'$©Úmatchz
|
2004-12-31© r r r r r" rB r- r. r, r ÚpytestÚraisesÚKeyError)r ÚrngrE r r3 r r r Útest_partial_slice« s z TestSlicing.test_partial_slicec C sp t dtddddd}tt t|¡|d}|d }t ||jd d
¡ t j
|
t d
|
d |d W 5 Q R X d S ) NÚHr? r) é r@ rA r z 2005-1-31é z^'2004-12-31 00'$rR z 2004-12-31 00) r r r r r" rB r- r. r, rU rV rW ©r rX rE r r r r Útest_partial_slice_dailyÁ s z$TestSlicing.test_partial_slice_dailyc C sª t dtdddddddd}tt t|¡|d}|d }t ||jd d
|
¡ |d }t ||jd d
¡ |d |jd kst t
|
j t dd |d W 5 Q R X d S )Nr r? r) é r r@ rA r rQ éð z 2005-1-1 20é< z2005-1-1 20:00z^'2004-12-31 00:15'$rR z2004-12-31 00:15rT r] r r r Útest_partial_slice_hourlyË s z%TestSlicing.test_partial_slice_hourlyc C s® t dtdddddddd}tt t|¡|d }|d
|
}t ||jd d
¡ |d }t ||jd d
¡ |t d |jd kst
|
t j t dd |d W 5 Q R X d S )NÚSr? r) é é; r r@ rA r z2005-1-1 23:59ra rQ z2005-1-1 23:59:00z^'2004-12-31 00:00:00'$rR z2004-12-31 00:00:00)r r r r r" rB r- r. r, r r rU rV rW r] r r r Útest_partial_slice_minutelyÙ s z'TestSlicing.test_partial_slice_minutelyc
|
C sØ t tddddddddddd }tt d¡|}t |d
|
|jd d
¡ t |d |jd d
¡ t |d |jd d
¡ t |d |jd d
¡ |td |jd ks²t t
|
j t dd |d W 5 Q R X d S )Nr? r) r re i6B )Ú microsecondr_ ZUS)r r r z2005-1-1 00:00r z2005-1-1 00:00:59z2005-1-1 00:01z2005-1-1 00:01:00z2005-1-1 00:00:59.999990z2005-1-1 00:00:00rR ) r r r r r" r- r. r, r r rU rV rW )r rX rE r r r Ú#test_partial_slice_second_precisionç s ýz/TestSlicing.test_partial_slice_second_precisionc C sÄ ddddddg}ddd d
|
d d g}t |d d
d D ]\}}td| }tdddddd}t|| ||| g}dd dg}td|i|tjd} | jj|kst t
|
||D ]l\}
|
} |
|
|| ¡} | d | } t | tjsÚt | | ksæt d| d}t jt|d | | W 5 Q R X q¨|d |
D ]}dtd dgdtdd gfD ]^\}}|| |¡} | d | } | d | } t | | ¡ t jt| d | | W 5 Q R X qBq"||d d
D ]n}|d |¡} | d | } t | tjsæt | d ksôt d| d}t jt|d | | W 5 Q R X q¶tt
|
|||d d
D ]|\}}|d td| }| |¡} d| d}t jt|d | d | W 5 Q R X t jt|d | | W 5 Q R X q@q2d S )Nz%Yz%Y-%mz%Y-%m-%dz %Y-%m-%d %Hz%Y-%m-%d %H:%Mz%Y-%m-%d %H:%M:%SrC ÚmonthÚdayÚhourÚminuteÚsecondr z1 iÜ r) r r' Úa)Zdtypez^'z'$rR )Ú enumerater r r r r Úint64r Ú
|
resolutionr ÚzipÚstrftimeÚ
|
isinstancerU rV rW Úslicer- r. r; )r ÚformatsZ resolutionsZrnumrq ÚunitZmiddater Úvaluesr Ú timestampr3 Z ts_stringr ÚmsgÚfmtÚelementZthesliceÚresÚtsr r r Ú test_partial_slicing_dataframeù s^ ú
|
$ "
|
z*TestSlicing.test_partial_slicing_dataframec C sÜ t ddddgddddgdddd gd
|
td d d d d}|jddgdd}t dggtdgdddgd}|jd }t ||¡ |jtd d dddf }|jd }t ||¡ |jd }|j d d
|
ddg¡}t ||¡ d S )NÚACCT1ZACCT2ÚABCZMNPZXYZr) r r' r )ÚACCOUNTÚTICKERÚvalú2013-06-19 09:30:00Z5TrH r r r T)Úappend)r r ©r Úcolumns)r
r )r! )r
r r )z
|
2013-06-19r r ) r r Z set_indexr r r- r< r r. r, Z droplevel)r r Zdf_multir3 r r r r Ú$test_partial_slicing_with_multiindexG s.
|
|
|
ý ú ÿ
|
ÿ
|
|
z0TestSlicing.test_partial_slicing_with_multiindexc C s¢ t tj dd¡tdddd ¡ }|d d
¡ }|d }|td }t ||¡ |td }|d }t ||¡ t |}|
|
d¡}|j td }t ||¡ d S )Niè z2000-1-1r r r* z2000-1-4) r r r rD r ÚstackÚcopyr r- r. Úxsr r< )r r0 Ús2r3 r Zdf2r r r Ú+test_partial_slicing_with_multiindex_seriese s
|
ÿ
|
|
z7TestSlicing.test_partial_slicing_with_multiindex_seriesc C sÎ t t d¡tddd}|dddg }td}tjtdd |dd
W 5 Q R X tjtd
|
d ||d
W 5 Q R X tjtdd |jdd
W 5 Q R X tjtd
|
d |j|d
W 5 Q R X d S ) Nr z
|
2014-01-01r r' r8 r z
|
2014-01-10z,Value based partial slicing on non-monotonicrR z"Timestamp\('2014-01-10 00:00:00'\)) r r r" r r rU rV rW r )r r0 Z nonmonotonicry r r r Ú(test_partial_slice_requires_monotonicityz s" ÿ ÿz4TestSlicing.test_partial_slice_requires_monotonicityc C sT t dgtddd}|jtdddd
}t ||¡ |jdd
}t ||¡ d S )NÚ1z2016-10-01T00:00:00z2016-10-01T23:59:59)r r ià r r) )r r r r r- r< r r r r Ú test_loc_datetime_length_one s þ z(TestSlicing.test_loc_datetime_length_oner z2018-12-02 21:50:00+00:00Úendz2018-12-02 21:52:00+00:00c C s t dddd}td|dgd}|||
}|jdd
d d
f }t ||¡ t|}t|}tjtd d
|
|||d d
d
W 5 Q R X tjtd d
|
| d ¡}|||
W 5 Q R X d S )Nz2018-12-02 14:50:00-07:00Z1min)r r r r) r: r r r' zBoth dates mustrR éüÿÿÿz1:00zThe index must be timezone)
|
r r r, r- r< ÚstrrU rV Ú
|
ValueErrorZ tz_localize)r r r r# r r r3 r r r Ú,test_getitem_with_datestring_with_UTC_offset s ý "
|
z8TestSlicing.test_getitem_with_datestring_with_UTC_offsetc C sT t dtditddddd}ttdtdddddd}|jd }t ||¡ d S )
|
Nr: r\ Ú2000ÚMrH r é )r r )r r: )r r+ r r r r- r. )r r r3 r r r r Útest_slice_reduce_to_series s ÿ
|
z'TestSlicing.test_slice_reduce_to_seriesN) Ú__name__Ú
|
__module__Ú __qualname__r r$ r6 r7 r= rF rU ÚmarkZ parametrizerI rN rP rY r^ rb rf rh r r r r r r Z to_pydatetimer r r r r r r
|
sZ
|
úþ
|
|
|
N
|
ýþ
|
ýþ r
|
)Ú__doc__r Únumpyr rU Zpandasr r r r r r r Zpandas._testingZ_testingr- r
|
r r r r Ú<module> s $
|
