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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
# -*- coding: utf-8 -*-
import os
import unittest
from StringIO import StringIO
import textwrap
import stringtemplate3
from antlr3.dottreegen import toDOT
from antlr3.treewizard import TreeWizard
from antlr3.tree import CommonTreeAdaptor
class TestToDOT(unittest.TestCase):
"""Test case for the toDOT function."""
def setUp(self):
self.adaptor = CommonTreeAdaptor()
self.tokens = [
"", "", "", "", "", "A", "B", "C", "D", "E", "ID", "VAR"
]
self.wiz = TreeWizard(self.adaptor, self.tokens)
def testNone(self):
"""toDOT()"""
treeST = stringtemplate3.StringTemplate(
template=(
"digraph {\n" +
" $nodes$\n" +
" $edges$\n" +
"}\n")
)
edgeST = stringtemplate3.StringTemplate(
template="$parent$ -> $child$\n"
)
tree = self.wiz.create("(A B (B C C) (B (C D D)))")
st = toDOT(tree, self.adaptor, treeST, edgeST)
result = st.toString()
expected = textwrap.dedent(
'''\
digraph {
n0 [label="A"];
n1 [label="B"];
n2 [label="B"];
n3 [label="C"];
n4 [label="C"];
n5 [label="B"];
n6 [label="C"];
n7 [label="D"];
n8 [label="D"];
n0 -> n1
n0 -> n2
n2 -> n3
n2 -> n4
n0 -> n5
n5 -> n6
n6 -> n7
n6 -> n8
}
'''
)
self.assertEqual(result, expected)
if __name__ == "__main__":
unittest.main(testRunner=unittest.TextTestRunner(verbosity=2))
|
