def test_count(self):
sieve = PrimeSieve(10)
sieve.run_sieve()
sieve.get_primes = lambda: (i for i in range(3))
output = self.get_print_results_output(sieve, False, 1, 1)
results = self.parse_results(output)
self.assertIn("Count", results)
self.assertEqual(results["Count"], "3")
sieve.get_primes = lambda: (i for i in range(10))
output = self.get_print_results_output(sieve, False, 1, 1)
results = self.parse_results(output)
self.assertIn("Count", results)
self.assertEqual(results["Count"], "10")
def test_prime_limit(self):
sieve = PrimeSieve(11)
sieve.run_sieve()
primes = list(sieve.get_primes())
self.assertEqual(primes, [2,3,5,7,11])
def test_limit_2(self):
sieve = PrimeSieve(2)
sieve.run_sieve()
primes = list(sieve.get_primes())
self.assertEqual(primes, [2])
def test_limit_1000(self):
sieve = PrimeSieve(1000)
sieve.run_sieve()
primes_len = self.count(sieve.get_primes())
self.assertEqual(primes_len, 168)
def test_limit_10(self):
sieve = PrimeSieve(10)
sieve.run_sieve()
primes = list(sieve.get_primes())
self.assertEqual(primes, [2,3,5,7])
def test_limit_with_odd_sqr(self):
sieve = PrimeSieve(25)
sieve.run_sieve()
primes = list(sieve.get_primes())
self.assertEqual(primes, [2,3,5,7,11,13,17,19,23])
