Don't run Docker workflow on push to forks (#2324)

[etc/vim.git] / tests / util.py

diff --git a/tests/util.py b/tests/util.py
index 3670952ba8c61da7fde79be42e81eb4c9a6f9eea..1e86a3f3d7873bff36b3dbcea9253f12af63d6b5 100644 (file)
--- a/tests/util.py
+++ b/tests/util.py
@@ -2,9 +2,12 @@ import os
 import unittest
 from pathlib import Path
 from typing import List, Tuple, Any
 import unittest
 from pathlib import Path
 from typing import List, Tuple, Any

-import black

 from functools import partial
 
 from functools import partial
 

+import black
+from black.output import out, err
+from black.debug import DebugVisitor
+

 THIS_DIR = Path(__file__).parent
 PROJECT_ROOT = THIS_DIR.parent
 EMPTY_LINE = "# EMPTY LINE WITH WHITESPACE" + " (this comment will be removed)"
 THIS_DIR = Path(__file__).parent
 PROJECT_ROOT = THIS_DIR.parent
 EMPTY_LINE = "# EMPTY LINE WITH WHITESPACE" + " (this comment will be removed)"


@@ -26,21 +29,21 @@ class BlackBaseTestCase(unittest.TestCase):
 
     def assertFormatEqual(self, expected: str, actual: str) -> None:
         if actual != expected and not os.environ.get("SKIP_AST_PRINT"):
 
     def assertFormatEqual(self, expected: str, actual: str) -> None:
         if actual != expected and not os.environ.get("SKIP_AST_PRINT"):

-            bdv: black.DebugVisitor[Any]
-            black.out("Expected tree:", fg="green")
+            bdv: DebugVisitor[Any]
+            out("Expected tree:", fg="green")

             try:
                 exp_node = black.lib2to3_parse(expected)
             try:
                 exp_node = black.lib2to3_parse(expected)

-                bdv = black.DebugVisitor()
+                bdv = DebugVisitor()

                 list(bdv.visit(exp_node))
             except Exception as ve:
                 list(bdv.visit(exp_node))
             except Exception as ve:

-                black.err(str(ve))
-            black.out("Actual tree:", fg="red")
+                err(str(ve))
+            out("Actual tree:", fg="red")

             try:
                 exp_node = black.lib2to3_parse(actual)
             try:
                 exp_node = black.lib2to3_parse(actual)

-                bdv = black.DebugVisitor()
+                bdv = DebugVisitor()

                 list(bdv.visit(exp_node))
             except Exception as ve:
                 list(bdv.visit(exp_node))
             except Exception as ve:

-                black.err(str(ve))
+                err(str(ve))

         self.assertMultiLineEqual(expected, actual)
 
 
         self.assertMultiLineEqual(expected, actual)