Parametrize test_mathtext_exceptions.

QuLogic · QuLogic · commit e82db62e169b · 2017-01-22T23:14:52.000-05:00
diff --git a/lib/matplotlib/tests/test_mathtext.py b/lib/matplotlib/tests/test_mathtext.py
@@ -4,8 +4,11 @@
 import six
 
 import io
+import re
 
 import numpy as np
+import pytest
+
 import matplotlib
 from matplotlib.testing.decorators import image_comparison, knownfailureif, cleanup
 import matplotlib.pyplot as plt
@@ -195,8 +198,10 @@ def test_fontinfo():
     table = font.get_sfnt_table("head")
     assert table['version'] == (1, 0)
 
-def test_mathtext_exceptions():
-    errors = [
+
+@pytest.mark.parametrize(
+    'math, msg',
+    [
         (r'$\hspace{}$', r'Expected \hspace{n}'),
         (r'$\hspace{foo}$', r'Expected \hspace{n}'),
         (r'$\frac$', r'Expected \frac{num}{den}'),
@@ -205,28 +210,47 @@ def test_mathtext_exceptions():
         (r'$\stackrel{}{}$', r'Expected \stackrel{num}{den}'),
         (r'$\binom$', r'Expected \binom{num}{den}'),
         (r'$\binom{}{}$', r'Expected \binom{num}{den}'),
-        (r'$\genfrac$', r'Expected \genfrac{ldelim}{rdelim}{rulesize}{style}{num}{den}'),
-        (r'$\genfrac{}{}{}{}{}{}$', r'Expected \genfrac{ldelim}{rdelim}{rulesize}{style}{num}{den}'),
+        (r'$\genfrac$',
+         r'Expected \genfrac{ldelim}{rdelim}{rulesize}{style}{num}{den}'),
+        (r'$\genfrac{}{}{}{}{}{}$',
+         r'Expected \genfrac{ldelim}{rdelim}{rulesize}{style}{num}{den}'),
         (r'$\sqrt$', r'Expected \sqrt{value}'),
         (r'$\sqrt f$', r'Expected \sqrt{value}'),
         (r'$\overline$', r'Expected \overline{value}'),
         (r'$\overline{}$', r'Expected \overline{value}'),
         (r'$\leftF$', r'Expected a delimiter'),
         (r'$\rightF$', r'Unknown symbol: \rightF'),
         (r'$\left(\right$', r'Expected a delimiter'),
-        (r'$\left($', r'Expected "\right"')
-        ]
-
+        (r'$\left($', r'Expected "\right"'),
+    ],
+    ids=[
+        'hspace without value',
+        'hspace with invalid value',
+        'frac without parameters',
+        'frac with empty parameters',
+        'stackrel without parameters',
+        'stackrel with empty parameters',
+        'binom without parameters',
+        'binom with empty parameters',
+        'genfrac without parameters',
+        'genfrac with empty parameters',
+        'sqrt without parameters',
+        'sqrt with invalid value',
+        'overline without parameters',
+        'overline with empty parameter',
+        'left with invalid delimiter',
+        'right with invalid delimiter',
+        'unclosed parentheses with sizing',
+        'unclosed parentheses without sizing',
+    ]
+)
+def test_mathtext_exceptions(math, msg):
     parser = mathtext.MathTextParser('agg')
 
-    for math, msg in errors:
-        try:
-            parser.parse(math)
-        except ValueError as e:
-            exc = str(e).split('\n')
-            assert exc[3].startswith(msg)
-        else:
-            assert False, "Expected '%s', but didn't get it" % msg
+    with pytest.raises(ValueError) as excinfo:
+        parser.parse(math)
+    excinfo.match(re.escape(msg))
+
 
 @cleanup
 def test_single_minus_sign():
