
    M/Ph                     N    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dS )    N)assert_equalassert_)_calc_nodewise_row_calc_nodewise_weight_calc_approx_inv_covRegularizedInvCovariancec                      t           j                            d           t           j                            d          } t	          | dd          }t          |j        d           d S )NA 2      sizer   {Gz?)   )nprandomseednormalr   r   shape)Xghats     s/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/statsmodels/stats/tests/test_regularized_covariance.pytest_calc_nodewise_rowr      sW    INN6
	g&&AaD))DT"""""    c                  ,   t           j                            d           t           j                            d          } t           j                            d          }t	          | |dd          }t          t          |t                               d S )Nr
   r   r   r   r   r   )r   r   r   r   r   r   
isinstancefloat)r   r   thats      r   test_calc_nodewise_weightr       sw    INN6
	g&&A9##D D!T22DJtU##$$$$$r   c                     t           j                            d           t           j                            d          } g }g }t	          d          D ]O}t          | |d          }t          | ||d          }|                    |           |                    |           Pt          t          j	        |          t          j	        |                    }t          |j        d           d S )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    INN6
	g&&AFF1XX  !!Q--$Qa66dd$RXf%5%5rx7G7GHHI&)))))r   c                     t           j                            d           t           j                            d          } t           j                            t          j        | j                            }t          |           }|	                                 t           j        
                    |                                |z
            }t          |dk                |	                    d           t          t          j        |                                dk              t          j        |dk              k               d S )	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    INN6
	g&&A
)--qs
$
$C%1---F
JJLLL9>>&//11C788DD3J JJSJBF6((**a/0026#(3C3CCDDDDDr   )numpyr   numpy.testingr   r   (statsmodels.stats.regularized_covariancer   r   r   r   r   r    r)   r7    r   r   <module>r<      s        / / / / / / / /4 4 4 4 4 4 4 4 4 4 4 4
# # #% % %* * *E E E E Er   