def test_export_natural_sort(self): StopTime.objects.create( trip=self.trip, arrival_time='6:00:00', departure_time='6:00:00', stop=self.stop, stop_sequence=1, stop_headsign='SC', pickup_type=2, drop_off_type=1, shape_dist_traveled=5.25) stop2 = Stop.objects.create( feed=self.feed, stop_id='SALOON', point="POINT(-117.1 36.5)") StopTime.objects.create(trip=self.trip, stop=stop2, stop_sequence=2) stop3 = Stop.objects.create( feed=self.feed, stop_id='GENERAL_STORE', point="POINT(-117.2 36.5)") StopTime.objects.create(trip=self.trip, stop=stop3, stop_sequence=3) stop4 = Stop.objects.create( feed=self.feed, stop_id='MORGUE', point="POINT(-117.2 36.6)") StopTime.objects.create( trip=self.trip, arrival_time='7:00:00', departure_time='7:00:00', stop=stop4, stop_sequence=4, stop_headsign='MORT') stop_times_txt = StopTime.export_txt(self.feed) self.assertEqual(stop_times_txt, """\ trip_id,arrival_time,departure_time,stop_id,stop_sequence,stop_headsign,\ pickup_type,drop_off_type,shape_dist_traveled STBA,06:00:00,06:00:00,STAGECOACH,1,SC,2,1,5.25 STBA,,,SALOON,2,,,, STBA,,,GENERAL_STORE,3,,,, STBA,07:00:00,07:00:00,MORGUE,4,MORT,,, """)
def test_export_stop_times_minimal(self): StopTime.objects.create( trip=self.trip, arrival_time='6:00:00', departure_time='6:00:00', stop=self.stop, stop_sequence=1) stop_times_txt = StopTime.export_txt(self.feed) self.assertEqual(stop_times_txt, """\ trip_id,arrival_time,departure_time,stop_id,stop_sequence STBA,06:00:00,06:00:00,STAGECOACH,1 """)
def test_export_stop_times_maximal(self): StopTime.objects.create( trip=self.trip, arrival_time='6:00:00', departure_time='6:00:00', stop=self.stop, stop_sequence=1, stop_headsign='SC', pickup_type=2, drop_off_type=1, shape_dist_traveled=5.25) stop_times_txt = StopTime.export_txt(self.feed) self.assertEqual(stop_times_txt, """\ trip_id,arrival_time,departure_time,stop_id,stop_sequence,stop_headsign,\ pickup_type,drop_off_type,shape_dist_traveled STBA,06:00:00,06:00:00,STAGECOACH,1,SC,2,1,5.25 """)
def test_export_stop_times_none(self): stop_times_txt = StopTime.export_txt(self.feed) self.assertFalse(stop_times_txt)