
    xf                     d    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	      Zy)
zG Unit tests for nonnegative least squares
Author: Uwe Schmitt
Sep 2008
    N)assert_)raises)nnls)arangedot)normc                       e Zd Zd Zd Zy)TestNNLSc                     t        d      j                  dd      }t        d      }t        ||      }t        ||      \  }}t	        |dk         t	        t        t        ||      |z
        dk         y )Ng      9@   g      @gHz>)r   reshaper   r   r   r   )selfaxyress        @/usr/lib/python3/dist-packages/scipy/optimize/tests/test_nnls.py	test_nnlszTestNNLS.test_nnls   sc    4L  A&3K!Ha3d
S1XaZ 4'(    c                     t         j                  j                  d      }|j                  d      }|j                  d      }t	        t
              5  t        ||d       d d d        y # 1 sw Y   y xY w)Ni  )d   r   )sizer      )maxiter)nprandomRandomStateuniformassert_raisesRuntimeErrorr   )r   rndmr   bs       r   test_maxiterzTestNNLS.test_maxiter   sa     yy$$T*LLjL)LLcL"<( 	"Aq!	" 	" 	"s   A++A4N)__name__
__module____qualname__r   r$    r   r   r
   r
      s    )"r   r
   )__doc__numpyr   numpy.testingr   pytestr   r    scipy.optimizer   r   r   numpy.linalgr   r
   r(   r   r   <module>r/      s(     ! *   " "r   