Python StopTime.import_txt примеры использования

Пример #1

Файл: stop_time.py Проект: cheekybastard/django-multi-gtfs

    def test_import_stop_times_txt_middle_times_optional(self):
        stop_times_txt = StringIO.StringIO("""\
trip_id,arrival_time,departure_time,stop_id,stop_sequence
STBA,6:00:00,6:00:00,STAGECOACH,1
STBA,,,STAGECOACH2,2
STBA,12:00:00,12:00:00,STAGECOACH3,3
""")
        stop2 = Stop.objects.create(feed=self.feed,
                                    stop_id='STAGECOACH2',
                                    lat="36.425288",
                                    lon="-117.133162")
        stop3 = Stop.objects.create(feed=self.feed,
                                    stop_id='STAGECOACH3',
                                    lat="36.425288",
                                    lon="-117.133162")
        StopTime.import_txt(stop_times_txt, self.feed)
        stoptime1 = StopTime.objects.get(stop=self.stop)
        self.assertEqual(str(stoptime1.arrival_time), '06:00:00')
        self.assertEqual(str(stoptime1.departure_time), '06:00:00')
        stoptime2 = StopTime.objects.get(stop=stop2)
        self.assertEqual(stoptime2.arrival_time, None)
        self.assertEqual(stoptime2.departure_time, None)
        stoptime3 = StopTime.objects.get(stop=stop3)
        self.assertEqual(str(stoptime3.arrival_time), '12:00:00')
        self.assertEqual(str(stoptime3.departure_time), '12:00:00')

Пример #2

    def test_import_stop_times_txt_bad_column_empty_OK(self):
        stop_times_txt = StringIO.StringIO("""\
trip_id,arrival_time,departure_time,stop_id,stop_sequence,drop_off_time
STBA,6:00:00,6:00:00,STAGECOACH,1,
""")
        StopTime.import_txt(stop_times_txt, self.feed)
        stoptime = StopTime.objects.get()
        self.assertEqual(stoptime.trip, self.trip)
        self.assertEqual(str(stoptime.arrival_time), '06:00:00')
        self.assertEqual(str(stoptime.departure_time), '06:00:00')
        self.assertEqual(stoptime.stop, self.stop)
        self.assertEqual(stoptime.stop_sequence, 1)
        self.assertEqual(stoptime.drop_off_type, '')

Пример #3

Файл: test_stop_time.py Проект: PolyglotOpenstreetmap/gtfsintegrate

    def test_import_stop_times_txt_duplicate(self):
        Stop.objects.create(
            feed=self.feed, stop_id='XXX',
            point="POINT(-117.133 36.425)")
        stop_times_txt = StringIO("""\
trip_id,arrival_time,departure_time,stop_id,stop_sequence
STBA,6:00:00,6:00:00,STAGECOACH,1
STBA,7:00:00,7:00:00,XXX,1
""")
        StopTime.import_txt(stop_times_txt, self.feed)
        stoptime = StopTime.objects.get()  # Just one
        self.assertEqual(stoptime.trip, self.trip)
        self.assertEqual(str(stoptime.arrival_time), '06:00:00')

Пример #4

Файл: stop_time.py Проект: brendancol/django-multi-gtfs

    def test_import_stop_times_txt_bad_column_empty_OK(self):
        stop_times_txt = StringIO.StringIO("""\
trip_id,arrival_time,departure_time,stop_id,stop_sequence,drop_off_time
STBA,6:00:00,6:00:00,STAGECOACH,1,
""")
        StopTime.import_txt(stop_times_txt, self.feed)
        stoptime = StopTime.objects.get()
        self.assertEqual(stoptime.trip, self.trip)
        self.assertEqual(str(stoptime.arrival_time), '06:00:00')
        self.assertEqual(str(stoptime.departure_time), '06:00:00')
        self.assertEqual(stoptime.stop, self.stop)
        self.assertEqual(stoptime.stop_sequence, 1)
        self.assertEqual(stoptime.drop_off_type, '')

Пример #5

    def test_import_stop_times_txt_minimal(self):
        stop_times_txt = StringIO.StringIO("""\
trip_id,arrival_time,departure_time,stop_id,stop_sequence
STBA,6:00:00,6:00:00,STAGECOACH,1
""")
        StopTime.import_txt(stop_times_txt, self.feed)
        stoptime = StopTime.objects.get()
        self.assertEqual(stoptime.trip, self.trip)
        self.assertEqual(str(stoptime.arrival_time), '06:00:00')
        self.assertEqual(str(stoptime.departure_time), '06:00:00')
        self.assertEqual(stoptime.stop, self.stop)
        self.assertEqual(stoptime.stop_sequence, 1)
        self.assertEqual(stoptime.stop_headsign, '')
        self.assertEqual(stoptime.pickup_type, '')
        self.assertEqual(stoptime.drop_off_type, '')
        self.assertEqual(stoptime.shape_dist_traveled, None)

Пример #6

Файл: stop_time.py Проект: alaw005/django-multi-gtfs

    def test_import_stop_times_txt_minimal(self):
        stop_times_txt = StringIO("""\
trip_id,arrival_time,departure_time,stop_id,stop_sequence
STBA,6:00:00,6:00:00,STAGECOACH,1
""")
        StopTime.import_txt(stop_times_txt, self.feed)
        stoptime = StopTime.objects.get()
        self.assertEqual(stoptime.trip, self.trip)
        self.assertEqual(str(stoptime.arrival_time), '06:00:00')
        self.assertEqual(str(stoptime.departure_time), '06:00:00')
        self.assertEqual(stoptime.stop, self.stop)
        self.assertEqual(stoptime.stop_sequence, 1)
        self.assertEqual(stoptime.stop_headsign, '')
        self.assertEqual(stoptime.pickup_type, '')
        self.assertEqual(stoptime.drop_off_type, '')
        self.assertEqual(stoptime.shape_dist_traveled, None)

Пример #7

Файл: stop_time.py Проект: alaw005/django-multi-gtfs

    def test_import_stop_times_txt_middle_times_optional(self):
        stop_times_txt = StringIO("""\
trip_id,arrival_time,departure_time,stop_id,stop_sequence
STBA,6:00:00,6:00:00,STAGECOACH,1
STBA,,,STAGECOACH2,2
STBA,12:00:00,12:00:00,STAGECOACH3,3
""")
        stop2 = Stop.objects.create(
            feed=self.feed, stop_id='STAGECOACH2',
            point="POINT(-117.133162 36.425288)")
        stop3 = Stop.objects.create(
            feed=self.feed, stop_id='STAGECOACH3',
            point="POINT(-117.133162 36.425288)")
        StopTime.import_txt(stop_times_txt, self.feed)
        stoptime1 = StopTime.objects.get(stop=self.stop)
        self.assertEqual(str(stoptime1.arrival_time), '06:00:00')
        self.assertEqual(str(stoptime1.departure_time), '06:00:00')
        stoptime2 = StopTime.objects.get(stop=stop2)
        self.assertEqual(stoptime2.arrival_time, None)
        self.assertEqual(stoptime2.departure_time, None)
        stoptime3 = StopTime.objects.get(stop=stop3)
        self.assertEqual(str(stoptime3.arrival_time), '12:00:00')
        self.assertEqual(str(stoptime3.departure_time), '12:00:00')