
    _Mh                     X    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          Z
dS )z)test sparse matrix construction functions    )assert_equal)
csr_matrix	csr_arraysparrayN)_extractc                   ,    e Zd 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/test/jupyter/venv/lib/python3.11/site-packages/scipy/sparse/tests/test_extract.pysetup_methodzTestExtract.setup_method   s   !ug!ug!ugsA3i  sA3i  sA3i  !uaUm$$!uaUm$$!uaUm$$!uaUm$${{{;;;{{{;<<{{{;;;{{{;<<>



    c                     | j         D ]h}t          j        |          \  }}}t          |||ff|j                  }t          |                                |                                           id S )N)shape)r   r   findr   r   r   toarray)r   AIJVBs         r   	test_findzTestExtract.test_find   sq     	3 	3AM!$$EAa1aU)17333Aaiikk2222	3 	3r   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   r"   r)   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   c                    | j         D ]}t          t          j        |          t                    sJ t          t          j        |          t                    sJ t          |          }t          t          j        |          t                    rJ t          t          j        |          t                    rJ d S )N)r   
isinstancer   r*   r   r0   r   )r   r   Ms      r   test_array_vs_matrixz TestExtract.test_array_vs_matrix-   s     	= 	=AhmA..88888hmA..888881A!(-"2"2G<<<<<!(-"2"2G<<<<<<	= 	=r   N)__name__
__module____qualname__r   r#   r-   r1   r5    r   r   r	   r	   
   se        
 
 
 3 3 3M M MM M M= = = = =r   r	   )__doc__numpy.testingr   scipy.sparser   r   r   numpyr+   r   r	   r9   r   r   <module>r>      s    / / & & & & & & 7 7 7 7 7 7 7 7 7 7     ! ! ! ! ! !)= )= )= )= )= )= )= )= )= )=r   