Also run mypy on test_black.py

This commit is contained in:
Łukasz Langa 2018-03-15 23:55:22 -07:00
parent fb84d8b2ee
commit 9fea478022
2 changed files with 6 additions and 5 deletions

View File

@ -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

View File

@ -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 @@ def test_report(self) -> None:
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 @@ def err(msg: str, **kwargs):
)
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")