017-6kyu-Is a number prime.py

def is_prime(n):
  if n <= 0 or n == 1:
    return False
  i = 2
  while (i <= n ** 0.5 ):
    if n % i == 0:
      return False
    i += 1
  return True