def test_invalid_number_of_primes_is_error(self):
with self.assertRaises(ValueError):
next(primes.gen("a"))
def test_value_of_tenth_prime(self):
self.assertEquals(list(primes.gen(10))[-1], 29)
def test_zero_number_of_primes_is_error(self):
with self.assertRaises(AssertionError):
next(primes.gen(0))
def test_number_of_primes_is_correct(self):
self.assertEquals(len(list(primes.gen(10))), 10)
