Exemple #1
0
def filter_filter_dt_test(mocker):
    mocker.patch('jormungandr.utils.get_timezone', return_value=pytz.timezone('UTC'))
    passages = [passage("10:00"), passage("11:00"), passage("12:00"), passage("13:00")]
    proxy = CustomProxy(passages)

    r = proxy.next_passage_for_route_point(None, from_dt=d2t(dt("12:00")))
    assert list(map(get_dt, r)) == [dt("12:00"), dt("13:00")]
Exemple #2
0
def filter_filter_dt_all_test(mocker):
    mocker.patch('jormungandr.utils.get_timezone', return_value=pytz.timezone('UTC'))
    """the filter will filter all, so we should not get an empty list but None"""
    passages = [passage("10:00"), passage("11:00"), passage("12:00"), passage("13:00")]
    proxy = CustomProxy(passages)

    r = proxy.next_passage_for_route_point(None, count=1, from_dt=d2t(dt("15:00")))
    assert r is None