Add tests

Author: harupy

lib/matplotlib/tests/test_sphinxext.py

@@ -95,12 +95,23 @@ 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))
-
-
-def build_sphinx_html(source_dir, doctree_dir, html_dir):
+    # 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,
+                      extra_args=["-D", "plot_html_show_source_link=0"])
+    assert "some_plots-1.py" not in [p.name for p in html_dir.iterdir()]
+
+
+def build_sphinx_html(source_dir, doctree_dir, html_dir, extra_args=None):
     # Build the pages with warnings turned into errors
+    extra_args = [] if extra_args is None else extra_args
     cmd = [sys.executable, '-msphinx', '-W', '-b', 'html',
-           '-d', str(doctree_dir), str(source_dir), str(html_dir)]
+           '-d', str(doctree_dir), str(source_dir), str(html_dir)] + extra_args
     proc = Popen(cmd, stdout=PIPE, stderr=PIPE, universal_newlines=True,
                  env={**os.environ, "MPLBACKEND": ""})
     out, err = proc.communicate()