def has_succesor(user_id, place_id, time): day = time.weekday() place = get_common_place_from_db(place_id) for suc in place["successors"]: trip = esdctp.get_common_trip_from_db(user_id, place_id, suc) for temp_hour in xrange(time.hour, esdctp.HOURS_IN_DAY): if trip.probabilites[day, temp_hour] > 0: return True return False
def get_succesor(user_id, place_id, time): temp = esmmc.Counter() day = time.weekday() place = get_common_place_from_db(place_id) for suc in place["successors"]: trip = esdctp.get_common_trip_from_db(user_id, place_id, suc) for temp_hour in xrange(time.hour, esdctp.HOURS_IN_DAY): counter_key = ("%s" % suc, temp_hour) temp[counter_key] = trip.probabilites[day, temp_hour] return boi.ObjectId(esmmc.sampleFromCounter(temp)[0])
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))