U
    »mœd’  ã                   @   s^   d Z ddlZddlmZ ddlmZ ddlm	Z	 ddlm
Z
mZ ddlmZ G dd	„ d	ƒZdS )
zG Unit tests for nonnegative least squares
Author: Uwe Schmitt
Sep 2008
é    N)Úassert_)Úraises)Únnls)ÚarangeÚdot)Únormc                   @   s   e Zd Zdd„ Zdd„ ZdS )ÚTestNNLSc                 C   sZ   t dƒ dd¡}t dƒ}t||ƒ}t||ƒ\}}t|dk ƒ ttt||ƒ| ƒdk ƒ d S )Ng      9@éÿÿÿÿé   g      @gH¯¼šò×z>)r   Zreshaper   r   r   r   )ÚselfÚaÚxÚyÚres© r   úW/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/scipy/optimize/tests/test_nnls.pyÚ	test_nnls   s    
zTestNNLS.test_nnlsc              	   C   sJ   t j d¡}|jdd�}|jdd�}ttƒ� t||dd� W 5 Q R X d S )NiÒ  )éd   r   )Úsizer   é   )Úmaxiter)ÚnpÚrandomZRandomStateÚuniformÚassert_raisesÚRuntimeErrorr   )r   Zrndmr   Úbr   r   r   Útest_maxiter   s
    
zTestNNLS.test_maxiterN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r   r      s   r   )Ú__doc__Únumpyr   Znumpy.testingr   Zpytestr   r   Zscipy.optimizer   r   r   Znumpy.linalgr   r   r   r   r   r   Ú<module>   s   