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 save_trips(tour_model): for trip in tour_model.common_trips: esdctq.save_common_trip(trip)