
    tKg                     P    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
y)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y)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| _        y )	N      r         )r   r   r   r   r   )r      r         )r   r      	   r   )r   Tcases)selfs    c/home/alanp/www/video.onchill/myenv/lib/python3.12/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 ]^  }t        j                  |      \  }}}t        |||ff|j                        }t        |j                         |j                                ` y )N)shape)r   r   findr   r   r   toarray)r   AIJVBs         r   	test_findzTestExtract.test_find   sR    AMM!$EAa1aU)1773Aaiik2 r   c           	          | j                   D ]]  }|j                         }dD ]F  }t        t        j                  ||      j                         t        j                  ||             H _ y N)r   r   r   r   )k)r   r   r   r   trilnpr   r   r"   r)   s       r   	test_trilzTestExtract.test_tril!   M    A		A'X]]1q199;RWWQ^L ( r   c           	          | j                   D ]]  }|j                         }dD ]F  }t        t        j                  ||      j                         t        j                  ||             H _ y r%   )r   r   r   r   triur+   r,   s       r   	test_triuzTestExtract.test_triu'   r.   r   c                 d   | 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              sJ  y )N)r   
isinstancer   r*   r   r0   r   )r   r   Ms      r   test_array_vs_matrixz TestExtract.test_array_vs_matrix-   s}    AhmmA.888hmmA.8881A!(--"2G<<<!(--"2G<<< r   N)__name__
__module____qualname__r   r#   r-   r1   r5    r   r   r	   r	   
   s    
 3MM=r   r	   )__doc__numpy.testingr   scipy.sparser   r   r   numpyr+   r   r	   r9   r   r   <module>r>      s!    / & 7 7  !)= )=r   