Add trove classifier for Python 3.7 support (#486)

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

diff --git a/tests/test_black.py b/tests/test_black.py
index 421e8712da130d9f592180c7afcd4b2a6ebab833..85ab2b6ba3e2375c09295eb3d65d9a0fe29fffac 100644 (file)
--- a/tests/test_black.py
+++ b/tests/test_black.py
@@ -99,8 +99,8 @@ class BlackRunner(CliRunner):
                 sys.stderr = TextIOWrapper(self.stderrbuf, encoding=self.charset)
                 yield output
             finally:
                 sys.stderr = TextIOWrapper(self.stderrbuf, encoding=self.charset)
                 yield output
             finally:

-                self.stdout_bytes = sys.stdout.buffer.getvalue()
-                self.stderr_bytes = sys.stderr.buffer.getvalue()
+                self.stdout_bytes = sys.stdout.buffer.getvalue()  # type: ignore
+                self.stderr_bytes = sys.stderr.buffer.getvalue()  # type: ignore

                 sys.stderr = hold_stderr
 
 
                 sys.stderr = hold_stderr
 
 


@@ -1226,6 +1226,7 @@ class BlackTestCase(unittest.TestCase):
             result = runner.invoke(
                 black.main, ["-", "--fast"], input=BytesIO(contents.encode("utf8"))
             )
             result = runner.invoke(
                 black.main, ["-", "--fast"], input=BytesIO(contents.encode("utf8"))
             )

+            self.assertEqual(result.exit_code, 0)

             output = runner.stdout_bytes
             self.assertIn(nl.encode("utf8"), output)
             if nl == "\n":
             output = runner.stdout_bytes
             self.assertIn(nl.encode("utf8"), output)
             if nl == "\n":