U
    3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   F/tmp/pip-unpacked-wheel-zrfo1fqw/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   Znumpyr	   Zsklearn.mixturer   r   markZparametrizer   r   r   r   r   r   <module>   s   

