def test_hops(self):
dao = Dao(DAO_URL, sql_logging=SQL_LOG)
dao.load_gtfs(MINI_GTFS)
# Get all hops
hops = dao.hops()
nhops = 0
for st1, st2 in hops:
self.assertTrue(st1.stop_sequence + 1 == st2.stop_sequence)
self.assertTrue(st1.trip == st2.trip)
nhops += 1
# 2 standard trips + 2 frequency generated (8 total)
self.assertTrue(nhops == 2 * 2 + 8 * 2)
# Get all hops with a distance <= 70km
hops = dao.hops(fltr=(dao.hop_second().shape_dist_traveled -
dao.hop_first().shape_dist_traveled <= 70000))
nhops1 = 0
for st1, st2 in hops:
self.assertTrue(st1.stop_sequence + 1 == st2.stop_sequence)
self.assertTrue(st1.trip == st2.trip)
self.assertTrue(
st2.shape_dist_traveled - st1.shape_dist_traveled <= 70000)
nhops1 += 1
# Get all hops with a distance > 70km
hops = dao.hops(fltr=(dao.hop_second().shape_dist_traveled -
dao.hop_first().shape_dist_traveled > 70000))
nhops2 = 0
for st1, st2 in hops:
self.assertTrue(st1.stop_sequence + 1 == st2.stop_sequence)
self.assertTrue(st1.trip == st2.trip)
self.assertTrue(
st2.shape_dist_traveled - st1.shape_dist_traveled > 70000)
nhops2 += 1
self.assertTrue(nhops == nhops1 + nhops2)
# Get all hops with a time +/- 1h
hops = dao.hops(fltr=(dao.hop_second().arrival_time -
dao.hop_first().departure_time >= 3600))
for st1, st2 in hops:
self.assertTrue(st2.arrival_time - st1.departure_time >= 3600)
hops = dao.hops(fltr=(dao.hop_second().arrival_time -
dao.hop_first().departure_time < 3600))
for st1, st2 in hops:
self.assertTrue(st2.arrival_time - st1.departure_time < 3600)
# Get hops with a delta of 2
hops = dao.hops(delta=2)
for st1, st2 in hops:
self.assertTrue(st1.stop_sequence + 2 == st2.stop_sequence)
self.assertTrue(st1.trip == st2.trip)
def test_hops(self):
dao = Dao(DAO_URL, sql_logging=SQL_LOG)
dao.load_gtfs(MINI_GTFS)
# Get all hops
hops = dao.hops()
nhops = 0
for st1, st2 in hops:
self.assertTrue(st1.stop_sequence + 1 == st2.stop_sequence)
self.assertTrue(st1.trip == st2.trip)
nhops += 1
# 2 standard trips + 2 frequency generated (8 total)
self.assertTrue(nhops == 2 * 2 + 8 * 2)
# Get all hops with a distance <= 70km
hops = dao.hops(fltr=(dao.hop_second().shape_dist_traveled - dao.hop_first().shape_dist_traveled <= 70000))
nhops1 = 0
for st1, st2 in hops:
self.assertTrue(st1.stop_sequence + 1 == st2.stop_sequence)
self.assertTrue(st1.trip == st2.trip)
self.assertTrue(st2.shape_dist_traveled - st1.shape_dist_traveled <= 70000)
nhops1 += 1
# Get all hops with a distance > 70km
hops = dao.hops(fltr=(dao.hop_second().shape_dist_traveled - dao.hop_first().shape_dist_traveled > 70000))
nhops2 = 0
for st1, st2 in hops:
self.assertTrue(st1.stop_sequence + 1 == st2.stop_sequence)
self.assertTrue(st1.trip == st2.trip)
self.assertTrue(st2.shape_dist_traveled - st1.shape_dist_traveled > 70000)
nhops2 += 1
self.assertTrue(nhops == nhops1 + nhops2)
# Get all hops with a time +/- 1h
hops = dao.hops(fltr=(dao.hop_second().arrival_time - dao.hop_first().departure_time >= 3600))
for st1, st2 in hops:
self.assertTrue(st2.arrival_time - st1.departure_time >= 3600)
hops = dao.hops(fltr=(dao.hop_second().arrival_time - dao.hop_first().departure_time < 3600))
for st1, st2 in hops:
self.assertTrue(st2.arrival_time - st1.departure_time < 3600)
# Get hops with a delta of 2
hops = dao.hops(delta=2)
for st1, st2 in hops:
self.assertTrue(st1.stop_sequence + 2 == st2.stop_sequence)
self.assertTrue(st1.trip == st2.trip)
def _test_one_gtfs(self, gtfs):
clear_mappers()
dao = Dao(DAO_URL, sql_logging=SQL_LOG)
dao.load_gtfs(gtfs)
# Check stop time normalization and interpolation
for trip in dao.trips(prefetch_stop_times=True):
stopseq = 0
n_stoptimes = len(trip.stop_times)
last_stop = None
distance = trip.stop_times[0].shape_dist_traveled
last_stoptime = None
last_interpolated_speed = None
for stoptime in trip.stop_times:
self.assertTrue(stoptime.stop_sequence == stopseq)
if stopseq == 0:
self.assertTrue(stoptime.arrival_time is None)
else:
self.assertTrue(stoptime.arrival_time is not None)
if stopseq == n_stoptimes - 1:
self.assertTrue(stoptime.departure_time is None)
else:
self.assertTrue(stoptime.departure_time is not None)
if last_stop is not None:
distance += orthodromic_distance(last_stop, stoptime.stop)
last_stop = stoptime.stop
if trip.shape is not None:
self.assertTrue(stoptime.shape_dist_traveled >= distance)
else:
self.assertAlmostEqual(stoptime.shape_dist_traveled,
distance, 1)
stopseq += 1
if stoptime.interpolated or (last_stoptime is not None
and last_stoptime.interpolated):
dist = stoptime.shape_dist_traveled - last_stoptime.shape_dist_traveled
time = stoptime.arrival_time - last_stoptime.departure_time
speed = dist * 1.0 / time
if last_interpolated_speed is not None:
self.assertAlmostEqual(speed, last_interpolated_speed,
2)
last_interpolated_speed = speed
if not stoptime.interpolated:
last_interpolated_speed = None
last_stoptime = stoptime
# Get all hops
hops = dao.hops()
nhops = 0
for st1, st2 in hops:
self.assertTrue(st1.stop_sequence + 1 == st2.stop_sequence)
self.assertTrue(st1.trip == st2.trip)
nhops += 1
# Get hops with a delta of 2
hops = dao.hops(delta=2)
nhops2 = 0
for st1, st2 in hops:
self.assertTrue(st1.stop_sequence + 2 == st2.stop_sequence)
self.assertTrue(st1.trip == st2.trip)
nhops2 += 1
ntrips = len(list(dao.trips()))
# Assume all trips have len > 2
self.assertTrue(nhops == nhops2 + ntrips)
# Test shape_dist_traveled on stoptimes
for trip in dao.trips():
# Assume no shapes for now
distance = 0.0
last_stop = None
for stoptime in trip.stop_times:
if last_stop is not None:
distance += orthodromic_distance(last_stop, stoptime.stop)
last_stop = stoptime.stop
if trip.shape:
self.assertTrue(stoptime.shape_dist_traveled >= distance)
else:
self.assertAlmostEqual(stoptime.shape_dist_traveled,
distance, 2)
# Test shape normalization
for shape in dao.shapes():
distance = 0.0
last_pt = None
ptseq = 0
for point in shape.points:
if last_pt is not None:
distance += orthodromic_distance(last_pt, point)
last_pt = point
self.assertAlmostEqual(point.shape_dist_traveled, distance, 2)
self.assertTrue(point.shape_pt_sequence == ptseq)
ptseq += 1
# Check zone-stop relationship
for zone in dao.zones(prefetch_stops=True):
for stop in zone.stops:
self.assertTrue(stop.zone == zone)
for stop in dao.stops():
if stop.zone:
self.assertTrue(stop in stop.zone.stops)
def _test_one_gtfs(self, gtfs):
dao = Dao(DAO_URL, sql_logging=SQL_LOG)
dao.load_gtfs(gtfs)
# Check stop time normalization and interpolation
for trip in dao.trips(prefetch_stop_times=True):
stopseq = 0
n_stoptimes = len(trip.stop_times)
last_stop = None
distance = trip.stop_times[0].shape_dist_traveled
last_stoptime = None
last_interpolated_speed = None
for stoptime in trip.stop_times:
self.assertTrue(stoptime.stop_sequence == stopseq)
if stopseq == 0:
self.assertTrue(stoptime.arrival_time is None)
else:
self.assertTrue(stoptime.arrival_time is not None)
if stopseq == n_stoptimes - 1:
self.assertTrue(stoptime.departure_time is None)
else:
self.assertTrue(stoptime.departure_time is not None)
if last_stop is not None:
distance += orthodromic_distance(last_stop, stoptime.stop)
last_stop = stoptime.stop
if trip.shape is not None:
self.assertTrue(stoptime.shape_dist_traveled >= distance)
else:
self.assertAlmostEqual(stoptime.shape_dist_traveled, distance, 1)
stopseq += 1
if stoptime.interpolated or (last_stoptime is not None and last_stoptime.interpolated):
dist = stoptime.shape_dist_traveled - last_stoptime.shape_dist_traveled
time = stoptime.arrival_time - last_stoptime.departure_time
speed = dist * 1.0 / time
if last_interpolated_speed is not None:
self.assertAlmostEqual(speed, last_interpolated_speed, 2)
last_interpolated_speed = speed
if not stoptime.interpolated:
last_interpolated_speed = None
last_stoptime = stoptime
# Get all hops
hops = dao.hops()
nhops = 0
for st1, st2 in hops:
self.assertTrue(st1.stop_sequence + 1 == st2.stop_sequence)
self.assertTrue(st1.trip == st2.trip)
nhops += 1
# Get hops with a delta of 2
hops = dao.hops(delta=2)
nhops2 = 0
for st1, st2 in hops:
self.assertTrue(st1.stop_sequence + 2 == st2.stop_sequence)
self.assertTrue(st1.trip == st2.trip)
nhops2 += 1
ntrips = len(list(dao.trips()))
# Assume all trips have len > 2
self.assertTrue(nhops == nhops2 + ntrips)
# Test shape_dist_traveled on stoptimes
for trip in dao.trips():
# Assume no shapes for now
distance = 0.0
last_stop = None
for stoptime in trip.stop_times:
if last_stop is not None:
distance += orthodromic_distance(last_stop, stoptime.stop)
last_stop = stoptime.stop
if trip.shape:
self.assertTrue(stoptime.shape_dist_traveled >= distance)
else:
self.assertAlmostEqual(stoptime.shape_dist_traveled, distance, 2)
# Test shape normalization
for shape in dao.shapes():
distance = 0.0
last_pt = None
ptseq = 0
for point in shape.points:
if last_pt is not None:
distance += orthodromic_distance(last_pt, point)
last_pt = point
self.assertAlmostEqual(point.shape_dist_traveled, distance, 2)
self.assertTrue(point.shape_pt_sequence == ptseq)
ptseq += 1
# Check zone-stop relationship
for zone in dao.zones(prefetch_stops=True):
for stop in zone.stops:
self.assertTrue(stop.zone == zone)
for stop in dao.stops():
if stop.zone:
self.assertTrue(stop in stop.zone.stops)
