diff --git a/src/parse.py b/src/parse.py index ba44eb4..283d4dc 100644 --- a/src/parse.py +++ b/src/parse.py @@ -11,3 +11,9 @@ if t.type not in [tokenize.TokenType.SPACE, tokenize.TokenType.NEWLINE]: output.append(t) return output + + +# Parses tokens +def parse(tokens): + stripped = remove_whitespace(tokens) + return stripped diff --git a/tests/test_parse.py b/tests/test_parse.py index 0d037e6..5982c36 100644 --- a/tests/test_parse.py +++ b/tests/test_parse.py @@ -31,3 +31,13 @@ def test_parse_remove_whitespace(test_data): (input, tokens) = test_data assert parse.remove_whitespace(input) == tokens + + +# Tests the parser wrapper works correctly +# We expect the following behaviour: +# - Whitespace tokens are removed +@given(lists(test_tokenize.draw_token_random())) +def test_parse_fuzz(tokens): + stripped = parse.remove_whitespace(tokens) + parsed = parse.parse(tokens) + assert stripped == parsed