def p_for_loop_infinite(p):
"""
statement : FOR LBRACK statement_list RBRACK
"""
p[0] = ast.While(ast.Primitive(True), p[3])
def p_while_loop(p):
"""
statement : WHILE expression LBRACK statement_list RBRACK
"""
p[0] = ast.While(p[2], p[4])