def test_square_3(self):
numbers = [1, 1, 1, 2, 5]
self.assertEqual(square(numbers), [1, 1, 1, 4, 25])
def test_square_1(self):
numbers = [1, 2, 3, 4, 5, 6]
self.assertEqual(square(numbers), [1, 4, 9, 16, 25, 36])
def test_square_2(self):
numbers = [10, 100, 1000]
self.assertEqual(square(numbers), [100, 10000, 1000000])