def test_second_prime(self):
self.assertEqual(prime(2), 3)
def test_sixth_prime(self):
self.assertEqual(prime(6), 13)
def test_first_prime(self):
self.assertEqual(prime(1), 2)
def prime_range(n):
"""Returns a list of the first n primes"""
return [prime(i) for i in range(1, n + 1)]
def test_there_is_no_zeroth_prime(self):
with self.assertRaisesWithMessage(ValueError):
prime(0)
def test_big_prime(self):
self.assertEqual(prime(10001), 104743)
def test_there_is_no_zeroth_prime(self):
with self.assertRaises(ValueError) as err:
prime(0)
self.assertEqual(type(err.exception), ValueError)
self.assertEqual(err.exception.args[0], "there is no zeroth prime")