U
|
®ý°dF ã ! @ s d dl Zd dlZd dlmZ d dlmZ d dlmZm Z m
|
Z
|
d dl m Z d@ddZejddd d
|
d d d dddddddddgddd Zej dde d ¡d d d d d gd d fde d d d!¡d d d d d gd d fdd d d ejd"gd d d d d gd#d fd$e d ¡ejd d d%d"gd dfd$e d d d!¡ejd d&d'd"gd dfd$d d d ejd"gejd d ejejgd#dfd(e d ¡ejd d d d%gd dfd(e d d d!¡ejd&d'd"d gd dfd(d d d ejd"gejd d ejejgd#dfd)e d ¡ejd d%d d*gd dfd)e d d d!¡ejd+d,d d'gd dfd)d d d ejd"gejd d%ejejgd#dfd-e d ¡ejd.d/d0d1gd dfd-e d d d!¡ejd2d*d d%gd dfd-d d d ejd"gejd.d/ejejgd#dfd3e d ¡ejge d.¡gd" d dfd3e d d d!¡ejge d ¡gd" d dfd3d d d ejd"gejge d.¡gd ejgd d#dfd4e d ¡ejd.d.d.d.gd dfd4e d d d!¡ejd d d d gd dfd4d d d ejd"gejd.d.ejejgd#dfd5e d ¡ejd.d/d0d1gd dfd5e d d d!¡ejd2d*d d%gd dfd5d d d ejd"gejd.d/ejejgd#dfg¡d6d7 Zd8d9 Zej d:de
|
d d d d d ge
|
d d d d d gd;d fd$e
|
ejd d"d'd&ge
|
ejd%d d*d2gd;dfd(e
|
ejd d d"d'ge
|
ejd d%d d*gd;dfd)e
|
ejd d'd d,ge
|
ejd"d&d<d=gd;dfd-e
|
ejd d%d d*ge
|
ejd d"d'd&gd;dfd3e
|
ejge d ¡gd" e
|
ejge d ¡gd" d;dfd4e
|
ejd d d d ge
|
ejd d d d gd;dfd5e
|
ejd d%d d*ge
|
ejd d"d'd&gd;dfg¡d>d? ZdS )Aé N)Ú DataError©Ú pandas_dtype)ÚNAÚ DataFrameÚSeriesc C s |dkrd| krd S t | S )NFÚintr )ÚdtypeÚ
|
coerce_int© r úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/window/test_dtypes.pyÚ get_dtype s r ÚobjectÚcategoryZint8Úint16Úint32Úint64Zuint8Zuint16Zuint32Zuint64Úfloat16Úfloat32Úfloat64úm8[ns]úM8[ns]údatetime64[ns, UTC])Úparamsc C s | j S )zDtypes for window tests)Úparam)Úrequestr r r Údtypes s r z4method, data, expected_data, coerce_int, min_periodsÚcounté é é Té
|
éþÿÿÿé FÚmaxé é é ÚminÚsumé é é Úmeanç à?g ø?g @g @é ZstdÚvarZmedianc C s t |t||dd}|jd||d}|dkr`| dkr`d} tjt| d t|| W 5 Q R X n.t|| }
|
t |d dd d |
} t |
|
| ¡ d S )
|
N)r
|
©r r ©Ú min_periodsÚstep©r r r r z No numeric types to aggregate©Úmatchr ) r r ÚrollingÚpytestÚraisesr ÚgetattrÚtmZassert_almost_equal) ÚmethodÚdataÚ expected_datar
|
r r3 r4 ÚserÚrolledÚmsgÚresultÚexpectedr r r Útest_series_dtypes5 s , rE c C sP t ddtg| d}|jd|d ¡ }t tjdtjgd d |
}t ||¡ d S )Nr r r1 r )r4 r. )r r r8 r- ÚnpÚnanr< Zassert_series_equal)Zany_signed_int_ea_dtyper4 r@ rC rD r r r Útest_series_nullable_intn s rH z"method, expected_data, min_periods)r r é é c
|
C s t t d¡ d¡t|d}|jd||d}|dkrh| dkrhd}tjt|d t || W 5 Q R X n.t || }t |d
|
dd d |
} t
|
|| ¡ d S ) Nr! )r r r1 r r2 r5 r z!Cannot aggregate non-numeric typer6 r ) r rF ÚarangeZreshaper r8 r9 r: r r; r< Zassert_frame_equal)
|
r= r? r r3 r4 ZdfrA rB rC rD r r r Útest_dataframe_dtypesv s - rL )N)ÚnumpyrF r9 Z pandas.errorsr Zpandas.core.dtypes.commonr Zpandasr r r Zpandas._testingZ_testingr< r Zfixturer ÚmarkZ parametrizerK rG ÚsqrtrE rH rL r r r r Ú<module> sÌ
|
ðÿ
|
" "( "( "( "(&*"û "( "ûâþ)
|
*(ý(ý(ý(ýþú (ý(ýÝþ,
|
