54 lines
1.3 KiB
Python
54 lines
1.3 KiB
Python
import antlr3
|
|
import testbase
|
|
import unittest
|
|
|
|
|
|
class t039labels(testbase.ANTLRTest):
|
|
def setUp(self):
|
|
self.compileGrammar()
|
|
|
|
|
|
def lexerClass(self, base):
|
|
class TLexer(base):
|
|
def recover(self, input, re):
|
|
# no error recovery yet, just crash!
|
|
raise
|
|
|
|
return TLexer
|
|
|
|
|
|
def parserClass(self, base):
|
|
class TParser(base):
|
|
def recover(self, input, re):
|
|
# no error recovery yet, just crash!
|
|
raise
|
|
|
|
return TParser
|
|
|
|
|
|
def testValid1(self):
|
|
cStream = antlr3.StringStream(
|
|
'a, b, c, 1, 2 A FOOBAR GNU1 A BLARZ'
|
|
)
|
|
|
|
lexer = self.getLexer(cStream)
|
|
tStream = antlr3.CommonTokenStream(lexer)
|
|
parser = self.getParser(tStream)
|
|
ids, w = parser.a()
|
|
|
|
self.assertEqual(len(ids), 6, ids)
|
|
self.assertEqual(ids[0].text, 'a', ids[0])
|
|
self.assertEqual(ids[1].text, 'b', ids[1])
|
|
self.assertEqual(ids[2].text, 'c', ids[2])
|
|
self.assertEqual(ids[3].text, '1', ids[3])
|
|
self.assertEqual(ids[4].text, '2', ids[4])
|
|
self.assertEqual(ids[5].text, 'A', ids[5])
|
|
|
|
self.assertEqual(w.text, 'GNU1', w)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|
|
|
|
|