o ht@sHdZddlZddlmZddlmZddlmZddl m Z ddZ dS)z'Test the minimum spanning tree functionN)assert_) csr_array)minimum_spanning_treecCsXgdgdgdgdgdg}t|}gdgdgdgdgdg}t|}t|}t|}|}t||d t||d t|d d }t||d tjddD]J}dtj||f}t|}t|}t |j |kt |d}d|||df<t|}t|}t ||f}d|||df<t||d q_dS)N)rrrr)rrrrr)rrr)rrrrr)rrrrr)rrrrr)rrrrr)rrrrrzIncorrect spanning tree found.zOriginal graph was modified.T) overwritez/Graph was not properly modified to contain MST.i)r r) npasarrayrrtoarraynptassert_array_equalrandomseedrnnzarangezeros)graphexpectedcsgraphmintree mintree_arrayNidxra/var/www/vscode/kcb/lib/python3.10/site-packages/scipy/sparse/csgraph/tests/test_spanning_tree.pytest_minimum_spanning_tree sV       r ) __doc__numpyr numpy.testingrtestingr scipy.sparserscipy.sparse.csgraphrr rrrrs