-
Notifications
You must be signed in to change notification settings - Fork 8
/
Copy pathdb_test.py
62 lines (44 loc) · 1.58 KB
/
db_test.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
# -*- coding: utf-8 -*-
"""
Created on Sat Jan 10 21:12:41 2015
@author: brian
"""
from traffic_estimation.Trip import Trip
from db_functions import db_main, db_travel_times, db_trip
from routing.Map import Map
from datetime import datetime
from traffic_estimation.TrafficEstimation import estimate_travel_times, load_trips
# Connect to the database
db_main.connect("db_functions/database.conf")
db_travel_times.drop_travel_time_table()
db_travel_times.create_travel_time_table()
# Load map
print("Loading map")
road_map = Map("nyc_map4/nodes.csv", "nyc_map4/links.csv")
print("Loading trips")
trips = db_trip.find_pickup_dt('2010-01-02 00:00:00', '2010-01-02 01:00:00')
print (len(trips))
# Get some travel times on the links
print("Estimating travel times")
estimate_travel_times(road_map, trips, max_iter=1, test_set=None, distance_weighting=None, model_idle_time=False)
# Now save these travel times with a bunch of different datetimes
for i in range(20):
size = db_travel_times.get_travel_time_table_size()
print str(i) + " ) " + str(size)
# Save travel times into DB
start_time = datetime.now()
print("Saving travel times")
db_travel_times.save_travel_times(road_map, start_time)
print("Done")
end_time = datetime.now()
print(str(end_time - start_time))
size = db_travel_times.get_travel_time_table_size()
print str(i) + " ) " + str(size)
"""
# Load travel times from the db
start_time = datetime.now()
print("Loading travel times")
db_travel_times.load_travel_times(road_map, start_time)
end_time = datetime.now()
print(str(end_time - start_time))
"""