Skip to content

Instantly share code, notes, and snippets.

@arturaz
Created November 7, 2012 14:46
Show Gist options
  • Save arturaz/4032014 to your computer and use it in GitHub Desktop.
Save arturaz/4032014 to your computer and use it in GitHub Desktop.
val parser = new Parser {
var builder: GraphBuilder = _
var agency: File = _
var stops: File = _
var routes: File = _
var trips: File = _
var stopTimes: File = _
var calendar: File = _
var calendarDates: Option[File] = _
var frequencies: Option[File] = _
var transfers: Option[File] = _
override def parse(
builder: GraphBuilder, agency: File, stops: File, routes: File,
trips: File, stopTimes: File, calendar: File,
calendarDates: Option[File], frequencies: Option[File],
transfers: Option[File]
) {
this.builder = builder
this.agency = agency
this.stops = stops
this.routes = routes
this.trips = trips
this.stopTimes = stopTimes
this.calendar = calendar
this.calendarDates = calendarDates
this.frequencies = frequencies
this.transfers = transfers
}
}
val builder = new GraphBuilder
it("should call #parse with all files when they can be found") {
val dir = new File("data/test/gtfs/from_directory/full")
def file(name: String) = new File(dir, name+".txt")
def optFile(name: String) = Some(new File(dir, name+".txt"))
val (
agency, stops, routes, trips, stopTimes, calendar, calendarDates,
frequencies, transfers
) = (
file("agency"), file("stops"), file("routes"), file("trips"),
file("stop_times"), file("calendar"),
optFile("calendar_dates"), optFile("frequencies"), optFile("transfers")
)
parser.parse(builder, dir)
(
parser.builder, parser.agency, parser.stops, parser.routes,
parser.trips, parser.stopTimes, parser.calendar, parser.calendarDates,
parser.frequencies, parser.transfers
) should be === (
builder, agency, stops, routes, trips, stopTimes, calendar,
calendarDates, frequencies, transfers
)
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment