gL i<ddlZddlZddlmZddlmZmZddlm Z ddl m Z ejjgZej dZej%ejj'e ej(e dk(xredej(ejj+d Gd d Zy) N)is_platform_arm) DataFrameSeries)Versionnumbaz0.61z&Segfaults on ARM platforms with numba )reasonignorec eZdZdZej j ej jdee de dde e ddgdZ ej je ej jdee de dde e ddgd Zej jd gd d Zy )TestEWMcBttdtdd}|jdjdj }t j td5|j|jddddy#1swYyxYw) Nab?z;Must call mean with update=None first before passing updatematchupdate) rrangeheadewmonlinepytestraises ValueErrormean)selfdf online_ewms e/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/pandas/tests/window/test_online.pytest_invalid_updatezTestEWM.test_invalid_updatesw U1XE!H5 6WWQZ^^C(//1 ]] O  / OO2771:O .  / / /s *"BBobjr rfoo)namec|jd||j}|||d}|jdjd||j|} t dD]} | j} t j | |jd| j|jd} t j | |jd| jy)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$s773v7CHHJ"'X8T  HHQK SVyS 9 V-V 0  q A__&F OOFHMM!$4 5__CHHQK_8F OOFHMM!$4 5     )rczttjgdd}|jd||||j } |||d} |j djd|||j d|j | } tdD]} | j } tj| | j d| j |jd |jd  } tj| | jd | jy) 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=s  HHV&   77 (  $& #(X8T HHQK S#jjm, V-V 0 q A__&F OOFHMM!$4 5__CHHQKejjQRm_TF OOFHMM!$4 5     r:method) aggregatestdcorrcovvarcttd}i}|dk(rd|d<tjtd5t |j dj|di|dddy#1swYyxYw) 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_raiseshspU2Y [ (F6N ]].6N O ; 0GCGGAJ%%' 0 :6 : ; ; ;s 0A66A?N)__name__ __module__ __qualname__r$rmarkslow parametrizerrrr9xfailrQrCrUrMr:r#r r s/ [[ [[ auQx896%(QV;WX* [[12 [[ auQx896%(QV;WX%3%N [[X'QR;S;r:r )numpyr@r pandas.compatrpandasrrpandas._testing_testingr2pandas.util.versionrrY single_cpu pytestmark importorskiprappendskipif __version__filterwarningsr rMr:r#rjs )'kk$$ % G$  KK!!"gfo5K/:K78I8I7JKH%V;V;&V;r: