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)