def test_negative(self): with self.assertRaises(TypeError): prime_numbers(-5)
def test_answer(self): self.assertEquals(prime_numbers(11), [1, 2, 3, 5, 7, 11])
def test_integer(self): with self.assertRaises(TypeError): prime_numbers([])
def test_zero_is_not_prime(self): self.assertNotIn(0, prime_numbers(5))
def test_one_is_prime(self): self.assertIn(1, prime_numbers(5))
def test_returns_list(self): self.assertIsInstance(prime_numbers(5), list)