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])