diff --git a/tests/test_lex.py b/tests/test_lex.py index 8771123..cee5661 100644 --- a/tests/test_lex.py +++ b/tests/test_lex.py @@ -57,10 +57,17 @@ def test_lex_expression_integer(): def test_lex_line_counting(): """ Make sure we can count lines / columns """ - entry = tokenize("(foo 2)")[0] + entry = tokenize("(foo (one two))")[0] assert entry.start_line == 1 assert entry.start_column == 1 assert entry.end_line == 1 - assert entry.end_column == 7 + assert entry.end_column == 15 + + entry = entry[1] + assert entry.start_line == 1 + assert entry.start_column == 6 + + assert entry.end_line == 1 + assert entry.end_column == 14