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]