def test_integer_literal_expression(self):
input = "5;"
l = lexer.new(input)
p = parser.new(l)
program = p.parse_program()
self.check_parse_errors(p)
if len(program.statements) != 1:
print("Program has not enough statements. Got={}".format(
len(program.statements)))
return False
stmt = program.statements[0]
if not isinstance(stmt, ast.ExpressionStatement):
print("stmt is not a ast.ExpressionStatement. Got={}".format(
type(stmt)))
return False
if not isinstance(stmt.expression, ast.IntegerLiteral):
print("exp not ast.IntegerLiteral. Got={}".format(
type(stmt.expression)))
return False
if stmt.expression.value != 5:
print("literal.Value not {}. Got={}").format(
'5', stmt.expression.value)
return False
if stmt.expression.token_literal() != "5":
print("literal.TokenLiteral not {}. Got={}").format(
'foobar', stmt.expression.token_literal())
return False
def start():
while True:
line = input(prompt)
l = lexer.new(line)
tok = l.next_token()
while tok.Type != tokens.EOF:
print(tok)
tok = l.next_token()
def test_let_statements(self):
tests = [
("boop x = 5;", "x", 5),
("boop y = 10;", "y", 10),
("boop foobar = 838383;", "foobar", 838383),
]
for t in tests:
with self.subTest(test_case=t):
l = lexer.new(t[0])
p = parser.new(l)
program = p.parse_program()
self.check_parse_errors(p)
stmt = program.statements[0]
if not self.check_let_statement(stmt, t[1]):
return
def test_return_statements(self):
tests = [
("bork 5;", 5),
("bork 10;", 10),
("bork 993322;", 993322),
]
for t in tests:
with self.subTest(test_case=t):
l = lexer.new(t[0])
p = parser.new(l)
program = p.parse_program()
self.check_parse_errors(p)
stmt = program.statements[0]
if not self.check_return_statement(stmt, t[1]):
return
def test_identifier_expression(self):
input = 'foobar;'
l = lexer.new(input)
p = parser.new(l)
program = p.parse_program()
self.check_parse_errors(p)
stmt = program.statements[0]
if not isinstance(stmt, ast.ExpressionStatement):
print("stmt is not a ast.ExpressionStatement. Got={}".format(
type(stmt)))
return False
if not isinstance(stmt.expression, ast.Identifier):
print("exp not ast.Identifier. Got={}".format(type(
stmt.expression)))
return False
if stmt.expression.token_literal() != 'foobar':
print("ident.TokenLiteral not {}. Got={}").format(
'foobar', stmt.expression.token_literal())
return True