U
|
ý°d. ã @ sB d dl m Z d dlmZ d dlZd dlZd dlZd dlmZm Z m
|
Z
|
d dl m Z d dlmZ d dlmZ e ej d¡eddd d
|
Zd d Zd d Zdd Zdd Zej de je j e j dd g¡dd Z dd Z dd Z ejj!d d d d Z"ej d d!i d fd!d"d id fd!d"d#iej#fd$i d#fd$d"d id#fd$d"d#iej#fg¡d%d& Z$ej d'd(ej#fd)ej#fd*d+d,g¡d-d. Z%d/d0 Z&d1d2 Z'ej d3d!i d#d d#gfd!d"d id#d d#gfd!d"d#id#ej#d#gfd!d"d4iej#ej#ej#gfd$i d#d#d#gfd$d"d id#d#d#gfd$d"d#id#ej#d#gfd$d"d4iej#ej#ej#gfg¡d5d6 Z(d7d8 Z)dS )9é )Údatetime)Ú methodcallerN)Ú DataFrameÚSeriesÚ Timestamp)ÚGrouper)Ú
|
date_rangeéè z1/1/2000©Úperiods©Úindexc C sj t dddd} t | ¡}dd }| |¡}t dd ¡ |¡}|j d¡|_|j d¡|_t ||¡ d S ) NÚAÚright©ÚfreqÚlabelÚclosedc S s | ¡ dd
S )Néýÿÿÿ)Z sort_values©Úx© r ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/resample/test_time_grouper.pyÚf s ztest_apply.<locals>.fc S s | j S ©N©Úyearr r r r Ú<lambda> ó z test_apply.<locals>.<lambda>r )r Ú test_seriesÚgroupbyÚapplyr Z droplevelÚtmÚassert_series_equal)ÚgrouperÚgroupedr ZappliedÚexpectedr r r Ú
|
test_apply s
|
|
r' c C sx t jtd d d
< t dd ¡ ¡ } tdddd}t |¡ ¡ }|j| _t || ¡ t d¡ ¡ }|j| _t || ¡ d S )Né c S s | j S r r r r r r r ' r z test_count.<locals>.<lambda>r r r )
|
ÚnpÚnanr r Úcountr r r" r# Úresample)r& r$ Úresultr r r Ú
|
test_count$ s r. c C s@ t jddd ¡ } t dd ¡ tj¡}| j|_t | |¡ d S )Nr r )r c S s | j S r r r r r r r 6 r z&test_numpy_reduction.<locals>.<lambda>) r r, Úprodr Zaggr) r r" r# ©r- r& r r r Útest_numpy_reduction3 s r1 c C sp d} t dd| d}tddd|d}td d
|
}| |¡\}}|j|d d }d d }| |¡}t |j|j¡ d S )Nr ú
|
2000-01-01ÚD)Ústartr r é é )ÚopenÚcloser ÚM©r F)Z
|
group_keysc S s | d | d S )Nr8 r7 r )Údfr r r r H s z test_apply_iteration.<locals>.f) r r r Z _get_grouperr r! r" Zassert_index_equalr ) ÚNÚindr; Útgr$ Ú_r% r r- r r r Útest_apply_iteration< s
|
|
r@ Úfuncc C s t | d¡S )Nr6 )r" ZmakeCustomIndex)Úmr r r r V r r c C sh d}| |}t |j}tdtj |¡i|d}d| d}tjt|d | t
|
dd¡ W 5 Q R X d S ) Nr6 Úar zVOnly valid with DatetimeIndex, TimedeltaIndex or PeriodIndex, but got an instance of 'ú')Úmatchr3 r: ) ÚtypeÚ__name__r r) ÚrandomÚrandnÚpytestZraisesÚ TypeErrorr r )rA Únr Únamer; Úmsgr r r Ú test_fails_on_no_datetime_indexP s
|
|
|
ÿrO c C s. d} t j | d¡}t|ddddgd}tdd d tdd d
|
tdd d tdd dtdd d gd |d < | td dd¡}t | tdd d ¡|d d d
¡ t | tdd d
|
¡|d d d
¡ t | tdd d ¡|d
|
d d
¡ t | tdd d¡|d d d
¡ t | tdd d ¡|dd d
¡ d S )Né é r ÚBÚCr3 ©ÚcolumnséÝ r5 r6 r( é Úkey©rX r )
|
r) rH rI r r r r r" Úassert_frame_equalZ get_group)rL Údatar; r% r r r Útest_aaa_group_orderg s"
|
|
|
|
|
ûú$$$$r\ c C sà t j dd¡}t|ddddgd}dd d
|
dd gd |d < t|ddddgd}td ddtd dd td dd
|
td ddtd dd gd |d < | d ¡}| td dd¡}t|| }t|| }tddd d d|_ t
|
||¡ dS ©z?Check TimeGrouper's aggregation is identical as normal groupby.rP rQ r rR rS r3 rT r5 r6 r( rW rX rV rY ú
|
2013-01-01©r4 r r rM N) r) rH rI r r r r Úgetattrr r r" Z assert_equal)Zresample_methodr[ Ú normal_dfÚdt_dfÚnormal_groupedÚ
|
dt_groupedr& Ú dt_resultr r r Útest_aggregate_normal} s$
|
|
|
|
|
ûú
|
rf z7if TimeGrouper is used included, 'nth' doesn't work yet)Úreasonc C sÜ t j dd¡} t| ddddgd}dd d
|
dd gd |d < t| ddddgd}td ddtd dd td dd
|
td ddtd dd gd |d < | d ¡}| td dd¡}| d
|
¡}tddd d d|_ | d
|
¡}t
|
||¡ dS r] ) r) rH rI r r r r Znthr r r" rZ )r[ ra rb rc rd r& re r r r Útest_aggregate_nth s$
|
|
|
|
|
ûú
|
|
|
rh zmethod, method_args, unitÚsumZ min_countr5 r/ c C sj t dgd tjgd tdddd}t| f|| d¡}t d|gtjd d
|
gd d d}t ||¡ d S ) Nr r6 Ú2017rQ r
|
r Z2dg z
|
2017-01-01z
|
2017-01-03Z2Dr: )
|
r r) r* r r r, ÚpdÚ DatetimeIndexr" r# )ÚmethodÚ method_argsÚunitÚsr- r& r r r Ú!test_resample_entirely_nat_window° s & ÿrq zfunc, fill_valueÚminÚmax)ri r )r/ r5 )r+ r c C s8 d}t j |d¡ d¡}t|ddddgd}d d
|
t jdd gd |d < t|ddddgd}td d d td d d
|
tjtd d dtd d d gd |d < | d ¡}| t
|
d dd¡}t || }t || } t|gd gdgddddgd}
|
t ||
|
g¡} | ¡ } tddd d d} | d ¡| _t | | ¡ | jjd ks4td S )NrP rQ Úint64r rR rS r3 rT r5 r6 rW rX rV rY r( )r rU r^ r_ )r) rH rI Úastyper r* r rk ÚNaTr r r` ÚconcatÚ
|
sort_indexr Ú
|
_with_freqr r" rZ rM ÚAssertionError) rA Z
|
fill_valuerL r[ ra rb rc rd Ú normal_resultre Úpadr& Zdtir r r Útest_aggregate_with_natÄ s0
|
|
|
|
ûú
|
r} c
|
C s d} t j | d¡ d¡}t|ddddgd}d d
|
t jdd gd |d < t|ddddgd}td d d td d d
|
tjtd d dtd d d gd |d < | d ¡}| t
|
d dd¡}| ¡ }| ¡ }t dgdgd}t ||g¡} | ¡ } tddd d d d ¡| _t | |¡ |jjd kstd S )NrP rQ rt r rR rS r3 rT r5 r6 rW rX rV rY r r( r r^ r_ )r) rH rI ru r r* r rk rv r r Úsizer rw rx r ry r r" r# rM rz )
|
rL r[ ra rb rc rd r{ re r| r& r r r Ú test_aggregate_with_nat_sizeê s<
|
|
|
|
ûú
|
ÿþ r c C sF t tddd} d}| |ks tt tdddd} d}| |ksBtd S )Nr ÚHrY zTimeGrouper(key='A', freq=<Hour>, axis=0, sort=True, dropna=True, closed='left', label='left', how='mean', convention='e', origin='start_day')r2 )rX r Úoriginz£TimeGrouper(key='A', freq=<Hour>, axis=0, sort=True, dropna=True, closed='left', label='left', how='mean', convention='e', origin=Timestamp('2000-01-01 00:00:00')))Úreprr rz r0 r r r Ú test_repr
|
s ÿ ÿr z$method, method_args, expected_valuesr6 c C s` t dtddddd}| d¡}tjdd d
|
gdd }t| f||}t ||d}t ||¡ d S ) Nr5 rj r6 r ©r r r Z30Tz2017-01-01T00:00:00z2017-01-01T00:30:00z2017-01-01T01:00:00r: )r r r, rk rl r r" r# )rm rn Zexpected_valuesrp Z resampledr r- r& r r r Útest_upsample_sum s
|
þ r
c C sP dddgdddgd} t | }tddd d
|
|d < | d ¡ d ¡ d ¡jdd}tjjdt dfdt dfdt dfdt dfdt dfdt dfdt dfdt dfdt dfdt dfdt dfdt dfdt d fdt d fdt d fdt dfgd d gd }t d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/gd0gd1 dg d|d2}t
|
||¡ d S )3Né
|
é é é2 é< )ZpriceÚvolumez
|
01/01/2018r( ÚWr Z week_startingr Z1DZlinear)rm z
|
2018-01-07z
|
2018-01-08z
|
2018-01-09z
|
2018-01-10z
|
2018-01-11z
|
2018-01-12z
|
2018-01-13z
|
2018-01-14z
|
2018-01-15z
|
2018-01-16z
|
2018-01-17z
|
2018-01-18z
|
2018-01-19z
|
2018-01-20z
|
2018-01-21)Únamesg $@g·mÛ¶mÛ#@gnÛ¶mÛ¶#@g%I$I#@gÛ¶mÛ¶m#@g$I$I#@gI$I$#@g #@g·mÛ¶mÛ"@gnÛ¶mÛ¶"@g%I$I"@gÛ¶mÛ¶m"@g$I$I"@gI$I$"@g "@g &@g I@é )r[ r ) r r Z set_indexr r, Z interpolaterk Z
|
MultiIndexÚ from_tuplesr r" rZ )Údr; r- Z expected_indr& r r r Ú!test_groupby_resample_interpolate6 sl
|
ÿþýÿ
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
ðíðíêr )*r Úoperatorr Únumpyr) rJ Zpandasrk r r r Zpandas._testingZ_testingr" Zpandas.core.groupby.grouperr Z pandas.core.indexes.datetimesr rH rI r r' r. r1 r@ ÚmarkZ parametrizeZ makeIntIndexZmakeStringIndexZmakeFloatIndexrO r\ rf Zxfailrh r* rq r} r r r
r r r r r Ú<module> st üþ
|
|
úþ
|
þ
|
" øþ
|
|
