U
|
®ý°dÌ ã @ s¢ d dl Zd dlZd dlmZmZmZmZmZm Z m
|
Z
|
d dl m Z d dlmZ d dlmZ ejdd Zejdd ZG d d
|
d
|
ZG d d d ZG d d dZdS )é N)Ú DataFrameÚIndexÚ
|
MultiIndexÚSeriesÚ TimestampÚ
|
date_rangeÚ to_datetime)Ú BaseIndexer)Ú get_groupbyc C sP t ddddddddddg
|
ddddddddddg
|
tdddd d
|
d d d d dg
|
dS )z0Frame for testing times argument in EWM groupby.ÚaÚbÚcr é é é ú
|
2020-01-01ú
|
2020-01-02ú
|
2020-01-10z
|
2020-01-22ú
|
2020-01-03z
|
2020-01-23ú
|
2020-01-04©ÚAÚBÚC)r r © r r úWd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/window/test_groupby.pyÚ times_frame s$ öÿýÿr c C s. t dgd dgd dgd t d¡dS © Nr é r é r é é( ©r r ©r ÚnpÚaranger r r r Ú
|
roll_frame+ s r& c @ s e Zd Zdd Zdd Zdd Zej ddd d
|
d d d dg¡dd Z ej dddg¡dd Z
|
ej ddddddg¡dd Z ej d d d gd d!gg¡d"d# Z ej dd d g¡d$d% Z ej dd d g¡d&d' Zej d(d d)d)gd)d*ggfd d)d!gd!d)ggfg¡d+d, Zd-d. Zd/d0 Zej d1d)d2gd3d4gg¡d5d6 Zd7d8 Zd9d: Zej d;d<d=d>g¡d?d@ ZdAdB ZdCdD ZdEdF ZdGdH Zej dId d
|
g¡dJdK ZdLdM ZdNdO Z dPdQ Z dRdS Z ej dTdUdVd dWfdXi fg¡dYdZ Z ej d[d\d]id]fd\d^id_fg¡d`da Z ej dbd2d4g¡dcdd Z!dedf Z"dgdh Z#didj Z$dkdl Z%dmdn Z&ej dodVdpdqd^d3d)d)d^d3d)d)gfdVdVdqd^d^d)d)d^d^d)d)gfd=d=dqd^d^d^d^d^d^d^d^gfd=d=drd^d^d^dsd^d^d^dsgfg¡dtdu Z'ej dve( )dwdxg¡dydzgg¡d{d| Z*d}d~ Z+ej ddgdddddgigdd_ge,d
e,de,d
e,dgddddgdgg¡dd Z-dd Z.dd Z/dd Z0dd Z1dd Z2d^S )Ú TestRollingc C s0 d}t jt|d |jddd W 5 Q R X d S )Nz4groupby\(\) got an unexpected keyword argument 'foo'©Úmatchr r )Úfoo)ÚpytestÚraisesÚ TypeErrorÚgroupby)Úselfr& Úmsgr r r Ú!test_groupby_unsupported_argument1 s z-TestRolling.test_groupby_unsupported_argumentc C s¢ | d¡}t|dd}|j dd ¡}| d¡ ¡ j}t ||¡ | d¡j ¡ }t ||¡ |j d¡ ¡ }t ||¡ |j |j¡ d¡ ¡ }t ||¡ d S )Nr ©Úbyc S s | d¡ ¡ S )Nr ©ÚrollingÚmean©Úxr r r Ú<lambda>: ó z*TestRolling.test_getitem.<locals>.<lambda>r ) r. r
|
r Úapplyr5 r6 ÚtmÚassert_series_equalr )r/ r& ÚgÚ g_mutatedÚexpectedÚresultr r r Ú test_getitem6 s
|
zTestRolling.test_getitemc C sd | d¡}|jddd}t|dd}|j dd ¡}|j ¡ }t ||¡ |j ¡ }t ||¡ d S )Nr r r ©Ú min_periodsr2 c S s | j ddd ¡ S )Nr r rC )r5 Úcountr7 r r r r9 M r: z3TestRolling.test_getitem_multiple.<locals>.<lambda>)r. r5 r
|
r r; rE r< r= )r/ r& r> Úrr? r@ rA r r r Útest_getitem_multipleH s
|
|
|
z!TestRolling.test_getitem_multipleÚfÚsumr6 ÚminÚmaxrE ÚkurtÚskewc sr |j ddd}|jdd}t| }| fdd¡}|jddd }t |d td
|
g¡}||_t
|
||¡ d S ) Nr F©Ú
|
group_keysé ©Úwindowc s t | d¡ S ©NrP ©Úgetattrr5 r7 ©rH r r r9 f r: z*TestRolling.test_rolling.<locals>.<lambda>r ©Zaxisr! © r. r5 rU r; Údropr Ú from_arraysÚrangeÚindexr< Úassert_frame_equal©r/ rH r& r> rF rA r@ Úexpected_indexr rV r Ú test_rollingU s zTestRolling.test_rollingÚstdÚvarc sv |j ddd}|jdd}t| dd}| fdd ¡}|jddd
|
}t |d td g¡}||_t
|
||¡ d S ) Nr FrN rP rQ r ©Zddofc s t | d¡ ddS )NrP r rc rT r7 rV r r r9 t r: z/TestRolling.test_rolling_ddof.<locals>.<lambda>rW r! rX r^ r rV r Útest_rolling_ddofn s z TestRolling.test_rolling_ddofÚ interpolationÚlinearÚlowerÚhigherÚmidpointÚnearestc st |j ddd}|jdd}|jd d}| fdd ¡}|jdd
|
d }t |d td g¡}||_t
|
||¡ d S ) Nr FrN rP rQ çÙ?©re c s | d¡jd dS )NrP rk rl )r5 Úquantiler7 rl r r r9
r: z3TestRolling.test_rolling_quantile.<locals>.<lambda>r rW r! ) r. r5 rm r; rY r rZ r[ r\ r< r] )r/ re r& r> rF rA r@ r_ r rl r Útest_rolling_quantile| s
|
ÿz!TestRolling.test_rolling_quantilezf, expected_valÚcorrr Úcovç à?c
|
C sô t tddgd dgd dddddgd d dd g¡}t tddddddgd
|
d ¡}t|jd d d¡||}tjg|gd d }t |d gtj dgd dgd dgd dgd t
|
tddd gddd gdd}t ||¡ d S )Né
|
r é r r rP )ÚvalueÚidx1Úidx2ru rv )rt rv r ©Úlevelrt é ©Únames©Úcolumnsr\ ) r r[ Ú set_indexrU r. r5 r$ Únanr rZ Úlistr< r] )r/ rH Z expected_valÚdfÚotherrA Ú expected_datar@ r r r Ú/test_rolling_corr_cov_other_same_size_as_groups s, ,ÿþ ÿ ýúý z;TestRolling.test_rolling_corr_cov_other_same_size_as_groupsc sV d¡}|jdd}t| } fdd}| |¡}tj|d< t ||¡ d S )Nr rP rQ c s t | d¡ S rS rT r7 ©rH r& r r Úfunc® s zITestRolling.test_rolling_corr_cov_other_diff_size_as_groups.<locals>.func)r. r5 rU r; r$ r r< r] ©r/ rH r& r> rF rA r r@ r r
r Ú/test_rolling_corr_cov_other_diff_size_as_groups§ s
|
|
|
z;TestRolling.test_rolling_corr_cov_other_diff_size_as_groupsc sN | d¡}|jdd}t|j dd} fdd}| |¡}t ||¡ d S )Nr rP rQ T©Úpairwisec s t | j d¡ ddS )NrP Tr )rU r r5 r7 rV r r r ¾ s z8TestRolling.test_rolling_corr_cov_pairwise.<locals>.func)r. r5 rU r r; r< r= r r rV r Ú test_rolling_corr_cov_pairwise· s
|
|
z*TestRolling.test_rolling_corr_cov_pairwisezfunc, expected_valuesç ð?ç @c C s¾ t ddddgddddgddddgd}| d ¡ d
|
¡}t|| }t dtjg |d dtjg dtjg |d dtjg d tjd d ddddddgd d d gdd}t ||¡ d S )NÚg1Úg2r r r ry rP ©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 rz ©r\ )
|
r r. r5 rU r$ r r Ú from_tuplesr< r] )r/ r Zexpected_valuesr ZrolrA r@ r r r Ú test_rolling_corr_cov_unorderedÄ s4
|
|
|
ýÿ þø
|
õûz+TestRolling.test_rolling_corr_cov_unorderedc sx |j ddd}|jdd}|jdd d}| fd d¡}|jdd
|
d }t |d td g¡}||_t ||¡ d S ) Nr FrN rP rQ c S s | ¡ S ©N©rI r7 r r r r9 í r: z0TestRolling.test_rolling_apply.<locals>.<lambda>©Úrawc s | d¡jdd dS )NrP c S s | ¡ S r r ©Úyr r r r9 î r: zBTestRolling.test_rolling_apply.<locals>.<lambda>.<locals>.<lambda>r )r5 r; r7 r r r r9 î r: r rW r! )
|
r. r5 r; rY r rZ r[ r\ r< r] )r/ r r& r> rF rA r@ r_ r r r Útest_rolling_applyè s z TestRolling.test_rolling_applyc C s® t dgd dgd dgd d}| d¡}t dd d
|
d d d g¡}dd g|_t tjddgd dg|d}|jdd ¡ }t
|
||¡ | ¡ |jdd ¡ }t
|
||¡ d S )Nr* r Úbarr ry r" r )r r )r rP )r rs )r* r )r* r )r* r ç @r r r| rQ ) r r. r r r{ r$ r r5 rI r< r] )r/ r r> Úmir@ rA r r r Ú test_rolling_apply_mutabilityö s $
|
ÿ
|
z)TestRolling.test_rolling_apply_mutabilityzexpected_value,raw_valueTç Fc C sp dd }t dddgdddgd}| d¡j d¡j||d}t|gd tjd dd gd
|
d d }t ||¡ d S ) Nc S s t t| tjS r )ÚintÚ
|
isinstancer$ Zndarrayr7 r r r Ú isnumpyarray s z6TestRolling.test_groupby_rolling.<locals>.isnumpyarrayr r r )Úidrt r£ r )©r r ©r r ©r r rz rt ©r\ Úname)
|
r r. rt r5 r; r r r r< r= )r/ Zexpected_valueZ raw_valuer¢ r rA r@ r r r Útest_groupby_rolling s ýz TestRolling.test_groupby_rollingc C s t tdd}| |¡jddd ¡ }t tjgd t d¡d}t
|
||¡ t tdd}| |¡jddd ¡ }t tjgd t d
|
¡d}t
|
||¡ t d gd d gd td d}| d ¡jddd ¡ }t tjdddtjtjddddtjg tjdd d gdd gd}t ||¡ t d gd d gd tdd}| d ¡jddd ¡ }t tjdddtjtjdddtjg
|
tjdd d gdd gd}t ||¡ d S )Nr ry Tr )ÚcenterrR rs )r¤ ©r r ©r r ©rP r )rs rP r rP )r¤ r« r¬ r r r é ©r r r é r é ) ©r r ©r r ©r r ©r r ©r rP ©r rs ©r ry ©r r° ©r r ©r r± )r rr rz )r\ r} rr )
|
r² r³ r´ rµ r¶ r· r¸ r¹ rº r» ) r r[ r. r5 r6 r$ r r r r< r= r r] )r/ ZseriesrA r@ r r r r Ú"test_groupby_rolling_center_center sH
|
þ
|
þ " òî " óïz.TestRolling.test_groupby_rolling_center_centerc C sî t tdddgd dgd tddd }| d
|
¡jdd d d dj ¡ }tddddddddddg
|
dtj dt
|
dfdt
|
dfdt
|
dfdt
|
dfdt
|
dfdt
|
d fdt
|
d fdt
|
d fdt
|
d fdt
|
dff
|
d
|
d gd d!}t ||¡ d S )"Nr r Úgroup_1ry Úgroup_2rP rr )ÚDateÚgbrt )ÚdatarÀ r¿ Tr )Úonrª rD r ç ø?r g @ç @g @ç @ç @rt r r r z
|
2020-01-05z
|
2020-01-06z
|
2020-01-07z
|
2020-01-08z
|
2020-01-09rz ©r¨ r\ ) r r r[ r. r5 rt r6 r r r r r< r= ©r/ r rA r@ r r r Ú test_groupby_rolling_center_on^ s@ ýÿ
|
ÿ
|
ÿ
|
|
|
|
|
|
|
|
|
|
ö óýz*TestRolling.test_groupby_rolling_center_onrD rs rP r c
|
C sò t dgd dgd tdd}d}| d¡j|d|d ¡ }| ¡ dd
|
g }d d d dddddddg
|
}dd |D }td|d }tjg| }|||d|
| } |||d|
| }
|
t dgd dgd | |
|
d} t
|
|| ¡ d S )Nr rr r r )ÚgrouprÁ rs rÊ T)rª rD rÁ r rà r rÄ r ç @g @rÅ rÆ ç @c S s g | ] }|d qS )g $@r ©Ú.0r8 r r r Ú
|
<listcomp> s zGTestRolling.test_groupby_rolling_center_min_periods.<locals>.<listcomp>r r ) r r[ r. r5 r6 Z reset_indexrK r$ r r< r] ) r/ rD r Ú window_sizerA Z
|
grp_A_meanZ
|
grp_B_meanZnum_nansZnansZgrp_A_expectedZgrp_B_expectedr@ r r r Ú'test_groupby_rolling_center_min_periods s& "
|
ÿÿ ÿz3TestRolling.test_groupby_rolling_center_min_periodsc C sÌ t ddddgddddgdd d
|
d gd }| d ¡d g d¡ ¡ }t tjtjdtjgd gtjdd d gdd}t ||¡ | d ¡d d¡ ¡ }t
|
tjtjdtjgtjdd d gdd d}t ||¡ d S )Nr r r r r rÄ r rr r é r r r )r¤ r« ©r r r¬ rz r| r§ ) r r. r5 rK r$ r r r r< r] r r= rÈ r r r Ú test_groupby_subselect_rolling s. "ÿ ÿý ÿûz*TestRolling.test_groupby_subselect_rollingc C s G dd dt }tddddddgd id
|
gd d gd d gd d}| |j¡j|d dd d ¡ }| |j¡jd d d ¡ }t ||¡ d S )Nc @ s e Zd ZdddZdS )zFTestRolling.test_groupby_rolling_custom_indexer.<locals>.SimpleIndexerr Nc S sH |d kr| j nd}tj|tjdd }| ¡ | j }|||dk < ||fS )Nr ©Údtyper )rÐ r$ r% Úint64Úcopy)r/ Z
|
num_valuesrD rª ÚclosedÚstepÚendÚstartr r r Úget_window_bounds¸ s
|
zXTestRolling.test_groupby_rolling_custom_indexer.<locals>.SimpleIndexer.get_window_bounds)r NNNN)Ú__name__Ú
|
__module__Ú __qualname__rÝ r r r r Ú SimpleIndexer· s úrá r r r rÄ r rË r r rs r r r )rÐ rC ©rR rD )r r r. r\ r5 rI r< r] )r/ rá r rA r@ r r r Ú#test_groupby_rolling_custom_indexerµ s ÿ ÿÿz/TestRolling.test_groupby_rolling_custom_indexerc C s¦ t tdtddddg tdgd d}| d¡jdd d
|
d d ¡ }ttjd dtjddgt j
|
dtdfgd dtdfgd dd gdd d}t ||¡ d S )Nry r r r ú
|
2019-01-01©Úcolumn1Zcolumn2rÊ ÚdaterÊ Ú1Drç Úleft©r rÙ ræ r r r r rz r§ © r r[ r r. r5 rI r r$ r r r r< r= rÈ r r r Ú'test_groupby_rolling_subset_with_closedÑ s(
|
üÿ ÿÿýù z3TestRolling.test_groupby_rolling_subset_with_closedc C s® t tdtddddg tdgd d}| d¡dd g jd
|
d d d d ¡ }ttjd dtjddgt j
|
dtdfgd dtdfgd dd gddd}t ||¡ d S )Nry r r r rä rå rÊ ræ rç rè ré rê r r r r rz r§ rë rÈ r r r Ú.test_groupby_subset_rolling_subset_with_closedé s4
|
üÿ
|
ÿÿÿÿýù z:TestRolling.test_groupby_subset_rolling_subset_with_closedr c C s~ t dddgtjdddgddgdd d
|
}t| |¡ d¡| }t tjtjd gtjd d dgd ddgdd d
|
}t ||¡ d S )Nr r )r r8 )r r )r ÚzÚ1Ú2rz r r§ r )r r r8 )r r r )r r rî )
|
r r r rU r. r5 r$ r r< r= )r/ r ZdsrA r@ r r r Ú"test_groupby_rolling_index_changed s$ ÿû ÿûz.TestRolling.test_groupby_rolling_index_changedc C sÞ t dg i}| d¡jdd ¡ }|jdd}tjtg ddtg ddgdd gd|_t
|
||¡ t g g d }| dd
|
g¡jdd ¡ }|jdd
|
gd}tjtg ddtg ddtg ddgdd
|
d gd|_t
|
||¡ d S ) NÚs1r rQ ©r} Úfloat64rÕ r× rz )rò Ús2rõ ) r r. r5 rI rY r Z from_productr r\ r< r] )r/ r@ rA r r r Ú test_groupby_rolling_empty_frame s& ÿ
|
|
|
ýúz,TestRolling.test_groupby_rolling_empty_framec C s6 t ddtddddgddtddddgdd tddd
|
dgd dtddd dgd d tddddggdddgd d¡}| d¡}| ¡ |d< |jddd}| dd ¡}t ddtdddddgddtdddddgd dtddd ddgdd tddd
|
ddgd d tdddddggddddgd ddg¡}t ||¡ d S )Nr r½ éã r r± r r ÚZr¾ r ÚHry ÚEr r\ rÊ Z eventTimeró Z count_to_dateZ10d©r c S s
|
| j d S )Nr )Úshape©r r r r r9 D r: z?TestRolling.test_groupby_rolling_string_index.<locals>.<lambda>r r rÄ ) r r r~ r. Zcumcountr5 r; r< r] )r/ r ÚgroupsZrolling_groupsrA r@ r r r Ú!test_groupby_rolling_string_index4 s6 ûø ÷
|
û
|
ø ÷
|
z-TestRolling.test_groupby_rolling_string_indexc C s| t ddgddgdjddd d¡ ¡ }t t ddgddgg¡dd gtjd
|
d gdd gd d }|jdd}t
|
||¡ d S )Nr r )r* r r* F)Úsortr r r )r r r¥ rz r| ró ) r r. r5 rJ r$ Úarrayr r rY r< r] )r/ rA r@ r r r Ú test_groupby_rolling_no_sortQ s ÿþÿý z(TestRolling.test_groupby_rolling_no_sortc C sÀ t tdtddddg tdddd}| d¡jd d
|
d d d ¡ }ttjddtjddgd t j
|
dt dfdt dfdt dfdt dfdt dfdt dfgdd
|
gdd}t ||¡ d S )Nry r r r Z20190101)rÛ Úperiodsrå rÊ Z3drç ré rê ræ r z
|
2018-12-27z
|
2018-12-29z
|
2018-12-31z
|
2018-12-28z
|
2018-12-30rä rz rÇ )r r[ r r. r5 rE r r$ r r r r r< r= rÈ r r r Ú$test_groupby_rolling_count_closed_onb s<
|
|
üÿ
|
ÿÿÿ
|
|
|
|
|
|
ú÷ýz0TestRolling.test_groupby_rolling_count_closed_on)r Úkwargsr5 r râ Ú expandingc C s¤ t ddgddgddgddgddggddgd}t| d¡|f | ¡ }t tjgd tjdtjddgd tjd
|
d d d dgdd gdd}|jdd}t
|
||¡ d S )Nr r r r r ró rs gÿ&¥ æ?r¯ r² r³ )r r )r r ©r rP rz r ) r rU r. Zsemr$ r r r rY r< r] )r/ r r r rA r@ r r r Útest_groupby_rolling_sem s ÿ ÿþ z$TestRolling.test_groupby_rolling_sem)ÚrollingsÚkeyr r Nr\ c C sp t tddtjgdddgdddgd}|dkr:| d¡}tjt| d d
|
| d ¡j d | W 5 Q R X d S )Nz2020-06-01 12:00z2020-06-01 14:00r r r r r\ r z values must not have NaTr( r Ú60min)r )
|
r r r$ r r~ r+ r, Ú
|
ValueErrorr. r5 )r/ r r
|
r r r r Ú"test_groupby_rolling_nans_in_index s ýÿ
|
z.TestRolling.test_groupby_rolling_nans_in_indexrO c C s dddgdddgg}t j|dd}tdddg|d}|jd d
|
g|d d¡ ¡ }td d dgt jdddgd d
|
d d
|
gdd}t ||¡ d S )NÚval1Úval2©ru rv rz r r r r ru rv rN r r rÄ )r r r r )r r r r ) r rZ r r. r5 r6 r r< r= )r/ rO Úarraysr\ ÚsrA r@ r r r Ú test_groupby_rolling_group_keys§ s ý
|
úþ z+TestRolling.test_groupby_rolling_group_keysc C s dddgdddgg}t j|dd}tdddgtdd|d }| d
|
d g¡ d¡ ¡ }td d ddgit jdddgd
|
d d
|
dgdd }t ||¡ d S )Nr r r rz r r r r" r ru r r r r r )r r r r )r r r r rv )
|
r rZ r r[ r. r5 r6 r r< r] ©r/ r r\ r rA r@ r r r Ú1test_groupby_rolling_index_level_and_column_label¾ s ý
|
úþ z=TestRolling.test_groupby_rolling_index_level_and_column_labelc C sd t t d¡ddgd d}| d¡ d¡ ¡ }tjdd d
|
d d d ddgdd gd}t |j
|
|¡ d S )NrÌ r r rP r¯ r r r¤ r¦ )r rP )r ry r« rÓ )r rs )r r° rz ) r r$ r% r. r5 r6 r r r< Úassert_index_equalr\ ©r/ r rA r_ r r r Ú)test_groupby_rolling_resulting_multiindexÒ s þz5TestRolling.test_groupby_rolling_resulting_multiindexc C s t t d¡ddgd ddddgd d}| dd g¡ d¡ ¡ }tjd
|
d d d ddddddddg dd d gd}t |j
|
|¡ d S )Ng (@r r ry r rP r r r )r r r )r r rP )r r r )r r r )r r ry )r r rr )r r r )r r rs )r r r± )r rP r )r rP r° )r rP r® rz ) r r$ r% r. r5 rI r r r< r r\ r r r r Ú*test_groupby_rolling_resulting_multiindex2ß s& *ôñz6TestRolling.test_groupby_rolling_resulting_multiindex2c C s t t d¡ddgd ddddgd d}|jddd }| d
|
¡ d¡ ¡ }tjd d d dddddgd
|
d dgd}t j
|
|j |dd d S )NrÌ r r rP r r r T)Úappendr )r r r )r r r )r rP r )r ry r )r r r )r r rP )r rs r )r r° rP rz Úequiv)Úexact) r r$ r% r~ r. r5 r6 r r r< r r\ r r r r Ú*test_groupby_rolling_resulting_multiindex3ö s *ø
|
õ z6TestRolling.test_groupby_rolling_resulting_multiindex3c C sJ |j ddd}| dd ¡j}|jdd}| dd ¡j}t ||¡ d S ) Nr FrN c S s | d¡ ¡ S rS ©r5 rI r7 r r r r9 r: zUTestRolling.test_groupby_rolling_object_doesnt_affect_groupby_apply.<locals>.<lambda>rP rQ c S s | d¡ ¡ S rS r r7 r r r r9 r: )r. r; r\ r5 r< r )r/ r& r> r@ Ú_rA r r r Ú7test_groupby_rolling_object_doesnt_affect_groupby_apply
|
s
|
zCTestRolling.test_groupby_rolling_object_doesnt_affect_groupby_apply)rR rD rÙ r@ r ré ÚrightrË c C s´ t ddddddddg}| ddddddddg¡j|||d d
|
¡}t tj|d d tt ddg¡d
|
dddddddggd
|
d
|
d
|
d
|
ddddgd
|
dddddddggd d}t ||¡ d S )Nr r r rP rs ry r° r )rR rD rÙ r rô rÕ )ÚlevelsÚcodesr ) r r. r5 rb r$ r r r< r] )r/ rR rD rÙ r@ r rA Zexpected_resultr r r Útest_groupby_rolling_var s"
|
ÿþÿ &þþz$TestRolling.test_groupby_rolling_varr} )r Ú )r r r r c C s| t ddggd ddggd ddggd |d}| d ¡}|jjd
|
d }| d ¡}| ¡ }d |jksjtt |j|¡ d S ) Nr r r r r r r ró r T)ÚdeeprP )
|
r r. ÚobjrØ r5 rI r} ÚAssertionErrorr< r] )r/ r} r r> Z original_objrF rA r r r Ú test_by_column_not_in_values+ s 2
|
|
z(TestRolling.test_by_column_not_in_valuesc C s ddddgddddgg}t j|dd}tddd d
|
d gi|d }|jd dd d¡ ¡ }ttjdtjdgt j ddddgdddgddd}t
|
||¡ d S )NÚFalconÚParrotÚCaptiveÚWild)ÚAnimalÚTyperz z Max Speedg `x@g àu@g >@g 4@r r rw r g @g I@)r* r* r, )r* r* r- )r+ r+ r, )r+ r+ r- r. r/ r§ ) r rZ r r. r5 rI r r$ r r r< r= r r r r Útest_groupby_level8 s&
|
|
þüù õ z TestRolling.test_groupby_levelzby, expected_datar£ Únumç Y@ç Àb@ç i@ú
|
2018-01-01ú
|
2018-01-02ç @o@)rç r1 c s¸ dddgdddgdddgdddgg}t |d d
|
d gd t d
|
d
|
< d
|
g¡ fd d|D } j|ddjddd ¡ }d ddddgi}| |¡ t | jd}t ||¡ d S )Nr r5 r2 r6 r4 r r3 r7 r£ rç r1 ró c s g | ]}t |qS r )rU )rÎ Úattrrý r r rÏ n s z3TestRolling.test_as_index_false.<locals>.<listcomp>F)Zas_indexr r râ r )
|
r r r~ r. r5 r6 Úupdater\ r< r] )r/ r3 r rÁ Zgp_byrA r@ r rý r Útest_as_index_falseP s$ ü ÿ
|
þz TestRolling.test_as_index_falsec
|
C sÎ t |¡j}d}t |d|¡}d|d< d}t ||¡}t j|d< d|d< t||d d¡}| d¡d jd dd
|
¡ }t
|
|dt t dgd dg |dd t dgd dg |dd g¡d } t || ¡ d S )Néè r r éÿÿÿÿg52·ôDE)r\ Úadl2r\ r= rr râ iç )rÖ r¨ rÇ )r$ rÖ ÚtypeÚrepeatr r r~ r. r5 r6 r r rZ r r< r= )
|
r/ Zany_int_numpy_dtypeÚtypÚsizeÚidxÚvalZarrr rA r@ r r r Útest_nan_and_zero_endpoints{ s4
|
þÿû þÿý
|
z'TestRolling.test_nan_and_zero_endpointsc sX ddddg}d t fdd|D }tjtdd | d
|
¡jd d d W 5 Q R X d S )Nr r r r r; c s( g | ] }t d | |d ddqS )r r é* )Útr8 r ©r rÍ ©Úsecr r rÏ s zBTestRolling.test_groupby_rolling_non_monotonic.<locals>.<listcomp>z.* must be monotonicr( r rF Z3s)r rR )r r+ r, r r. r5 )r/ Zshuffledr r rH r Ú"test_groupby_rolling_non_monotonic s ÿz.TestRolling.test_groupby_rolling_non_monotonicc C sÆ dddgdddgdddgdddgd d
|
dgd d dgd d dgd ddgd ddgd ddgg
|
}t |dddgd}t|d |d< | d¡}| d¡ d¡ dd ¡}| d¡jdddd ¡ }t ||¡ d S )NZDavidz1/1/2015éd z1/5/2015iô z 5/30/2015é2 z 7/25/2015ZRyanz1/4/2014z 1/19/2015z 3/31/2016ZJoez7/1/2015z9/9/2015z
|
10/15/2015r¨ rç Úamount)rÁ r} c S s | d¡d ¡ S )NÚ180DrM r r7 r r r r9 À r: z4TestRolling.test_groupby_monotonic.<locals>.<lambda>rN rû )
|
r r Z sort_valuesr~ r. r; r5 rI r< r= )r/ rÁ r r@ rA r r r Útest_groupby_monotonic¥ s, ö
|
|
ÿþÿ z"TestRolling.test_groupby_monotonicc C s t dddd}tdgd dgd dgd t ||f¡t d
|
¡d }| d ¡ d ¡ dd ¡}| d ¡jdd dj
|
¡ }t ||¡ d S )Nz2016-01-01 09:30:00r r )rÜ r Úfreqr r r r r r! r r r c S s | d¡d ¡ S )NÚ4sr r4 r7 r r r r9 Õ r: zJTestRolling.test_datelike_on_monotonic_within_each_group.<locals>.<lambda>rQ rû ) r r r$ Z concatenater% r~ r. r; r5 r r6 r< r= )r/ Údatesr r@ rA r r r Ú,test_datelike_on_monotonic_within_each_groupÅ s ýÿ ÿz8TestRolling.test_datelike_on_monotonic_within_each_groupc C sd t dgd dgd dd dD d tdd}tjtd d
|
| d ¡jd d d W 5 Q R X d S )Nr r r c S s g | ]}t |d d qS )r rG )rÎ Úyearr r r rÏ ß s zPTestRolling.test_datelike_on_not_monotonic_within_each_group.<locals>.<listcomp>)iä iå r÷ ry r z&Each group within B must be monotonic.r( r Z365Dr rû )r r[ r+ r, r r. r5 )r/ r r r r Ú0test_datelike_on_not_monotonic_within_each_groupÚ s ýÿz<TestRolling.test_datelike_on_not_monotonic_within_each_group)3rÞ rß rà r1 rB rG r+ ÚmarkÚ parametrizer` rd rn 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) r0 r r: rD rJ rO rS rU r r r r r' 0 sÎ ùþ
|
|
ÿ
|
|
|
|
&þ
|
|
A#
|
|
þ
|
ÿ
|
|
üþ
|
ÿ
|
|
ü
|
ùþþþ
|
r' c
|
@ s e Zd Zejdd Zej ddddddd d
|
g¡d d Zej dd dg¡dd Z ej ddddddg¡dd Z
|
ej dddg¡dd Z d d Z d S ) Ú TestExpandingc C s. t dgd dgd dgd t d¡dS r r# )r/ r r r Úframeè s zTestExpanding.framerH rI r6 rJ rK rE rL rM c sn |j ddd}| ¡ }t| }| fdd¡}|jddd}t |d tdg¡}||_t
|
||¡ d S ) Nr FrN c s t | ¡ S r ©rU r r7 rV r r r9 ô r: z.TestExpanding.test_expanding.<locals>.<lambda>r rW r! © r. r rU r; rY r rZ r[ r\ r< r] ©r/ rH rY r> rF rA r@ r_ r rV r Útest_expandingì s z TestExpanding.test_expandingra rb c sr |j ddd}| ¡ }t| dd}| fdd¡}|jddd }t |d td
|
g¡}||_t
|
||¡ d S ) Nr FrN r rc c s t | ¡ ddS )Nr rc rZ r7 rV r r r9 r: z3TestExpanding.test_expanding_ddof.<locals>.<lambda>r rW r! r[ r\ r rV r Útest_expanding_ddofü s z!TestExpanding.test_expanding_ddofre rf rg rh ri rj c sp |j ddd}| ¡ }|jd d}| fdd¡}|jddd }t |d td
|
g¡}||_t
|
||¡ d S ) Nr FrN rk rl c s | ¡ jd dS )Nrk rl )r rm r7 rl r r r9 r: z7TestExpanding.test_expanding_quantile.<locals>.<lambda>r rW r! ) r. r rm r; rY r rZ r[ r\ r< r] )r/ re rY r> rF rA r@ r_ r rl r Útest_expanding_quantile
|
s
|
ÿz%TestExpanding.test_expanding_quantilero rp c
|
s² d¡}| ¡ }t| } fdd}| |¡}ttddttdd }tj|j|df< tj|d< t
|
||¡ t|j d d
|
} fd d } | | ¡}t ||¡ d S ) Nr c s t | ¡ S r rZ r7 ©rH rY r r Úfunc_0# s z5TestExpanding.test_expanding_corr_cov.<locals>.func_0r é= éH éq r Tr c s t | j ¡ ddS )NTr )rU r r r7 rV r r Úfunc_12 s z5TestExpanding.test_expanding_corr_cov.<locals>.func_1) r. r rU r; r r[ r$ r Zilocr< r] r r= )
|
r/ rH rY r> rF rA ra r@ Znull_idxre r r` r Útest_expanding_corr_cov s
|
|
|
|
z%TestExpanding.test_expanding_corr_covc st |j ddd}| ¡ }|jdd d}| fdd¡}|jddd }t |d td
|
g¡}||_t ||¡ d S ) Nr FrN c S s | ¡ S r r r7 r r r r9 = r: z4TestExpanding.test_expanding_apply.<locals>.<lambda>r c s | ¡ jdd dS )Nc S s | ¡ S r r r r r r r9 > r: zFTestExpanding.test_expanding_apply.<locals>.<lambda>.<locals>.<lambda>r )r r; r7 r r r r9 > r: r rW r! )
|
r. r r; rY r rZ r[ r\ r< r] )r/ r rY r> rF rA r@ r_ r r r Útest_expanding_apply8 s z"TestExpanding.test_expanding_applyN) rÞ rß rà r+ ÚfixturerY rV rW r] r^ r_ rf rg r r r r rX ç s"
|
ÿ
|
|
ÿ
|
|
rX c
|
@ s e Zd Zej ddddddggdejdd d
|
ggd ejd d dggg¡dd Zej ddejdddggdejd ddggg¡dd Z dd Z
|
dd Z d d Z d S ) ÚTestEWMzmethod, expected_datar6 r gUUUUUUå?g·mÛ¶mÛö?g""""""@ra g»¹øÛ æ?g¦´ÿÖî?gët ë©Õò?rb rq gnÛ¶mÛ¶í?g¾â+¾â+ö?c C sh t dgd tdd}t| d¡jdd| }t d|itjdd d
|
d gdd gd d }t ||¡ d S )Nr rP r" r r ©Úcomr r² r³ r´ rµ rz r ) r r[ rU r. Úewmr r r< r] ©r/ Úmethodr r rA r@ r r r Ú test_methodsH s
|
üùþ zTestEWM.test_methodsro r r rp gÒr Ú¶í?gß¿yqâ+ö?c s t dgd tdd}t| d¡jdd }t d|itjdd d
|
d gdd d gd d }t ||¡ | d¡ fdd¡}t ||¡ d S )Nr rP r" r r rj r )r r r )r r r )r r r )r r r rz r c s t | jdd S )Nr rj )rU rl r7 ©rn r r r9 x r: z/TestEWM.test_pairwise_methods.<locals>.<lambda>)
|
r r[ rU r. rl r r r< r] r; rm r rp r Útest_pairwise_methodsb s üùþ z TestEWM.test_pairwise_methodsc C s| d}| d¡}| d¡j||d ¡ }tddddd dd
|
d dd d g
|
itjddddddddddg
|
dd gdd}t ||¡ d S )Nú23 daysr r ©ÚhalflifeÚtimesr r gCusñ·=à?g;¬pËGRð?gÀZµkBø?gÕ> (â?g8KÉró?gÒ§Uôæä?gAó9w» ó?r² rµ )r ry )r r± )r r r r¹ )r r )r rs )r r rz r ) Úpopr. rl r6 r r r r< r] )r/ r rt ru rA r@ r r r Ú
|
test_times{ sB
|
öÿö óñ zTestEWM.test_timesc C sN d}| d¡}| d¡}|j||d ¡ }|j||jd ¡ }t ||¡ d S )Nrr r r rs )rv r. rl r6 Úvaluesr< r] )r/ r rt ru rÀ rA r@ r r r Útest_times_array¢ s
|
|
zTestEWM.test_times_arrayc C s t dddddgtddddtdd}| d¡jd d
|
d }| ¡ }t tddddd gd d tjdddddgttdgdd gdd}t
|
||¡ |d ¡ }t d gd tjdddddgtddddgdd
|
gddd}t ||¡ | ¡ }t
|
||¡ d S )Nr r z2021-9-1rs rù )r rP )r£ Ú timestampr r£ Z1Hrz rû r )rz r rz r r r§ ) r r r[ r. r5 rE r rZ r r< r] r r= )r/ r ÚgrprA Z expected_dfZexpected_seriesr r r Ú"test_dont_mutate_obj_after_slicing« s@ ýÿ þ ÿû þû÷ z*TestEWM.test_dont_mutate_obj_after_slicingN) rÞ rß rà r+ rV rW r$ r ro rq rw ry r| r r r r ri G s ýþ
|
"þ
|
' ri )Únumpyr$ r+ Zpandasr r r r r r r Zpandas._testingZ_testingr< Zpandas.api.indexersr Zpandas.core.groupby.groupbyr
|
rh r r& r' rX ri r r r r Ú<module> s& $
|
|
>`
|
