U
|
ý°dô ã @ sJ d dl Zd dlZd dlmZmZmZmZmZ d dl m
|
Z G dd dZ dS )é N)Ú DataFrameÚ PeriodIndexÚSeriesÚ
|
date_rangeÚ period_rangec @ s e Zd Zdd Zdd Zdd Zej de e
|
g¡dd Z ej de e
|
g¡d
|
d Z ej de e
|
g¡d d Z ej de e
|
g¡dd Zdd ZdS )ÚTestPeriodIndexc C sÞ t dddddgdd}ttj t|¡|d}| ¡ }|d }|dd
}t ||¡ d|d d
< |rtt ||¡ n|dd
dk ¡ st
|
t dddddgdd}ttj t|¡|d}|d }||dk }t ||¡ d S )
|
NiÐ i× iÙ zA-JUN©Úfreq©ÚindexZ2007é é ) r r ÚnpÚrandomZrandnÚlenÚcopyÚtmÚassert_series_equalÚallÚAssertionError)ÚselfZusing_copy_on_writeÚidxÚtsÚoriginalÚresultÚexpected© r úgd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/period/test_partial_slicing.pyÚ0test_getitem_periodindex_duplicates_string_slice s z@TestPeriodIndex.test_getitem_periodindex_duplicates_string_slicec C sJ t dddddgdd}ttj t|¡|d ¡ }|d |d
|
ksFtd S ) NZ2Q05Z3Q05Z4Q05Z1Q06Z2Q06ÚQr r
|
Z05Q4é )r r r r Úrandr Zcumsumr )r ÚpiÚserr r r Ú'test_getitem_periodindex_quarter_string& s z7TestPeriodIndex.test_getitem_periodindex_quarter_stringc C sj t dddd}ttj t|¡|d}|d }|dd
}t ||¡ |d }|dd
|
}t ||¡ d S ) Nz1/1/10z12/31/12ÚM)ÚstartÚendr r
|
Z2010r é Z2011é )r r r r r! r r r )r r" ÚsÚresÚexpr r r Útest_pindex_slice_index, s z'TestPeriodIndex.test_pindex_slice_indexÚ
|
make_rangec
|
C s |dddd}d}dddd d
|
g}|D ]*}t jt|d ||d
W 5 Q R X q$ttj t|¡|d }t |dd
|d d
¡ t |dd
|d d
¡ t |dd
|dd
¡ t |dd
|dd
¡ d d
|
g}|D ]*}t jt|d ||d
W 5 Q R X qäd S )Nú
|
2013/01/01ÚDi ©r& r ÚperiodsúBslice indices must be integers or None or have an __index__ methodÚ2014ú2013/02ú
|
2013/01/02ú 2013/02/01 9Hú2013/02/01 09:00©Úmatchr
|
r z
|
2013/01/05é é im ©
|
ÚpytestÚraisesÚ TypeErrorr r r r! r r r )r r. r ÚmsgÚvaluesÚvr* Úinvalidr r r Útest_range_slice_day6 s( û z$TestPeriodIndex.test_range_slice_dayc
|
C sú |dddd}d}dddd d
|
g}|D ]*}t jt|d ||d
W 5 Q R X q$ttj t|¡|d }t |d d
|dd
¡ t |dd
|dd
¡ t |dd
|dd
¡ t |d d
|d d
¡ dD ]}t ||d
|¡ qÜd S )Nz2013/01/01 09:00:00ÚSi r1 r3 r4 r5 r6 r7 r8 r9 r
|
z2013/01/01 09:05z2013/01/01 09:10i, i z2013/01/01 10:00z2013/01/01 10:05i ix z2013/01/01 10Hz2013/01/01 09:30iD )r/ z2013/01Z2013r= )r r. r rA rB rC r* Údr r r Útest_range_slice_secondsT s$ û z(TestPeriodIndex.test_range_slice_secondsc C sü |dddd}t ddd tdD i|d}t t|g dd dgd
|
}|d d ¡|d< t |d d
|¡ t |d d
|jd d
¡ t |dd
|jd d
¡ t |dd
|¡ t |dd
|¡ t |dd
|¡ t |dd
|¡ d S )Nú
|
2013/10/01r0 é
|
r1 Úunitsc S s g | ] }d | qS ©éd r ©Ú.0Úir r r Ú
|
<listcomp>t s z@TestPeriodIndex.test_range_slice_outofbounds.<locals>.<listcomp>r
|
r )r ÚcolumnsÚint64z
|
2013/09/01ú
|
2013/09/30z
|
2013/10/02r z
|
2013/10/15z
|
2013/10/17z2013-06z2013-09z2013-11z2013-12)r ÚrangeÚtypeZastyper Zassert_frame_equalZiloc)r r. r ZdfÚemptyr r r Ú test_range_slice_outofboundso s z,TestPeriodIndex.test_range_slice_outofboundsc C s6 |dddd}t ddd tdD i|d}t ||¡}d t|j d
|
}tjt|d | d d ¡ W 5 Q R X tjt|d |
|
d d ¡ W 5 Q R X tjt|d |dd
W 5 Q R X tjt|d |d d
W 5 Q R X tjt|d |j dd
W 5 Q R X tjt|d |j d d
W 5 Q R X d S )NrI r0 rJ r1 rK c S s g | ] }d | qS rL r rN r r r rQ s z?TestPeriodIndex.test_maybe_cast_slice_bound.<locals>.<listcomp>r
|
z cannot do slice indexing on z( with these indexers \[foo\] of type strr9 ZfooÚleftrT ) r rU r Úget_objrV Ú__name__r> r? r@ Z_maybe_cast_slice_boundZget_slice_boundÚloc)r r. Zframe_or_seriesr ÚobjrA r r r Útest_maybe_cast_slice_bound s" ÿz+TestPeriodIndex.test_maybe_cast_slice_boundc C sz t dddd}| d¡}tt d¡|d}ttdddttd d
|
d }|| }|j}tjdd dd d d dddddddg tj d}|| j
|
dk ¡ st || j
|
dk ¡ r°t | d¡}t ||¡ || } |jd }t || ¡ |d }t || ¡ tjdgtj d}
|
|d j
|
dkr*|d jd ks.t | d¡}t ||
|
¡ ||
|
} |jd }t || ¡ |d }t || ¡ d S )Nz
|
2014-01-01é Z30D)r2 r r0 r
|
r r r r< r é r; é é é é é é é )ZdtypeiÞ r4 é iß zMay 2015)r Z to_periodr r ZarangeÚlistrU r ÚarrayZintpÚyearr r ÚanyZget_locr Zassert_numpy_array_equalr\ r Úmonth) r Zdtir" Z ser_montonicZshufflerr# ZnidxZ indexer_2014r r Zindexer_may2015r r r Ú.test_partial_slice_doesnt_require_monotonicity s8
|
ÿ
|
|
$
|
|
z>TestPeriodIndex.test_partial_slice_doesnt_require_monotonicityN)r[ Ú
|
__module__Ú __qualname__r r$ r- r> ÚmarkZ parametrizer r rE rH rX r^ rn r r r r r s
|
|
|
|
|
r ) Únumpyr r> Zpandasr r r r r Zpandas._testingZ_testingr r r r r r Ú<module> s
|
