From: Łukasz Langa <lukasz@langa.pl>
Date: Fri, 16 Mar 2018 06:55:22 +0000 (-0700)
Subject: Also run mypy on test_black.py
X-Git-Url: https://git.madduck.net/etc/vim.git/commitdiff_plain/9fea4780228dc8e9ceb53ddb18fe1a87f6e05546?pf=etc

Also run mypy on test_black.py
---

diff --git a/.travis.yml b/.travis.yml
index 5870f83..804b39d 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -6,7 +6,7 @@ before_script:
 - pip install -e .
 script:
 - python setup.py test
-- if [[ $TRAVIS_PYTHON_VERSION == '3.6' ]]; then mypy black.py; fi
+- if [[ $TRAVIS_PYTHON_VERSION == '3.6' ]]; then mypy black.py tests/test_black.py; fi
 notifications:
   on_success: change
   on_failure: always
diff --git a/tests/test_black.py b/tests/test_black.py
index 1dda5fc..c11fca0 100644
--- a/tests/test_black.py
+++ b/tests/test_black.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python3
 from functools import partial
 from pathlib import Path
-from typing import List, Tuple
+from typing import Any, List, Tuple
 import unittest
 from unittest.mock import patch
 
@@ -46,6 +46,7 @@ class BlackTestCase(unittest.TestCase):
 
     def assertFormatEqual(self, expected: str, actual: str) -> None:
         if actual != expected:
+            bdv: black.DebugVisitor[Any]
             black.out('Expected tree:', fg='green')
             try:
                 exp_node = black.lib2to3_parse(expected)
@@ -161,10 +162,10 @@ class BlackTestCase(unittest.TestCase):
         out_lines = []
         err_lines = []
 
-        def out(msg: str, **kwargs):
+        def out(msg: str, **kwargs: Any) -> None:
             out_lines.append(msg)
 
-        def err(msg: str, **kwargs):
+        def err(msg: str, **kwargs: Any) -> None:
             err_lines.append(msg)
 
         with patch("black.out", out), patch("black.err", err):
@@ -223,7 +224,7 @@ class BlackTestCase(unittest.TestCase):
             )
             self.assertEqual(report.return_code, 123)
 
-    def test_is_python36(self):
+    def test_is_python36(self) -> None:
         node = black.lib2to3_parse("def f(*, arg): ...\n")
         self.assertFalse(black.is_python36(node))
         node = black.lib2to3_parse("def f(*, arg,): ...\n")