run tests faster

Signed-off-by: harupy <[email protected]>

Author: harupy

lib/matplotlib/tests/test_sphinxext.py

@@ -95,16 +95,28 @@ def plot_directive_file(num):
     # contents are the same)
     assert plot_directive_file(5).stat().st_mtime > modification_times[3]
     assert filecmp.cmp(range_6, plot_file(5))
-    # Make sure that source scripts are generated by default
-    assert "some_plots-1.py" in [p.name for p in html_dir.iterdir()]
-    # Make sure that source scripts are NOT generated when
-    # `plot_html_show_source_link` is set to False
-    html_dir = source_dir / '_build' / 'html_no_source'
-    # `plot_html_show_source_link=0` is equivalent to
-    # `plot_html_show_source_link=False`
-    build_sphinx_html(source_dir, doctree_dir, html_dir,
+
+
+def test_plot_html_show_source_link(tmpdir):
+    source_dir = Path(tmpdir) / 'src'
+    source_dir.mkdir()
+    shutil.copyfile(Path(__file__).parent / 'tinypages' / "conf.py", source_dir / "conf.py")
+    shutil.copytree(Path(__file__).parent / 'tinypages' / "_static", source_dir / "_static")
+    doctree_dir = source_dir / 'doctrees'
+    (source_dir / 'index.rst').write_text("""
+.. plot::
+
+    plt.plot(range(4))
+""")
+    # Make sure source scripts are created by default
+    html_dir1 = source_dir / '_build' / 'html1'
+    build_sphinx_html(source_dir, doctree_dir, html_dir1)
+    assert "index-1.py" in [p.name for p in html_dir1.iterdir()]
+    # Make sure source scripts are NOT created when `plot_html_show_source_link` is False
+    html_dir2 = source_dir / '_build' / 'html2'
+    build_sphinx_html(source_dir, doctree_dir, html_dir2,
                       extra_args=["-D", "plot_html_show_source_link=0"])
-    assert "some_plots-1.py" not in [p.name for p in html_dir.iterdir()]
+    assert "index-1.py" not in [p.name for p in html_dir2.iterdir()]
 
 
 def build_sphinx_html(source_dir, doctree_dir, html_dir, extra_args=None):