def test_stop_station(self):
dao = Dao()
f1 = FeedInfo("F1")
sa = Stop("F1", "SA", "Station A", 45.0000, 0.0000)
sa1 = Stop("F1", "SA1", "Stop A1", 45.0001, 0.0001)
sa1.parent_station_id = 'SA'
sb = Stop("F1", "SB", "Station B", 45.0002, 0.0002)
sb1 = Stop("F1", "SB1", "Stop B1", 45.0003, 0.0003)
sb1.parent_station_id = 'SB'
sb2 = Stop("F1", "SB2", "Stop B2", 45.0002, 0.0003)
sb2.parent_station_id = 'SB'
a1 = Agency("F1", "A1", "Agency 1", "url1", "Europe/Paris")
r1 = Route("F1", "R1", "A1", Route.TYPE_BUS)
c1 = Calendar("F1", "C1")
t1 = Trip("F1", "T1", "R1", "C1")
st1a = StopTime("F1", "T1", "SA1", 0, None, 3600, 0.0)
st1b = StopTime("F1", "T1", "SB1", 1, 3800, None, 100.0)
dao.add_all([f1, sa, sa1, sb, sb1, sb2, a1, r1, c1, t1, st1a, st1b])
stops = list(dao.stops(fltr=(Agency.agency_id == 'A1')))
self.assertTrue(len(stops) == 2)
self.assertTrue(sa1 in stops)
self.assertTrue(sb1 in stops)
stops = list(dao.stops(fltr=(Stop.parent_station_id == 'SA')))
self.assertTrue(len(stops) == 1)
self.assertTrue(sa1 in stops)
stops = list(dao.stops(fltr=(Stop.parent_station_id == 'SB')))
self.assertTrue(len(stops) == 2)
self.assertTrue(sb1 in stops)
self.assertTrue(sb2 in stops)
def test_complex_queries(self):
dao = Dao(DAO_URL, sql_logging=SQL_LOG)
dao.load_gtfs(DUMMY_GTFS)
# Get the list of departures:
# 1) from "Porte de Bourgogne"
# 2) on 4th July
# 3) between 10:00 and 14:00
# 4) on route type BUS
# 5) not the last of trip (only departing)
porte_bourgogne = dao.stop("BBG")
july4 = CalendarDate.ymd(2016, 7, 4)
from_time = gtfstime(10, 00)
to_time = gtfstime(14, 00)
departures = dao.stoptimes(
fltr=(StopTime.stop == porte_bourgogne) & (StopTime.departure_time >= from_time) & (StopTime.departure_time <= to_time)
& (Route.route_type == Route.TYPE_BUS) & (func.date(CalendarDate.date) == july4.date),
prefetch_trips=True)
n = 0
for dep in departures:
self.assertTrue(dep.stop == porte_bourgogne)
self.assertTrue(july4 in dep.trip.calendar.dates)
self.assertTrue(dep.trip.route.route_type == Route.TYPE_BUS)
self.assertTrue(dep.departure_time >= from_time and dep.departure_time <= to_time)
n += 1
self.assertTrue(n > 10)
# Plage is a stop that is used only in summer (hence the name!)
plage = dao.stop("BPG")
# Get the list of stops used by some route:
# 1) All-year round
route_red = dao.route("BR")
stoplist_all = list(dao.stops(fltr=Trip.route == route_red))
# 2) Only in january
from_date = CalendarDate.ymd(2016, 1, 1)
to_date = CalendarDate.ymd(2016, 1, 31)
stoplist_jan = list(dao.stops(
fltr=(Trip.route == route_red) & (func.date(CalendarDate.date) >= from_date.date) & (func.date(CalendarDate.date) <= to_date.date)))
# Now, make some tests
self.assertTrue(len(stoplist_all) > 5)
self.assertTrue(plage in stoplist_all)
self.assertFalse(plage in stoplist_jan)
stoplist = list(stoplist_all)
stoplist.remove(plage)
self.assertTrue(set(stoplist) == set(stoplist_jan))
# Get all routes passing by the set of stops
routes = dao.routes(fltr=or_(StopTime.stop == stop for stop in stoplist_jan))
stopset = set()
for route in routes:
for trip in route.trips:
for stoptime in trip.stop_times:
stopset.add(stoptime.stop)
self.assertTrue(set(stoplist_jan).issubset(stopset))
def test_broken(self):
exception = False
try:
dao = Dao("")
dao.load_gtfs(BROKEN_GTFS, lenient=False)
except KeyError:
exception = True
self.assertTrue(exception)
dao = Dao("")
dao.load_gtfs(BROKEN_GTFS, lenient=True)
# The following are based on BROKEN GTFS content,
# that is the entities count minus broken ones.
self.assertTrue(len(dao.routes()) == 4)
self.assertTrue(len(list(dao.stops())) == 12)
self.assertTrue(len(dao.calendars()) == 2)
self.assertTrue(len(list(dao.trips())) == 104)
self.assertTrue(len(dao.stoptimes()) == 500)
self.assertTrue(len(dao.fare_attributes()) == 2)
self.assertTrue(len(dao.fare_rules()) == 4)
# This stop has missing coordinates in the broken file
stop00 = dao.stop('FUR_CREEK_RES3')
self.assertAlmostEquals(stop00.stop_lat, 0.0, 5)
self.assertAlmostEquals(stop00.stop_lon, 0.0, 5)
def test_broken(self):
exception = False
try:
clear_mappers()
dao = Dao("")
dao.load_gtfs(BROKEN_GTFS, lenient=False)
except KeyError:
exception = True
self.assertTrue(exception)
clear_mappers()
dao = Dao("")
dao.load_gtfs(BROKEN_GTFS, lenient=True)
# The following are based on BROKEN GTFS content,
# that is the entities count minus broken ones.
self.assertTrue(len(dao.routes()) == 4)
self.assertTrue(len(list(dao.stops())) == 12)
self.assertTrue(len(dao.calendars()) == 2)
self.assertTrue(len(list(dao.trips())) == 104)
self.assertTrue(len(dao.stoptimes()) == 500)
self.assertTrue(len(dao.fare_attributes()) == 2)
self.assertTrue(len(dao.fare_rules()) == 4)
# This stop has missing coordinates in the broken file
stop00 = dao.stop('FUR_CREEK_RES3')
self.assertAlmostEquals(stop00.stop_lat, 0.0, 5)
self.assertAlmostEquals(stop00.stop_lon, 0.0, 5)
def test_stop_station_multi_feed(self):
dao = Dao()
fa = FeedInfo("FA")
fb = FeedInfo("FB")
sa = Stop("FA",
"S",
"Station A",
45.0,
0.0,
location_type=Stop.TYPE_STATION)
sa1 = Stop("FA", "S1", "Stop A1", 45.0, 0.0, parent_station_id="S")
sa2 = Stop("FA", "S2", "Stop A2", 45.0, 0.1, parent_station_id="S")
sa3 = Stop("FA", "S3", "Stop A3", 45.0, 0.2)
sb = Stop("FB",
"S",
"Station B",
45.0,
0.0,
location_type=Stop.TYPE_STATION)
sb1 = Stop("FB", "S1", "Stop B1", 45.0, 0.0, parent_station_id="S")
sb2 = Stop("FB", "S2", "Stop B2", 45.0, 0.1, parent_station_id="S")
dao.add_all([fa, fb, sa, sa1, sa2, sa3, sb1, sb2, sb])
sa = dao.stop("S", feed_id="FA")
self.assertTrue(sa.stop_name == "Station A")
self.assertTrue(len(sa.sub_stops) == 2)
for ssa in sa.sub_stops:
self.assertTrue(ssa.stop_name.startswith("Stop A"))
self.assertTrue(ssa.parent_station.stop_name == "Station A")
sa1 = dao.stop("S1", feed_id="FA")
self.assertTrue(sa1.stop_name == "Stop A1")
self.assertTrue(sa1.parent_station.stop_name == "Station A")
self.assertTrue(len(list(dao.stops())) == 7)
def test_stop_station_multi_feed(self):
dao = Dao()
fa = FeedInfo("FA")
fb = FeedInfo("FB")
sa = Stop("FA", "S", "Station A", 45.0, 0.0, location_type=Stop.TYPE_STATION)
sa1 = Stop("FA", "S1", "Stop A1", 45.0, 0.0, parent_station_id="S")
sa2 = Stop("FA", "S2", "Stop A2", 45.0, 0.1, parent_station_id="S")
sa3 = Stop("FA", "S3", "Stop A3", 45.0, 0.2)
sb = Stop("FB", "S", "Station B", 45.0, 0.0, location_type=Stop.TYPE_STATION)
sb1 = Stop("FB", "S1", "Stop B1", 45.0, 0.0, parent_station_id="S")
sb2 = Stop("FB", "S2", "Stop B2", 45.0, 0.1, parent_station_id="S")
dao.add_all([ fa, fb, sa, sa1, sa2, sa3, sb1, sb2, sb ])
sa = dao.stop("S", feed_id="FA")
self.assertTrue(sa.stop_name == "Station A")
self.assertTrue(len(sa.sub_stops) == 2)
for ssa in sa.sub_stops:
self.assertTrue(ssa.stop_name.startswith("Stop A"))
self.assertTrue(ssa.parent_station.stop_name == "Station A")
sa1 = dao.stop("S1", feed_id="FA")
self.assertTrue(sa1.stop_name == "Stop A1")
self.assertTrue(sa1.parent_station.stop_name == "Station A")
self.assertTrue(len(list(dao.stops())) == 7)
def test_areas(self):
dao = Dao()
f1 = FeedInfo("F1")
s1 = Stop("F1", "S1", "Stop 1", 45.0, 0.0)
s2 = Stop("F1", "S2", "Stop 2", 45.1, 0.1)
s3 = Stop("F1", "S3", "Stop 3", 45.2, 0.2)
dao.add_all([ f1, s1, s2, s3 ])
# Rectangular area
stops = list(dao.stops(fltr=dao.in_area(RectangularArea(0, 0, 1, 1))))
self.assertTrue(len(stops) == 0)
stops = list(dao.stops(fltr=dao.in_area(RectangularArea(-90, -180, 90, 180))))
self.assertTrue(len(stops) == 3)
stops = list(dao.stops(fltr=dao.in_area(RectangularArea(45.05, 0.05, 45.15, 0.15))))
self.assertTrue(len(stops) == 1)
self.assertTrue(stops[0].stop_id == 'S2')
def test_areas(self):
dao = Dao()
f1 = FeedInfo("F1")
s1 = Stop("F1", "S1", "Stop 1", 45.0, 0.0)
s2 = Stop("F1", "S2", "Stop 2", 45.1, 0.1)
s3 = Stop("F1", "S3", "Stop 3", 45.2, 0.2)
dao.add_all([f1, s1, s2, s3])
# Rectangular area
stops = list(dao.stops(fltr=dao.in_area(RectangularArea(0, 0, 1, 1))))
self.assertTrue(len(stops) == 0)
stops = list(
dao.stops(fltr=dao.in_area(RectangularArea(-90, -180, 90, 180))))
self.assertTrue(len(stops) == 3)
stops = list(
dao.stops(
fltr=dao.in_area(RectangularArea(45.05, 0.05, 45.15, 0.15))))
self.assertTrue(len(stops) == 1)
self.assertTrue(stops[0].stop_id == 'S2')
def test_demo(self):
dao = Dao(DAO_URL, sql_logging=False)
dao.load_gtfs(DUMMY_GTFS)
print("List of stops named '...Bordeaux...':")
stops_bordeaux = list(dao.stops(fltr=(Stop.stop_name.ilike('%Bordeaux%')) & (Stop.location_type == Stop.TYPE_STOP)))
for stop in stops_bordeaux:
print(stop.stop_name)
print("List of routes passing by those stops:")
routes_bordeaux = dao.routes(fltr=or_(StopTime.stop == stop for stop in stops_bordeaux))
for route in routes_bordeaux:
print("%s - %s" % (route.route_short_name, route.route_long_name))
july4 = CalendarDate.ymd(2016, 7, 4)
print("All departures from those stops on %s:" % (july4.as_date()))
departures = list(dao.stoptimes(fltr=(or_(StopTime.stop == stop for stop in stops_bordeaux)) & (StopTime.departure_time != None) & (func.date(CalendarDate.date) == july4.date)))
print("There is %d departures" % (len(departures)))
for departure in departures:
print("%30.30s %10.10s %-20.20s > %s" % (departure.stop.stop_name, fmttime(departure.departure_time), departure.trip.route.route_long_name, departure.trip.trip_headsign))
print("Number of departures and time range per stop on %s:" % (july4.as_date()))
departure_by_stop = defaultdict(list)
for departure in departures:
departure_by_stop[departure.stop].append(departure)
for stop, deps in departure_by_stop.items():
min_dep = min(d.departure_time for d in deps)
max_dep = max(d.departure_time for d in deps)
print("%30.30s %3d departures (from %s to %s)" % (stop.stop_name, len(deps), fmttime(min_dep), fmttime(max_dep)))
# Compute the average distance and time to next stop by route type
ntd = [ [0, 0, 0.0] for type in range(0, Route.TYPE_FUNICULAR + 1) ]
for departure in departures:
# The following is guaranteed to succeed as we have departure_time == Null for last stop time in trip
next_arrival = departure.trip.stop_times[departure.stop_sequence + 1]
hop_dist = next_arrival.shape_dist_traveled - departure.shape_dist_traveled
hop_time = next_arrival.arrival_time - departure.departure_time
route_type = departure.trip.route.route_type
ntd[route_type][0] += 1
ntd[route_type][1] += hop_time
ntd[route_type][2] += hop_dist
for route_type in range(0, len(ntd)):
n, t, d = ntd[route_type]
if n > 0:
print("The average distance to the next stop on those departures for route type %d is %.2f meters" % (route_type, d / n))
print("The average time in sec to the next stop on those departures for route type %d is %s" % (route_type, fmttime(t / n)))
def test_clusterizer(self):
dao = Dao(DAO_URL, sql_logging=SQL_LOG)
dao.load_gtfs(DUMMY_GTFS)
# Merge stops closer than 300m together
sc = SpatialClusterizer(300.0)
for stop in dao.stops():
sc.add_point(stop)
sc.clusterize()
# for cluster in sc.clusters():
# print("---CLUSTER: %d stops" % (len(cluster)))
# for stop in cluster:
# print("%s %s" % (stop.stop_id, stop.stop_name))
gare1 = dao.stop("GBSJT")
gare2 = dao.stop("GBSJ")
gare3 = dao.stop("GBSJB")
self.assertTrue(sc.in_same_cluster(gare1, gare2))
self.assertTrue(sc.in_same_cluster(gare1, gare3))
self.assertTrue(sc.in_same_cluster(gare2, gare3))
bq = dao.stop("BQ")
bq1 = dao.stop("BQA")
bq2 = dao.stop("BQD")
self.assertTrue(sc.in_same_cluster(bq, bq1))
self.assertTrue(sc.in_same_cluster(bq, bq2))
bs = dao.stop("BS")
bs1 = dao.stop("BS1")
bs2 = dao.stop("BS2")
self.assertTrue(sc.in_same_cluster(bs, bs1))
self.assertTrue(sc.in_same_cluster(bs, bs2))
self.assertFalse(sc.in_same_cluster(gare1, bq))
self.assertFalse(sc.in_same_cluster(gare1, bs))
self.assertFalse(sc.in_same_cluster(gare3, bs2))
bjb = dao.stop("BJB")
self.assertFalse(sc.in_same_cluster(bjb, gare1))
self.assertFalse(sc.in_same_cluster(bjb, bs))
self.assertFalse(sc.in_same_cluster(bjb, bq))
def test_non_overlapping_feeds(self):
dao = Dao(DAO_URL, sql_logging=SQL_LOG)
# Load twice the same data under two distinct namespaces
dao.load_gtfs(DUMMY_GTFS, feed_id='A')
dao.load_gtfs(DUMMY_GTFS, feed_id='B')
# Check that each feed only return it's own data
feed_a = dao.feed('A')
self.assertTrue(feed_a.feed_id == 'A')
feed_b = dao.feed('B')
self.assertTrue(feed_b.feed_id == 'B')
self.assertTrue(len(dao.agencies()) == 4)
self.assertTrue(len(feed_a.agencies) == 2)
self.assertTrue(len(feed_b.agencies) == 2)
self.assertTrue(len(feed_a.routes) * 2 == len(dao.routes()))
self.assertTrue(len(feed_b.routes) * 2 == len(dao.routes()))
self.assertTrue(len(feed_a.stops) * 2 == len(list(dao.stops())))
self.assertTrue(len(feed_b.stops) * 2 == len(list(dao.stops())))
self.assertTrue(len(feed_a.calendars) * 2 == len(dao.calendars()))
self.assertTrue(len(feed_b.calendars) * 2 == len(dao.calendars()))
self.assertTrue(len(feed_a.trips) * 2 == len(list(dao.trips())))
self.assertTrue(len(feed_b.trips) * 2 == len(list(dao.trips())))
# -*- coding: utf-8 -*-
"""
Created on Tue May 2 21:23:46 2017
@author: luxn
"""
from gtfslib.dao import Dao
# DAO ist ein Data Access Object
dao = Dao("gtfs-sample.db.sqlite")
for stop in dao.stops():
print(stop.stop_name)
"""
So eine Ausgabe sollte dann kommen:
(C:\Anaconda3) D:\Github\geoinf\gtfs>python sample.py
Amargosa Valley (Demo)
Nye County Airport (Demo)
Bullfrog (Demo)
Doing Ave / D Ave N (Demo)
E Main St / S Irving St (Demo)
Furnace Creek Resort (Demo)
North Ave / D Ave N (Demo)
North Ave / N A Ave (Demo)
Stagecoach Hotel & Casino (Demo)
"""
def test_demo(self):
dao = Dao(DAO_URL, sql_logging=False)
dao.load_gtfs(DUMMY_GTFS)
print("List of stops named '...Bordeaux...':")
stops_bordeaux = list(
dao.stops(fltr=(Stop.stop_name.ilike('%Bordeaux%'))
& (Stop.location_type == Stop.TYPE_STOP)))
for stop in stops_bordeaux:
print(stop.stop_name)
print("List of routes passing by those stops:")
routes_bordeaux = dao.routes(fltr=or_(StopTime.stop == stop
for stop in stops_bordeaux))
for route in routes_bordeaux:
print("%s - %s" % (route.route_short_name, route.route_long_name))
july4 = CalendarDate.ymd(2016, 7, 4)
print("All departures from those stops on %s:" % (july4.as_date()))
departures = list(
dao.stoptimes(fltr=(or_(StopTime.stop == stop
for stop in stops_bordeaux))
& (StopTime.departure_time != None)
& (func.date(CalendarDate.date) == july4.date)))
print("There is %d departures" % (len(departures)))
for departure in departures:
print("%30.30s %10.10s %-20.20s > %s" %
(departure.stop.stop_name, fmttime(departure.departure_time),
departure.trip.route.route_long_name,
departure.trip.trip_headsign))
print("Number of departures and time range per stop on %s:" %
(july4.as_date()))
departure_by_stop = defaultdict(list)
for departure in departures:
departure_by_stop[departure.stop].append(departure)
for stop, deps in departure_by_stop.items():
min_dep = min(d.departure_time for d in deps)
max_dep = max(d.departure_time for d in deps)
print("%30.30s %3d departures (from %s to %s)" %
(stop.stop_name, len(deps), fmttime(min_dep),
fmttime(max_dep)))
# Compute the average distance and time to next stop by route type
ntd = [[0, 0, 0.0] for type in range(0, Route.TYPE_FUNICULAR + 1)]
for departure in departures:
# The following is guaranteed to succeed as we have departure_time == Null for last stop time in trip
next_arrival = departure.trip.stop_times[departure.stop_sequence +
1]
hop_dist = next_arrival.shape_dist_traveled - departure.shape_dist_traveled
hop_time = next_arrival.arrival_time - departure.departure_time
route_type = departure.trip.route.route_type
ntd[route_type][0] += 1
ntd[route_type][1] += hop_time
ntd[route_type][2] += hop_dist
for route_type in range(0, len(ntd)):
n, t, d = ntd[route_type]
if n > 0:
print(
"The average distance to the next stop on those departures for route type %d is %.2f meters"
% (route_type, d / n))
print(
"The average time in sec to the next stop on those departures for route type %d is %s"
% (route_type, fmttime(t / n)))
def test_gtfs_data(self):
dao = Dao(DAO_URL, sql_logging=False)
dao.load_gtfs(DUMMY_GTFS)
# Check feed
feed = dao.feed()
self.assertTrue(feed.feed_id == "")
self.assertTrue(feed.feed_publisher_name == "Mecatran")
self.assertTrue(feed.feed_publisher_url == "http://www.mecatran.com/")
self.assertTrue(feed.feed_contact_email == "*****@*****.**")
self.assertTrue(feed.feed_lang == "fr")
self.assertTrue(len(dao.agencies()) == 2)
self.assertTrue(len(dao.routes()) == 3)
self.assertTrue(len(feed.agencies) == 2)
self.assertTrue(len(feed.routes) == 3)
# Check agencies
at = dao.agency("AT")
self.assertTrue(at.agency_name == "Agency Train")
self.assertTrue(len(at.routes) == 1)
ab = dao.agency("AB")
self.assertTrue(ab.agency_name == "Agency Bus")
self.assertTrue(len(ab.routes) == 2)
# Check calendars
week = dao.calendar("WEEK")
self.assertTrue(len(week.dates) == 253)
summer = dao.calendar("SUMMER")
self.assertTrue(len(summer.dates) == 42)
mon = dao.calendar("MONDAY")
self.assertTrue(len(mon.dates) == 49)
sat = dao.calendar("SAT")
self.assertTrue(len(sat.dates) == 53)
for date in mon.dates:
self.assertTrue(date.dow() == 0)
for date in sat.dates:
self.assertTrue(date.dow() == 5)
for date in week.dates:
self.assertTrue(date.dow() >= 0 and date.dow() <= 4)
for date in summer.dates:
self.assertTrue(date >= CalendarDate.ymd(2016, 7, 1) and date <= CalendarDate.ymd(2016, 8, 31))
empty = dao.calendars(func.date(CalendarDate.date) == datetime.date(2016, 5, 1))
# OR USE: empty = dao.calendars(CalendarDate.date == "2016-05-01")
self.assertTrue(len(empty) == 0)
july4 = CalendarDate.ymd(2016, 7, 4)
summer_mon = dao.calendars(func.date(CalendarDate.date) == july4.date)
n = 0
for cal in summer_mon:
self.assertTrue(july4 in cal.dates)
n += 1
self.assertTrue(n == 3)
# Check stops
sbq = dao.stop("BQ")
self.assertAlmostEqual(sbq.stop_lat, 44.844, places=2)
self.assertAlmostEqual(sbq.stop_lon, -0.573, places=2)
self.assertTrue(sbq.stop_name == "Bordeaux Quinconces")
n = 0
for stop in dao.stops(Stop.stop_name.like("Gare%")):
self.assertTrue(stop.stop_name.startswith("Gare"))
n += 1
self.assertTrue(n == 7)
n = 0
for stop in dao.stops(fltr=dao.in_area(RectangularArea(44.7, -0.6, 44.9, -0.4))):
self.assertTrue(stop.stop_lat >= 44.7 and stop.stop_lat <= 44.9 and stop.stop_lon >= -0.6 and stop.stop_lon <= -0.4)
n += 1
self.assertTrue(n == 16)
for station in dao.stops(Stop.location_type == Stop.TYPE_STATION):
self.assertTrue(station.location_type == Stop.TYPE_STATION)
self.assertTrue(len(station.sub_stops) >= 2)
for stop in station.sub_stops:
self.assertTrue(stop.parent_station == station)
# Check zones
z_inexistant = dao.zone("ZX")
self.assertTrue(z_inexistant is None)
z1 = dao.zone("Z1")
self.assertEquals(16, len(z1.stops))
z2 = dao.zone("Z2")
self.assertEquals(4, len(z2.stops))
# Check transfers
transfers = dao.transfers()
self.assertTrue(len(transfers) == 3)
transfers = dao.transfers(fltr=(dao.transfer_from_stop().stop_id == 'GBSJB'))
self.assertTrue(len(transfers) == 1)
self.assertTrue(transfers[0].from_stop.stop_id == 'GBSJB')
# Check routes
tgv = dao.route("TGVBP")
self.assertTrue(tgv.agency == at)
self.assertTrue(tgv.route_type == 2)
r1 = dao.route("BR")
self.assertTrue(r1.route_short_name == "R1")
self.assertTrue(r1.route_long_name == "Bus Red")
n = 0
for route in dao.routes(Route.route_type == 3):
self.assertTrue(route.route_type == 3)
n += 1
self.assertTrue(n == 2)
# Check trip for route
n = 0
trips = dao.trips(fltr=Route.route_type == Route.TYPE_BUS)
for trip in trips:
self.assertTrue(trip.route.route_type == Route.TYPE_BUS)
n += 1
self.assertTrue(n > 20)
# Check trips on date
trips = dao.trips(fltr=func.date(CalendarDate.date) == july4.date, prefetch_calendars=True)
n = 0
for trip in trips:
self.assertTrue(july4 in trip.calendar.dates)
n += 1
self.assertTrue(n > 30)
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)
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_transfers(self):
dao = Dao()
f1 = FeedInfo("F1")
s1 = Stop("F1", "S1", "Stop 1", 45.0000, 0.0000)
s2 = Stop("F1", "S2", "Stop 2", 45.0001, 0.0001)
s3 = Stop("F1", "S3", "Stop 3", 45.0002, 0.0002)
t12 = Transfer("F1", "S1", "S2")
t21 = Transfer("F1", "S2", "S1")
t23 = Transfer("F1",
"S2",
"S3",
transfer_type=Transfer.TRANSFER_TIMED,
min_transfer_time=180)
t32 = Transfer("F1",
"S3",
"S2",
transfer_type=Transfer.TRANSFER_TIMED,
min_transfer_time=120)
t13 = Transfer("F1", "S1", "S3", transfer_type=Transfer.TRANSFER_NONE)
a1 = Agency("F1", "A1", "Agency 1", "url1", "Europe/Paris")
a2 = Agency("F1", "A2", "Agency 2", "url2", "Europe/London")
r1 = Route("F1", "R1", "A1", Route.TYPE_BUS)
r2 = Route("F1", "R2", "A2", Route.TYPE_BUS)
c1 = Calendar("F1", "C1")
t1 = Trip("F1", "T1", "R1", "C1")
t2 = Trip("F1", "T2", "R2", "C1")
st1a = StopTime("F1", "T1", "S1", 0, None, 3600, 0.0)
st1b = StopTime("F1", "T1", "S2", 1, 3800, None, 100.0)
st2a = StopTime("F1", "T2", "S1", 0, None, 4600, 0.0)
st2b = StopTime("F1", "T2", "S3", 1, 4800, None, 100.0)
dao.add_all([
f1, s1, s2, s3, t12, t21, t23, t32, t13, a1, a2, r1, r2, c1, t1,
t2, st1a, st1b, st2a, st2b
])
self.assertTrue(len(dao.transfers()) == 5)
timed_transfers = dao.transfers(
fltr=(Transfer.transfer_type == Transfer.TRANSFER_TIMED))
self.assertTrue(len(timed_transfers) == 2)
for transfer in timed_transfers:
self.assertTrue(transfer.transfer_type == Transfer.TRANSFER_TIMED)
s1_from_transfers = dao.transfers(
fltr=(dao.transfer_from_stop().stop_name == "Stop 1"))
self.assertTrue(len(s1_from_transfers) == 2)
for transfer in s1_from_transfers:
self.assertTrue(transfer.from_stop.stop_name == "Stop 1")
s1_fromto_transfers = dao.transfers(
fltr=((dao.transfer_from_stop().stop_name == "Stop 1")
| (dao.transfer_to_stop().stop_name == "Stop 1")))
self.assertTrue(len(s1_fromto_transfers) == 3)
for transfer in s1_fromto_transfers:
self.assertTrue(transfer.from_stop.stop_name == "Stop 1"
or transfer.to_stop.stop_name == "Stop 1")
s1 = dao.stop("S1", feed_id="F1")
self.assertTrue(len(s1.from_transfers) == 2)
self.assertTrue(len(s1.to_transfers) == 1)
for transfer in s1.from_transfers:
if transfer.to_stop.stop_id == "S2":
self.assertTrue(
transfer.transfer_type == Transfer.TRANSFER_DEFAULT)
elif transfer.to_stop.stop_id == "S3":
self.assertTrue(
transfer.transfer_type == Transfer.TRANSFER_NONE)
a1_stops = list(dao.stops(fltr=(Agency.agency_id == 'A1')))
self.assertTrue(len(a1_stops) == 2)
self.assertTrue(s1 in a1_stops)
self.assertTrue(s2 in a1_stops)
