a
    ¶Àhá  ã                   @   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ÚfitZn_iter_)r   ÚrngÚXr   © r   ú^/var/www/html/assistant/venv/lib/python3.9/site-packages/sklearn/mixture/tests/test_mixture.pyÚtest_gaussian_mixture_n_iter
   s    
r   c                 C   sd   t j d¡}| dd¡}| jdd d}tjt|d |  |¡ W d  ƒ n1 sV0    Y  dS )	z*Check error when n_components <= n_samplesr   r   r   é   )Zn_componentsz"Expected n_samples >= n_components)ÚmatchN)	r	   r
   r   r   r   ÚpytestZraisesÚ
ValueErrorr   )r   r   r   Úmsgr   r   r   Ú6test_mixture_n_components_greater_than_n_samples_error   s    r   )
Únumpyr	   r   Zsklearn.mixturer   r   ÚmarkZparametrizer   r   r   r   r   r   Ú<module>   s   

