from exploits.graphs.min_span_tree import kruskal
from test.exploits.dummy_output import DummyOutput


def test_small_kruskal():
    output = DummyOutput()
    n_nodes = 10
    kruskal.options['n_nodes'] = n_nodes
    kruskal.run(output)
    assert len(output) == n_nodes


def test_large_kruskal():
    output = DummyOutput()
    n_nodes = 1000
    kruskal.options['n_nodes'] = n_nodes
    kruskal.run(output)
    assert len(output) == n_nodes
