def test_for_negative_input(self): result = prime_nums(-6) self.assertEqual( result, 'Cannot generate prime numbers for negative integers')
def test_integer_zero(self): result = prime_nums(0) self.assertEqual(result, 'Zero or one cannot be a prime number')
def test_returns_a_list_of_generated_prime_numbers(self): result = prime_nums(10) self.assertEqual( result, [3, 5, 7, 9], 'Should return a list of prime number from zero to 10')
def test_for_invalid_input(self): with self.assertRaises(ValueError): result = prime_nums('h')