mirror of
https://github.com/PyCQA/flake8.git
synced 2026-04-04 20:26:53 +00:00
28 lines
666 B
Python
28 lines
666 B
Python
|
|
import textwrap
|
|
import _ast
|
|
|
|
from twisted.trial import unittest
|
|
|
|
from pyflakes import checker
|
|
|
|
|
|
class Test(unittest.TestCase):
|
|
|
|
def flakes(self, input, *expectedOutputs, **kw):
|
|
ast = compile(textwrap.dedent(input), "<test>", "exec",
|
|
_ast.PyCF_ONLY_AST)
|
|
w = checker.Checker(ast, **kw)
|
|
outputs = [type(o) for o in w.messages]
|
|
expectedOutputs = list(expectedOutputs)
|
|
outputs.sort()
|
|
expectedOutputs.sort()
|
|
self.assert_(outputs == expectedOutputs, '''\
|
|
for input:
|
|
%s
|
|
expected outputs:
|
|
%s
|
|
but got:
|
|
%s''' % (input, repr(expectedOutputs), '\n'.join(map(str, w.messages))))
|
|
return w
|
|
|