import testmods.simple_pkg.b def subtract_one_add_two(n): return testmods.simple_pkg.b.subtract_one(n) + 2