1*16467b97STreehugger Robotimport antlr3 2*16467b97STreehugger Robotimport testbase 3*16467b97STreehugger Robotimport unittest 4*16467b97STreehugger Robot 5*16467b97STreehugger Robot 6*16467b97STreehugger Robotclass t032subrulePredict(testbase.ANTLRTest): 7*16467b97STreehugger Robot def setUp(self): 8*16467b97STreehugger Robot self.compileGrammar() 9*16467b97STreehugger Robot 10*16467b97STreehugger Robot 11*16467b97STreehugger Robot def parserClass(self, base): 12*16467b97STreehugger Robot class TParser(base): 13*16467b97STreehugger Robot def recover(self, input, re): 14*16467b97STreehugger Robot # no error recovery yet, just crash! 15*16467b97STreehugger Robot raise 16*16467b97STreehugger Robot 17*16467b97STreehugger Robot return TParser 18*16467b97STreehugger Robot 19*16467b97STreehugger Robot 20*16467b97STreehugger Robot def testValid1(self): 21*16467b97STreehugger Robot cStream = antlr3.StringStream( 22*16467b97STreehugger Robot 'BEGIN A END' 23*16467b97STreehugger Robot ) 24*16467b97STreehugger Robot 25*16467b97STreehugger Robot lexer = self.getLexer(cStream) 26*16467b97STreehugger Robot tStream = antlr3.CommonTokenStream(lexer) 27*16467b97STreehugger Robot parser = self.getParser(tStream) 28*16467b97STreehugger Robot events = parser.a() 29*16467b97STreehugger Robot 30*16467b97STreehugger Robot 31*16467b97STreehugger Robot @testbase.broken("DFA tries to look beyond end of rule b", Exception) 32*16467b97STreehugger Robot def testValid2(self): 33*16467b97STreehugger Robot cStream = antlr3.StringStream( 34*16467b97STreehugger Robot ' A' 35*16467b97STreehugger Robot ) 36*16467b97STreehugger Robot 37*16467b97STreehugger Robot lexer = self.getLexer(cStream) 38*16467b97STreehugger Robot tStream = antlr3.CommonTokenStream(lexer) 39*16467b97STreehugger Robot parser = self.getParser(tStream) 40*16467b97STreehugger Robot events = parser.b() 41*16467b97STreehugger Robot 42*16467b97STreehugger Robot 43*16467b97STreehugger Robotif __name__ == '__main__': 44*16467b97STreehugger Robot unittest.main() 45