diff --git a/tests/test_parse.py b/tests/test_parse.py index 0e86344..1edcde6 100644 --- a/tests/test_parse.py +++ b/tests/test_parse.py @@ -263,8 +263,11 @@ assert tokens[out_pos].location.column == soup.tokens[in_pos].location.column in_pos += 1 out_pos += 1 + all_lines = soup.code.split("\n") assert tokens[out_pos].type == "EOF" assert tokens[out_pos].location.file == "" + assert tokens[out_pos].location.line == len(all_lines) + assert tokens[out_pos].location.column == len(all_lines[-1]) # General fuzz test, make sure the parser doesn't fall apart and spew