o
    ?Hh                     @   s`   d dl Zd dlZd dlmZmZ ejde e gdd Zejde e gdd Z	dS )    N)BayesianGaussianMixtureGaussianMixture	estimatorc                 C   sD   t jd}|dd}d}| j|d | | | j|ks J d S )Nr   
         )max_iter)nprandomRandomStaterand
set_paramsfitn_iter_)r   rngXr    r   b/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/sklearn/mixture/tests/test_mixture.pytest_gaussian_mixture_n_iter
   s   
r   c                 C   sf   t jd}|dd}| jdd d}tjt|d | | W d   dS 1 s,w   Y  dS )	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   msgr   r   r   6test_mixture_n_components_greater_than_n_samples_error   s   "r   )
numpyr	   r   sklearn.mixturer   r   markparametrizer   r   r   r   r   r   <module>   s   

