o
    Eb!                     @   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 g dg dg dgt g dg dg dgjg| _d S )	N      r         )r   r   r   r   r   )r	      r         )r   r      	   r   )r   Tcases)self r   A/usr/lib/python3/dist-packages/scipy/sparse/tests/test_extract.pysetup_method   s   
zTestExtract.setup_methodc                 C   s@   | j D ]}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   D   | j D ]}| }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       
$zTestExtract.test_trilc                 C   r   r   )r   r   r   r   Ztriur"   r#   r   r   r   	test_triu&   r&   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    