StoneX interview question

Python check for a prime