結果 : write a function to check prime number in python