def f(a, b):
    return a+b

assert f(1, 2) == 3