Python语言判断是否为素数

import math
def isPrimeNumber(n):
    if n<2:
        return False   
    else:
        for i in range(2,n):
            if n%i==0:
                return False
        else:
            return True
if __name__=="__main__":
    num=eval(input("请输入一个整数:"))
    print("问:整数{}是素数吗？\n答:{}".format(num,isPrimeNumber(num)))