def test_total_primes(self):
self.assertEqual(euler.total_primes(1), [2])
self.assertEqual(euler.total_primes(2), [2, 3])
self.assertEqual(euler.total_primes(3), [2, 3, 5])
self.assertEqual(euler.total_primes(6), [2, 3, 5, 7, 11, 13])
def euler_problem7(find_nth_prime):
return euler.total_primes(find_nth_prime)[-1]