def test_can_retrieve_coords():
stations = Stations()
station = stations.find("FRFEV")
coords = station.coords
assert len(coords) == 2
assert coords[0] == 44.9288345
assert coords[1] == -0.4963844
def test_can_compute_distance_between_two_stations():
stations = Stations()
origin = stations.find("FRFEV")
destination = stations.find("FRJFU")
distance = origin.distance_to(destination)
assert distance > 456
assert distance < 457
def test_can_find_station_by_name():
stations = Stations()
assert stations.find("Le Porage")
assert stations.find("LE PORAGE")
assert stations.find("le porage")
# Accents
assert stations.find("Jeumont Frontière")
assert stations.find("Jeumont Frontiere")
def segments():
stations = Stations()
departure_station = stations.find("FRFEV")
arrival_station = stations.find("FRJFU")
segments = [
Segment(
Transport("TER", "TER", "0000", "TRAIN"),
departure_station,
arrival_station,
dt.datetime(2019, 1, 1, 0),
dt.datetime(2019, 1, 1, 1, 30),
),
Segment(
Transport("TER", "TER", "0000", "TRAIN"),
departure_station,
arrival_station,
dt.datetime(2019, 1, 1, 1, 45),
dt.datetime(2019, 1, 1, 2, 15),
),
]
return tuple(segments)
import datetime as dt
import pytz
from locomotive.api.gc import Client
from locomotive.api.requests import BoardRequest
from locomotive.stores import Stations
stations = Stations()
client = Client(stations)
def test_board_request_1():
req = BoardRequest(stations.find("Lyon Part-Dieu"), "departure")
res = client.board_request(req)
assert len(res) > 0
req = BoardRequest(stations.find("Lyon Part-Dieu"), "arrival")
res = client.board_request(req)
assert len(res) > 0
def test_can_load_default_data_if_no_path_provided():
stations = Stations()
assert stations.count() > 0
def test_raises_exception_when_cannot_find_station_by_a_given_id():
stations = Stations()
with pytest.raises(StationNotFoundException):
stations.find_or_raise("This station ID doesn't exist")
def test_can_find_station_by_id():
stations = Stations()
assert stations.find("FRFEV")
assert stations.find("frfev")
assert stations.find("frFeV")