Update test_primes.py