Update test

MikhailRyazanov · web-flow · commit 3e625e9c5cd9 · 2022-10-17T23:29:38.000+03:00
diff --git a/lib/matplotlib/tests/test_sphinxext.py b/lib/matplotlib/tests/test_sphinxext.py
@@ -18,6 +18,7 @@ def test_tinypages(tmpdir):
     source_dir = Path(tmpdir) / 'src'
     shutil.copytree(Path(__file__).parent / 'tinypages', source_dir)
     html_dir = source_dir / '_build' / 'html'
+    img_dir = html_dir / '_images'
     doctree_dir = source_dir / 'doctrees'
     # Build the pages with warnings turned into errors
     cmd = [sys.executable, '-msphinx', '-W', '-b', 'html',
@@ -35,7 +36,7 @@ def test_tinypages(tmpdir):
     build_sphinx_html(source_dir, doctree_dir, html_dir)
 
     def plot_file(num):
-        return html_dir / f'some_plots-{num}.png'
+        return img_dir / f'some_plots-{num}.png'
 
     def plot_directive_file(num):
         # This is always next to the doctree dir.
@@ -58,8 +59,8 @@ def plot_directive_file(num):
 
     assert b'# Only a comment' in html_contents
     # check plot defined in external file.
-    assert filecmp.cmp(range_4, html_dir / 'range4.png')
-    assert filecmp.cmp(range_6, html_dir / 'range6.png')
+    assert filecmp.cmp(range_4, img_dir / 'range4.png')
+    assert filecmp.cmp(range_6, img_dir / '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
@@ -113,13 +114,13 @@ def test_plot_html_show_source_link(tmpdir):
     # 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()]
+    assert len(list(html_dir1.glob("**/index-1.py"))) == 1
     # 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 "index-1.py" not in [p.name for p in html_dir2.iterdir()]
+    assert len(list(html_dir2.glob("**/index-1.py"))) == 0
 
 
 def build_sphinx_html(source_dir, doctree_dir, html_dir, extra_args=None):
