From 9fea4780228dc8e9ceb53ddb18fe1a87f6e05546 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=C5=81ukasz=20Langa?= Date: Thu, 15 Mar 2018 23:55:22 -0700 Subject: [PATCH] Also run mypy on test_black.py --- .travis.yml | 2 +- tests/test_black.py | 9 +++++---- 2 files changed, 6 insertions(+), 5 deletions(-) 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") -- 2.39.5