Split code at plt.show().
(text, function_name)
| 375 | |
| 376 | |
| 377 | def _split_code_at_show(text, function_name): |
| 378 | """Split code at plt.show().""" |
| 379 | |
| 380 | is_doctest = contains_doctest(text) |
| 381 | if function_name is None: |
| 382 | parts = [] |
| 383 | part = [] |
| 384 | for line in text.split("\n"): |
| 385 | if ((not is_doctest and line.startswith('plt.show(')) or |
| 386 | (is_doctest and line.strip() == '>>> plt.show()')): |
| 387 | part.append(line) |
| 388 | parts.append("\n".join(part)) |
| 389 | part = [] |
| 390 | else: |
| 391 | part.append(line) |
| 392 | if "\n".join(part).strip(): |
| 393 | parts.append("\n".join(part)) |
| 394 | else: |
| 395 | parts = [text] |
| 396 | return is_doctest, parts |
| 397 | |
| 398 | |
| 399 | # ----------------------------------------------------------------------------- |
no test coverage detected
searching dependent graphs…