add test_primes.py