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)
def test_stop_detector_on_mock_trip(self, device_id = 'fake_device_1', trip_id = '010414_00168'): remove_from_redis([device_id]) 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) reports = generate_mock_reports(device_id=device_id, trip_id=trip_id, nostop_percent=0.05, day=day) tracker_id = device_id for i, report in enumerate(reports): add_report(tracker_id, report=report) if (i % 100) == 0: print i stop_detector.print_tracked_stop_times(tracker_id) self.evaluate_detected_stop_times(tracker_id, trip_id) remove_from_redis([device_id]) print 'done'
def test_tracker_on_mock_device(self, device_id = 'fake_device_1', trip_ids = ['010414_00168'], 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 for report in reports[::2]: del report.my_loc_mock matched_trip = self.track_mock_reports(reports, tracker_id) if matched_trip: gtfs.models.Trip.objects.filter(trip_id = matched_trip)[0].print_stoptimes() self.assertEquals(matched_trip, trip_id) stop_detector_test.remove_from_redis(tracker_id)