def test_graph_with_attributes(self): g = Graph( [ (ATTR, "foo", "1"), (ATTR, "title", "Testing Attrs"), (NODE, "a", {"color": "green"}), (NODE, "c", {}), (NODE, "b", {"label": "Beta!"}), (EDGE, "b", "c", {}), (EDGE, "a", "b", {"color": "blue"}), (ATTR, "bar", "true"), ] ) self.assertEqual( g.nodes, [ Node("a", {"color": "green"}), Node("c", {}), Node("b", {"label": "Beta!"}), ], ) self.assertEqual( g.edges, [Edge("b", "c", {}), Edge("a", "b", {"color": "blue"})] ) self.assertEqual(g.attrs, {"foo": "1", "title": "Testing Attrs", "bar": "true"})
def test_graph_with_one_node_with_keywords(self): g = Graph([(NODE, "a", {"color": "green"})]) self.assertEqual(g.nodes, [Node("a", {"color": "green"})]) self.assertEqual(g.edges, []) self.assertEqual(g.attrs, {})
def test_graph_with_one_node(self): g = Graph([(NODE, "a", {})]) self.assertEqual(g.nodes, [Node("a", {})]) self.assertEqual(g.edges, []) self.assertEqual(g.attrs, {})