U
    /d!                     @   sB   d Z ddlmZ ddlmZ ddlZddlmZ G dd dZdS )z)test sparse matrix construction functions    )assert_equal)
csr_matrixN)_extractc                   @   s,   e Zd Zdd Zdd Zdd Zdd Zd	S )
TestExtractc                 C   s   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 dddddgdddddgddd	d
dggt dddddgdddddgddd	d
dggjg| _d S )N      r                     	   )r   Tcases)self r   C/tmp/pip-unpacked-wheel-9gxwnfpp/scipy/sparse/tests/test_extract.pysetup_method   s    *,zTestExtract.setup_methodc                 C   s@   | j D ]4}t|\}}}t| t||f|f|jd qd S )N)shape)r   r   findr   toarrayr   r   )r   AIJVr   r   r   r      s    
zTestExtract.findc                 C   sD   | j D ]8}| }dD ]&}ttj||d tj||d qqd S N)r   r   r   r   )k)r   r   r   r   Ztrilnpr   r   Br    r   r   r   	test_tril    s    
zTestExtract.test_trilc                 C   sD   | j D ]8}| }dD ]&}ttj||d tj||d qqd S r   )r   r   r   r   Ztriur!   r"   r   r   r   	test_triu&   s    
zTestExtract.test_triuN)__name__
__module____qualname__r   r   r$   r%   r   r   r   r   r   
   s   r   )	__doc__Znumpy.testingr   Zscipy.sparser   Znumpyr!   r   r   r   r   r   r   <module>   s
   