U
|
ý°dª ã @ sä d dl m Z d dlZd dlZd dlZd dlZd dlmZmZ d dl m
|
Z
|
d dl m Z d dl Zd dl mZmZmZ d dlmZ d dlmZ d dlmZmZmZ d d lm Z d d
|
l m Z e ¡ d d Z ej d d Z!G dd dZ"dS )é )ÚdatetimeN)ÚDAYSÚMONTHS)ÚIncompatibleFrequency)ÚInvalidIndexError)Ú DataFrameÚSeriesÚ Timestamp)Ú
|
date_range)ÚPeriodÚ PeriodIndexÚ period_range)Ú_get_period_range_edges)Úoffsetsc C s t S )N)r © r r ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/resample/test_period_index.pyÚ_index_factory! s r c C s dS )NÚpir r r r r Ú _series_name& s r c @ s e Zd Zej ddddg¡ej ddddg¡d d
|
Zd d Zej dd dddg¡ej ddddg¡ej dddiddig¡dd Zej de ¡ej dddg¡ej dd d g¡ej d d d d!g¡d"d# Z
|
d$d% Z ej d&d'd(d)d*g¡d+d, Z ej dd dg¡d-d. Z d/d0 Zd1d2 Zej de ¡ej d3d d d!g¡ej d4d d g¡d5d6 Zej d3d d g¡ej d4d d g¡d7d8 Zd9d: Zej d;d!d<d=d<d>gfd?d@dAgfg¡dBdC ZdDdE ZdFdG ZdHdI ZdJdK ZdLdM ZdNdO ZdPdQ ZdRdS Zej dTe ¡ej d3d d g¡ej d4d d g¡dUdV Z dWdX Z ej de ¡dYdZ Z ej d[d d g¡d\d] Z d^d_ Z!d`da Z"ej ddbg¡ej ddddg¡dcdd Z#dedf Z$dgdh Z%didj Z&dkdl Z'dmdn Z(dodp Z)dqdr Z*dsdt Z+dudv Z,ejj-dwdxdydz Z.ej d{d|d}d~dg¡dd Z/ej d{dddd
dg¡dd Z0dd Z1dd Z2ej dddg¡ej dddg¡dd Z3ej de4j5de4j5ddgdddddgfe4j5e4j5de4j5e4j5e4j5dde4j5e4j5g
|
ddddddddddg
|
fg¡ej d d¡de6j7ddgfd¢dd£gfd¤d¥dgfg¡d¦d§ Z8d¨d© Z9ej dªd«d¬dd®d¯d°d±d²d³d´dµd¶d·d¸d¹dºg¡d»d¼ Z:ej d½d¾d¿dÀdÁdÂdÃg¡dÄdÅ Z;dÆdÇ Z<dS )ÈÚTestPeriodIndexÚfreqZ2DÚ1HÚ2HÚkindÚperiodNÚ timestampc
|
C s |}|dkr | ¡ |¡ ¡ }nP|jd j dd}|jd |jj j dd}t|||dd}| ¡ |¡ |¡}|j||d ¡ } t | |¡ d S ) Nr r Ústart©ÚhowéÿÿÿÿÚleft)r Úendr Ú inclusive©r )
|
Ú to_timestampÚresampleÚasfreqÚindexr r
|
ÚreindexÚ to_periodÚtmÚassert_almost_equal)
|
ÚselfZseries_and_framer r ÚobjÚexpectedr r! Ú new_indexÚresultr r r Ú test_asfreq, s zTestPeriodIndex.test_asfreqc C sÎ |}t |jd jdd|jd jdddd}| ¡ j|dd}|jdd d
|
jdd}t ||¡ | d ¡}t |jd jdd|jd jdddd}| ¡ j|d d}|jdd d
|
jd d}t ||¡ d S ) Nr r r r r ©r g @)Z
|
fill_valuer r# Úvalueç @)
|
r
|
r' r$ r( r% r& r* Úassert_series_equalZto_frameÚassert_frame_equal)r, ÚseriesÚsr/ r. r0 Úframer r r Útest_asfreq_fill_value= s$ ý
|
ýz&TestPeriodIndex.test_asfreq_fill_valueÚHÚ12HÚWÚkwargsÚonÚdateÚlevelÚdc C sr t jt|t jd}t||dtjj||gddgdd}d}tj t
|
|d |j |fd |i| W 5 Q R X d S )
|
N©Údtype)r@ ÚaÚvrB )Únames©r' zResampling from level= or on= selection with a PeriodIndex is not currently supported, use \.set_index\(\.\.\.\) to explicitly set index©Úmatchr ) ÚnpÚarangeÚlenÚint64r ÚpdZ
|
MultiIndexZ from_arraysÚpytestÚraisesÚNotImplementedErrorr% )r, r' r r r> ÚrngÚdfÚmsgr r r Útest_selectionT s þÿz TestPeriodIndex.test_selectionÚmonthÚmethÚffillÚbfillÚconvr r! ÚtargÚDÚBÚMc C sX |ddd| d}t |j||d| }|j||d}| ||¡ ¡ }t ||¡ d S )Nú1/1/1990z
|
12/31/1991úA-r2 ©Ú
|
conventionr )Úgetattrr% r$ r& r) r* r5 ) r, r\ r[ rX rW Úsimple_period_range_seriesÚtsr0 r. r r r Útest_annual_upsample_casesg s
|
z*TestPeriodIndex.test_annual_upsample_casesc C sz |dddd}| d¡ ¡ }| |jj¡ ¡ }tdddd|_t ||¡ t | d¡ ¡ |¡ t | d¡ ¡ |¡ d S )Nr` ú 6/30/1995r_ r2 úa-decrE )r% ÚmeanÚgroupbyr' Úyearr r* r5 ©r, re rf r0 r. r r r Útest_basic_downsampleu s z%TestPeriodIndex.test_basic_downsamplezrule,expected_error_msg)ri z<YearEnd: month=12>)zq-marz <QuarterEnd: startingMonth=3>)r_ z
|
<MonthEnd>)zw-thuz<Week: weekday=3>c C sF |dddd}d| d}t jt|d | |¡ ¡ W 5 Q R X d S )Nr` rh zw-wedr2 z3Frequency <Week: weekday=2> cannot be resampled to z&, as they are not sub or super periodsrI )rP rQ r r% rj )r, re ZruleZexpected_error_msgrf rU r r r Útest_not_subperiod s
|
|
ÿz"TestPeriodIndex.test_not_subperiodc C s^ |dddd}| d¡ ¡ }|j |dd ¡ }|j|dd}| |d ¡ |¡}t ||¡ d S )
|
Nr` rh r_ r2 ri r! rb r rY )r% rj rY r$ r& r) r* r5 )r, r re rf r0 Ú resampledr. r r r Útest_basic_upsample s z#TestPeriodIndex.test_basic_upsamplec C sb t dddd}ttj t|¡|}|jdddjdd }| d¡j |j
|
d
|
dd }t ||¡ d S ) Nú1/1/2000é ÚA©Úperiodsr r_ r! rb é )ÚlimitrY )Úmethodrx ) r r rK ÚrandomÚrandnrM r% rY r& r( r' r* r5 ©r, rS rf r0 r. r r r Útest_upsample_with_limit s
|
z(TestPeriodIndex.test_upsample_with_limitc
|
C s° |dddd}t d|i}| d¡ ¡ }|d d¡ ¡ }t |d |¡ tdddd}td d
|
d d g|d }| d¡ ¡ }tdddd}|jdddj|dd} t || ¡ d S )Nr` ú
|
12/31/1995úA-DECr2 rE r] Ú2000Z2003é rw é é rH r_ ú2000-01z2003-12r r rY ©ry ) r r% rY r* r5 r r r& r( )
|
r, re rf rT ÚrdfÚexprS r0 Úex_indexr. r r r Útest_annual_upsample¦ s z$TestPeriodIndex.test_annual_upsampleÚtargetrc c C sX d| }|dd|d}|j ||d ¡ }|j||d}| |d¡ ¡ }t ||¡ d S )NúQ-r` r~ r2 rb r rY ©r% rY r$ r& r) r* r5 ) r, rW r rc re r rf r0 r. r r r Útest_quarterly_upsample¶ s
|
z'TestPeriodIndex.test_quarterly_upsamplec C sN |dddd}|j ||d ¡ }|j||d}| |d¡ ¡ }t ||¡ d S )Nr` r~ r_ r2 rb r rY r )r, r rc re rf r0 r. r r r Útest_monthly_upsampleà s
|
z%TestPeriodIndex.test_monthly_upsamplec C s t tdtddddddd}tj|dd
< ttd
|
d td d gdd }t ddg|d}| ¡ jd dd ¡ }t
|
||¡ |jd dd ¡ }t
|
||¡ d S )Néd Z20130101r8 Úidx)r rv ÚnameÚfloat©r' rD é
|
é z2013-01-01 00:00ÚTz2013-01-01 00:01)r g @A@g àS@rH r r# ) r Úranger
|
rK Únanr r r) r% rj r* r5 )r, r8 r' r. r0 Zresult2r r r Útest_resample_basicÌ s ýþ z#TestPeriodIndex.test_resample_basiczfreq,expected_valsé é é Z2Mé< é( c C sP t dtdddd}| |¡ ¡ }td|t|d}t ||d}t ||¡ d S )Nr r r ©r rv rH ©r r rv )r r r% ÚcountrM r* r5 )r, r Z expected_valsr7 r0 Úexpected_indexr. r r r Útest_resample_countÞ s ÿ z#TestPeriodIndex.test_resample_countc C s@ t tdtddddd}|}t| d¡| }t ||¡ d S )Nr r r_ ©r rv r rH )r r r rd r% r* r5 )r, Zresample_methodr7 r. r0 r r r Útest_resample_same_freqë s z'TestPeriodIndex.test_resample_same_freqc C sF d}t jt|d* ttdtddddd d¡ ¡ W 5 Q R X d S ) NzcFrequency <MonthEnd> cannot be resampled to <Week: weekday=6>, as they are not sub or super periodsrI r r r_ r¤ rH r= )rP rQ r r r r r% rj )r, rU r r r Útest_resample_incompat_freqó s ÿ ÿþz+TestPeriodIndex.test_resample_incompat_freqc C s t d¡}tdddddt jd}tdddddt jd}t||dd }td|d
|
}| |¡}|jd d d ¡ }t ||d dt
|
¡ }td|d
|
}t ||¡ d S )NúAmerica/Los_AngeleséÝ é r r ©rl rW ÚdayÚhourÚminuteÚtzinforw r; r2 rH r] r r# ©r r! r ç ð?)ÚpytzÚtimezoner Úutcr
|
r Ú
|
tz_convertr% rj r r ÚDayr* r5 © r, Zlocal_timezoner r! r' r7 r0 r¢ r. r r r Ú test_with_local_timezone_pytzý s
|
|
z-TestPeriodIndex.test_with_local_timezone_pytzc C sh t dtdddddd}| d¡ ¡ }t d tjdd
|
gddd d}t ||¡ |jj t
|
d¡ksdt d S ) Nrw z
|
2017-01-01é0 r; z
|
US/Eastern©rv r ÚtzrH r] ç @z
|
2017-01-02)rº r ) r r
|
r% rj rO Ú DatetimeIndexr* r5 r' rº r± r² ÚAssertionError)r, r8 r0 r. r r r Útest_resample_with_pytz s ÿ ÿþ z'TestPeriodIndex.test_resample_with_pytzc C s¤ d}t dddddtj ¡ d}t dddddtj ¡ d}t||dd d
|
}td|d }| |¡}|jd d d ¡ }t ||d d dt
|
¡ }td|d }t ||¡ d S )Nz dateutil/America/Los_Angelesr¨ r© r r rª rw r; r )r r rH r] r r# )r r! r r r° )r Údateutilrº Ztzutcr
|
r r´ r% rj r r rµ r* r5 r¶ r r r Ú!test_with_local_timezone_dateutil" s2 ÿ ÿ
|
ÿ z1TestPeriodIndex.test_with_local_timezone_dateutilc C sÌ t dddd}tt t|¡|d}| d¡}tj|jdd|_| d¡
|
¡ }t ||¡ t dd d
|
d }| d ¡ d ¡}tttt||d}| tjdd¡ ¡ }t dd dd ddd}t |j|¡ d S )Nz
|
2017-03-12z2017-03-12 1:45:00Z15Tr2 rH z
|
US/PacificZ900Sz
|
2017-10-10z
|
2017-10-20r r¯ ÚUTCzAmerica/Sao_Paulo)Údatar' Z1Dz
|
2017-10-09r] Z shift_forwardr )r r! r rº Z nonexistentr" )r
|
r rK ÚzerosrM Ú tz_localizerO r¼ r' r% rj r* r5 r´ r Úlistr rk ZGrouperr¡ Úassert_index_equal)r, r' r8 r. r0 rT r r r Ú'test_resample_nonexistent_time_bin_edge= s&
|
úz7TestPeriodIndex.test_resample_nonexistent_time_bin_edgec C sD t ddddd}tt t|¡|d}| d¡ ¡ }t ||¡ d S )Nz2014-10-25 22:00:00z2014-10-26 00:30:00Z30Tz Europe/London©r rº rH ) r
|
r rK rà rM r% rj r* r5 )r, r r. r0 r r r Ú%test_resample_ambiguous_time_bin_edgeU s ÿz5TestPeriodIndex.test_resample_ambiguous_time_bin_edgec C s\ t tjdddtddddd}| d¡ ¡ }| d¡ ¡ d¡ ¡ d¡}t ||¡ d S ) Nr rN rC z
|
2010-01-01ÚQru rH r_ )
|
r rK rL r
|
r% rY ÚlastZastyper* r5 )r, r8 rË Zbothr r r Ú!test_fill_method_and_how_upsample^ s þ z1TestPeriodIndex.test_fill_method_and_how_upsampler« c C sX d| }|dd|d}|j ||d ¡ }|j||d}| |d¡ ¡ }t ||¡ d S )NzW-r` r~ r2 rb r rY r ) r, r« r rc re r rf r0 r. r r r Útest_weekly_upsampleh s
|
z$TestPeriodIndex.test_weekly_upsamplec C sF |dddd}|j ddd ¡ }|jdd d¡ ¡ }t ||¡ d S )
|
Nr` r~ r_ r2 r r r# r r )r% rj r$ r* r5 rm r r r Útest_resample_to_timestampss s z+TestPeriodIndex.test_resample_to_timestampsc C s |ddd| d}| d| ¡ ¡ }|jddd}t|jd dd¡|jd
|
dd ¡d| d}|j| dd ¡d d}||_t ||¡ d S )NÚ1990Ú1992ra r2 r r] r r r r r! r8 rY r
) r% rY r$ r r' r& r( r* r5 )r, re rW rf Zquar_tsÚstampsZqdatesr. r r r Útest_resample_to_quarterlyz s ýz*TestPeriodIndex.test_resample_to_quarterlyr c C sN |dddd}|j d|d ¡ }|jd|d}|j|jdd }t ||¡ d S )
|
NrÏ rÐ zA-JUNr2 zQ-MARrb r rY r
)r% rY r& r( r' r* r5 )r, re r rf r0 r. r r r Ú$test_resample_to_quarterly_start_end s
|
z4TestPeriodIndex.test_resample_to_quarterly_start_endc C sb t ddddgdd}ttj d¡|d}| ¡ }| d¡ ¡ }| d¡ ¡ d¡}t
|
||¡ d S ) NéÐ éÕ é× iÙ rt r2 r rH ) r r rK rz r{ r$ r% rY r) r* r5 )r, rS r8 rÑ Zfilledr. r r r Útest_resample_fill_missing s z*TestPeriodIndex.test_resample_fill_missingc C sZ t dddddgdd}ttj d¡|d}d}tjt|d | d¡ ¡ W 5 Q R X d S )
|
NrÔ rÕ rÖ rt r2 rs rH z8Reindexing only valid with uniquely valued Index objectsrI )
|
r r rK rz r{ rP rQ r r% rY )r, rS r8 rU r r r Útest_cant_fill_missing_dups¢ s
|
z+TestPeriodIndex.test_cant_fill_missing_dupsÚ5minc C sl t dddd}ttj t|¡|d}| ¡ |¡ ¡ }|dkrJ| |¡}|j||d ¡ }t
|
||¡ d S )Nrr z1/5/2000r r2 rH r r# ) r r rK rz r{ rM r$ r% rj r) r* r5 )r, r r rS rf r. r0 r r r Útest_resample_5minute© s
|
z%TestPeriodIndex.test_resample_5minutec C s |dddd}| d¡ ¡ }| d¡ tdd¡}t ||¡ |dd}|j ddd ¡ }tdd
|
dd}|jddd |¡}t ||¡ d S ) Nrr z2/1/2000r^ r2 r] z1/3/2000r; r8 rb z2/1/2000 23:00r )r% r& r( r r* r5 )r, re rf r0 r. Zexp_rngr r r Ú"test_upsample_daily_business_daily´ s
|
z2TestPeriodIndex.test_upsample_daily_business_dailyc C sd t dddd}tt d¡|d}|d d
}| d¡ t¡}| d¡ t¡j|j }t
|
||¡ d S ) Nz1/1/2012rÙ iè r r rH z2012-01-04 06:55Ú10min) r
|
r rK Úarrayr% ÚapplyrM Úlocr' r* r5 )r, Údrr8 Zsubsetr0 r. r r r Ú test_resample_irregular_sparseÁ s z.TestPeriodIndex.test_resample_irregular_sparsec C s| t dddd}ttj t|¡|d}| d¡ ¡ }| ¡ ¡ sDt
|
| d¡ ¡ ¡ d d
}| d¡ ¡ }t ||¡ d S )Nrr r zW-WEDru rH zW-THUr )r
|
r rK rz r{ rM r% r& ZisnaÚallr½ rY r* r5 r| r r r Útest_resample_weekly_all_naË s z+TestPeriodIndex.test_resample_weekly_all_nac C sð t ddd}ttt||d}| d¡}| d¡}| d¡ ¡ }| ¡ }dd |j
|
¡ D |_ | d¡ ¡ d¡}t j |j dd
|
|_ t ||¡ | d ¡ ¡ }t d d ddd}tddg|d} | jd ddd ¡ }t ddd dd}
|
tdg|
|
d} t || ¡ | jd dd ¡ }tddd d}
|
tdg|
|
d} t || ¡ t d d d d} | d ¡} t| d}tj t| ¡|d < t tj t| ¡¡|d!< t| d"¡ ¡ d | d"¡ ¡ d! d#d d!gd$} | d"¡ tjtjd#¡jd d!gd$}t || ¡ d S )%Nz 2012-4-13z2012-5-1)r r! rH rÁ r§ r= c S s g | ]}|j d dqS )N)r® )Úreplace©Ú.0Úxr r r Ú
|
<listcomp>á s z>TestPeriodIndex.test_resample_tz_localized.<locals>.<listcomp>r2 r] z2001-09-20 15:59z2001-09-20 16:00r zAustralia/SydneyrÈ r rw Úright©ÚclosedÚlabelz
|
2001-09-21r¹ g ø?r r# z
|
2001-09-20ru z1/1/2011i N r; ÚESTÚfirstÚsecondrt )rî rï )Úcolumns)r
|
r r rM rÄ r´ r% rj Úcopyr' Z to_pydatetimerO r¼ r* r5 r r rK rz r{ ZcumsumÚsumZaggr( r6 ) r, rà rf Zts_utcZts_localr0 Zts_local_naiver r r8 r r. rS r r r Útest_resample_tz_localized× s\
|
|
ÿ ÿ
|
|
þû
|
ÿþÿz*TestPeriodIndex.test_resample_tz_localizedc C sÀ t tj d¡tddddd}tj|d< |jddd d
|
¡ }|d d
jddd d
|
¡ }t ||¡ |jdddd
|
¡ }|d d
jdddd
|
¡ }tddd d}t
|
|j |¡ t ||¡ d S ) Né z 1/1/2012 9:30Z1minr rH r rÜ r ré rê r r ) r rK rz r{ r
|
r r% rj r* r5 rÆ r' )r, r8 r0 r r r r r Útest_closed_left_corner s
|
þ
|
z'TestPeriodIndex.test_closed_left_cornerc C sT t dddd}tt d¡|d}| d¡ ¡ }| ¡ d¡ ¡ ¡ }t ||¡ d S )NZ2000Q1r úQ-DECru rH rt )
|
r r rK rL r% rj r$ r) r* r5 )r, rS rf r0 r r r r Útest_quarterly_resampling( s
|
z)TestPeriodIndex.test_quarterly_resamplingc C sX t dddd}t|}dd t|D }t|ddd d
|
d g|d }|jd ddd ¡ d S )Nz8/6/2012z 8/26/2012r] r¯ c S s g | ]}|gd qS )rs r rå r r r rè 4 s zATestPeriodIndex.test_resample_weekly_bug_1726.<locals>.<listcomp>ÚopenÚhighÚlowÚcloseZvol)rð r' zW-MONr rê )r
|
rM r r r% rî )r, ÚindÚnr rT r r r Ú test_resample_weekly_bug_17260 s
|
z-TestPeriodIndex.test_resample_weekly_bug_1726c C s¸ t ddg¡ d¡ d¡}tddg|d}|jdd d d
|
¡ ¡ }d d d ddddddddddg }t j|dd d¡}t j|dd}tddddddddddddd g |d}t
|
||¡ d S ) Nl æZÑ! l ²NìEà! rÁ zAmerica/Chicagor rw rH Z12hré rê z2016-03-09 12:00:00-06:00z2016-03-10 00:00:00-06:00z2016-03-10 12:00:00-06:00z2016-03-11 00:00:00-06:00z2016-03-11 12:00:00-06:00z2016-03-12 00:00:00-06:00z2016-03-12 12:00:00-06:00z2016-03-13 00:00:00-06:00z2016-03-13 13:00:00-05:00z2016-03-14 01:00:00-05:00z2016-03-14 13:00:00-05:00z2016-03-15 01:00:00-05:00z2016-03-15 13:00:00-05:00T)r³ r2 r° r» ) rO r¼ rÄ r´ r r% rË rY Ú to_datetimer* r6 )r, r' rT r0 Zexpected_index_valuesr. r r r Ú"test_resample_with_dst_time_change: s> ÿþÿóÿ þz2TestPeriodIndex.test_resample_with_dst_time_changec C sn t t dd¡tjd}| d¡ ¡ }| d¡ ¡ d¡ ¡ }|jd tdksRt |jd |jd ksjt d S )NZ20000101Z20000201r ZBMSr^ r Z20000103)
|
r rO Z bdate_rangerK Úfloat64r% rj r' r r½ )r, Z
|
timeseriesZres1Zres2r r r Útest_resample_bms_2752] s
|
ÿz&TestPeriodIndex.test_resample_bms_2752z6Commented out for more than 3 years. Should this work?)Úreasonc C sl t dddd}tt d¡|d}t dddd }|jdd
|
d |¡}|jd d }| d¡ ¡ }t
|
||¡ d S )Nr r r_ ru rH z
|
2000-01-01z
|
2000-03-31r] r2 r! r rZ r
) r r rK rL r& r( Zfillnar% rj r* r5 )r, rS rf Z exp_indexr. r0 r r r Ú test_monthly_convention_spang s z,TestPeriodIndex.test_monthly_convention_spanzfrom_freq, to_freq)r] r_ )rÊ rt )r_ rÊ )r] r= c C sT t dd|d}ttj t|d¡|}| |¡ ¡ }t ||j|ddd ¡ ¡ d S )Nú 8/15/2012r r¤ rw ré rê ©
|
r
|
r rK rz r{ rM r% rj r* r6 ©r, Z from_freqZto_freqr rT rp r r r Ú test_default_right_closed_labelu s ÿz/TestPeriodIndex.test_default_right_closed_label)r] ZMS)rÊ ZAS)r_ ZQS)r; r] )r r; c C sT t dd|d}ttj t|d¡|}| |¡ ¡ }t ||j|ddd ¡ ¡ d S )Nr r r¤ rw r rê r r r r r Ú test_default_left_closed_label s ÿz.TestPeriodIndex.test_default_left_closed_labelc C sL t dddd}ttj t|¡|d}| d¡ ¡ }t |d | ¡ ¡ d S )Nz
|
2012-01-01z
|
2012-12-31r_ r¯ rH rt r )
|
r r rK rz r{ rM r% rj r* r+ )r, r' r8 r0 r r r Útest_all_values_single_bin s z*TestPeriodIndex.test_all_values_single_binc C sP t tj dd¡tdddd}| d¡ ¡ }tj|j dd
¡ |j dd
¡ gd d
|
j
|
}tj t dt d gdd |_ t ||¡ td dd}t d dddddgd ddddddgd | |¡d ¡ }td ddd}t d d d d d dgd |d}| d¡ ¡ }t ||¡ t d d d d d!dgd |d}| d¡ ¡ }t ||¡ d S )"Nr r z2000-1-1)rv rH Z5Dr rs r )Zaxisz2000-1-6r2 z2001-5-4é r éP éZ r é )ZREST_KEYZ
|
DLY_TRN_QTZ DLY_SLS_AMTZCOOP_DLY_TRN_QTZCOOP_DLY_SLS_AMTrw éF r é2 r Z7Dru é rô i i¼ i0 i )r rK rz r{ r
|
r% rj rO ÚconcatZilocr r¼ r r' r* r6 ÚappendZ
|
sort_indexr¡ rò )r, rT r0 r. r' r r r Ú(test_evenly_divisible_with_no_extra_bins sx 0 ÿ ûÿ ÷ ûÿ ÷öë
|
ûÿ ÷
|
õ ûÿ ÷
|
õ z8TestPeriodIndex.test_evenly_divisible_with_no_extra_binszfreq, period_mult)r; é )r< rw c C s| t dddd}ttt||d}| ¡ |¡ ¡ |¡}t d||t| d}| |¡}|j||d ¡ }t
|
||¡ d S )Nr r] r r rH r# ) r r r rM r$ r% Zohlcr) r( r* r6 ) r, r Z period_multr r r8 r. r/ r0 r r r Útest_upsampling_ohlcÜ s
|
z$TestPeriodIndex.test_upsampling_ohlczperiods, valuesú1970-01-01 00:00:00z1970-01-01 00:00:02z1970-01-01 00:00:03rw r rs é r© r é é é é zfreq, expected_valuesÚ1sZ2sg "@Z3sg @c
|
C sT t |dd}t||d}tdt||d}t||d}| |¡ ¡ } t | |¡ d S )NÚSr2 rH r ru )r r r rM r% rj r* r6 )
|
r, rv Úvaluesr Zexpected_valuesr' r9 r¢ r. r0 r r r Útest_resample_with_natë s ( ÿ z&TestPeriodIndex.test_resample_with_natc C sf t tjgd dd}tdddg|dgd}t g |jd}t|dgd d
|
}| d ¡ ¡ }t ||¡ d S ) Nr r r2 rw rs rE )r' rð )r r r )r' rð rD r ) r rO ÚNaTr r r% rj r* r6 )r, r r9 r¢ r. r0 r r r Útest_resample_with_only_nat s z+TestPeriodIndex.test_resample_with_only_natz$start,end,start_freq,end_freq,offset)Ú19910905ú19910909 03:00r; Ú24HÚ10H)r# ú19910909 12:00r; r% r& )r# z19910909 23:00r; r% r& )ú19910905 10:00Ú19910909r; r% r& )r( ú19910909 10:00r; r% r& )r# r* r; r% r& )ú19910905 12:00r) r; r% r& )r+ r$ r; r% r& )r+ r' r; r% r& )r+ r' r; r% Z34H)r+ r' r; Ú17Hr& )r+ r' r; r, Ú3H)r+ z 19910909 1:00r; r_ r- )r# z19910913 06:00r r% r& )r# z19910905 01:39ZMinÚ5MinÚ3Min)r# z19910905 03:18Z2Minr. r/ c C s t dt|||dd}|t t|¡ }|j||d ¡ }| |¡}| ¡ j||d ¡ }|dkrp|j d ¡|_t
|
||¡ d S )Nr r2 rH )Úoffsetr_ ) r r rK rL rM r% rj r$ r' Z
|
_with_freqr* r5 ) r, r r! Z
|
start_freqZend_freqr0 r8 r0 r. r r r Útest_resample_with_offset& s
|
z)TestPeriodIndex.test_resample_with_offsetz"first,last,freq,exp_first,exp_last)r# Ú19920406r] r# r2 )z19910905 00:00ú19920406 06:00r] r# r2 )ú19910905 06:00r3 r; r4 r3 )Z19910906r2 r_ z1991-09ú1992-04)Z19910831Z19920430r_ ú1991-08r5 )r6 r5 r_ r6 r5 c C sZ t |}t |}t ||d}t ||d}tjj |¡}t|||}||f}||ksVtd S )Nr2 )r rO ZtseriesZ frequenciesZ to_offsetr r½ )r, rî rË r Z exp_firstZexp_lastr0 r. r r r Útest_get_period_range_edgesH s z+TestPeriodIndex.test_get_period_range_edgesc C st t dddd}t d¡}tj|dd
< t|| ¡ }| d¡jdd}td tjgtd
|
d gd d d}t
|
||¡ d S )NZ2018r_ r r r rÊ r )Z min_countr4 Z2018Q1Z2018Q2rö r2 rH ) r
|
rK Zonesr r r) r% rò r r* r5 )r, r' r r8 r0 r. r r r Útest_sum_min_counte s
|
ÿz"TestPeriodIndex.test_sum_min_count)=Ú__name__Ú
|
__module__Ú __qualname__rP ÚmarkZ parametrizer1 r: rV r rg rn ro rq r} r r r r r£ r¥ r¦ r· r¾ rÀ rÇ rÉ rÌ r rÍ rÎ rÒ rÓ r× rØ rÚ rÛ rá rã ró rõ r÷ rþ r r Zxfailr r r r
|
r r rO r! rK ÚNaNr r" r1 r7 r8 r r r r r + s6
|
üþ
|
|
|
|
ÿ
|
|
|
|
|
|
|
|
=
|
#
|
|
|
ÿ
|
þ
|
F û ø ö óõþ
|
|
ýþ ðþ
|
ôþ
|
r )#r r¿ ÚnumpyrK rP r± Z pandas._libs.tslibs.ccalendarr r Zpandas._libs.tslibs.periodr Z pandas.errorsr ZpandasrO r r r Zpandas._testingZ_testingr* Z pandas.core.indexes.datetimesr
|
Zpandas.core.indexes.periodr r r Zpandas.core.resampler Zpandas.tseriesr Zfixturer r r r r r r Ú<module> s&
|
|
|
