a 鴫j<@sddlZddlZddlmZddlmZmZddlm Z ddl m Z ej jgZedZeej je eje dkoxedejdej d Gd d d ZdS) N)is_platform_arm) DataFrameSeries)Versionnumbaz0.61z&Segfaults on ARM platforms with numba )reasonignorec @seZdZddZejjejdee de dde e dddgdd Z ejj e d ejdee de dde e dddgd d Zejd gdddZdS)TestEWMcCsnttdtdd}|dd}tjtdd"|j|ddWdn1s`0YdS) Nab?z;Must call mean with update=None first before passing updatematchupdate) rrangeheadewmonlinepytestraises ValueErrormean)selfZdf online_ewmr^/var/www/html/laksyashrimp/venv/lib/python3.9/site-packages/pandas/tests/window/test_online.pytest_invalid_updateszTestEWM.test_invalid_updateobjr r Zfoo)namec Cs|jd||d}|||d}|djd||dj|d} tdD]J} | } t| |d| j|dd} t| |d| qFdS)Nr)adjust ignore_nanogilparallelnopythonr engine_kwargsr) rrrrrtm assert_equaltailreset) rr"r'r(r)r$r%expectedr+r_resultrrr test_online_vs_non_online_mean$s   z&TestEWM.test_online_vs_non_online_mean)rcCsttjgddd}|jd||||d} |||d} |djd|||d|dj| d} tdD]R} | } t | | d| j| d | d d } t | | d | qjdS) N)z 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) rnparrayrrrrrr-r.r/r0)rr"r'r(r)r$r%Zhalflife_with_timesr5r1r+rr2r3rrr test_update_times_mean=s<    zTestEWM.test_update_times_meanmethod) aggregateZstdZcorrZcovvarcCstttd}i}|dkr$dd|d<tjtdd.t|d|fi|Wdn1sf0YdS) N r:cSs|S)Nr)xrrr mz=TestEWM.test_ewm_notimplementederror_raises..funcz.* is not implemented.rr)rrrrNotImplementedErrorgetattrrr)rr9Zserkwargsrrr #test_ewm_notimplementederror_raiseshs   z+TestEWM.test_ewm_notimplementederror_raisesN)__name__ __module__ __qualname__r!rmarkZslowZ parametrizerrrr4ZxfailrAr8rDrrrr r s &  & 'r )numpyr6rZ pandas.compatrZpandasrrZpandas._testingZ_testingr-Zpandas.util.versionrrHZ single_cpuZ pytestmarkZ importorskiprappendZskipif __version__filterwarningsr rrrr s