def test_formula_fn_registry_args(self):
f = Formula()
@f(1)
def a(b, c):
return b + c
self.assertEqual(f.get_fn_by_name('a')[f.__ARGS_KEY__][1], ['b', 'c'])
def test_formula_fn_registry_call(self):
f = Formula()
@f(1)
def a(b):
return b ** 2
self.assertEqual(f.get_fn_by_name('a')[f.__CALL_KEY__][1](2), 4)
