
    -it                     F    S r SSKrSSKJr  SSKJr  SSKJr  SSK	J
r
  S rg)z'Test the minimum spanning tree function    N)assert_)	csr_array)minimum_spanning_treec                     / SQ/ SQ/ SQ/ SQ/ SQ/n [         R                  " U 5      n / SQ/ SQ/ SQ/ SQ/ SQ/n[         R                  " U5      n[        U 5      n[        U5      nUR	                  5       n[
        R                  " XAS	5        [
        R                  " UR	                  5       U S
5        [        USS9n[
        R                  " UR	                  5       US5        [         R                  R                  S5        S H  nS[         R                  R                  XU45      -   n [        U 5      n[        U5      n[        UR                  U:  5        [         R                  " US-
  5      nSXUS-   4'   [        U 5      n[        U5      n[         R                  " XU45      nSXUS-   4'   [
        R                  " UR	                  5       US	5        M     g )N)r      r   r   r   )r   r   r   r   r   )r   r   r         )r   r   r   r   r   )r   r   r	   r   r   )r   r   r   r   r   )r   r   r   r   r	   )r   r   r   r   r   zIncorrect spanning tree found.zOriginal graph was modified.T)	overwritez/Graph was not properly modified to contain MST.i  )r	   
            r   )npasarrayr   r   toarraynptassert_array_equalrandomseedr   nnzarangezeros)graphexpectedcsgraphmintreemintree_arrayNidxs          `/var/www/html/venv/lib/python3.13/site-packages/scipy/sparse/csgraph/tests/test_spanning_tree.pytest_minimum_spanning_treer!   	   s    	E
 JJuE 	H
 zz(#H G#G,GOO%M=;= 7??,e&( $Gt<G7??,h9; IINN4 BII$$aV,,E" (0a  ii!n#a%iE"'0 88QF# c!ew0(,	.+     )__doc__numpyr   numpy.testingr   testingr   scipy.sparser   scipy.sparse.csgraphr   r!    r"   r    <module>r*      s    -  !  " 69.r"   