o iC@sHdZddlZddlmZddlmZddlmZddl m Z ddZ dS)z'Test the minimum spanning tree functionN)assert_) csr_matrix)minimum_spanning_treecCsTgdgdgdgdgdg}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) npasarrayrrnptassert_array_equaltoarrayrandomseedrnnzarangezeros)graphexpectedcsgraphmintreeNidxrh/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/scipy/sparse/csgraph/tests/test_spanning_tree.pytest_minimum_spanning_tree sT        r) __doc__numpyr numpy.testingrtestingr scipy.sparserscipy.sparse.csgraphrrrrrrs