TST: update tests to account for functions that plot

tacaswell · tacaswell · commit 4030ff49d595 · 2022-11-23T17:10:04.000-05:00
diff --git a/lib/matplotlib/tests/test_sphinxext.py b/lib/matplotlib/tests/test_sphinxext.py
@@ -81,7 +81,7 @@ def plot_directive_file(num):
     assert b'# Only a comment' in html_contents
     # check plot defined in external file.
     assert filecmp.cmp(range_4, img_dir / 'range4.png')
-    assert filecmp.cmp(range_6, img_dir / 'range6.png')
+    assert filecmp.cmp(range_6, img_dir / 'range6_range6.png')
     # check if figure caption made it into html file
     assert b'This is the caption for plot 15.' in html_contents
     # check if figure caption using :caption: made it into html file
@@ -95,6 +95,8 @@ def plot_directive_file(num):
     # Plot 21 is range(6) plot via an include directive. But because some of
     # the previous plots are repeated, the argument to plot_file() is only 17.
     assert filecmp.cmp(range_6, plot_file(17))
+    # plot 22 is from the range6.py file again, but a different function
+    assert filecmp.cmp(range_10, img_dir / 'range6_range10.png')
 
     # Modify the included plot
     contents = (tmp_path / 'included_plot_21.rst').read_bytes()
diff --git a/lib/matplotlib/tests/tinypages/range6.py b/lib/matplotlib/tests/tinypages/range6.py
@@ -11,3 +11,10 @@ def range6():
     plt.figure()
     plt.plot(range(6))
     plt.show()
+
+
+def range10():
+    """The function that should be executed."""
+    plt.figure()
+    plt.plot(range(10))
+    plt.show()
diff --git a/lib/matplotlib/tests/tinypages/some_plots.rst b/lib/matplotlib/tests/tinypages/some_plots.rst
@@ -168,8 +168,11 @@ scenario:
 
    plt.figure()
    plt.plot(range(4))
- 
+
 Plot 21 is generated via an include directive:
 
 .. include:: included_plot_21.rst
 
+Plot 22 uses a different specific function in a file with plot commands:
+
+.. plot:: range6.py range10
