Create prime_test.py