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)