Exemplos de fill_stop_times em Python

Exemplo n.º 1

Arquivo: populate_pb_test.py Projeto: mbounabi/kirin

def test_populate_pb_skipped_for_detour_stop_times_status():
    st_added_status = StopTimeUpdate({'id': 'id1'},
                                     dep_status='deleted_for_detour',
                                     arr_status='deleted_for_detour')
    pb_stop_time = gtfs_realtime_pb2.TripUpdate.StopTimeUpdate()

    fill_stop_times(pb_stop_time, st_added_status)

    assert pb_stop_time.departure.Extensions[
        kirin_pb2.stop_time_event_status] == kirin_pb2.DELETED_FOR_DETOUR
    assert pb_stop_time.arrival.Extensions[
        kirin_pb2.stop_time_event_status] == kirin_pb2.DELETED_FOR_DETOUR

Exemplo n.º 2

Arquivo: populate_pb_test.py Projeto: patochectp/kirin

def test_populate_pb_deleted_stop_times_status():
    st_added_status = StopTimeUpdate({"id": "id1"},
                                     dep_status="delete",
                                     arr_status="delete")
    pb_stop_time = gtfs_realtime_pb2.TripUpdate.StopTimeUpdate()

    fill_stop_times(pb_stop_time, st_added_status)

    assert pb_stop_time.departure.Extensions[
        kirin_pb2.stop_time_event_status] == kirin_pb2.DELETED
    assert pb_stop_time.arrival.Extensions[
        kirin_pb2.stop_time_event_status] == kirin_pb2.DELETED

Exemplo n.º 3

Arquivo: populate_pb_test.py Projeto: mbounabi/kirin

def test_populate_pb_no_status_stop_times_status():
    st_no_status = StopTimeUpdate({'id': 'id1'},
                                  dep_status='none',
                                  arr_status='none')
    pb_stop_time = gtfs_realtime_pb2.TripUpdate.StopTimeUpdate()

    fill_stop_times(pb_stop_time, st_no_status)

    assert pb_stop_time.departure.Extensions[
        kirin_pb2.stop_time_event_status] == kirin_pb2.SCHEDULED
    assert pb_stop_time.arrival.Extensions[
        kirin_pb2.stop_time_event_status] == kirin_pb2.SCHEDULED

Exemplo n.º 4

Arquivo: populate_pb_test.py Projeto: patochectp/kirin

def test_populate_pb_added_for_detour_stop_times_status():
    st_added_status = StopTimeUpdate({"id": "id1"},
                                     dep_status="added_for_detour",
                                     arr_status="added_for_detour")
    pb_stop_time = gtfs_realtime_pb2.TripUpdate.StopTimeUpdate()

    fill_stop_times(pb_stop_time, st_added_status)

    assert pb_stop_time.departure.Extensions[
        kirin_pb2.stop_time_event_status] == kirin_pb2.ADDED_FOR_DETOUR
    assert pb_stop_time.arrival.Extensions[
        kirin_pb2.stop_time_event_status] == kirin_pb2.ADDED_FOR_DETOUR