U
|
®ý°dî ã @ s d dl mZ d dlZd dlZd dlZd dlm m m Z
|
d dl m Z d dl mZ dd Ze Zejed ed dd d
|
Zd d Zd d ZdS )é )ÚpartialN)ÚSeriesc C s dt jfdt jfgdd dD dt jfdt jfdt jfd t jfd
|
t jfg d d d D d d dD } tt | }|d |d dS )NÚroll_sumÚ roll_meanc S s& g | ] }d | dt tj|dfqS )z roll_var(ú))Úddof)r Úwindow_aggregationsZroll_var)Ú.0r © r
|
úcd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/window/test_cython_aggregations.pyÚ
|
<listcomp> s ÿz-_get_rolling_aggregations.<locals>.<listcomp>)r é Ú roll_skewÚ roll_kurtÚ roll_median_cÚroll_maxÚroll_minc S s: g | ]2}t jD ]&}d | d| dtt j||dfqqS )zroll_quantile(ú,r )ÚquantileÚ interpolation)r Zinterpolation_typesr Z roll_quantile)r r r r
|
r
|
r r " s øýþ)g-C ëâ6?g à?g §èH.ÿï?c S sP g | ]H}t j ¡ D ]8}d D ].}d| d| d| dtt j|||dfqqqS )©TFz
|
roll_rank(r r )Ú
|
percentileÚmethodÚ ascending)r Zrolling_rank_tiebreakersÚkeysr Z roll_rank)r r r r r
|
r
|
r r . s
|
öüþr r r )ÚidsÚparams)
|
r r r r r r r r ÚlistÚzip)Znamed_roll_aggsZunzippedr
|
r
|
r Ú_get_rolling_aggregations s2 þþü ûø ÷ñ
|
öåÿ, r r r )r r c C s | j S )z/Make a rolling aggregation function as fixture.)Úparam)Úrequestr
|
r
|
r Úrolling_aggregationE s r" c C s ddddddgf\}}}}}t jdd| t jd}t j|||t jd}|| }t j|t jd} t| ||| || |}
|
t| ||||| } t | |
|
¡ d S )Nr r é é é é ©Zdtype) ÚnpÚarangeÚfloat64Úint64ÚarrayÚint32r ÚtmÚ assert_equal) r" ÚminpÚstepÚwidthÚsizeZ selectionÚvaluesÚendÚstartZselarrÚresultÚexpectedr
|
r
|
r Ú-test_rolling_aggregation_boundary_consistencyM s r9 c s> d\}}d| d }t jd|d t jd}tjj|||d
< t j||d < tjj||d |d
< t jd|| gt j dt j||gt j d t j fdd t
|
t D t j d}t | | |}t j|| t jd}t jdt | |t j d}|| } t | ||| |}
|
t |
|
j¡ ¡ s.td
|
t |
|
|¡ d S ) N)r é r% r: r r' r# r c s( g | ] }t | | D ]}|qqS r
|
)Úrange)r ÚiÚj©r5 r6 r
|
r r d s zAtest_rolling_aggregation_with_unused_elements.<locals>.<listcomp>z"Not all expected values are finite)r( r) r* ÚsysÚ
|
float_infoÚminÚnanÚmaxr, r+ r; Úlenr- r Úisfiniter4 ÚallÚAssertionErrorr. r/ ) r" r0 r2 r3 r4 Úlocr7 Zcompact_valuesZ compact_startZ compact_endr8 r
|
r> r Ú-test_rolling_aggregation_with_unused_elementsY s* þ ÿrI )Ú functoolsr r? Únumpyr( ZpytestZ pandas._libs.window.aggregationsZ_libsZwindowZ aggregationsr Zpandasr Zpandas._testingZ_testingr. r Z_rolling_aggregationsZfixturer" r9 rI r
|
r
|
r
|
r Ú<module> s 5 ÿ
|
|
