def test_first_few_primes(self): primes = solution.primes() self.assertEqual(2, next(primes)) self.assertEqual(3, next(primes)) self.assertEqual(5, next(primes)) self.assertEqual(7, next(primes)) self.assertEqual(11, next(primes)) self.assertEqual(13, next(primes))
def test_primes(self): primes = solution.primes() first_five = list(islice(primes, 5)) self.assertEqual(first_five, [2, 3, 5, 7, 11])
def test_primes(): assert primes(6) == [2, 3, 5, 7, 11, 13]