MNT: simplify and correct docs in tk test decorator

tacaswell · tacaswell · commit 3f14f749be90 · 2022-02-02T19:15:34.000-05:00
diff --git a/lib/matplotlib/tests/test_backend_tk.py b/lib/matplotlib/tests/test_backend_tk.py
@@ -20,14 +20,14 @@ def _isolated_tk_test(success_count, func=None):
 
     TkAgg tests seem to have interactions between tests, so isolate each test
     in a subprocess. See GH#18261
-
-    The decorated function must be fully self-contained, and thus perform
-    all the imports it needs.
     """
 
     if func is None:
         return functools.partial(_isolated_tk_test, success_count)
 
+    if "MPL_TEST_ESCAPE_HATCH" in os.environ:
+        return func
+
     @functools.wraps(func)
     def test_func():
         if (sys.platform == "linux" and
@@ -53,10 +53,8 @@ def test_func():
             assert not [line for line in proc.stderr.splitlines()
                         if "OpenGL" not in line]
             assert proc.stdout.count("success") == success_count
-    if "MPL_TEST_ESCAPE_HATCH" in os.environ:
-        return func
-    else:
-        return test_func
+
+    return test_func
 
 
 @_isolated_tk_test(success_count=6)  # len(bad_boxes)
