Is that Prime number?

OUTPUT

prime? : 1
False

OUTPUT

prime? : 2
True

OUTPUT

prime? : 3
True

OUTPUT

prime? : 37
True

OUTPUT

prime? : 45
False

Solution

def is_prime(num):
    if num == 1:
        return False
    else:
        is_prime = True
        for i in range(2,num):
            if(num%i == 0):
                is_prime = False
                break
        return is_prime

num = int(input("prime? :"))
print(is_prime(num))