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])