def test_display_nodes():
from simple_graph import SimpleGraph
instance = SimpleGraph()
instance.add_node("waffles")
instance.add_node("waffles2")
instance.add_node("waffles3")
for waffle in ['waffles', 'waffles2', 'waffles3']:
assert waffle in instance.nodes()
def test_nodes():
from simple_graph import SimpleGraph
new_graph = SimpleGraph()
new_graph.graph = {'A': ['A', 'B', 'C']}
assert new_graph.nodes() == ['A']
def test_sg_nodes():
"""Ensure nodes returns a correct list of nodes in graph."""
from simple_graph import SimpleGraph
test_sg = SimpleGraph()
test_node = test_sg.nodes()
assert test_node == []
