spec/support/journey_pattern_helper.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

module Support
  module JourneyPatternHelper
    def generate_journey_pattern_costs distance, time
      costs = {}
      (journey_pattern.stop_points.size - 1).times do |i|
        start, finish = journey_pattern.stop_points[i..i+1]
        costs["#{start.stop_area_id}-#{finish.stop_area_id}"] = {
          distance: (distance.respond_to?(:call) ? distance.call(i) : distance),
          time: (time.respond_to?(:call) ? time.call(i) : time)
        }
      end
      costs
    end
  end
end

RSpec.configure do | config |
  config.include Support::JourneyPatternHelper, type: :model
end