Check Primality (PracticePython)

import math
def main():

    while True:
        print(isPrime(int(input('Enter a number to check if it is prime: '))))

def isPrime(n):

    ans = [num for num in range(2,int(math.sqrt(n)+1)) if n % num == 0]
    return not bool(ans) 
    
main()