U
|
®ý°d5 ã @ sà d dl Zd dlZd dlmZ d dlmZ dd Zdd Z ej
|
ddd ej ej g¡d
|
d Zej
|
d d d g¡dd Zej
|
d d d g¡dd Zej
|
d d d g¡dd Zej
|
d d d g¡dd Zdd Zdd Zdd ZdS ) é N)ÚSeriesc C s | ¡ ¡ ¡ S ©N)ZnotnaÚall©Úx© r úud:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/window/moments/test_moments_consistency_expanding.pyÚno_nans s r c C s | ¡ ¡ ¡ S r )Zisnullr r r r r Úall_na s r
|
Úfc C s t | ¡ S r )r Úsum)Úvr r r Ú<lambda> ó r c C st |t jkr>t|s>t|r(|js(|dks>| j tjj dd¡ |j
|
|d ¡ }|j
|
|dj |dd}t ||¡ d S )Nr z'np.sum has different behavior with NaNs)Úreason©Ú min_periodsT)ÚfuncÚraw)Únpr r r
|
ÚemptyÚnodeZ
|
add_markerÚpytestÚmarkZxfailÚ expandingÚapplyÚtmÚ assert_equal)ÚrequestÚall_datar r Zexpanding_f_resultZexpanding_apply_f_resultr r r Ú)test_expanding_apply_consistency_sum_nans s"
|
ÿÿÿ ÿ ÿr Úddofé c C sl | j |dj|d}|dk ¡ ¡ r(t|dkrh| | j |d ¡ }| j |d ¡ }t |||| ¡ d S ©Nr ©r! r )r ÚvarÚanyÚAssertionErrorÚmeanr r )r r r! Úvar_xZmean_x2Úmean_xr r r Ú test_moments_consistency_var s r+ c C sz | j |d ¡ }| j |dj|d}|dk ¡ ¡ r8t| tj }d||t|dk< |dkrjtj||dk < t ||¡ d S )Nr r$ r ç r" é )
|
r Úcountr% r& r' r ÚnanÚmaxr r )Úconsistent_datar r! Úcount_xr) Úexpectedr r r Ú%test_moments_consistency_var_constant, s
|
r4 c C s | j |dj|d}|dk ¡ ¡ r(t| j |dj|d}|dk ¡ ¡ rPtt ||| ¡ | j |dj| |d}|dk ¡ ¡ rtt ||¡ d S r# )r r% r& r' Ústdr r Úcov)r r r! r) Ústd_xZcov_x_xr r r Ú&test_expanding_consistency_var_std_cov: s r8 c C s | | j |dj|d}| j |dj|d}| j |dj|d}| j |dj| |d}t |d|| | ¡ | j |d | ¡}| j |dj|d}| j |dj|d} t |||| ¡ |dkr| j |d ¡ }
|
| j |d ¡ } | | j |d ¡ } t || |
|
| ¡ d S )Nr r$ g à?r )r r% r6 r r Úcorrr5 r( ) Z series_datar r! Z var_x_plus_yr) Zvar_yZcov_x_yZcorr_x_yr7 Zstd_yr* Zmean_yZmean_x_times_yr r r Ú*test_expanding_consistency_series_cov_corrL s ÿ
|
ÿr: c C sF | j |d ¡ }| j |d ¡ | j |d ¡ }t || d¡¡ d S )Nr Úfloat64)r r( r r. r r Zastype)r r Úresultr3 r r r Ú test_expanding_consistency_meanj s ÿÿr= c C s | ¡ ¡ }| j |d ¡ }| j |d | ¡}t| tr@| ¡ n
|
| ¡ ¡ }| tj }|||t|dk< t
|
||¡ tj|d d
< t
|
||¡ d S )Nr r" ) r r. r( r9 Ú
|
isinstancer r0 r r/ r r )r1 r r2 r* Zcorr_x_xÚexpr3 r r r Ú#test_expanding_consistency_constants s ÿ
|
|
ý
|
r@ c C s^ | j |d ¡ }| j |djdd}| ¡ ¡ | ¡ ¡ d dtj¡ }t ||| ¡ d S )Nr r r$ g ð?r, )r r% r. Úreplacer r/ r r )r r Zvar_unbiased_xZ var_biased_xZvar_debiasing_factors_xr r r Ú0test_expanding_consistency_var_debiasing_factors s
|
ÿ þrB )Únumpyr r Zpandasr Zpandas._testingZ_testingr r r
|
r Z parametrizeZnansumr r r+ r4 r8 r: r= r@ rB r r r r Ú<module> s$
|
|
|
|
|
|
