U
|
ý°d-N ã @ s d dl mZ d dlZd dlZd dlmZ d dlmZ d dl Z
|
d dl m Z m Z m Z mZmZ d dlmZ d dlmZ e dgd d gd
|
d gd e d ¡deddd ddZe dd Zdd Zdd Zdd Zdd Zd d Z d d Z ej d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0g¡d1d2 Z d3d4 Z!ej d!d5d6g¡d7d8 Z"d9d: Z#d;d< Z$d=d> Z%d?d@ Z&dAdB Z'dCdD Z(dEdF Z)dGdH Z*dIdJ Z+ej dKdLgdLdMgg¡dNdO Z,ej dPdQdRg¡dSdT Z-dUdV Z.ej dKdLgdLdMgg¡dWdX Z/dYdZ Z0d[d\ Z1d]d^ Z2d_d` Z3dadb Z4dS )cé )ÚdedentN)Úis_platform_windows)Ú
|
async_mark)Ú DataFrameÚIndexÚSeriesÚTimedeltaIndexÚ Timestamp)Ú
|
date_rangeé é é é é é é( ©ÚAÚBú1/1/2000Ús©ÚfreqÚperiods©Úindexc
|
à sd ddl m} td}| |¡I d H t d ¡, |d t| j dd¡ W 5 Q R X W 5 Q R X d S )Nr )ÚprovisionalcompleterzZ import pandas._testing as tm
|
s = tm.makeTimeSeries()
|
rs = s.resample("D")
|
Úignorezrs.r ) ZIPython.core.completerr r Zrun_codeÚtmZassert_produces_warningÚlistZ CompleterZ completions)Úipr Úcode© r" úcd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/resample/test_resampler_grouper.pyÚ"test_tab_complete_ipython6_warning s ÿ
|
r$ c C s dddgdddgdddgddd gd
|
ddgdd d gd d dgdd dgdd d gdd dgg
|
} t | dddgd}t |j¡|_dd }| d¡ |¡}| d¡ d¡ d¡ ¡ }t
|
||¡ t t dddd ddddgd d d dgd d¡}d d }| d!¡ |¡}| d!¡ d"¡ ¡ }t
|
||¡ d S )#Nz
|
2010-01-01r r z
|
2010-01-02r z
|
2010-01-05r z
|
2010-01-10é z
|
2010-01-13r é z
|
2010-01-03z
|
2010-01-04r z
|
2010-01-11z
|
2010-01-14ÚdateÚidZscore©Úcolumnsc S s | d¡ d¡ ¡ S )Nr' ÚD)Ú set_indexÚresampleÚasfreq©Úxr" r" r# Úf_0A s z'test_deferred_with_groupby.<locals>.f_0r+ ú
|
2016-01-01é ÚW)Ústartr r é ©r' ÚgroupÚvalc S s | d¡ ¡ S )NÚ1D)r- Úffillr/ r" r" r# Úf_1P s z'test_deferred_with_groupby.<locals>.f_1r8 r: ) r ÚpdÚ to_datetimer' ÚgroupbyÚapplyr, r- r. r Úassert_frame_equalr
|
r; )ÚdataÚdfr1 ÚexpectedÚresultr< r" r" r# Útest_deferred_with_groupby. s: ö
|
|
ýÿúrF c C sr t d¡} | j dd ¡}| d¡j ¡ }t ||¡ | j d¡ ¡ }t ||¡ | d¡ ¡ j}t ||¡ d S )Nr c S s | d¡ ¡ S ©NÚ2s)r- Úmeanr/ r" r" r# Ú<lambda>[ ó z test_getitem.<locals>.<lambda>rH )Ú
|
test_framer? r r@ r- rI r Úassert_series_equal)ÚgrD rE r" r" r# Ú test_getitemX s
|
rO c C s¢ ddddddg} t | tdddd}| d ¡ d
|
¡}|d ¡ }tddgtjjdt dfdt d fgd d gd d d}t
|
||¡ |d ¡ }t
|
||¡ d S )Nr r )r( Úbuyerr r r2 ©r r r( r: rP z
|
2016-01-02©Únames)r Úname) r r
|
r? r- Úcountr r= Ú
|
MultiIndexÚ from_tuplesr r rM )rB rC ÚrrE rD r" r" r# Útest_getitem_multipleg s þú rY c C sf t tdtddddd} | d¡ d¡ d ¡d
|
¡ }| d¡jd dd d
|
¡ }t ||¡ d S ) NZaabbbz1-1-2016r& rQ r )r( r' rB r' r( Ú2DrB ©Úon) r r r
|
r, r? r- Úsumr rM )rC ÚexprE r" r" r# Ú)test_groupby_resample_on_api_with_getitem| s ÿ r_ c C s^ d} d\}}d}t ||dd}ttj t|¡|d}|||
}tj| d}| |¡ d¡}|||
}| |¡ d¡} t
|
j t dd t |j| j¡ W 5 Q R X td
|
}
|
tj| |
|
d } | | ¡ d¡} | ||
} | | ¡ d¡} t | | ¡ td
|
t d¡d }tj| |d }| |¡ d¡}|||
}| |¡ d¡}t ||¡ t | |¡ d S ) NZ1399min)z1/1/2000 00:00:00z1/31/2000 00:00z1/15/2000 00:00:00Z1231min©r r rU zIndex are different©Úmatchr )r Úorigini0u )r
|
r ÚnpÚrandomÚrandnÚlenr= ÚGrouperr? ÚaggÚpytestÚraisesÚAssertionErrorr Úassert_index_equalr r rM Z Timedelta)r r5 ÚendZmiddleÚrngÚtsZts2Zsimple_grouperZcount_tsZ count_ts2rc Zadjusted_grouperZadjusted_count_tsZadjusted_count_ts2Z origin_futureZadjusted_grouper2Zadjusted2_count_tsZadjusted2_count_ts2r" r" r# Útest_groupby_with_origin s2 rq c
|
C sl t dddd} ttd| d d¡ ¡ }tdddddd d gtjd
|
d d d dddgdddd}t ||¡ d S )Nr r ÚT©r r r Z20sr r r z2000-01-01 00:00:00z2000-01-01 00:00:20z2000-01-01 00:00:40z2000-01-01 00:01:00z2000-01-01 00:01:20z2000-01-01 00:01:40z2000-01-01 00:02:00zdatetime64[ns]Z20S)Údtyper ) r
|
r Úranger- Únearestr= Z DatetimeIndexr rM )r rE rD r" r" r# Ú test_nearest¯ s$ ù õþrw ÚfÚfirstÚlastÚmedianZsemr] rI ÚminÚmaxÚsizerU rv Zbfillr; r. Zohlcc sB t d¡}| d¡}t| }| fdd¡}t ||¡ d S )Nr rH c s t | d¡ S rG ©Úgetattrr- r/ ©rx r" r# rJ á rK z test_methods.<locals>.<lambda>)rL r? r- r r@ r Z assert_equal©rx rN rX rE rD r" r r# Ú test_methodsÈ s
|
|
|
r c C s> t d¡} | d¡}|j ¡ }| j dd ¡}t ||¡ d S )Nr rH c S s | d¡ ¡ S rG )r- Únuniquer/ r" r" r# rJ ê rK z&test_methods_nunique.<locals>.<lambda>)rL r? r- r r r@ r rM )rN rX rE rD r" r" r# Útest_methods_nuniqueå s
|
|
|
|
r
ZstdÚvarc sF t d¡}| d¡}t| dd}| fdd¡}t ||¡ d S )Nr rH r ©Zddofc s t | d¡ ddS )NrH r r r r/ r r" r# rJ ó rK z&test_methods_std_var.<locals>.<lambda>)rL r? r- r r@ r rA r r" r r# Útest_methods_std_varî s
|
|
|
r c C sl t d¡} | d¡}| d¡ ¡ }dd }| |¡}t ||¡ dd }| |¡}| d¡}t ||¡ d S )Nr rH c S s | d¡ ¡ S rG )r- r] r/ r" r" r# r1 þ s ztest_apply.<locals>.f_0c S s | d¡ dd ¡S )NrH c S s | ¡ S ©N©r] )Úyr" r" r# rJ rK z)test_apply.<locals>.f_1.<locals>.<lambda>)r- r@ r/ r" r" r# r< s ztest_apply.<locals>.f_1Úint64)rL r? r- r] r@ r rA Zastype)rN rX rD r1 rE r< r" r" r# Ú
|
test_apply÷ s
|
|
|
|
|
r c C sª t dddd} tdtj t| ¡i| d}dd }| tjd d¡ |¡}|
|
d ¡ |¡}t ||¡ |d jtjd dd
|
d |¡}|d
|
d ¡ |¡}t ||¡ d S ) Nz1-1-2015z12-31-15r+ r` Úcol1©rB r c S s t ddgddgd}|S )Nr r ÚaÚbr )r )r0 r r" r" r# rx s z(test_apply_with_mutated_index.<locals>.fÚMF)Z
|
group_keys)r
|
r rd re Zrandrg r? r= rh r@ r- r rA rM )r rC rx rD rE r" r" r# Ú test_apply_with_mutated_index s r c s¦ t j ddg¡} tdddd}tt dgd ¡ dd ¡|| d
|
}d d |jD | d ¡
|
fdd¡}td ddgg tddd dt j ddg¡d
|
}t ||¡ d S )N)r r Ú Úone)r r ÚiZtwoz
|
2017-01-01Z15Minr )r5 r r r é r ©r r* c S s& i | ] }||d dkr t jnt jqS )r r )rd r] rI )Ú.0Úcolr" r" r# Ú
|
<dictcomp>& s z1test_apply_columns_multilevel.<locals>.<dictcomp>ÚHc s | j | S r ©rT r/ ©Zagg_dictr" r# rJ ' rK z/test_apply_columns_multilevel.<locals>.<lambda>g Z1H) r= rV rW r
|
r rd ÚarrayZreshaper* r- r@ r rA )ÚcolsÚindrC rE rD r" r r# Ú test_apply_columns_multilevel! s " ÿýr¢ c C s t dddd} t| dddddgdddddgdd }| d
|
¡jd d d ¡ }tjddddgtjdt t ddddg¡¡g}tj
|
j |d
|
d gd}tddddgddddgd|d}t ||¡ d S )Nz
|
2000-01-01rZ r& r r r r )Úcol0r )r rB r£ Z1WÚleft)Úlabel©rt z
|
1999-12-26z
|
2000-01-02rR r )r
|
r r? r- r] rd r r r= r> rV Ú from_arraysr rA )r rC rE ÚmiZmindexrD r" r" r# Ú test_resample_groupby_with_label2 s &ÿþ ÿr© c C s t } tdddgdd}| d¡ d¡ ¡ }|jjdks:tt |jj
|
d |¡ | d¡ d¡ ¡ }|jjdksrtt |jj
|
d |¡ d S ) Nr r r r r rH r r ) rL r r? r- rI r Znlevelsrl r rm ÚlevelsZrolling)rC rD rE r" r" r# Ú test_consistency_with_windowF s r« c C sn t tj dd¡tdtddddd} | ¡ }dd d
|
g|_| d ¡ ¡ }| d ¡ ¡ }|j|_t
|
||¡ d S ) Nr r Zaaaú
|
2012-01-01r rs ©r* r r r ÚcZ5s) r rd re rf r r
|
Úcopyr* r- r{ r rA )rC Údf2rD rE r" r" r# Ú test_median_duplicate_columnsS s ý r± c C s t tdtdddtddddd} | d¡ d d
|
¡}td d d dgtddddd}t ||¡ | d¡ dd
|
¡}t ||¡ d S )Né
|
r )r r r¬ Z20minrs r r c S s
|
| j ¡ S r )r r] ©r8 r" r" r# rJ k rK z0test_apply_to_one_column_of_df.<locals>.<lambda>r r é é r3 c S s | d ¡ S )Nr r r³ r" r" r# rJ r rK )r ru r
|
r- r@ r r rM ©rC rE rD r" r" r# Ú test_apply_to_one_column_of_dfc s þ
|
ÿ r· c C s t ddddddddgddddddd d
|
gd d d dddd dgd} t | d ¡| d< | d¡jddd}|dg ¡ }| ddi¡}t ||¡ d S )NZcat_1Zcat_2r& r é r r3 é r² é2 z2019-2-1z
|
2018-02-03z 2020-3-11z2019-2-2z 2018-12-4z
|
2020-12-12)ÚcatÚnumr' r' r» ÚYr[ r¼ r] ) r r= r> r? r- r] ri r rA )rC Ú resampledrD rE r" r" r# Útest_resample_groupby_aggv s6 ø
|
øôÿr¿ c C s t d} tdgdgdt| gddd}| d¡ d ¡d
|
}| d d g¡}tdd ggtjjd| fgddgdd d gd}t
|
||¡ d S )Nz2021-02-28 00:00:00ÚbetaéE )ÚclassÚvaluer' r r r r rà r] r~ r rR r ) r r r r? r- ri r= rV rW r rA )rp rC r¾ rE rD r" r" r# Ú"test_resample_groupby_agg_listlike s ýrÄ Úkeysr r c C s t g ddgtg d}| | ¡jt d¡d ¡ }t ddgdj| ddjtg d d
|
}t| d krp| d |j _
|
t ||¡ d S ) Nr r r ú00:00:01©Zruler) F©ÚdropT)Úappendr r ) r r r? r- r= Ú to_timedeltarI r, rg r rT r rA ©rÅ rC rE rD r" r" r# Ú
|
test_empty¨ s ÿ þÿ rÍ Ú consolidateTFc C sø t dddd}td|tddd}td|tdd }tj||gd
|
d }| rV| ¡ }| d g¡jd dd ¡ }tj j
|
dgd dgd t dddgd ¡gd dgd}tdgd dgd dddgd dgd t j gd d|d}t ||¡ d S )Nz
|
2020-01-01é r+ rs r r9 )Úkeyr' r Ú
|
col_objectr )rÐ r' r T)Z ignore_indexrÐ r4 r' r[ r z
|
2020-01-05z
|
2020-01-12z
|
2020-01-19r rR r r& r )rÐ r rÑ r )r
|
r ru r= ÚconcatZ _consolidater? r- r| rV r§ r> rd Únanr rA )rÎ ÚdatesZdf1r° rC rE ÚidxrD r" r" r# Ú.test_resample_groupby_agg_object_dtype_all_nan¸ s* þû ýúrÖ c C s° t tdddddddddddgddddd d
|
ddgd d } | d ¡jddddg ¡ }t dddddgitdtdfdtdfdtdfdtdfgddd}t ||¡ d S ) Nr2 r ©r5 r r r r% r& r r r² r7 )rB r8 rZ r' r[ r9 ç @g @g @g @ú
|
2016-01-03ú
|
2016-01-05ú
|
2016-01-07©r8 r' ©rB rT r ) r r
|
r? r- rI r r r rA r¶ r" r" r# Ú'test_groupby_resample_with_list_of_keysÖ s*
|
ýÿ
|
ÿ
|
|
|
|
üùürÞ c C s® t g dddgd}t |d ¡|d< | d¡}| | ¡jt d¡d ¡ }t dddgdj| dd}t |d ¡|d< |jdd d d
|
}t| d kr| d |j _
|
t ||¡ d S ) Nr r r' r) rÆ rÇ FrÈ T)rÊ rÉ r r ) r r= r> r, r? r- rË rI rg r rT r rA rÌ r" r" r# Útest_resample_no_indexñ s
|
rß c C s¬ t tt dddg¡ddd} | dddg¡jt d ¡d
|
¡ }t dd ddg¡}t tjt j
|
ddgt j d |gddddgddd dggd dgdd}t j ||t d d S )Nz2018-01-01 00:00:00z2018-01-01 12:00:00z2018-01-02 00:00:00r' r r r r z06:00:00rÇ z2018-01-01 06:00:00r¦ r r )rª ÚcodesrS )Zcheck_index_type)r r r= r> r? r- rË rI rV rd r Zintpr rA r )rC rE r rD r" r" r# Útest_resample_no_columns s0 ÿüÿ"üÿýÿ rá c C s® t dgd dgd dgd dgd t d¡dtddddd } | d
|
¡ d ¡ ¡ }tdtj j
|
dt d fdt d fdt d fdt d fgd
|
d gdd }t ||¡ d S )Nr r r r r z31/12/2000 18:00r r r r r+ z
|
2000-12-31z
|
2001-01-01rR )r rd Úaranger
|
r? r- r~ r r= rV rW r r rM r¶ r" r" r# Ú)test_groupby_resample_size_all_index_same s" 2 þ
|
|
|
|
üùþ rã c
|
C s´ t ddddddddgddddddddgd ttd
|
d d d dd} | d¡ d¡dg ¡ }t dddddgitdtd
|
fdtdfdtdfdtdfgddd}t ||¡ d S ) Nr r r r3 r& rµ r r6 ©r8 r9 r2 r r× r' r r r8 rZ r9 ç @ç @ç @rØ rÙ rÚ rÛ rÜ rÝ ©
|
r r r
|
r? r- rI r r r rA r¶ r" r" r# Ú0test_groupby_resample_on_index_with_list_of_keys6 s0 þ
|
þû
|
|
ÿ
|
|
|
|
üùüré c C sä t ddddddddgddddddddgdd dd
|
dd
|
dd
|
gddddddddgd ttd d
|
d ddd} | d¡ d¡ddg ¡ }t ddddgddddgdtdtd fdtd fdtd fdtd fgd d d}t ||¡ d S )!Nr r r r3 r& rµ r r6 r% r )r8 Ú first_valÚ
|
second_valZ third_valr2 r× r' r r r8 rZ rê rë rå ræ rç rØ g @g @)rê rë rÙ rÚ rÛ rÜ rÝ rè r¶ r" r" r# Ú>test_groupby_resample_on_index_with_list_of_keys_multi_columnsT s4 ü
|
þù
|
|
þ
|
|
|
|
üùûrì c
|
C sz t ddddddddgddddddddgd ttd
|
d d d dd} tjtdd | d¡ d¡dg ¡ W 5 Q R X d S )Nr r r r3 r& rµ r r6 rä r2 r r× r' r r zColumns not foundra r8 rZ Zval_not_in_dataframe) r r r
|
rj rk ÚKeyErrorr? r- rI )rC r" r" r# Ú?test_groupby_resample_on_index_with_list_of_keys_missing_columnu s þ
|
þû
|
rî )5Útextwrapr Únumpyrd rj Z pandas.compatr Z pandas.util._test_decoratorsr Zpandasr= r r r r r Zpandas._testingZ_testingr Z pandas.core.indexes.datetimesr
|
râ rL r$ rF rO rY r_ rq rw ÚmarkZ parametrizer r
r r r r¢ r© r« r± r· r¿ rÄ rÍ rÖ rÞ rß rá rã ré rì rî r" r" r" r# Ú<module> s ( þ
|
*
|
)ñþ
|
|
$
|
|
|
!
|
