U
    ½mœdû  ã                   @   sh   d dl Z d dlZd dlmZ d dlmZ e j deƒ eƒ g¡dd„ ƒZe j deƒ eƒ g¡dd„ ƒZ	dS )	é    N)ÚGaussianMixture)ÚBayesianGaussianMixtureÚ	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   ú[/home/sam/Atlas/atlas_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   )
r   Únumpyr	   Zsklearn.mixturer   r   ÚmarkZparametrizer   r   r   r   r   r   Ú<module>   s   

