Capgemini interview question

Prime number program in python