def test_first_four_primes(self):
"""Test a middle case, four primes
"""
self.assertEqual(primes.first_n_primes(4), [2, 3, 5, 7])
def test_only_one_prime(self):
"""Test the corner case where only the first prime (2) is requested
"""
self.assertEqual(primes.first_n_primes(1), [2])
def test_first_three_primes(self):
"""Test the corner case where only the two frist primes are requested
"""
self.assertEqual(primes.first_n_primes(3), [2, 3, 5])
def test_first_10_primes(self):
"""Test an easy case: calculate the first ten primes.
"""
ten_primes = [2, 3, 5, 7, 11, 13, 17, 19, 23, 29]
self.assertEqual(primes.first_n_primes(10), ten_primes)
