Python- Is number pronic，如果不是，则查找最近的pronic数字

def is_pronic_number(num):
    n = int(num**0.5)
    return n*(n+1) == num

def find_closest_pronic_number(num):
    if is_pronic_number(num):
        return num
    else:
        lower_num = upper_num = num

        while not is_pronic_number(lower_num):
            lower_num -= 1

        while not is_pronic_number(upper_num):
            upper_num += 1

        if num - lower_num < upper_num - num:
            return lower_num
        else:
            return upper_num