o
    Eb                     @   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      @gHz>)r   Zreshaper   r   r   r   )selfaxyres r   @/usr/lib/python3/dist-packages/scipy/optimize/tests/test_nnls.py	test_nnls   s   
zTestNNLS.test_nnlsc                 C   s`   t jd}|jdd}|jdd}tt t||dd W d    d S 1 s)w   Y  d S )Ni  )d   r   )sizer      )maxiter)npZrandomZRandomStateZ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__Znumpyr   Znumpy.testingr   Zpytestr   r   Zscipy.optimizer   r   r   Znumpy.linalgr   r   r   r   r   r   <module>   s    