def test_add_default(self):
g = dotp.DotGraph()
g.add_default_node_attr(color="red", label="a")
self.assertEqual(len(g.default_node_attr), 2)
g.add_default_edge_attr(color="red")
self.assertEqual(len(g.default_edge_attr), 1)
g.add_default_graph_attr(color="red")
self.assertEqual(len(g.default_graph_attr), 1)
def test_add_edge_to_subgraph(self):
g = dotp.DotGraph()
s = g.add_subgraph('subG')
self.assertEqual(len(g), 0)
s.add_edge(1, 2)
self.assertEqual(len(s), 2)
self.assertEqual(len(g), 2)
g.add_edge(3, 4)
self.assertEqual(len(s), 2)
self.assertEqual(len(g), 4)
def test_addequalnodes(self):
g = dotp.DotGraph()
g.add_node('a', label="test1")
g.add_node('a', style="filled")
g.add_node('a', label="test2", texmode="math")
n = g.get_node('a')
self.assertEqual(len(list(g.nodes)), 1)
self.assertEqual(len(n.attr), 3)
self.assertEqual(n.attr['label'], 'test2')
self.assertEqual(set(n.attr.keys()), set(['label', 'style', 'texmode']))
def test_add_default_node_subgraph(self):
g = dotp.DotGraph()
g.add_default_node_attr(color="red")
g.add_node('a')
s = g.add_subgraph('S')
s.add_default_node_attr(style='test')
n = s.add_node('b')
self.failUnless('color' in n.attr)
self.assertEqual(n.attr['color'], "red")
self.failUnless('style' in n.attr)
nn = g.add_node(2)
self.failIf('style' in nn.attr)
def test_add_default_node(self):
g = dotp.DotGraph()
g.add_default_node_attr(color="red")
n = g.add_node('a')
self.failUnless('color' in n.attr)
self.assertEqual(n.attr['color'], "red")
g.add_default_node_attr(color="blue", label="b")
n = g.add_node(2)
self.failUnless('color' in n.attr)
self.assertEqual(n.attr['color'], "blue")
self.failUnless('label' in n.attr)
self.assertEqual(n.attr['label'], "b")
def test_add_subgraph(self):
g = dotp.DotGraph()
s = g.add_subgraph('subG')
self.assertEqual(s.name, 'subG')
def test_add_edge(self):
g = dotp.DotGraph()
e = g.add_edge('a', 'b', label='test')
e2 = g.add_edge('a', 'd', label='test2')
self.assertEqual(len(g), 3)
self.assertEqual(len(g.edges), 2)
def test_add_nonstring_nodes(self):
g = dotp.DotGraph()
n = g.add_node(1)
self.assertEqual(n.name, '1')
n = g.add_node(3.14)
self.assertEqual(n.name, '3.14')
def test_add_node(self):
g = dotp.DotGraph()
na = g.add_node('a')
nb = g.add_node('b')
self.assertEqual(len(list(g.nodes)), 2)
def test_create_with_attributes(self):
g = dotp.DotGraph(style='fancy', label='testgraph')
self.assertEqual(len(g.attr), 2)
self.assertEqual(g.attr['style'], 'fancy')
self.assertEqual(g.attr['label'], 'testgraph')
def test_create(self):
g = dotp.DotGraph('mygraph', strict=False, directed=True)
self.failUnless(g.name == "mygraph")
self.assertEqual(g.strict, False)
self.assertEqual(g.directed, True)
