CampusX SDE Sheet | Write a program to print whether a given number is prime number or not

Write a program to print whether a given number is prime number or not

'''
Write a program to print whether a given number is prime number or not
'''

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

num = int(input("enter a number: "))
if is_prime(num):
    print(f"{num} is a prime number")
else:
    print(f"{num} is not a prime number")