diff --git a/tests/test_token.py b/tests/test_token.py index 9176286..0ac8098 100644 --- a/tests/test_token.py +++ b/tests/test_token.py @@ -72,7 +72,7 @@ # Draws a bool token @composite def draw_token_bool(draw): - location = draw(draw_token_location()) + location = static_token_location() if draw(booleans()): value = "True" else: @@ -83,7 +83,7 @@ # Draws a keyword token @composite def draw_token_keyword(draw): - location = draw(draw_token_location()) + location = static_token_location() value = draw(sampled_from(keywords)) return Token(value, location) @@ -105,7 +105,7 @@ # Draws an unknown token @composite def draw_token_unknown(draw): - location = draw(draw_token_location()) + location = static_token_location() value = draw(text(min_size=1)) assume(value not in literals) assume(value not in keywords)