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)
