diff --git a/tests/parse/test_text.py b/tests/parse/test_text.py index 1149048..3404c86 100644 --- a/tests/parse/test_text.py +++ b/tests/parse/test_text.py @@ -23,6 +23,7 @@ from tests.test_syntax import ( draw_token_by_value, draw_syntax_random, + draw_syntax_not_token, draw_syntax_token, ) @@ -43,14 +44,6 @@ return token -# Draws a random syntax that isn't a token -@composite -def draw_syntax_not_token(draw): - token = draw(draw_syntax_random()) - assume(token.type != SyntaxType.TOKEN) - return token - - # Draws tokens to make a valid text string and its value @composite def draw_syntax_text_valid(draw): diff --git a/tests/test_syntax.py b/tests/test_syntax.py index f6446dd..7e0b2d7 100644 --- a/tests/test_syntax.py +++ b/tests/test_syntax.py @@ -193,6 +193,15 @@ return draw(one_of(strategies)) +# Draws a random syntax that isn't a token +@composite +def draw_syntax_random_not_token(draw): + strategies = [ + draw_syntax_text(), + ] + return draw(one_of(strategies)) + + # Test syntax structure @template_test_structure( Syntax,