
    (ph                     P    S r SSKJr  SSKJrJrJr  SSKrSSKJ	r	   " S S5      r
g)z)test sparse matrix construction functions    )assert_equal)
csr_matrix	csr_arraysparrayN)_extractc                   2    \ rS rSrS rS rS rS rS rSr	g)	TestExtract
   c                    [        SS//5      [        SS//5      [        SS//5      [        S/S//5      [        S/S//5      [        S/S//5      [        SS/SS//5      [        SS/SS//5      [        SS/SS//5      [        SS/SS//5      [        / SQ/ SQ/ SQ/5      [        / SQ/ SQ/ SQ/5      R                  /U l        g )	N      r         )r   r   r   r   r   )r      r         )r   r      	   r   )r   Tcases)selfs    R/var/www/html/venv/lib/python3.13/site-packages/scipy/sparse/tests/test_extract.pysetup_methodTestExtract.setup_method   s    !ug!ug!ugsA3i sA3i sA3i !uaUm$!uaUm$!uaUm$!uaUm${;{;<{;{;<>>

    c                     U R                    H[  n[        R                  " U5      u  p#n[        XBU44UR                  S9n[        UR                  5       UR                  5       5        M]     g )N)shape)r   r   findr   r   r   toarray)r   AIJVBs         r   	test_findTestExtract.test_find   sN    AMM!$EA1U)1773Aaiik2 r   c           
          U R                    HZ  nUR                  5       nS HA  n[        [        R                  " XS9R                  5       [
        R                  " X#S95        MC     M\     g N)r   r   r   r   )k)r   r   r   r   trilnpr   r    r$   r,   s       r   	test_trilTestExtract.test_tril!   I    A		A'X]]1199;RWWQ^L ( r   c           
          U R                    HZ  nUR                  5       nS HA  n[        [        R                  " XS9R                  5       [
        R                  " X#S95        MC     M\     g r(   )r   r   r   r   triur.   r/   s       r   	test_triuTestExtract.test_triu'   r2   r   c                    U R                    H  n[        [        R                  " U5      [        5      (       d   e[        [        R
                  " U5      [        5      (       d   e[        U5      n[        [        R                  " U5      [        5      (       a   e[        [        R
                  " U5      [        5      (       d  M   e   g )N)r   
isinstancer   r-   r   r4   r   )r   r    Ms      r   test_array_vs_matrix TestExtract.test_array_vs_matrix-   s    AhmmA.8888hmmA.88881A!(--"2G<<<<!(--"2G<<<< r   )r   N)
__name__
__module____qualname____firstlineno__r   r%   r0   r5   r:   __static_attributes__ r   r   r	   r	   
   s    
 3MM=r   r	   )__doc__numpy.testingr   scipy.sparser   r   r   numpyr.   r   r	   rA   r   r   <module>rF      s!    / & 7 7  !)= )=r   