
     j                        d dl Zd dlZd dlmZ d dlmZ d Zd Z	ej
                            dd ej        ej        g          d             Zej
                            dd d	g          d
             Zej
                            dd d	g          d             Zej
                            dd d	g          d             Zej
                            dd d	g          d             Zd Zd Zd ZdS )    N)Seriesc                 r    |                                                                                                  S N)notnaallxs    C:\Users\Terasoftware\OneDrive\Desktop\faahhh\fyndo\fyndo\venv\Lib\site-packages\pandas/tests/window/moments/test_moments_consistency_expanding.pyno_nansr      s$    7799==??       c                 r    |                                                                                                  S r   )isnullr   r   s    r
   all_nar      s&    88::>>!!!r   fc                 D    t          |                                           S r   )r   sum)vs    r
   <lambda>r      s     r   c                    |t           j        u r^t          |          sOt          |          r|j        s|dk    s3|                     t          j                            d                     |	                    |                                          }|	                    |          
                    |d          }t          j        ||           d S )Nr   z'np.sum has different behavior with NaNs)reasonmin_periodsT)funcraw)npr   r   r   emptyapplymarkerpytestmarkxfail	expandingapplytmassert_equal)requestall_datar   r   expanding_f_resultexpanding_apply_f_results         r
   )test_expanding_apply_consistency_sum_nansr)      s    BF{ x   	8	%-^	8Ca	 !!)R!SS   "+++DDHHJJ'11k1JJPPD  Q     O&(@AAAAAr   ddof   c                    |                      |                              |          }|dk                                                                     rJ |dk    rp| | z                       |                                          }|                      |                                          }t	          j        ||||z  z
             d S d S Nr   r*   r   )r!   varanymeanr#   r$   )r&   r   r*   var_xmean_x2mean_xs         r
   test_moments_consistency_varr5       s    ;77;;;FFE	  $$&&&&&qy <h&11k1JJOOQQ###<<AACC
w&6/:;;;;;	< <r   c                    |                      |                                          }|                      |                              |          }|dk                                                                    rJ | t          j        z  }d||t          |d          k    <   |dk    rt          j        ||dk     <   t          j        ||           d S )Nr   r.   r           r+      )	r!   countr/   r0   r   nanmaxr#   r$   )consistent_datar   r*   count_xr2   expecteds         r
   %test_moments_consistency_var_constantr?   ,   s    ''K'@@FFHHG%%+%>>BBBMME 	  $$&&&&&'H/2HWK+++,qy ' "1OE8$$$$$r   c                 f   |                      |                              |          }|dk                                                                     rJ |                      |                              |          }|dk                                                                     rJ t	          j        |||z             |                      |                              | |          }|dk                                                                     rJ t	          j        ||           d S r-   )r!   r/   r0   stdr#   r$   cov)r&   r   r*   r2   std_xcov_x_xs         r
   &test_expanding_consistency_var_std_covrE   :   s   ;77;;;FFE	  $$&&&&&;77;;;FFE	  $$&&&&& OE55=)))  [ 99==hT=RRG!  ""&&((((( OE7#####r   c                    | | z                        |                              |          }|                      |                              |          }|                      |                              |          }|                      |                              | |          }t          j        |d||z
  |z
  z             |                      |                              |           }|                      |                              |          }|                      |                              |          }	t          j        ||||	z  z             |dk    r|                      |                                          }
|                      |                                          }| | z                       |                                          }t          j        |||
|z  z
             d S d S )Nr   r.   g      ?r   )r!   r/   rB   r#   r$   corrrA   r1   )series_datar   r*   var_x_plus_yr2   var_ycov_x_ycorr_x_yrC   std_yr4   mean_ymean_x_times_ys                r
   *test_expanding_consistency_series_cov_corrrP   L   s    
{	"--+-FFJJPTJUU  !!k!::>>D>IIE!!k!::>>D>IIE###<<@@SW@XXG OGSL5$85$@ABBB $$$==BB;OOH!!k!::>>D>IIE!!k!::>>D>IIEOHg7888qy E &&;&??DDFF&&;&??DDFF;&11k1JJOOQQ 	 	6F?!CDDDDDE Er   c                 H   |                      |                                          }|                      |                                          |                      |                                          z  }t	          j        ||                    d                     d S )Nr   float64)r!   r1   r   r9   r#   r$   astype)r&   r   resultr>   s       r
   test_expanding_consistency_meanrU   j   s    K88==??F{337799




5
5
;
;
=
=	>  OFHOOI6677777r   c                 T   |                                                                  }|                      |                                          }|                      |                              |           }t	          | t
                    r|                                 n%|                                                                 }| t          j        z  }|||t          |d          k    <   t          j
        ||           t          j        |d d <   t          j
        ||           d S )Nr   r+   )r!   r9   r1   rG   
isinstancer   r;   r   r:   r#   r$   )r<   r   r=   r4   corr_x_xexpr>   s          r
   #test_expanding_consistency_constantrZ   s   s   ''))//11G&&;&??DDFFF(([(AAFFWWH ov..	)  ""&&((  'H/2HWK+++,OFH%%% &HQQQKOHh'''''r   c                    |                      |                                          }|                      |                              d          }|                                                                  |                                                                  dz
                      dt          j                  z  }t          j        |||z             d S )Nr   r   r.   g      ?r7   )r!   r/   r9   replacer   r:   r#   r$   )r&   r   var_unbiased_xvar_biased_xvar_debiasing_factors_xs        r
   0test_expanding_consistency_var_debiasing_factorsr`      s    ''K'@@DDFFN%%+%>>BBBJJL&002288::""$$s*gc26 ONL3J$JKKKKKr   )numpyr   r   pandasr   pandas._testing_testingr#   r   r   r   parametrizenansumr   r)   r5   r?   rE   rP   rU   rZ   r`    r   r
   <module>rh      s                   ! ! !" " " 88")RVLMMB B NMB !Q((< < )(< !Q((
% 
% )(
% !Q(($ $ )($" !Q((E E )(E:8 8 8( ( (,L L L L Lr   