def test_negative(self):
self.assertEqual(
PrimeGenerator(-1).prime_generator(),
"Input should be a positive integer")
self.assertEqual(
PrimeGenerator(-50).prime_generator(),
"Input should be a positive integer")
def test_non_integer(self):
self.assertEqual(
PrimeGenerator("test").prime_generator(),
"Input should be an integer")
self.assertEqual(
PrimeGenerator([1, 2, 3]).prime_generator(),
"Input should be an integer")
def test_primes(self):
self.assertEqual(PrimeGenerator(10).prime_generator(), [2, 3, 5, 7])
self.assertEqual(
PrimeGenerator(20).prime_generator(), [2, 3, 5, 7, 11, 13, 17, 19])
def test_one(self):
self.assertEqual(
PrimeGenerator(1).prime_generator(), "One is not prime")
def test_zero(self):
self.assertEqual(
PrimeGenerator(0).prime_generator(), "Zero is not prime")
def test_non_negative(self):
self.assertEqual(
PrimeGenerator(-5).prime_generator(),
"Number should be a positive integer")
def test_non_integer(self):
self.assertEqual(
PrimeGenerator('a').prime_generator(), "Enter an integer")
def test_one(self):
self.assertEqual(
PrimeGenerator(1).prime_generator(), "1 is not a prime number")
def test_zero(self):
self.assertEqual(
PrimeGenerator(0).prime_generator(), "0 is not a prime number")