input: Write a python function to calculate the sum of all divisors of a number.
  assert divisor_sum(16) == 31
  assert divisor_sum(10) == 18
  assert divisor_sum(15) == 24
output: def divisor_sum(n):
      return sum(i for i in range(1, n + 1) if n % i == 0)