def test_tracker_on_mock_device(self,
device_id='fake_device_1',
trip_ids=['010714_00115'],
remove_some_locations=True):
if not isinstance(trip_ids, list):
trip_ids = [trip_ids]
tracker_id = device_id
stop_detector_test.remove_from_redis(tracker_id)
reports = []
for trip_id in trip_ids:
day = datetime.datetime.strptime(trip_id.split('_')[0], '%d%m%y')
now = ot_utils.get_localtime_now(
) # we want to get the correct timezone so we take it from get_localtime_now()
day = now.replace(year=day.year, month=day.month, day=day.day)
trip_reports = generate_mock_reports(
device_id=device_id,
trip_id=trip_id,
nostop_percent=0.05,
day=day)
reports += trip_reports
if remove_some_locations:
for report in reports[::2]:
del report.my_loc_mock
logger.addFilter(
MessageExcludeFilter(
'skipped because it has not location data'))
matched_trips = self.track_mock_reports(reports, tracker_id)
for matched_trip in matched_trips:
timetable.services.get_trip(matched_trip).print_stoptimes()
self.assertEquals(len(matched_trips), len(trip_ids))
self.assertEquals(sorted(matched_trips), sorted(trip_ids))
stop_detector_test.remove_from_redis(tracker_id)
get_redis_client,
load_by_key,
save_by_key)
import stop_detector_test
import stop_detector
import trip_matcher
import trip_ground_truth
from alg_logger import MessageExcludeFilter
import train_tracker_test
def run_tracker_on_trips_without_ground_truth(device_ids=None):
trip_suffixes_list = []
if not device_ids:
device_ids = [x for x in trip_ground_truth.data if not trip_ground_truth.data[x]]
stop_detector_test.remove_from_redis(device_ids)
for i in xrange(len(device_ids)):
device_id = device_ids[i]
tracker_id, trips = train_tracker_test.track_device(device_id, do_preload_reports=True, report_limit=100)
print 'DEVICE_ID=%s' % device_id
for trip_id in trips:
timetable.services.print_trip_stop_times(trip_id)
stop_detector.print_tracked_stop_times(device_id)
stop_detector_test.remove_from_redis(device_ids)
if __name__ == '__main__':
logger.addFilter(MessageExcludeFilter('saving disabled!!!'))
run_tracker_on_trips_without_ground_truth(['ofer_3138f247524647e4'])