def testCreation(self):
common_trip = esdctp.make_new_common_trip()
common_trip.user_id = self.testUserId
common_trip.start_loc = self.testStart.location
common_trip.end_loc = self.testEnd.location
common_trip.trips = []
self.assertIsNotNone(common_trip.start_loc)
self.assertIsNotNone(common_trip.end_loc)
self.assertIsNotNone(common_trip.trips)
def testCreation(self):
common_trip = esdctp.make_new_common_trip()
common_trip.user_id = self.testUserId
common_trip.start_loc = self.testStart.location
common_trip.end_loc = self.testEnd.location
common_trip.trips = []
self.assertIsNotNone(common_trip.start_loc)
self.assertIsNotNone(common_trip.end_loc)
self.assertIsNotNone(common_trip.trips)
def testSaveAndRecieve(self):
common_trip = esdctp.make_new_common_trip()
common_trip.user_id = self.testUserId
common_trip.start_place = self.testStart.get_id()
common_trip.end_place = self.testEnd.get_id()
common_trip.start_loc = self.testStart.location
common_trip.end_loc = self.testEnd.location
common_trip.probabilites = np.zeros((24, 7))
common_trip.trips = []
esdctp.save_common_trip(common_trip)
new_trip = esdctp.get_common_trip_from_db(self.testUserId, self.testStart.get_id(), self.testEnd.get_id())
self.assertEqual(new_trip.user_id, common_trip.user_id)
self.assertEqual(gj.dumps(new_trip.start_loc), gj.dumps(common_trip.start_loc))
self.assertEqual(gj.dumps(new_trip.end_loc), gj.dumps(common_trip.end_loc))
def testSaveAndRecieve(self):
common_trip = esdctp.make_new_common_trip()
common_trip.user_id = self.testUserId
common_trip.start_place = self.testStart.get_id()
common_trip.end_place = self.testEnd.get_id()
common_trip.start_loc = self.testStart.location
common_trip.end_loc = self.testEnd.location
common_trip.probabilites = np.zeros((24, 7))
common_trip.trips = []
esdctp.save_common_trip(common_trip)
new_trip = esdctp.get_common_trip_from_db(self.testUserId,
self.testStart.get_id(),
self.testEnd.get_id())
self.assertEqual(new_trip.user_id, common_trip.user_id)
self.assertEqual(gj.dumps(new_trip.start_loc),
gj.dumps(common_trip.start_loc))
self.assertEqual(gj.dumps(new_trip.end_loc),
gj.dumps(common_trip.end_loc))
def testGetSuccessor(self):
logging.debug("size of db is %s" % edb.get_common_place_db().find().count())
self.assertIsNotNone(edb.get_common_place_db().find_one({"_id": self.testEnd.get_id()}))
probs = np.zeros( (7, 24) )
probs[self.time0.weekday(), 3] = 10
props = {
"user_id" : self.testUserId,
"start_place" : self.testStart.get_id(),
"end_place" : self.testEnd.get_id(),
"start_loc" : self.testStart.location,
"end_loc" : self.testEnd.location,
"probabilites" : probs,
"trips" : ()
}
trip = esdctp.make_new_common_trip(props)
esdctp.save_common_trip(trip)
suc = esdcpq.get_succesor(self.testUserId, self.testStart.get_id(), self.time0)
self.assertEqual(suc, self.testEnd.get_id())
def testGetSuccessor(self):
logging.debug("size of db is %s" % edb.get_common_place_db().find().count())
self.assertIsNotNone(edb.get_common_place_db().find_one({"_id": self.testEnd.get_id()}))
probs = np.zeros( (7, 24) )
probs[self.time0.weekday(), 3] = 10
props = {
"user_id" : self.testUserId,
"start_place" : self.testStart.get_id(),
"end_place" : self.testEnd.get_id(),
"start_loc" : self.testStart.location,
"end_loc" : self.testEnd.location,
"probabilites" : probs,
"trips" : (),
"start_times": [],
"durations": []
}
trip = esdctp.make_new_common_trip(props)
esdctp.save_common_trip(trip)
suc = esdcpq.get_succesor(self.testUserId, self.testStart.get_id(), self.time0)
self.assertEqual(suc, self.testEnd.get_id())
def get_common_trips(user_id):
db = edb.get_common_trip_db()
trip_jsons = db.find({"user_id" : user_id})
return [esdctq.make_new_common_trip(trip_json) for trip_json in trip_jsons]
def get_common_trips(user_id):
db = edb.get_common_trip_db()
trip_jsons = db.find({"user_id": user_id})
return [esdctq.make_new_common_trip(trip_json) for trip_json in trip_jsons]
