def test_tokenstream_expect(): # expect s = TokenStream(iter((Token('a', 1), Token('b', 2), Token('c', 3)))) assert s.expect('a') == Token('a', 1) assert s.expect('b', 2) == Token('b', 2) pytest.raises(AssertionError, s.expect, 'e') pytest.raises(AssertionError, s.expect, 'c', 5)