input: Write a function to create a list of the squares of numbers from 1 to n using a list comprehension.
  assert square_numbers(5) == [1, 4, 9, 16, 25]
  assert square_numbers(3) == [1, 4, 9]
  assert square_numbers(7) == [1, 4, 9, 16, 25, 36, 49]
output: def square_numbers(n):
      return [i * i for i in range(1, n + 1)]