
     h!                     P    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dS )z)test sparse matrix construction functions    )assert_equal)
csr_matrixN)_extractc                   &    e Zd Zd Zd Zd Zd ZdS )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| _        d S )	N      r         )r	   r
   r   r   r   )r      r         )r   r      	   r   )r   Tcases)selfs    [/var/www/html/Sam_Eipo/venv/lib/python3.11/site-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}%%KKK<==KKK<==?



    c                     | j         D ]T}t          j        |          \  }}}t          |                                t          ||f|f|j                             Ud S )N)shape)r   r   findr   toarrayr   r   )r   AIJVs        r   r   zTestExtract.find   sh     	L 	LAM!$$EAaj1Q%!'&J&J&JKKKK	L 	Lr   c           	          | j         D ]e}|                                }dD ]L}t          t          j        ||                                          t          j        ||                     Mfd S N)r   r	   r
   r   )k)r   r   r   r   trilnpr   r   Br%   s       r   	test_trilzTestExtract.test_tril         	M 	MA		A' M MX]1q11199;;RWQ^^^LLLLM	M 	Mr   c           	          | j         D ]e}|                                }dD ]L}t          t          j        ||                                          t          j        ||                     Mfd S 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 L L
M M MM M M M Mr   r   )	__doc__numpy.testingr   scipy.sparser   numpyr'   r   r   r2   r   r   <module>r7      s    / / & & & & & & # # # # # #     ! ! ! ! ! ! M  M  M  M  M  M  M  M  M  Mr   