1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
import pygraphviz as pgv
stringify = pgv.testing.stringify
def test_subgraph():
G = pgv.AGraph()
G.add_edge(1, 2)
s = G.add_subgraph((1, 2), label="foo")
ans = """strict graph { { graph [label=foo]; 1 -- 2; } }"""
assert stringify(G) == ans
def test_subgraph_cluster():
G = pgv.AGraph(label="foo")
s = G.subgraph("cluster_a", label="<Hello<BR/>World>")
s.add_node("sa")
G.add_node("a")
ans = """strict graph { graph [label=foo];
{ graph [label=<Hello<BR/>World>]; sa; }
a; }"""
assert stringify(G) == " ".join(ans.split())
def test_subgraph_cluster_attribute():
G = pgv.AGraph()
s = G.subgraph(name="cluster_a")
s.node_attr["foo"] = "bar"
G.add_node("a")
G.node_attr["foo"] = "baz"
ans = """strict graph { node [foo=baz];
subgraph cluster_a {
graph [foo=bar];
}
a; }"""
assert stringify(G) == " ".join(ans.split())
|
