import math as mt def isPrime(num): if (num < 2 or num % 2 == 0): return num == 2 for i in range(3, mt.ceil(mt.sqrt(num + 1))): if (num % i == 0): return False return True