def f(a, b): if a: x = 1 if a+b: x = 1 if a: x = 1 elif b: x = 2 if a: x = 1 elif b: x = 2 else: x = 3