
    xf!                     H    d Z ddlmZ ddlmZ ddlZddlmZ  G d d      Zy)z)test sparse matrix construction functions    )assert_equal)
csr_matrixN)_extractc                   $    e Zd Zd Zd Zd Zd Zy)TestExtractc                    t        ddgg      t        ddgg      t        ddgg      t        dgdgg      t        dgdgg      t        dgdgg      t        ddgddgg      t        ddgddgg      t        ddgddgg      t        ddgddgg      t        g dg dg dg      t        g dg dg dg      j                  g| _        y )	N      r         )r	   r
   r   r   r   )r      r         )r   r      	   r   )r   Tcases)selfs    A/usr/lib/python3/dist-packages/scipy/sparse/tests/test_extract.pysetup_methodzTestExtract.setup_method   s    1w1w1wQCy!QCy!QCy!1qe}%1qe}%1qe}%1qe}%K<=K<=??

    c                     | j                   D ]N  }t        j                  |      \  }}}t        |j	                         t        ||f|f|j                               P y )N)shape)r   r   findr   toarrayr   r   )r   AIJVs        r   r   zTestExtract.find   sN     	LAMM!$EAaj1Q%!''&JK	Lr   c           	          | j                   D ]]  }|j                         }dD ]F  }t        t        j                  ||      j                         t        j                  ||             H _ y N)r   r	   r
   r   )k)r   r   r   r   trilnpr   r   Br%   s       r   	test_trilzTestExtract.test_tril    [     	MA		A' MX]]1q199;RWWQ^LM	Mr   c           	          | j                   D ]]  }|j                         }dD ]F  }t        t        j                  ||      j                         t        j                  ||             H _ y r!   )r   r   r   r   triur'   r(   s       r   	test_triuzTestExtract.test_triu&   r+   r   N)__name__
__module____qualname__r   r   r*   r.    r   r   r   r   
   s    
 L
MMr   r   )	__doc__numpy.testingr   scipy.sparser   numpyr'   r   r   r2   r   r   <module>r7      s     / & #  ! M  Mr   