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")]
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