X-Git-Url: https://git.madduck.net/etc/vim.git/blobdiff_plain/1d45f6e6a11675ee1ee5a3b0c3664cd7feec532b..f7fd36b2289d18ac3cfedc8e160cc965e7cb2209:/tests/test_black.py diff --git a/tests/test_black.py b/tests/test_black.py index d2359cb..a4d2382 100644 --- a/tests/test_black.py +++ b/tests/test_black.py @@ -150,7 +150,7 @@ class BlackTestCase(unittest.TestCase): tmp_file = Path(black.dump_to_file(source)) try: self.assertTrue(ff(tmp_file, write_back=black.WriteBack.YES)) - with open(tmp_file) as f: + with open(tmp_file, encoding="utf8") as f: actual = f.read() finally: os.unlink(tmp_file) @@ -174,7 +174,14 @@ class BlackTestCase(unittest.TestCase): sys.stdout = hold_stdout os.unlink(tmp_file) actual = actual.rstrip() + "\n" # the diff output has a trailing space - self.assertEqual(expected, actual) + if expected != actual: + dump = black.dump_to_file(actual) + msg = ( + f"Expected diff isn't equal to the actual. If you made changes " + f"to expression.py and this is an anticipated difference, " + f"overwrite tests/expression.diff with {dump}" + ) + self.assertEqual(expected, actual, msg) @patch("black.dump_to_file", dump_to_stderr) def test_fstring(self) -> None: