distance_presenter_distance.rb

class DistancePresenter
  def initialize(unit)
    @unit = unit
  end
  
  def to_s
    "The distance between Warsaw and Paris is #{distance} #{unit}"
  end
  
  private
  attr_reader :unit
  
  def distance
    if unit == 'km'
      1591
    else
      998
    end
  end
end