26 lines
524 B
Python
26 lines
524 B
Python
# 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 |