input: Write a Python function to calculate the sum of cubes of the first n odd natural numbers.
  assert cube_sum(2) == 28
  assert cube_sum(3) == 153
  assert cube_sum(4) == 496
output: def cube_sum(n):
      return sum((2*i+1)**3 for i in range(n))