def test_negative_input(self):
with self.assertRaises(ValueError):
prime_generator(-5)
def test_correct_output(self):
self.assertEqual(prime_generator(5), [2, 3, 5])
def test_invalid_input(self):
with self.assertRaises(TypeError):
prime_generator("string")
def test_one_not_prime(self):
self.assertNotIn(1, prime_generator(8))
def test_two_is_prime(self):
self.assertIn(2, prime_generator(10))
def test_output_is_list(self):
self.assertIsInstance(prime_generator(10), list)