Add support for custom python cell magics (#2744)

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

diff --git a/tests/util.py b/tests/util.py
index 84e98bb0fbde883da8d61545c935c7904641c232..8755111f7c57716c76a8394711e2608cb621490b 100644 (file)
--- a/tests/util.py
+++ b/tests/util.py
@@ -9,7 +9,7 @@ from typing import Any, Iterator, List, Optional, Tuple
 import black
 from black.debug import DebugVisitor
 from black.mode import TargetVersion
 import black
 from black.debug import DebugVisitor
 from black.mode import TargetVersion

-from black.output import err, out
+from black.output import diff, err, out

 
 THIS_DIR = Path(__file__).parent
 DATA_DIR = THIS_DIR / "data"
 
 THIS_DIR = Path(__file__).parent
 DATA_DIR = THIS_DIR / "data"


@@ -47,6 +47,9 @@ def _assert_format_equal(expected: str, actual: str) -> None:
         except Exception as ve:
             err(str(ve))
 
         except Exception as ve:
             err(str(ve))
 

+    if actual != expected:
+        out(diff(expected, actual, "expected", "actual"))
+

     assert actual == expected
 
 
     assert actual == expected