160 lines
3.7 KiB
Python
160 lines
3.7 KiB
Python
import antlr3
|
|
import testbase
|
|
import unittest
|
|
import textwrap
|
|
|
|
|
|
class t022scopes(testbase.ANTLRTest):
|
|
def setUp(self):
|
|
self.compileGrammar()
|
|
|
|
|
|
def parserClass(self, base):
|
|
class TParser(base):
|
|
def emitErrorMessage(self, msg):
|
|
# report errors to /dev/null
|
|
pass
|
|
|
|
def reportError(self, re):
|
|
# no error recovery yet, just crash!
|
|
raise re
|
|
|
|
return TParser
|
|
|
|
|
|
def testa1(self):
|
|
cStream = antlr3.StringStream('foobar')
|
|
lexer = self.getLexer(cStream)
|
|
tStream = antlr3.CommonTokenStream(lexer)
|
|
parser = self.getParser(tStream)
|
|
parser.a()
|
|
|
|
|
|
def testb1(self):
|
|
cStream = antlr3.StringStream('foobar')
|
|
lexer = self.getLexer(cStream)
|
|
tStream = antlr3.CommonTokenStream(lexer)
|
|
parser = self.getParser(tStream)
|
|
|
|
self.assertRaises(antlr3.RecognitionException, parser.b, False)
|
|
|
|
|
|
def testb2(self):
|
|
cStream = antlr3.StringStream('foobar')
|
|
lexer = self.getLexer(cStream)
|
|
tStream = antlr3.CommonTokenStream(lexer)
|
|
parser = self.getParser(tStream)
|
|
parser.b(True)
|
|
|
|
|
|
def testc1(self):
|
|
cStream = antlr3.StringStream(
|
|
textwrap.dedent('''\
|
|
{
|
|
int i;
|
|
int j;
|
|
i = 0;
|
|
}
|
|
'''))
|
|
|
|
lexer = self.getLexer(cStream)
|
|
tStream = antlr3.CommonTokenStream(lexer)
|
|
parser = self.getParser(tStream)
|
|
symbols = parser.c()
|
|
|
|
self.assertEqual(
|
|
symbols,
|
|
set(['i', 'j'])
|
|
)
|
|
|
|
|
|
def testc2(self):
|
|
cStream = antlr3.StringStream(
|
|
textwrap.dedent('''\
|
|
{
|
|
int i;
|
|
int j;
|
|
i = 0;
|
|
x = 4;
|
|
}
|
|
'''))
|
|
|
|
lexer = self.getLexer(cStream)
|
|
tStream = antlr3.CommonTokenStream(lexer)
|
|
parser = self.getParser(tStream)
|
|
|
|
self.assertRaisesRegex(RuntimeError, r'x', parser.c)
|
|
|
|
|
|
def testd1(self):
|
|
cStream = antlr3.StringStream(
|
|
textwrap.dedent('''\
|
|
{
|
|
int i;
|
|
int j;
|
|
i = 0;
|
|
{
|
|
int i;
|
|
int x;
|
|
x = 5;
|
|
}
|
|
}
|
|
'''))
|
|
|
|
lexer = self.getLexer(cStream)
|
|
tStream = antlr3.CommonTokenStream(lexer)
|
|
parser = self.getParser(tStream)
|
|
symbols = parser.d()
|
|
|
|
self.assertEqual(
|
|
symbols,
|
|
set(['i', 'j'])
|
|
)
|
|
|
|
|
|
def teste1(self):
|
|
cStream = antlr3.StringStream(
|
|
textwrap.dedent('''\
|
|
{ { { { 12 } } } }
|
|
'''))
|
|
|
|
lexer = self.getLexer(cStream)
|
|
tStream = antlr3.CommonTokenStream(lexer)
|
|
parser = self.getParser(tStream)
|
|
res = parser.e()
|
|
|
|
self.assertEqual(res, 12)
|
|
|
|
|
|
def testf1(self):
|
|
cStream = antlr3.StringStream(
|
|
textwrap.dedent('''\
|
|
{ { { { 12 } } } }
|
|
'''))
|
|
|
|
lexer = self.getLexer(cStream)
|
|
tStream = antlr3.CommonTokenStream(lexer)
|
|
parser = self.getParser(tStream)
|
|
res = parser.f()
|
|
|
|
self.assertIsNone(res)
|
|
|
|
|
|
def testf2(self):
|
|
cStream = antlr3.StringStream(
|
|
textwrap.dedent('''\
|
|
{ { 12 } }
|
|
'''))
|
|
|
|
lexer = self.getLexer(cStream)
|
|
tStream = antlr3.CommonTokenStream(lexer)
|
|
parser = self.getParser(tStream)
|
|
res = parser.f()
|
|
|
|
self.assertIsNone(res)
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|