zmc
2023-12-22 9fdbf60165db0400c2e8e6be2dc6e88138ac719a
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
U
®ý°d
ã@s˜ddlZddlZddlmZmZmZddlmm    Z
ddl m Z m Z ddlmZejjeƒoheƒpheƒddZe
 d¡ej d¡Gdd    „d    ƒƒƒZdS)
éN)Úis_ci_environmentÚis_platform_macÚis_platform_windows)Ú    DataFrameÚSeriesz`On GHA CI, Windows can fail with 'Windows fatal exception: stack overflow' and macOS can timeout)ÚreasonZnumbaÚignorec    @sºeZdZdd„Zejjej dee    dƒe    dƒdœƒe
e    dƒddg¡dd    „ƒƒZ ejj e d
ej dee    dƒe    dƒdœƒe
e    dƒddg¡d d „ƒƒZej d dddddg¡dd„ƒZdS)ÚTestEWMc    CsZttdƒtdƒdœƒ}| d¡ d¡ ¡}tjtdd|j| d¡dW5QRXdS)    Né©ÚaÚbéçà?z;Must call mean with update=None first before passing update©Úmatché©Úupdate)    rÚrangeÚheadÚewmÚonlineÚpytestÚraisesÚ
ValueErrorÚmean)ÚselfZdfÚ
online_ewm©rúVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/window/test_online.pyÚtest_invalid_updatesþzTestEWM.test_invalid_updateÚobjr
r Zfoo)Únamec Cs–|jd||d ¡}|||dœ}| d¡jd||dj|d}    tdƒD]J}
|     ¡} t | | d¡¡|    j| d¡d} t | | d¡¡|     ¡qFdS)Nr)ÚadjustÚ    ignore_na©ÚnogilÚparallelÚnopythonr©Ú engine_kwargsér)    rrrrrÚtmÚ assert_equalÚtailÚreset) rr"r'r(r)r$r%Úexpectedr+rÚ_Úresultrrr Útest_online_vs_non_online_mean's  
ÿþÿ z&TestEWM.test_online_vs_non_online_mean)rcCsÈttjdddddgddƒ}|jd||||d     ¡}    |||d
œ}
| d ¡jd||| d ¡|d    j|
d } td ƒD]R} |  ¡} t     | |     d ¡¡| j| 
d ¡| 
d ¡d} t     | |     
d ¡¡|   ¡qpdS)Nz
2020-01-01z
2020-01-05z
2020-01-07z
2020-01-17z
2020-01-21zdatetime64[ns])Zdtyper)r$r%ÚtimesZhalflifer&rr*r,)rZ update_times) rÚnpÚarrayrrrrrr-r.r/r0)rr"r'r(r)r$r%Zhalflife_with_timesr5r1r+rr2r3rrr Útest_update_times_mean@s< þÿû
 
úøÿ zTestEWM.test_update_times_meanÚmethodÚ    aggregateZstdZcorrZcovÚvarc    Cs\ttdƒƒ}i}|dkr$dd„|d<tjtdd t| d¡ ¡|ƒf|ŽW5QRXdS)    Né
r:cSs|S)Nr)Úxrrr Ú<lambda>póz=TestEWM.test_ewm_notimplementederror_raises.<locals>.<lambda>Úfuncz.* is not implemented.rr)rrrrÚNotImplementedErrorÚgetattrrr)rr9ZserÚkwargsrrr Ú#test_ewm_notimplementederror_raisesks   z+TestEWM.test_ewm_notimplementederror_raisesN)Ú__name__Ú
__module__Ú __qualname__r!rÚmarkZslowZ parametrizerrrr4ZxfailrAr8rDrrrr r    s    $ÿ  $ÿ 'r    )Únumpyr6rZ pandas.compatrrrZpandas.util._test_decoratorsÚutilZ_test_decoratorsÚtdZpandasrrZpandas._testingZ_testingr-rHZskipifZ
pytestmarkZ
skip_if_noÚfilterwarningsr    rrrr Ú<module>s þ