def test_add_multiple_edge_weight():
g = WeightedGraph()
g.add_node(u"A")
g.add_edge(u"A", u"C", 10)
g.add_edge(u"C", u"D", 5)
g.add_edge(u"A", u"B", 1)
g.add_edge(u"C", u"A", 5)
assert g.edgeWeights[(u"A", u"C")] == 10
assert g.edgeWeights[(u"C", u"A")] == 5
def test_adj():
g = WeightedGraph()
g.add_edge(u"A", u"C")
g.add_node(u"B")
print g.edges()
assert g.adjacent(u"A", u"C") is True
assert g.adjacent(u"C", u"A") is True
assert g.adjacent(u"B", u"C") is False
assert g.adjacent(u"B", u"A") is False
with pytest.raises(KeyError):
g.adjacent(u"A", u"D")
def test_add_edge_weight():
g = WeightedGraph()
g.add_node(u"A")
g.add_edge(u"A", u"C", 10)
assert g.edgeWeights[(u"A", u"C")] == 10
def test_has_node():
g = WeightedGraph()
g.add_node(u"A")
g.add_node(u"B")
assert g.has_node(u"A") is True
assert g.has_node(u"C") is False
def test_add_edge_one_new_node():
g = WeightedGraph()
g.add_node(u"A")
g.add_edge(u"A", u"C")
assert u"C" in g.nodes()
assert (u"A", u"C") in g.edges()
def test_add_edge():
g = WeightedGraph()
g.add_node(u"A")
g.add_node(u"B")
g.add_edge(u"A", u"B")
assert (u"A", u"B") in g.edges()
def test_add_node():
g = WeightedGraph()
g.add_node("A")
assert "A" in g.nodes()
