
    >qj                     ^    d Z ddl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 Zy)	z)test sparse matrix construction functions    N)assert_equal)
csr_matrix	csr_arraysparray)_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    ^/opt/rentech/trading_bot/.venv/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   sW     	3AMM!$EAa1aU)1773Aaiik2	3r   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' MX]]1q199;RWWQ^LM	M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	   c                  :   t        j                  ddgddgg      } t        j                  d      5  t	        j
                  |        d d d        t        j                  d      5  t	        j                  |        d d d        y # 1 sw Y   >xY w# 1 sw Y   y xY w)Nr   r   r   r   z.*switching.*sparse array int)match)r+   arraypytestdeprecated_callr   r*   r0   )r   s    r   =test_deprecated_warnings_output_defaults_switch_from_spmatrixr?   8   s    
1a&1a&!"A			&E	F a			&E	F a   s   B&BBB)__doc__r=   numpy.testingr   scipy.sparser   r   r   numpyr+   r   r	   r?   r9   r   r   <module>rD      s*    /  & 7 7  !)= )=Xr   