def test_prime_number(self):
expected = [2, 3, 5]
self.assertEqual(prime_number(7), expected)
def test_no_less_than_two(self):
result = prime_number(1)
self.assertFalse(result)
def test_negative_number(self):
result = prime_number(-7)
self.assertFalse(result)
def test_check_start_point(self):
self.assertNotIn(1, prime_number(7))