rmotr.com assignment 1

find_next_prime.py

def find_next_prime(n):
    '''
    >>> find_next_prime(6)
    7
    >>> find_next_prime(10)
    11
    >>> find_next_prime(11)
    13
    '''
    def is_prime(p):
        for i in range(2, p - 1):
            if p % i == 0:
                return False
        return True

    j = n
    while True:
        j += 1
        if is_prime(j):
            return j