Пиши только выражение — оно подставится после return.
Все тесты
одинаковы для локальной и серверной проверки
Тест 1
Ввод: solution([1, 2, 3])
Ожидалось: [1, 4, 9]
Тест 2
Ввод: solution([0, 4, 5])
Ожидалось: [0, 16, 25]
Тест 3
Ввод: solution([-2, -3])
Ожидалось: [4, 9]
Тест 4
Ввод: solution([0, 1, 2])
Ожидалось: [0, 1, 4]
Тест 5
Ввод: solution([-1, -2])
Ожидалось: [1, 4]
Тест 6
Ввод: solution([])
Ожидалось: []
Локальная проверка и отправка в рейтинг используют один и тот же набор тестов. Если есть расхождение, причина обычно в лимитах времени или окружении раннера.