diff --git a/tests/test_tokenize.py b/tests/test_tokenize.py index 807d7c1..a5f5d41 100644 --- a/tests/test_tokenize.py +++ b/tests/test_tokenize.py @@ -6,6 +6,7 @@ booleans, characters, composite, + integers, just, lists, sampled_from, @@ -47,6 +48,29 @@ assert (location1 == location2) == equals +# Draws a random symbol +@composite +def draw_symbol(draw): + value = draw(text()) + location = draw(draw_symbol_location()) + return tokenize.Symbol(value, location) + + +# Test symbol getters +@given(text(), draw_symbol_location()) +def test_tokenize_symbol_getters(value, location): + test = tokenize.Symbol(value, location) + assert test.value == value + assert test.location == location + + +# Test symbol equals +@given(draw_symbol(), draw_symbol()) +def test_tokenize_symbol_equality(symbol1, symbol2): + equals = symbol1.value == symbol2.value and symbol1.location == symbol2.location + assert (symbol1 == symbol2) == equals + + # Draws a tokenizer non-whitespace symbol @composite def draw_symbol_nonwhitespace(draw):