def split_code_at_show(text):
parts = []<EOL>is_doctest = contains_doctest(text)<EOL>part = []<EOL>for line in text.split("<STR_LIT:\n>"):<EOL><INDENT>if (not is_doctest and line.strip() == '<STR_LIT>') or(is_doctest and line.strip() == '<STR_LIT>'):<EOL><INDENT>part.append(line)<EOL>parts.append("<STR_LIT:\n>".join(part))<EOL>part = []<EOL><DEDENT>else:<EOL><INDENT>part.append(line)<EOL><DEDENT><DEDENT>if "<STR_LIT:\n>".join(part).strip():<EOL><INDENT>parts.append("<STR_LIT:\n>".join(part))<EOL><DEDENT>return parts<EOL>
Split code at plt.show()