def testCreation(self): # This is mostly just a sanity check db = get_trip_db() db.remove() create_fake_trips() list_of_cluster_data = eamtcp.main() tm = create_tour_model('test_user', list_of_cluster_data) self.assertEquals(len(tm.get_top_trips(1)), 1) tour = tm.build_tour_model() self.assertEquals(len(tour), 7)
def make_tour_model_from_fake_data(fake_user_id): estg.create_fake_trips(fake_user_id, True) make_tour_model_from_raw_user_data(fake_user_id)
def get_fake_data(user_name): # Call with a username unique to your database tg.create_fake_trips(user_name, True) return eamtcp.main(user_name, old=False)
def get_fake_data(user_name): # Call with a username unique to your database tg.create_fake_trips(user_name, True) return eamtcp.main(user_name)