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())
示例#4
0
 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)
示例#6
0
def save_trips(tour_model):
    for trip in tour_model.common_trips:
        esdctq.save_common_trip(trip)