class TestParser(TestBase):
def setUp(self):
self.parser = Parser()
self.row = {'VAR': 1}
TestBase.setUp(self)
def _check_func(self, parse_result):
agg, functions = parse_result
for func in functions:
self.assertEqual(func.func.func_name, 'eval')
return functions[0]
def test_parse_formula(self):
func = self._check_func(
self.parser.parse_formula('VAR'))
self.assertEqual(func(self.row, self.parser.context), 1)
def test_bnf(self):
result = self.parser._build_bnf()
self.assertNotEqual(self.parser.bnf, None)
def test_parse_formula_with_var(self):
func = self._check_func(
self.parser.parse_formula('VAR + 1'))
self.assertEqual(func(self.row, self.parser.context), 2)
def test_parse_formula_bad_formula(self):
bad_formulas = [
'=BAD +++ FOR',
'2 +>+ 1',
'1 ** 2',
]
for bad_formula in bad_formulas:
self.assertRaises(ParseError, self.parser.parse_formula,
bad_formula)
