diff options
Diffstat (limited to 'tests/extensions/test_while_used.py')
-rw-r--r-- | tests/extensions/test_while_used.py | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/tests/extensions/test_while_used.py b/tests/extensions/test_while_used.py new file mode 100644 index 000000000..3bf081a29 --- /dev/null +++ b/tests/extensions/test_while_used.py @@ -0,0 +1,25 @@ +"""Tests for the pylint checker in :mod:`pylint.extensions.while +""" + +import astroid + +from pylint.extensions.while_used import WhileChecker +from pylint.testutils import CheckerTestCase, Message + + +class TestWhileUsed(CheckerTestCase): + + CHECKER_CLASS = WhileChecker + + def test_while_used(self) -> None: + node = astroid.extract_node( + """ + def f(): + i = 0 + while i < 10: + i += 1 + """ + ).body[1] + + with self.assertAddsMessages(Message("while-used", node=node)): + self.checker.visit_while(node) |