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)