
    !g                     L    d dl Zd dlmZmZ d dlmZmZmZm	Z	 d Z
d Zd Zd Zy)    N)assert_equalassert_)_calc_nodewise_row_calc_nodewise_weight_calc_approx_inv_covRegularizedInvCovariancec                      t         j                  j                  d       t         j                  j                  d      } t	        | dd      }t        |j                  d       y )NA 2      sizer   {Gz?)   )nprandomseednormalr   r   shape)Xghats     p/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/statsmodels/stats/tests/test_regularized_covariance.pytest_calc_nodewise_rowr      sE    IINN6
		g&AaD)DT"    c                     t         j                  j                  d       t         j                  j                  d      } t         j                  j                  d      }t	        | |dd      }t        t        |t                     y )Nr
   r   r   r   r   r   )r   r   r   r   r   r   
isinstancefloat)r   r   thats      r   test_calc_nodewise_weightr       s_    IINN6
		g&A99#D D!T2DJtU#$r   c                     t         j                  j                  d       t         j                  j                  d      } g }g }t	        d      D ]?  }t        | |d      }t        | ||d      }|j                  |       |j                  |       A t        t        j                  |      t        j                  |            }t        |j                  d       y )Nr
   r   r   r   r   )r   r   )r   r   r   r   ranger   r   appendr   arrayr   r   )r   ghat_lthat_lir   r   	theta_hats          r   test_calc_approx_inv_covr)      s    IINN6
		g&AFF1X !!Q-$Qa6dd	
 %RXXf%5rxx7GHI&)r   c                  b   t         j                  j                  d       t         j                  j                  d      } t         j                  j                  t        j                  | j                              }t        |       }|j                          t         j                  j                  |j                         |z
        }t        |dk         |j                  d       t        t        j                  |j                         dk(        t        j                  |dk(        kD         y )	Nr
   r   r   )exogg?g      ?)alphar   )r   r   r   r   linalginvcovTr   fitnormapprox_inv_covr   sum)r   r.   regcovdiffs       r   test_fitr7   (   s    IINN6
		g&A
))--qss
$C%1-F
JJL99>>&//1C78DD3J JJSJBFF6((*a/0266#(3CCDr   )numpyr   numpy.testingr   r   (statsmodels.stats.regularized_covariancer   r   r   r   r   r    r)   r7    r   r   <module>r<      s*     /4 4
#%*Er   