diff options
Diffstat (limited to 'tests/checkers/unittest_format.py')
| -rw-r--r-- | tests/checkers/unittest_format.py | 78 |
1 files changed, 59 insertions, 19 deletions
diff --git a/tests/checkers/unittest_format.py b/tests/checkers/unittest_format.py index 4023207bb..93beda9f4 100644 --- a/tests/checkers/unittest_format.py +++ b/tests/checkers/unittest_format.py @@ -37,7 +37,7 @@ import astroid from pylint import lint, reporters from pylint.checkers.format import FormatChecker -from pylint.testutils import CheckerTestCase, Message, _tokenize_str +from pylint.testutils import CheckerTestCase, TestMessage, _tokenize_str class TestMultiStatementLine(CheckerTestCase): @@ -50,7 +50,9 @@ class TestMultiStatementLine(CheckerTestCase): """ ) self.checker.config.single_line_if_stmt = False - with self.assertAddsMessages(Message("multiple-statements", node=stmt.body[0])): + with self.assertAddsMessages( + TestMessage("multiple-statements", node=stmt.body[0]) + ): self.visitFirst(stmt) self.checker.config.single_line_if_stmt = True with self.assertNoMessages(): @@ -62,7 +64,9 @@ class TestMultiStatementLine(CheckerTestCase): pass """ ) - with self.assertAddsMessages(Message("multiple-statements", node=stmt.body[0])): + with self.assertAddsMessages( + TestMessage("multiple-statements", node=stmt.body[0]) + ): self.visitFirst(stmt) def testSingleLineClassStmts(self) -> None: @@ -72,7 +76,9 @@ class TestMultiStatementLine(CheckerTestCase): """ ) self.checker.config.single_line_class_stmt = False - with self.assertAddsMessages(Message("multiple-statements", node=stmt.body[0])): + with self.assertAddsMessages( + TestMessage("multiple-statements", node=stmt.body[0]) + ): self.visitFirst(stmt) self.checker.config.single_line_class_stmt = True with self.assertNoMessages(): @@ -84,7 +90,9 @@ class TestMultiStatementLine(CheckerTestCase): """ ) self.checker.config.single_line_class_stmt = False - with self.assertAddsMessages(Message("multiple-statements", node=stmt.body[0])): + with self.assertAddsMessages( + TestMessage("multiple-statements", node=stmt.body[0]) + ): self.visitFirst(stmt) self.checker.config.single_line_class_stmt = True with self.assertNoMessages(): @@ -96,10 +104,14 @@ class TestMultiStatementLine(CheckerTestCase): """ ) self.checker.config.single_line_class_stmt = False - with self.assertAddsMessages(Message("multiple-statements", node=stmt.body[0])): + with self.assertAddsMessages( + TestMessage("multiple-statements", node=stmt.body[0]) + ): self.visitFirst(stmt) self.checker.config.single_line_class_stmt = True - with self.assertAddsMessages(Message("multiple-statements", node=stmt.body[0])): + with self.assertAddsMessages( + TestMessage("multiple-statements", node=stmt.body[0]) + ): self.visitFirst(stmt) def testTryExceptFinallyNoMultipleStatement(self) -> None: @@ -133,7 +145,9 @@ class TestMultiStatementLine(CheckerTestCase): def concat2(arg1: str) -> str: ... """ stmt = astroid.extract_node(code) - with self.assertAddsMessages(Message("multiple-statements", node=stmt.body[0])): + with self.assertAddsMessages( + TestMessage("multiple-statements", node=stmt.body[0]) + ): self.visitFirst(stmt) @@ -163,20 +177,40 @@ class TestSuperfluousParentheses(CheckerTestCase): def testCheckKeywordParensHandlesUnnecessaryParens(self) -> None: cases = [ - (Message("superfluous-parens", line=1, args="if"), "if (foo):", 0), - (Message("superfluous-parens", line=1, args="if"), "if ((foo, bar)):", 0), - (Message("superfluous-parens", line=1, args="if"), "if (foo(bar)):", 0), - (Message("superfluous-parens", line=1, args="not"), "not (foo)", 0), - (Message("superfluous-parens", line=1, args="not"), "if not (foo):", 1), - (Message("superfluous-parens", line=1, args="if"), "if (not (foo)):", 0), - (Message("superfluous-parens", line=1, args="not"), "if (not (foo)):", 2), + (TestMessage("superfluous-parens", line=1, args="if"), "if (foo):", 0), ( - Message("superfluous-parens", line=1, args="for"), + TestMessage("superfluous-parens", line=1, args="if"), + "if ((foo, bar)):", + 0, + ), + ( + TestMessage("superfluous-parens", line=1, args="if"), + "if (foo(bar)):", + 0, + ), + (TestMessage("superfluous-parens", line=1, args="not"), "not (foo)", 0), + ( + TestMessage("superfluous-parens", line=1, args="not"), + "if not (foo):", + 1, + ), + ( + TestMessage("superfluous-parens", line=1, args="if"), + "if (not (foo)):", + 0, + ), + ( + TestMessage("superfluous-parens", line=1, args="not"), + "if (not (foo)):", + 2, + ), + ( + TestMessage("superfluous-parens", line=1, args="for"), "for (x) in (1, 2, 3):", 0, ), ( - Message("superfluous-parens", line=1, args="if"), + TestMessage("superfluous-parens", line=1, args="if"), "if (1) in (1, 2, 3):", 0, ), @@ -199,8 +233,14 @@ class TestSuperfluousParentheses(CheckerTestCase): def testPositiveSuperfluousParensWalrusOperatorIf(self) -> None: """Test positive superfluous parens cases with the walrus operator""" cases = [ - (Message("superfluous-parens", line=1, args="if"), "if ((x := y)):\n"), - (Message("superfluous-parens", line=1, args="not"), "if not ((x := y)):\n"), + ( + TestMessage("superfluous-parens", line=1, args="if"), + "if ((x := y)):\n", + ), + ( + TestMessage("superfluous-parens", line=1, args="not"), + "if not ((x := y)):\n", + ), ] for msg, code in cases: with self.assertAddsMessages(msg): |