# reflected_mod.py def reflected_mod(n, mod): r = n % (2 * mod) return mod - abs(mod - r) for i in range(20): print(f"{i} -> {reflected_mod(i, 5)}")