def test_formula_set_params(self): f = Formula() f({'a': 2, 'b': 5}) self.assertEqual(f('a'), 2) self.assertEqual(f.get_param('b'), 5)
def test_calc(self): f = Formula() calc = Calc(f) @f(1) def a(b, c): return b + c @f(2) def b(c): return c * 2 f({'c': 4}) calc('a') self.assertEqual(f('a'), 12) self.assertEqual(f.get_param('b'), 8)