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 . - pip install -e .
script: script:
- python setup.py test - 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: notifications:
on_success: change on_success: change
on_failure: always on_failure: always

View File

@ -1,7 +1,7 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
from functools import partial from functools import partial
from pathlib import Path from pathlib import Path
from typing import List, Tuple from typing import Any, List, Tuple
import unittest import unittest
from unittest.mock import patch from unittest.mock import patch
@ -46,6 +46,7 @@ class BlackTestCase(unittest.TestCase):
def assertFormatEqual(self, expected: str, actual: str) -> None: def assertFormatEqual(self, expected: str, actual: str) -> None:
if actual != expected: if actual != expected:
bdv: black.DebugVisitor[Any]
black.out('Expected tree:', fg='green') black.out('Expected tree:', fg='green')
try: try:
exp_node = black.lib2to3_parse(expected) exp_node = black.lib2to3_parse(expected)
@ -161,10 +162,10 @@ def test_report(self) -> None:
out_lines = [] out_lines = []
err_lines = [] err_lines = []
def out(msg: str, **kwargs): def out(msg: str, **kwargs: Any) -> None:
out_lines.append(msg) out_lines.append(msg)
def err(msg: str, **kwargs): def err(msg: str, **kwargs: Any) -> None:
err_lines.append(msg) err_lines.append(msg)
with patch("black.out", out), patch("black.err", err): with patch("black.out", out), patch("black.err", err):
@ -223,7 +224,7 @@ def err(msg: str, **kwargs):
) )
self.assertEqual(report.return_code, 123) 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") node = black.lib2to3_parse("def f(*, arg): ...\n")
self.assertFalse(black.is_python36(node)) self.assertFalse(black.is_python36(node))
node = black.lib2to3_parse("def f(*, arg,): ...\n") node = black.lib2to3_parse("def f(*, arg,): ...\n")