def test_add_edge():
from simple_graph import SimpleGraph
instance = SimpleGraph()
instance.add_node("waffles")
instance.add_node("waffles2")
instance.add_edge('waffles', 'waffles2')
assert 'waffles2' in instance.graph_dict['waffles']
def test_sg_adjacent_false():
"""Ensure adjacent returns false if a and b don't share an edge."""
from simple_graph import SimpleGraph
test_sg = SimpleGraph()
test_sg.add_node("a")
test_sg.add_node("b")
assert test_sg.adjacent("a", "b") is False
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 make_graph_three():
my_graph = SimpleGraph()
my_graph.add_node('a')
my_graph.add_node('b')
my_graph.add_node('c')
my_graph.add_edge('a', 'c')
my_graph.add_edge('b', 'a')
return my_graph
def test_sg_add_node_error():
"""Ensure duplicate node raises IndexError."""
from simple_graph import SimpleGraph
test_sg = SimpleGraph()
test_sg.add_node("node_a")
with pytest.raises(IndexError) as message:
test_sg.add_node("node_a")
assert "That node already exists dumbo." in str(message)
def test_sg_adjace_error_two():
"""Ensure adjacent raises an error if either node doesn't exist."""
from simple_graph import SimpleGraph
test_sg = SimpleGraph()
test_sg.add_node("a")
with pytest.raises(IndexError) as message:
test_sg.adjacent("a", "b")
assert "Second argument is not in the graph." in str(message)
def test_add_node():
"""Test add node."""
from simple_graph import SimpleGraph
instance = SimpleGraph()
instance.add_node("waffles")
instance.add_node("waffles2")
instance.add_node("waffles3")
assert 'waffles' in instance.graph_dict
assert 'waffles2' in instance.graph_dict
assert 'waffles3' in instance.graph_dict
def fixture_graph():
"""Create a fixture."""
from simple_graph import SimpleGraph
obj = SimpleGraph()
obj.add_node('A')
obj.add_node('B')
obj.add_node('C')
obj.add_node('D')
obj.add_node('E')
obj.add_node('F')
obj.add_node('G')
obj.add_node('H')
obj.add_edge('A', 'B', 45000000)
obj.add_edge('A', 'C', .5)
obj.add_edge('A', 'D', 8)
obj.add_edge('C', 'E', 5)
obj.add_edge('C', 'F', 1)
obj.add_edge('F', 'E', 2)
obj.add_edge('D', 'G', 7)
obj.add_edge('G', 'H')
return obj
def test_add_node():
from simple_graph import SimpleGraph
new_graph = SimpleGraph()
new_graph.add_node('A')
assert list(new_graph.graph.keys()) == ['A']
assert new_graph.graph['A'] == {}
def test_has_node():
from simple_graph import SimpleGraph
instance = SimpleGraph()
instance.add_node("waffles")
assert instance.has_node("waffles")
def test_del_non_existent_node2():
from simple_graph import SimpleGraph
instance = SimpleGraph()
instance.add_node("waffles")
with pytest.raises(ValueError):
instance.del_edge('waffles', 'waffles2')
def test_del_node():
from simple_graph import SimpleGraph
instance = SimpleGraph()
instance.add_node("waffles")
instance.del_node('waffles')
assert instance.graph_dict == {}
def make_graph_no_edges():
my_graph = SimpleGraph()
my_graph.add_node('a')
my_graph.add_node('b')
my_graph.add_node('c')
return my_graph
def test_sg_add_node(data, result):
"""Ensure new node is added to the graph."""
from simple_graph import SimpleGraph
test_sg = SimpleGraph()
test_sg.add_node(data)
assert test_sg.graph[data] == result
def test_sg_has_node_true():
"""Ensure has_node returns true when called with node in graph."""
from simple_graph import SimpleGraph
test_sg = SimpleGraph()
test_sg.add_node("a")
assert test_sg.has_node("a") is True
