U
|
®ý°dA ã @ s~ d dl Zd dlZd dlmZmZmZ d dlmZ d dl m
|
m Z d dl mZmZmZmZ d dlmZ ejje o|e p|e ddZejddgd d
|
d Zejd i gd i gdi gdi gdi gdi gddd igdi gddd igg d dd Ze d¡ej d¡G dd dZ e d¡G dd dZ e d¡d d Z e d¡d d Z e d¡ejj!ej d¡G d d! d!Z"dS )"é N)Úis_ci_environmentÚis_platform_macÚis_platform_windows)ÚNumbaUtilError)Ú DataFrameÚSeriesÚoption_contextÚ to_datetimez`On GHA CI, Windows can fail with 'Windows fatal exception: stack overflow' and macOS can timeout)ÚreasonÚsingleÚtable)Úparamsc C s | j S )z3method keyword in rolling/expanding/ewm constructor©Úparam©Úrequest© r úUd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/window/test_numba.pyÚmethod s r ÚsumÚmeanZmedianÚmaxÚminÚvarZddofÚstdc C s | j S ©Nr r r r r Ú$arithmetic_numba_supported_operators# s r ÚnumbaÚignorec @ sV e Zd Zej dddg¡dd Zej dee d¡eddddej
|
ej d d
|
d
|
d
|
g
|
ddddej
|
ej dd
|
d
|
d
|
g
|
ej
|
ej
|
dd ddddddg
|
gj e edd d e dddej dddd
|
ge dddej
|
dddd
|
gg¡dd Zej dee d¡e edd d g¡dd Zej dddg¡dd Zej ddd
|
ddgd i gg¡d d Zd d Zd!S )"Ú
|
TestEngineÚjitTFc C s dd }|r dd l }| |¡}|||d} d}
|
ttd} | jd||dj||
|
d | d
|
d } | jd||dj|d |
|
d
|
d } t | | ¡ d S )Nc W s$ d}|D ] }||7 }qt | ¡| S )Nr ©Únpr )ÚxÚargsZarg_sumÚargr r r Úf: s
|
z0TestEngine.test_numba_vs_cython_apply.<locals>.fr ©ÚnogilÚparallelÚnopython©é é
|
r, )ÚcenterÚstepr T)r$ ÚengineÚ engine_kwargsÚrawÚcython)r0 r$ r2 ©r r r ÚrangeÚrollingÚapplyÚtmÚassert_series_equal)Úselfr r( r) r* r. r/ r& r r1 r$ ÚsÚresultÚexpectedr r r Útest_numba_vs_cython_apply8 s*
|
ÿ ÿz%TestEngine.test_numba_vs_cython_applyÚdataé é é é é Úfoo)Únameé r- é r, c C sd |\}}|||d} |j d|d}
|
t|
|
|f d| d|} t|
|
|f ddi|} t | | ¡ d S )Nr' rC ©r/ r ©r0 r1 r0 r3 )r6 Úgetattrr8 Ú assert_equal) r: r? r( r) r* r r/ r Úkwargsr1 Úrollr< r= r r r Ú$test_numba_vs_cython_rolling_methodsQ s
|
ÿÿz/TestEngine.test_numba_vs_cython_rolling_methodsc C sl |\}}|||d}t t d¡}| ¡ } t| |f d|d|}
|
t| |f ddi|} t |
|
| ¡ d S )Nr' r@ r rJ r0 r3 )r r" ÚeyeÚ expandingrK r8 rL ) r: r? r( r) r* r r rM r1 Úexpandr< r= r r r Ú&test_numba_vs_cython_expanding_methodsu s
|
ÿÿz1TestEngine.test_numba_vs_cython_expanding_methodsc C sà dd }dd }|r0dd l }| |¡}| |¡}|||d} ttdjd|d }
|
|
|
j|d
|
| d d } |
|
j|d d d} t | | ¡ |
|
j|d
|
| d d } |
|
j|d d d} t | | ¡ |
|
j|d
|
| d d } |
|
j|d d d} t | | ¡ d S )Nc S s t | ¡d S )NrB r! ©r# r r r Úfunc_1 s z+TestEngine.test_cache_apply.<locals>.func_1c S s t | ¡d S )Nr@ )r" r rT r r r Úfunc_2 s z+TestEngine.test_cache_apply.<locals>.func_2r r' r- r, rI r T)r0 r1 r2 r3 ©r0 r2 r4 ) r: r r( r) r* r/ rU rV r r1 rN r< r= r r r Útest_cache_apply s@
|
|
ÿ ÿ ÿzTestEngine.test_cache_applyzwindow,window_kwargsr6 r )ÚwindowÚ min_periodsrQ c C s¶ dd }|||d}t ddddgi} t| |f d|i|j|dd|d d
|
}
|
t dd d d gi} t |
|
| ¡ t| |f d|i|j|dd|d d
|
}
|
t dd d d gi} t |
|
| ¡ d S )Nc S s t | ¡| S r ©r" r )Úvaluesr# r r r Úadd· s z,TestEngine.test_dont_cache_args.<locals>.add©r* r( r) Úvaluer r Tr )rH )r2 r0 r1 r$ ç ð?r+ ç @)r rK r7 r8 Úassert_frame_equal) r: rY Z window_kwargsr( r) r* r r] r1 Údfr< r= r r r Útest_dont_cache_args« s* ÿ ÿz TestEngine.test_dont_cache_argsc sº d dd fdd} d}t ddddgi}| d¡j|dd |d
|
}t dd d d gi}t ||¡ d d}| d¡j|dd |d
|
}t dd d d gi}t ||¡ d S ) NFTc s S r r rT ©r( r* r) r r ÚfuncÏ s z6TestEngine.test_dont_cache_engine_kwargs.<locals>.funcr^ r_ r rH r )r2 r0 r1 ra r` ©r r6 r7 r8 rb )r: rf r1 rc r< r= r re r Ú test_dont_cache_engine_kwargsÈ s0
|
ÿ
|
ÿz(TestEngine.test_dont_cache_engine_kwargsN)Ú__name__Ú
|
__module__Ú __qualname__ÚpytestÚmarkÚ parametrizer> r r" rP ÚnanÚinfÚTr r5 rO rS rX rd rh r r r r r 4 s@
|
ýÿõþ
|
ÿ
|
|
# þþ
|
r c @ sÊ e Zd Zejjddd dd gddgdej dd d
|
g¡d d Zejjdd d dd gddgdej dd d
|
g¡dd Zej dddg¡ej dd d
|
g¡dd Zej dddg¡dd Z dS )ÚTestEWMÚgrouperc C s | S r r rT r r r Ú<lambda>æ ó zTestEWM.<lambda>c C s
|
| d¡S ©NÚA©ÚgroupbyrT r r r rt æ ru ÚNonery )Úidsr r r c C sT t ddddgtdd}tjtdd" t||jdd|d d
|
W 5 Q R X d S ) NÚaÚbrB ©rw ÚBzengine must be either©Úmatchr` ©ÚcomrE ©r0 ©r r5 rl ÚraisesÚ
|
ValueErrorrK Úewm©r: rs r rc r r r Útest_invalid_engineå s zTestEWM.test_invalid_enginec C s | S r r rT r r r rt ï ru c C s
|
| d¡S rv rx rT r r r rt ï ru c C sZ t ddddgtdd}tjtdd( t||jdd|d d
|
d id W 5 Q R X d S ) Nr| r} rB r~ zcython engine does notr r` r r3 r* TrJ r
r r r r Útest_invalid_engine_kwargsî s ÿz"TestEWM.test_invalid_engine_kwargsc C s t dtdi}|dkr"dd }nddddg|d< d d }|d
|
krFd }||jd ||d } |||d}
|
t| |d|
|
d} t| |dd} t | | ¡ d S )Nr rB rz c S s | S r r rT r r r rt ru z.TestEWM.test_cython_vs_numba.<locals>.<lambda>r| r} rw c S s
|
| d¡S rv rx rT r r r rt ru r Tr` )r ÚadjustÚ ignore_nar' r rJ r3 r )r r5 r rK r8 rb ) r: rs r r( r) r* r r rc r r1 r< r= r r r Útest_cython_vs_numbaù s
|
z TestEWM.test_cython_vs_numbac C sª t dddddddgi}|dkr*dd }n dd }d d
|
d d
|
d
|
d g|d < d }td d ddddg}||j|d||d} |||d}
|
| jd|
|
d} | jdd} t | | ¡ d S )Nr r rH r, rz c S s | S r r rT r r r rt ru z4TestEWM.test_cython_vs_numba_times.<locals>.<lambda>c S s
|
| d¡S rv rx rT r r r rt ru r| r} rw z23 daysz
|
2020-01-01z
|
2020-01-02z
|
2020-01-10z
|
2020-02-23z
|
2020-01-03T)Úhalflifer r Útimesr' r rJ r3 r )r r r r r8 rb ) r: rs r( r) r* r rc r r r r1 r< r= r r r Útest_cython_vs_numba_times s2
|
úÿ
|
ÿ z"TestEWM.test_cython_vs_numba_timesN)
|
ri rj rk rl rm rn r r r r r r r r rr ã s& ÿ ÿ rr c C sf dd } t td}tdd | d¡j| d dd}W 5 Q R X | d¡j| ddd}t ||¡ d S ) Nc S s t | ¡d S )Nr, r! rT r r r r& 2 s z!test_use_global_config.<locals>.fr- zcompute.use_numbaTr, rW r )r r5 r r6 r7 r8 r9 )r& r; r< r= r r r Útest_use_global_config0 s r c C sF t jtdd. ttd d¡jdd ddiddd W 5 Q R X d S )
|
Nz"numba does not support kwargs withr rH c S s | S r r rT r r r rt @ ru z.test_invalid_kwargs_nopython.<locals>.<lambda>r| r T)rM r0 r2 )rl r r r r5 r6 r7 r r r r Ú test_invalid_kwargs_nopython< s ÿr c @ s| e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Ze j
|
d e d¡e d¡e d¡g¡e j
|
dddg¡dd ZdS )ÚTestTableMethodc C sH dd }t jtdd( ttdjdddj|dd d
|
W 5 Q R X d S ) Nc S s t j| ddd S ©Nr ©ÚaxisrH r[ rT r r r r& J s z7TestTableMethod.test_table_series_valueerror.<locals>.fz1method='table' not applicable for Series objects.r rH r )r r TrW )rl r r r r5 r6 r7 )r: r& r r r Ú test_table_series_valueerrorI s ÿ ÿz,TestTableMethod.test_table_series_valueerrorc C sÒ |\}}|||d} t t d¡}
|
|
|
jdd|d|d} |dkrztjt| dd t| |f | d
|
d | W 5 Q R X nT|
|
jdd |d|d} t| |f | d
|
d |} t| |f | d
|
d |}t | |¡ d S ) Nr' rC r, r r ©r r rZ r/ ©r r ú not supportedr r ©r1 r0 r )
|
r r" rP r6 rl r ÚNotImplementedErrorrK r8 rb )r: r r( r) r* r r/ r rM r1 rc Z
|
roll_tableZ roll_singler< r= r r r Ú!test_table_method_rolling_methodsT s@
|
ÿÿ ÿ
|
ÿÿ
|
ÿÿz1TestTableMethod.test_table_method_rolling_methodsc C sv |||d}dd }t t d¡}|jdd|d|dj|d |d
|
d } |jdd |d|dj|d |d
|
d }
|
t | |
|
¡ d S ) Nr' c S s t j| ddd S r r[ rT r r r r& w s z:TestTableMethod.test_table_method_rolling_apply.<locals>.frC r, r r r Tr ©r2 r1 r0 r )r r" rP r6 r7 r8 rb ) r: r r( r) r* r/ r1 r& rc r< r= r r r Ú test_table_method_rolling_applyt s8 ÿ þ ÿ þz/TestTableMethod.test_table_method_rolling_applyc C s dd }t dddgdddgddd gdd
|
d gg}|jdd d |dj|ddd}t dddgdddgdddgdddggd d |
}t ||¡ d S )Nc S sh t d| jd f¡}| d d
d d
f | d d
df jdd| d d
df ¡ |d d
d d
f< |S )NrH r, r r )r" ÚonesÚshaper )r# Zarrr r r Ú weighted_mean s PzNTestTableMethod.test_table_method_rolling_weighted_mean.<locals>.weighted_meanrH r, g333333ã?rC gÙ?rB gÉ?r@ gffffffæ?r r )r rZ r/ Tr )r2 r0 r` ra gÍÌÌÌÌÌü?g©kí}ªª
|
@g©kí}ªª@g1°ãø?rA rg )r: r/ r£ rc r< r= r r r Ú'test_table_method_rolling_weighted_mean s" ( ÿüÿùz7TestTableMethod.test_table_method_rolling_weighted_meanc
|
C sj |||d}dd }t t d¡}|jd|dj|d|dd }|jd
|
|dj|d|dd } t || ¡ d S ) Nr' c S s t j| ddd S r r[ rT r r r r& s z<TestTableMethod.test_table_method_expanding_apply.<locals>.frC r ©r r Tr r r )r r" rP rQ r7 r8 rb )
|
r: r r( r) r* r1 r& rc r< r= r r r Ú!test_table_method_expanding_apply s ÿ ÿz1TestTableMethod.test_table_method_expanding_applyc C sÆ |\}}|||d}t t d¡} | jd|d}
|
|dkrttjt| dd t|
|
|f |dd | W 5 Q R X nN| jd
|
|d} t|
|
|f |dd |} t| |f |dd |} t | | ¡ d S ) Nr' rC r r¥ r r r r r r )
|
r r" rP rQ rl r r rK r8 rb )r: r r( r) r* r r rM r1 rc Z expand_tableZ expand_singler< r= r r r Ú#test_table_method_expanding_methods¦ s4
|
ÿÿ
|
ÿÿ
|
ÿÿz3TestTableMethod.test_table_method_expanding_methodsr? rC )r, rC )rC r, r r r c C s` |||d}t |}t|jdd|d||dd} t|jdd|d||dd}
|
t | |
|
¡ d S )Nr' rH r )r r r r r r )r rK r r8 rb ) r: r? r r r( r) r* r1 rc r< r= r r r Útest_table_method_ewm¾ s ÿ ÿz%TestTableMethod.test_table_method_ewmN)ri rj rk r r r r¤ r¦ r§ rl rm rn r" rP r¡ r¨ r r r r r D s $r )#Únumpyr" rl Z pandas.compatr r r Z pandas.errorsr Z pandas.util._test_decoratorsÚutilZ_test_decoratorsÚtdZpandasr r r r Zpandas._testingZ_testingr8 rm ZskipifZ
|
pytestmarkZfixturer r Z
|
skip_if_noÚfilterwarningsr rr r r Zslowr r r r r Ú<module> sN þ
|
|
|
÷ÿ
|
|
-L
|
|
|
|
