U
    }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@td S )Nr   
         )max_iter)nprandomRandomStaterand
set_paramsfitZn_iter_AssertionError)r   rngXr    r   F/tmp/pip-unpacked-wheel-ig1s1lm8/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   )
Znumpyr	   r   Zsklearn.mixturer   r   markZparametrizer   r   r   r   r   r   <module>   s   

