Python Code Challenge

209 - Write a function to test if a number is prime and write test cases for it.