Viết chương trình nhập vào số tự nhiên n. Kiểm tra n có là số nguyên tố hay không?

import math

# Nhập dữ liệu từ bàn phím

n = int(input("Nhập vào số tự nhiên n: "))

def kiem_tra_so_nguyen_to(n):

    # Số nguyên tố phải lớn hơn 1

    if n < 2:

        return False

    # Kiểm tra các ước từ 2 đến căn bậc hai của n

    for i in range(2, int(math.sqrt(n)) + 1):

        if n % i == 0:

            return False # Nếu chia hết cho số nào đó thì không phải số nguyên tố

    return True # Nếu không chia hết cho số nào thì là số nguyên tố

# Kết luận

if kiem_tra_so_nguyen_to(n):

    print(f"{n} là số nguyên tố.")

else:

    print(f"{n} không phải là số nguyên tố.")