1324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport antlr3 2324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport testbase 3324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport unittest 4324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 5324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverclass t015calc(testbase.ANTLRTest): 6324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver def setUp(self): 7324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver self.compileGrammar() 8324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 9324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 10324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver def _evaluate(self, expr, expected, errors=[]): 11324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver cStream = antlr3.StringStream(expr) 12324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver lexer = self.getLexer(cStream) 13324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver tStream = antlr3.CommonTokenStream(lexer) 14324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver parser = self.getParser(tStream) 15324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver result = parser.evaluate() 16324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver assert result == expected, "%r != %r" % (result, expected) 17324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver assert len(parser.reportedErrors) == len(errors), parser.reportedErrors 18324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 19324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 20324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver def testValid01(self): 21324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver self._evaluate("1 + 2", 3) 22324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 23324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 24324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver def testValid02(self): 25324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver self._evaluate("1 + 2 * 3", 7) 26324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 27324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 28324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver def testValid03(self): 29324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver self._evaluate("10 / 2", 5) 30324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 31324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 32324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver def testValid04(self): 33324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver self._evaluate("6 + 2*(3+1) - 4", 10) 34324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 35324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 36324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver def testMalformedInput(self): 37324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver self._evaluate("6 - (2*1", 4, ["mismatched token at pos 8"]) 38324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 39324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver # FIXME: most parse errors result in TypeErrors in action code, because 40324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver # rules return None, which is then added/multiplied... to integers. 41324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver # evaluate("6 - foo 2", 4, ["some error"]) 42324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 43324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 44324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverif __name__ == '__main__': 45324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver unittest.main() 46324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 47