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'])