ZuciTech interview question

Write a Python function to check whether the number is prime or not