3 OdC@sFdZddlZddlmZddljZddlmZddl m Z ddZ dS)z'Test the minimum spanning tree functionN)assert_) csr_matrix)minimum_spanning_treec Csdddddgdddddgdddddgdddddgdddddgg}tj|}dddddgdddddgdddddgdddddgdddddgg}tj|}t|}t|}tj|j|dtj|j|dt|dd}tj|j|d tjjd xdD]}dtjj||f}t|}t|}t |j |ktj |d}d|||df<t|}t|}tj ||f}d|||df<tj|j|dqWdS)NrzIncorrect spanning tree found.zOriginal graph was modified.T) overwritez/Graph was not properly modified to contain MST.i )rr r r ) npZasarrayrrnptZassert_array_equalZtodenserandomseedrZnnzZarangeZzeros)graphexpectedZcsgraphZmintreeNidxra/var/www/html/virt/lib64/python3.6/site-packages/scipy/sparse/csgraph/tests/test_spanning_tree.pytest_minimum_spanning_tree sF                 r) __doc__Znumpyr Z numpy.testingrZtestingrZ scipy.sparserZscipy.sparse.csgraphrrrrrrs