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_edge(self):
g = Graph([(EDGE, "a", "b", {})])
self.assertEqual(g.nodes, [])
self.assertEqual(g.edges, [Edge("a", "b", {})])
self.assertEqual(g.attrs, {})