U
    ‘Ç-eà  ã                   @   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@t‚d S )Nr   é
   é   é   )Úmax_iter)ÚnpÚrandomÚRandomStateÚrandÚ
set_paramsÚfitZn_iter_ÚAssertionError)r   ÚrngÚXr   © r   úc/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/sklearn/mixture/tests/test_mixture.pyÚtest_gaussian_mixture_n_iter
   s    
r   c              	   C   sP   t j d¡}| dd¡}| jdd d}tjt|d |  |¡ W 5 Q R X 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   

