STY: condense code a bit

tacaswell · tacaswell · commit cd4c42b1c295 · 2019-10-03T22:52:53.000-07:00
diff --git a/lib/matplotlib/testing/decorators.py b/lib/matplotlib/testing/decorators.py
@@ -381,6 +381,7 @@ def test_plot(fig_test, fig_ref):
             fig_test.subplots().plot([1, 3, 5])
             fig_ref.subplots().plot([0, 1, 2], [1, 3, 5])
     """
+    POK = inspect.Parameter.POSITIONAL_OR_KEYWORD
 
     def decorator(func):
         import pytest
@@ -410,18 +411,14 @@ def wrapper(*args, ext, **kwargs):
                     for param in sig.parameters.values()
                     if param.name not in {"fig_test", "fig_ref"}
                 ]
-                + [
-                    inspect.Parameter(
-                        "ext", inspect.Parameter.POSITIONAL_OR_KEYWORD
-                    )
-                ]
+                + [inspect.Parameter("ext", POK)]
             )
         )
         wrapper.__signature__ = new_sig
 
         # reach a bit into pytest internals to hoist the marks from
         # our wrapped function
-        new_marks = getattr(func, 'pytestmark', []) + wrapper.pytestmark
+        new_marks = getattr(func, "pytestmark", []) + wrapper.pytestmark
         wrapper.pytestmark = new_marks
 
         return wrapper
