import busbus
from busbus.provider.lawrenceks import LawrenceTransitProvider

import arrow
import pytest


@pytest.fixture(scope='module')
def lawrenceks_provider():
    return LawrenceTransitProvider()


def test_43_to_eaton_hall(lawrenceks_provider):
    stop = lawrenceks_provider.get(busbus.Stop, u'15TH_SPAHR_WB')
    route = lawrenceks_provider.get(busbus.Route, u'RT_43')
    assert len(list(lawrenceks_provider.arrivals.where(
        stop=stop, route=route,
        start_time=arrow.get('2015-03-10T14:00:00-05:00'),
        end_time=arrow.get('2015-03-10T16:00:00-05:00')))) == 13

Add a simple test case for LawrenceTransitProvider
Add a simple test case for LawrenceTransitProvider
