diff options
Diffstat (limited to 'pylint/test/unittest_checker_base.py')
| -rw-r--r-- | pylint/test/unittest_checker_base.py | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/pylint/test/unittest_checker_base.py b/pylint/test/unittest_checker_base.py index fb595e3..b58d8d9 100644 --- a/pylint/test/unittest_checker_base.py +++ b/pylint/test/unittest_checker_base.py @@ -398,6 +398,21 @@ class IterableTest(CheckerTestCase): with self.assertNoMessages(): self.checker.visit_call(node) + def test_non_iterable_in_listcomp(self): + node = test_utils.extract_node(""" + [x for x in 123] + """) + message = Message('not-an-iterable', node=node, args='123') + with self.assertAddsMessages(message): + self.checker.visit_listcomp(node) + # TODO: more tests + + def test_non_iterable_in_generator(self): + node = test_utils.extract_node("__(x for x in 123)") + message = Message('not-an-iterable', node=node, args='123') + with self.assertAddsMessages(message): + self.walk(node.root()) + # TODO: more tests if __name__ == '__main__': |