def test_define(self): exprs.clear() exprs.define("test_expr", "a + b") vexprs = visit.Expressions() self.assertEqual(1, len(vexprs)) self.assertEqual("test_expr", vexprs[0][0]) self.assertEqual("a + b", vexprs[0][1]) exprs.define("test_expr", "b*b") vexprs = visit.Expressions() self.assertEqual("test_expr", vexprs[0][0]) self.assertEqual("b*b", vexprs[0][1])
def exists(ename): """ Checks if an expression with the given name exists. """ expr_names = [expr[0] for expr in visit.Expressions()] if ename in expr_names: return True else: return False
def clear(): """ Deletes all existing expressions. """ expr_names = [expr[0] for expr in visit.Expressions()] for ename in expr_names: try: visit.DeleteExpression(ename) except: # must be a database expression pass
def test_clear(self): exprs.clear() self.assertEqual(0, len(visit.Expressions()))