
    {Kg                         d dl Zd dlZd dlmZmZ ej                  j                  d e        e       g      d        Zej                  j                  d e        e       g      d        Z	y)    N)BayesianGaussianMixtureGaussianMixture	estimatorc                     t         j                  j                  d      }|j                  dd      }d}| j	                  |       | j                  |       | j                  |k(  sJ y )Nr   
         )max_iter)nprandomRandomStaterand
set_paramsfitn_iter_)r   rngXr
   s       f/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/sklearn/mixture/tests/test_mixture.pytest_gaussian_mixture_n_iterr   
   s]     ))


"CQAH(+MM!(((    c                    t         j                  j                  d      }|j                  dd      }| j	                  d       d}t        j                  t        |      5  | j                  |       ddd       y# 1 sw Y   yxY w)	z*Check error when n_components <= n_samplesr   r   r      )n_componentsz"Expected n_samples >= n_components)matchN)	r   r   r   r   r   pytestraises
ValueErrorr   )r   r   r   msgs       r   6test_mixture_n_components_greater_than_n_samples_errorr      sc     ))


"CQAb)
.C	z	-a 
.	-	-s   !A<<B)
numpyr   r   sklearn.mixturer   r   markparametrizer   r    r   r   <module>r%      st      D '8:Q:S&TU) V) '8:Q:S&TU Vr   