ttj<ddlZddlZddlmZddlmZmZddlm Z ddl m Z ej jgZejdZeej e eje dko edejej d Gd d ZdS) N)is_platform_arm) DataFrameSeries)Versionnumbaz0.61z&Segfaults on ARM platforms with numba )reasonignorec &eZdZdZejjejdee de dde e ddgdZ ej e ejdee de dde e ddgd Zejd gd d Zd S)TestEWMcttdtdd}|dd}t jtd5||dddddS#1swxYwYdS) Nab?z;Must call mean with update=None first before passing updatematchupdate) rrangeheadewmonlinepytestraises ValueErrormean)selfdf online_ewms ]/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/pandas/tests/window/test_online.pytest_invalid_updatezTestEWM.test_invalid_updates U1XXE!HH55 6 6WWQZZ^^C((//11 ] O    / / OO2771::O . . .  / / / / / / / / / / / / / / / / / /s*B99B=B=objr rfoo)namecJ|d||}|||d}|dd|||} t dD]} | } t j| |d| |d} t j| |d| dS)Nr)adjust ignore_nanogilparallelnopythonr engine_kwargsr) rrrrrtm assert_equaltailreset) r r%r,r-r.r)r*expectedr0r"_results r#test_online_vs_non_online_meanz&TestEWM.test_online_vs_non_online_mean$s 773v7CCHHJJ"'X8TT  HHQKK SVyS 9 9 V-V 0 0  q  A__&&F OFHMM!$4$4 5 5 5__CHHQKK_88F OFHMM!$4$4 5 5 5         )rcttjgdd}|d||||} |||d} |dd|||d|| } tdD]} | } tj | | d| | d | d  } tj | | d | dS) N)z 2020-01-01z 2020-01-05z 2020-01-07z 2020-01-17z 2020-01-21zdatetime64[ns])dtyper)r)r*timeshalflifer+rr/r1)r update_times) rnparrayrrrrrr2r3r4r5)r r%r,r-r.r)r*halflife_with_timesr=r6r0r"r7r8s r#test_update_times_meanzTestEWM.test_update_times_mean=sv HVVV&      77 (   $&& #(X8TT HHQKK S#jjmm, V-V 0 0 q  A__&&F OFHMM!$4$4 5 5 5__CHHQKKejjQRmm_TTF OFHMM!$4$4 5 5 5         r:method) aggregatestdcorrcovvarc8ttd}i}|dkrd|d<tjtd5t |d|di|ddddS#1swxYwYdS) N rEc|S)N)xs r#z=TestEWM.test_ewm_notimplementederror_raises..msqr:funcz.* is not implemented.rrrM)rrrrNotImplementedErrorgetattrrr)r rDserkwargss r##test_ewm_notimplementederror_raisesz+TestEWM.test_ewm_notimplementederror_raiseshsU2YY [ ([F6N ].6N O O O ; ; 0GCGGAJJ%%'' 0 0 : :6 : : : ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ;srjs\ ))))))''''''k$ % G$$  K!""ggfoo5K//:K:KK8IKKH%%V;V;V;V;V;V;V;&%V;V;V;r: