3
���h� � @ sz d dl Zd dlmZ d dljZdd� Zdd� Zdd� Z d d
� Z
d d � Z d
d� Z dd� Z
dd� Zdd� Zdd� Zdd� ZdS )� N)�Seriesc sd d� ��fdd� }|| �}|j td �dfdf }|jjd�|_|| d | d �}tj||dd� d S )Nc s t t | � �f ����|�S )N)�getattr)�obj�obj2)�dispatch�kwargs�name� �</tmp/pip-build-5_djhm0z/pandas/pandas/tests/window/common.py�
get_result s z)check_pairwise_moment.<locals>.get_result� � F)Z check_names)N)�loc�slice�indexZ droplevel�tm�assert_series_equal)�framer r r r �result�expectedr )r r r r
�check_pairwise_moment s r c K s t | j|f|�|�|�S )N)r Zewm)�A�B�comr r r r r
�ew_func s r c C sP t ||d| dd�}tj|jd d� �j� s.t�tj|jdd � �j� sLt�d S )N� r
)r r r r � min_periods� )r �np�isnan�values�all�AssertionError�any)r r r r r r r
�check_binary_ew s r$ c C s� t ||d| |d�}tj|jd d� �j� s.t�tj|jdd � �j� sLt�tg tjd�}t ||d| |d�}t j
||� t tdg�tdg�d| |d�}t j
|ttj g�� d S )Nr )r r � )Zdtype�2 g �?) r r r r r! r" r# r �float64r r �NaN)r r r r r �emptyr r r
�check_binary_ew_min_periods s r* c C s* || �}|r&|| �}t j||jd�� d S )Nr' )r � assert_equalZastype)�x�meanZ mock_mean�mean_xr r r r
� moments_consistency_mock_mean1 s r/ c C s� || �}|| �}|| | �}|r~t | t�r0| j� n
| j� j� } | tj }
| |
|t|d�k< tj||
� tj|
d d �< tj||
� d S )Nr )�
isinstancer �maxr �nanr r+ ) r, � is_constantr �countr- �corr�count_xr. Zcorr_x_x�expr r r r
� moments_consistency_is_constant; s
r8 c C s8 |r4|r4|r4|| �}|| �}|| �}t j||| � d S )N)r r+ )r, �
var_biased� var_unbiasedZvar_debiasing_factorsZvar_unbiased_xZ var_biased_xZvar_debiasing_factors_xr r r
�)moments_consistency_var_debiasing_factorsN s
r; c
C s� || �}|| �}x�||gD ]�} | | �}
|
dk j � j � s<t�| |krd|| | �} tj|
| || � |r|
dkj � j � s~t�| tj } d| |t|d�k< | |kr�tj| |dk < tj|
| � qW d S )Nr g r � )r# r" r r+ r r2 r1 )
r, r3 r r4 r- r: r9 r6 r. �var�var_xZmean_x2r r r r
� moments_consistency_var_dataY s
r? c C sn xh||f||fgD ]T\}}|| �}|| �}|dk j � j � s@t�|dk j � j � sVt�tj||| � qW d S )Nr )r# r" r r+ ) r, � std_unbiasedr: �
std_biasedr9 �stdr= r> �std_xr r r
� moments_consistency_std_dataq s rD c C sp xj||f||fgD ]V\}}|| �}|dk j � j � s8t�|r|| | �}|dk j � j � s\t�tj||� qW d S )Nr )r# r" r r+ ) r, � cov_unbiasedr: �
cov_biasedr9 �covr= r> Zcov_x_xr r r
� moments_consistency_cov_data| s
rH c C s t | t��r| } || �}
| j� j| j� �s*|| | �} || | �} tj| | � x�|||f|||fgD ]�\}
}}|| �}|
| �}|r`|| | �}|| | �}tj||� || | �}|| �}tj|d|| | � |
| �}tj| ||| � ||kr`|| �}|| | �}tj|||
| � q`W d S )Ng �?)r0 r Zisna�equalsr r+ )r, r5 r- rA r@ rE r: r9 rF �yr. Zcorr_x_yZcorr_y_xrB r= rG r> rC Zcov_x_yZcov_y_xZ var_x_plus_yZvar_yZstd_yZmean_yZmean_x_times_yr r r
� moments_consistency_series_data� s2
rK )Znumpyr Zpandasr Zpandas._testingZ_testingr r r r$ r* r/ r8 r; r? rD rH rK r r r r
�<module> s
|