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
|
import antlr3
import testbase
import unittest
class t002lexer(testbase.ANTLRTest):
def setUp(self):
self.compileGrammar()
def lexerClass(self, base):
class TLexer(base):
def emitErrorMessage(self, msg):
# report errors to /dev/null
pass
def reportError(self, re):
# no error recovery yet, just crash!
raise re
return TLexer
def testValid(self):
stream = antlr3.StringStream('01')
lexer = self.getLexer(stream)
token = lexer.nextToken()
self.assertEqual(token.type, self.lexerModule.ZERO)
token = lexer.nextToken()
self.assertEqual(token.type, self.lexerModule.ONE)
token = lexer.nextToken()
self.assertEqual(token.type, self.lexerModule.EOF)
def testMalformedInput(self):
stream = antlr3.StringStream('2')
lexer = self.getLexer(stream)
try:
token = lexer.nextToken()
self.fail()
except antlr3.NoViableAltException as exc:
self.assertEqual(exc.unexpectedType, '2')
if __name__ == '__main__':
unittest.main()
|