# HumanEval/150 # Loki Mode Multi-Agent Solution # Attempts: 1 # Passed: True def x_or_y(n, x, y): """A simple program which should return the value of x if n is a prime number and should return the value of y otherwise. Examples: for x_or_y(7, 34, 12) == 34 for x_or_y(15, 8, 5) == 5 """ if n <= 1: return y if n == 2: return x if n % 2 == 0: return y i = 3 while i * i <= n: if n % i == 0: return y i += 2 return x