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")