MNT: make the tests easier to read

tacaswell · tacaswell · commit 791c99a82bf4 · 2019-09-01T15:11:13.000-04:00
diff --git a/lib/matplotlib/tests/test_cm.py b/lib/matplotlib/tests/test_cm.py
@@ -8,18 +8,19 @@ def test_double_register_cmap():
 
 
 def test_register_cmap():
-    mcm.register_cmap('viridis'[::-1], mcm.get_cmap('viridis'))
-    assert mcm.get_cmap('viridis'[::-1]) is mcm.get_cmap('viridis')
+    target = 'viridis'[::-1]
+    mcm.register_cmap(target, mcm.get_cmap('viridis'))
+    assert mcm.get_cmap(target) is mcm.get_cmap('viridis')
 
     with pytest.warns(UserWarning):
-        mcm.register_cmap('viridis'[::-1], mcm.get_cmap('viridis'))
+        mcm.register_cmap(target, mcm.get_cmap('viridis'))
 
 
-    mcm.unregister_cmap('viridis'[::-1])
+    mcm.unregister_cmap(target)
     with pytest.raises(ValueError):
-        mcm.get_cmap('viridis'[::-1])
+        mcm.get_cmap(target)
     # test that second time is error free
-    mcm.unregister_cmap('viridis'[::-1])
+    mcm.unregister_cmap(target)
 
 
 def test_unregister_cmap():
