diff --git a/tests/test_parse.py b/tests/test_parse.py index 8171e10..dc60985 100644 --- a/tests/test_parse.py +++ b/tests/test_parse.py @@ -1,5 +1,5 @@ from hypothesis import given -from hypothesis.strategies import text +from hypothesis.strategies import text, booleans from src import parse @@ -35,3 +35,17 @@ tokenizer = parse.Tokenizer(code, "") tokens = tokenizer.tokenize() assert tokens[0].type == "EOF" + + +# Test that we can make booleans using True and False +@given(booleans()) +def test_lexer_boolean(bool): + if bool == True: + code = "True" + else: + code = "False" + tokenizer = parse.Tokenizer(code, "") + tokens = tokenizer.tokenize() + assert tokens[0].type == "bool" + assert tokens[0].value == bool + assert tokens[1].type == "EOF"