aboutsummaryrefslogtreecommitdiff
path: root/tests/extensions/test_while_used.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/extensions/test_while_used.py')
-rw-r--r--tests/extensions/test_while_used.py25
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)