def test_adjacent(): g = Graph() 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_has_node(): g = Graph() 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(): g = Graph() 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_edge_one_new_node(): g = Graph() 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_node(): g = Graph() g.add_node("A") assert "A" in g.nodes()