def test_nodes_of_filled_weight_graph_has_all_nodes(num):
"""Test that nodes lists all the nodes in a graph."""
from weight_graph import Graph
g = Graph()
for x in range(num):
g.add_node(x)
assert len(g.nodes()) == num
assert sorted(g.nodes()) == list(range(num))
def test_dublicate_values_to_a_weight_graph_adds_some_nodes(num):
"""Test that adding duplicate values to the graph add only unique items."""
from weight_graph import Graph
g = Graph()
for x in range(num):
g.add_node(x % 5)
assert len(g.nodes()) == 5 if num > 5 else num
def test_adding_unique_values_to_a_weight_graph_adds_all_nodes(num):
"""Test that adding unique values to the graph adds all of them."""
from weight_graph import Graph
g = Graph()
for x in range(num):
g.add_node(x)
assert len(g.nodes()) == num
def test_nodes(node, result):
"""Test to check if all nodes are there."""
from weight_graph import Graph
g = Graph()
for idx in node:
g.add_node(idx)
assert g.nodes() == result