Add machinery for svg-only mathtext tests.

anntzer · anntzer · commit 15f690582683 · 2021-01-05T07:23:22.000+01:00
diff --git a/lib/matplotlib/testing/compare.py b/lib/matplotlib/testing/compare.py
@@ -181,7 +181,7 @@ def __call__(self, orig, dest):
                 # just be reported as a regular exception below).
                 "DISPLAY": "",
                 # Do not load any user options.
-                "INKSCAPE_PROFILE_DIR": os.devnull,
+                "INKSCAPE_PROFILE_DIR": self._tmpdir.name,
             }
             # Old versions of Inkscape (e.g. 0.48.3.1) seem to sometimes
             # deadlock when stderr is redirected to a pipe, so we redirect it
@@ -233,6 +233,15 @@ def __del__(self):
             self._tmpdir.cleanup()
 
 
+class _SVGWithMatplotlibFontsConverter(_SVGConverter):
+    def __call__(self, orig, dest):
+        if not hasattr(self, "_tmpdir"):
+            self._tmpdir = TemporaryDirectory()
+            shutil.copytree(cbook._get_data_path("fonts/ttf"),
+                            Path(self._tmpdir.name, "fonts"))
+        return super().__call__(orig, dest)
+
+
 def _update_converter():
     try:
         mpl._get_executable_info("gs")
@@ -254,6 +263,7 @@ def _update_converter():
 #: extension to png format.
 converter = {}
 _update_converter()
+_svg_with_matplotlib_fonts_converter = _SVGWithMatplotlibFontsConverter()
 
 
 def comparable_formats():
@@ -303,7 +313,12 @@ def convert(filename, cache):
                 return str(newpath)
 
         _log.debug("For %s: converting to png.", filename)
-        converter[path.suffix[1:]](path, newpath)
+        convert = converter[path.suffix[1:]]
+        if path.suffix == ".svg":
+            contents = path.read_text()
+            if 'style="font:' in contents:  # for svg.fonttype = none.
+                convert = _svg_with_matplotlib_fonts_converter
+        convert(path, newpath)
 
         if cache_dir is not None:
             _log.debug("For %s: caching conversion result.", filename)
diff --git a/lib/matplotlib/tests/test_mathtext.py b/lib/matplotlib/tests/test_mathtext.py
@@ -112,6 +112,8 @@
     r'$\left(X\right)_{a}^{b}$',  # github issue 7615
     r'$\dfrac{\$100.00}{y}$',  # github issue #1888
 ]
+svg_only_math_tests = [
+]
 
 digits = "0123456789"
 uppercase = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
@@ -167,8 +169,6 @@
         for set in chars:
             font_tests.append(wrapper % set)
 
-font_tests = list(filter(lambda x: x[1] is not None, enumerate(font_tests)))
-
 
 @pytest.fixture
 def baseline_images(request, fontset, index):
@@ -192,6 +192,32 @@ def test_mathtext_rendering(baseline_images, fontset, index, test):
              horizontalalignment='center', verticalalignment='center')
 
 
+cur_svg_only_math_tests = list(
+    filter(lambda x: x[1] is not None, enumerate(svg_only_math_tests)))
+
+
+@pytest.mark.parametrize('index, test', cur_svg_only_math_tests,
+                         ids=[str(idx) for idx, _ in cur_svg_only_math_tests])
+@pytest.mark.parametrize('fontset', ['all'])
+@pytest.mark.parametrize('baseline_images', ['mathtext1'], indirect=True)
+@image_comparison(
+    baseline_images=None, extensions=['svg'],
+    savefig_kwarg={
+        'metadata': {  # Minimize svg size.
+            'Creator': None, 'Date': None, 'Format': None, 'Type': None}})
+def test_mathtext_rendering_svg_only(baseline_images, fontset, index, test):
+    mpl.rcParams['svg.fonttype'] = 'none'
+    fig = plt.figure(figsize=(5.25, 5.25))
+    fig.patch.set_visible(False)  # Minimize svg size.
+    fontsets = ['cm', 'stix', 'stixsans', 'dejavusans', 'dejavuserif']
+    for i, fontset in enumerate(fontsets):
+        fig.text(0.5, (i + .5) / len(fontsets), test, math_fontfamily=fontset,
+                 horizontalalignment='center', verticalalignment='center')
+
+
+font_tests = list(filter(lambda x: x[1] is not None, enumerate(font_tests)))
+
+
 @pytest.mark.parametrize('index, test', font_tests,
                          ids=[str(index) for index, _ in font_tests])
 @pytest.mark.parametrize('fontset',
