module ArrayExtensions

# File lib/array_extensions.rb, line 4
def sum(identity = 0, &block)
  return identity unless size > 0

  if block_given?
    map(&block).sum
  else
    inject { |sum, element| sum + element }
  end
end