gtfs-utils/test/read-stop-times.js

Utilities to process GTFS data sets.

'use strict'

const test = require('tape')
const {readFilesFromFixture} = require('./lib')

const readCsv = require('../read-csv')
const readStopTimes = require('../lib/read-stop-times')

const readFile = (file) => {
	return readCsv(require.resolve('sample-gtfs-feed/gtfs/' + file + '.txt'))
}

const noFilter = {
	trip: () => true,
	stopTime: () => true,
	frequenciesRow: () => true,
}

test('read-stop-times works', async (t) => {
	const stopTimes = readStopTimes(readFile, noFilter)
	const res = []
	for await (const st of stopTimes) res.push(st)

	t.deepEqual(res, [
		{
			tripId: 'a-downtown-all-day',
			routeId: 'A',
			serviceId: 'all-day',
			shapeId: 'a-downtown-all-day-s0',
			stops: [ 'airport', 'museum', 'center' ],
			arrivals: [ 55380, 55800, 56100 ],
			departures: [ 55440, 55860, 56160 ],
			headwayBasedStarts: [],
			headwayBasedEnds: [],
			headwayBasedHeadways: []
		},
		{
			tripId: 'a-outbound-all-day',
			routeId: 'A',
			serviceId: 'all-day',
			shapeId: 'a-outbound-all-day-s0',
			stops: [ 'center', 'museum', 'airport' ],
			arrivals: [ 61980, 62400, 62700 ],
			departures: [ 62040, 62460, 62760 ],
			headwayBasedStarts: [],
			headwayBasedEnds: [],
			headwayBasedHeadways: []
		},
		{
			tripId: 'b-downtown-on-weekends',
			routeId: 'B',
			serviceId: 'on-weekends',
			shapeId: '',
			stops: [ 'airport', 'lake', 'center' ],
			arrivals: [ 47580, 48120, 48600 ],
			departures: [ 47640, 48240, 48660 ],
			headwayBasedStarts: [],
			headwayBasedEnds: [],
			headwayBasedHeadways: []
		},
		{
			tripId: 'b-downtown-on-working-days',
			routeId: 'B',
			serviceId: 'on-working-days',
			shapeId: '',
			stops: [ 'airport', 'lake', 'center', 'airport', 'lake', 'center' ],
			arrivals: [ 32100, 32520, 33120, 47580, 48000, 48600 ],
			departures: [ 32160, 32640, 33180, 47640, 48120, 48660 ],
			headwayBasedStarts: [],
			headwayBasedEnds: [],
			headwayBasedHeadways: []
		},
		{
			tripId: 'b-outbound-on-weekends',
			routeId: 'B',
			serviceId: 'on-weekends',
			shapeId: '',
			stops: [ 'center', 'lake', 'airport' ],
			arrivals: [ 65580, 66120, 66600 ],
			departures: [ 65640, 66240, 66660 ],
			headwayBasedStarts: [],
			headwayBasedEnds: [],
			headwayBasedHeadways: []
		},
		{
			tripId: 'b-outbound-on-working-days',
			routeId: 'B',
			serviceId: 'on-working-days',
			shapeId: '',
			stops: [ 'center', 'lake', 'airport' ],
			arrivals: [ 65580, 66000, 66600 ],
			departures: [ 65640, 66120, 66660 ],
			headwayBasedStarts: [ 54000 ],
			headwayBasedEnds: [ 57600 ],
			headwayBasedHeadways: [ 600 ]
		},
		{
			tripId: 'c-downtown-all-day',
			routeId: 'C',
			serviceId: 'all-day',
			shapeId: '',
			stops: [ 'airport', 'center' ],
			arrivals: [ 55620, 55980 ],
			departures: [ 55680, 56100 ],
			headwayBasedStarts: [],
			headwayBasedEnds: [],
			headwayBasedHeadways: []
		},
		{
			tripId: 'during-dst-1',
			routeId: 'D',
			serviceId: 'during-dst',
			shapeId: '',
			stops: [ 'airport', 'center' ],
			arrivals: [ 93300, 93780 ],
			departures: [ 93480, 93900 ],
			headwayBasedStarts: [],
			headwayBasedEnds: [],
			headwayBasedHeadways: [],
		},
	])
})

test('read-stop-times: handles DST switch properly', async (t) => {
	const readFile = readFilesFromFixture('daylight-saving-time')
	const stopTimes = readStopTimes(readFile, noFilter)

	const res = []
	for await (const st of stopTimes) res.push(st)
	t.deepEqual(res, [{
		tripId: 'A1',
		routeId: 'A',
		serviceId: 'sA',
		shapeId: undefined,
		stops: ['1', '2'],
		arrivals: [7140, 10740],
		departures: [7260, 10860],
		headwayBasedStarts: [],
		headwayBasedEnds: [],
		headwayBasedHeadways: [],
	}, {
		tripId: 'B1',
		routeId: 'B',
		serviceId: 'sB',
		shapeId: undefined,
		stops: ['2', '1'],
		arrivals: [7140, 10740],
		departures: [7260, 10860],
		headwayBasedStarts: [],
		headwayBasedEnds: [],
		headwayBasedHeadways: [],
	}])

	t.end()
})