GoGuardian interview question

Python checking if a number is prime or not