Parametrize colormap reversing test.

QuLogic · QuLogic · commit d700f039a6f1 · 2017-01-30T00:05:56.000-05:00
This is a bit nicer for debugging.
diff --git a/lib/matplotlib/tests/test_colors.py b/lib/matplotlib/tests/test_colors.py
@@ -608,19 +608,19 @@ def test_pandas_iterable():
     assert_array_equal(cm1.colors, cm2.colors)
 
 
-def test_colormap_reversing():
+@pytest.mark.parametrize('name', cm.cmap_d)
+def test_colormap_reversing(name):
     """Check the generated _lut data of a colormap and corresponding
     reversed colormap if they are almost the same."""
-    for name in cm.cmap_d:
-        with warnings.catch_warnings(record=True) as w:
-            warnings.simplefilter('always')
-            cmap = plt.get_cmap(name)
-        assert len(w) == (1 if name in ('spectral', 'spectral_r') else 0)
-        cmap_r = cmap.reversed()
-        if not cmap_r._isinit:
-            cmap._init()
-            cmap_r._init()
-        assert_array_almost_equal(cmap._lut[:-3], cmap_r._lut[-4::-1])
+    with warnings.catch_warnings(record=True) as w:
+        warnings.simplefilter('always')
+        cmap = plt.get_cmap(name)
+    assert len(w) == (1 if name in ('spectral', 'spectral_r') else 0)
+    cmap_r = cmap.reversed()
+    if not cmap_r._isinit:
+        cmap._init()
+        cmap_r._init()
+    assert_array_almost_equal(cmap._lut[:-3], cmap_r._lut[-4::-1])
 
 
 @cleanup
