input: Write a python function to partition a list into sublists of a given size.
  assert partition([1, 2, 3, 4, 5, 6, 7, 8, 9, 10], 3) == [[1, 2, 3], [4, 5, 6], [7, 8, 9], [10]]
  assert partition([1, 2, 3, 4, 5], 2) == [[1, 2], [3, 4], [5]]
  assert partition([1, 2, 3], 1) == [[1], [2], [3]]
output: def partition(lst, size):
      return [lst[i:i + size] for i in range(0, len(lst), size)]