]>
Commit | Line | Data |
---|---|---|
1 | import ast | |
2 | import textwrap | |
3 | import unittest | |
4 | ||
5 | from pyflakes import checker | |
6 | ||
7 | __all__ = ['TestCase', 'skip', 'skipIf'] | |
8 | ||
9 | skip = unittest.skip | |
10 | skipIf = unittest.skipIf | |
11 | ||
12 | ||
13 | class TestCase(unittest.TestCase): | |
14 | ||
15 | withDoctest = False | |
16 | ||
17 | def flakes(self, input, *expectedOutputs, **kw): | |
18 | tree = ast.parse(textwrap.dedent(input)) | |
19 | if kw.get('is_segment'): | |
20 | tree = tree.body[0] | |
21 | kw.pop('is_segment') | |
22 | w = checker.Checker(tree, withDoctest=self.withDoctest, **kw) | |
23 | outputs = [type(o) for o in w.messages] | |
24 | expectedOutputs = list(expectedOutputs) | |
25 | outputs.sort(key=lambda t: t.__name__) | |
26 | expectedOutputs.sort(key=lambda t: t.__name__) | |
27 | self.assertEqual(outputs, expectedOutputs, '''\ | |
28 | for input: | |
29 | {} | |
30 | expected outputs: | |
31 | {!r} | |
32 | but got: | |
33 | {}'''.format(input, expectedOutputs, '\n'.join([str(o) for o in w.messages]))) | |
34 | return w |