API: only split the code on fully-dedented plt.show calls

tacaswell · tacaswell · commit 3c90d8f6d54c · 2022-11-23T16:18:54.000-05:00
Also remove the () from the search so `plt.show(block=True)` will also
be caught.

The reason for this change is so that calling `plt.show()` inside of a function
will not try to split the code into independently executed chunks.
diff --git a/lib/matplotlib/sphinxext/plot_directive.py b/lib/matplotlib/sphinxext/plot_directive.py
@@ -348,7 +348,7 @@ def _split_code_at_show(text):
     is_doctest = contains_doctest(text)
     part = []
     for line in text.split("\n"):
-        if (not is_doctest and line.strip() == 'plt.show()') or \
+        if (not is_doctest and line.startswith('plt.show')) or \
                (is_doctest and line.strip() == '>>> plt.show()'):
             part.append(line)
             parts.append("\n".join(part))
