<commit_before>import busbus
from busbus.provider.lawrenceks import LawrenceTransitProvider

import arrow
import pytest


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


def test_agency_phone_e164(lawrenceks_provider):
    agency = next(lawrenceks_provider.agencies)
    assert agency.phone_e164 == '+17858644644'


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<commit_msg>Add another stop to test_43_to_eaton_hall<commit_after>import busbus
from busbus.provider.lawrenceks import LawrenceTransitProvider

import arrow
import pytest


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


def test_agency_phone_e164(lawrenceks_provider):
    agency = next(lawrenceks_provider.agencies)
    assert agency.phone_e164 == '+17858644644'


@pytest.mark.parametrize('stop_id,count', [
    (u'15TH_SPAHR_WB', 13),
    (u'SNOW_HALL_WB', 14),
])
def test_43_to_eaton_hall(lawrenceks_provider, stop_id, count):
    stop = lawrenceks_provider.get(busbus.Stop, stop_id)
    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')))) == count
{'fraction_non_alphanumeric': 0.11775147928994083, 'fraction_numerical': 0.07455621301775148, 'mean_word_length': 5.65748031496063, 'pattern_counts': {'":': 0, '<': 3, '<?xml version=': 0, '>': 3, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '5543980', 'n_tokens_mistral': 696, 'n_tokens_neox': 611, 'n_words': 84}