input: Write a python function to print the squares of all the numbers from 1 to a given number.
  assert square_numbers(3)==[1, 4, 9]
  assert square_numbers(5)==[1, 4, 9, 16, 25]
  assert square_numbers(1)==[1]
output: def square_numbers(n):
      return [i * i for i in range(1, n + 1)]