o
    `^h                     @   sJ   d Z ddlmZ ddlmZmZ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_matrix	csr_arraysparrayN)_extractc                   @   s4   e Zd Zd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   ]/home/air/shanriGPT/back/venv/lib/python3.10/site-packages/scipy/sparse/tests/test_extract.pysetup_method   s   
zTestExtract.setup_methodc                 C   sH   | j D ]}t|\}}}t|||ff|jd}t| |  qd S )N)shape)r   r   findr   r   r   toarray)r   AIJVBr   r   r   	test_find   s
   
zTestExtract.test_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   trilnpr   r   r   r%   r   r   r   	test_tril!      
$zTestExtract.test_trilc                 C   r    r!   )r   r   r   r   triur'   r(   r   r   r   	test_triu'   r*   zTestExtract.test_triuc                 C   sh   | j D ].}tt|tsJ tt|tsJ t|}tt|tr'J tt|tr1J qd S )N)r   
isinstancer   r&   r   r+   r   )r   r   Mr   r   r   test_array_vs_matrix-   s   
z TestExtract.test_array_vs_matrixN)__name__
__module____qualname__r   r   r)   r,   r/   r   r   r   r   r   
   s    r   )__doc__numpy.testingr   scipy.sparser   r   r   numpyr'   r   r   r   r   r   r   <module>   s    