def METHOD_NAME(code_str, arena):
    arena.PushSource(source.MainFile('pgen2_main'))
    line_reader = reader.StringLineReader(code_str, arena)
    line_lexer = lexer.LineLexer(arena)
    lex = lexer.Lexer(line_lexer, line_reader)
    return lex