1 import antlr3 2 import testbase 3 import unittest 4 5 class t015calc(testbase.ANTLRTest): 6 def setUp(self): 7 self.compileGrammar() 8 9 10 def _evaluate(self, expr, expected, errors=[]): 11 cStream = antlr3.StringStream(expr) 12 lexer = self.getLexer(cStream) 13 tStream = antlr3.CommonTokenStream(lexer) 14 parser = self.getParser(tStream) 15 result = parser.evaluate() 16 self.assertEqual(result, expected) 17 self.assertEqual(len(parser.reportedErrors), len(errors), 18 parser.reportedErrors) 19 20 21 def testValid01(self): 22 self._evaluate("1 + 2", 3) 23 24 25 def testValid02(self): 26 self._evaluate("1 + 2 * 3", 7) 27 28 29 def testValid03(self): 30 self._evaluate("10 / 2", 5) 31 32 33 def testValid04(self): 34 self._evaluate("6 + 2*(3+1) - 4", 10) 35 36 37 def testMalformedInput(self): 38 self._evaluate("6 - (2*1", 4, ["mismatched token at pos 8"]) 39 40 # FIXME: most parse errors result in TypeErrors in action code, because 41 # rules return None, which is then added/multiplied... to integers. 42 # evaluate("6 - foo 2", 4, ["some error"]) 43 44 45 if __name__ == '__main__': 46 unittest.main() 47 48
