diff options
Diffstat (limited to 'tests/checkers/unittest_imports.py')
-rw-r--r-- | tests/checkers/unittest_imports.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/tests/checkers/unittest_imports.py b/tests/checkers/unittest_imports.py index 6bc3ae2c1..0f1242749 100644 --- a/tests/checkers/unittest_imports.py +++ b/tests/checkers/unittest_imports.py @@ -25,7 +25,7 @@ import astroid from pylint.checkers import imports from pylint.interfaces import UNDEFINED -from pylint.testutils import CheckerTestCase, Message, set_config +from pylint.testutils import CheckerTestCase, TestMessage, set_config REGR_DATA = os.path.join(os.path.dirname(__file__), "..", "regrtest_data", "") @@ -54,7 +54,7 @@ class TestImportsChecker(CheckerTestCase): ).body[0] with self.assertAddsMessages( - Message("import-outside-toplevel", node=node, args="pylint") + TestMessage("import-outside-toplevel", node=node, args="pylint") ): self.checker.visit_import(node) @@ -110,7 +110,7 @@ class TestImportsChecker(CheckerTestCase): import foo, bar """ ) - msg = Message("multiple-imports", node=node, args="foo, bar") + msg = TestMessage("multiple-imports", node=node, args="foo, bar") with self.assertAddsMessages(msg): self.checker.visit_import(node) @@ -128,7 +128,7 @@ class TestImportsChecker(CheckerTestCase): Test that duplicate imports on single line raise 'reimported'. """ node = astroid.extract_node("from time import sleep, sleep, time") - msg = Message(msg_id="reimported", node=node, args=("sleep", 1)) + msg = TestMessage(msg_id="reimported", node=node, args=("sleep", 1)) with self.assertAddsMessages(msg): self.checker.visit_importfrom(node) @@ -136,7 +136,7 @@ class TestImportsChecker(CheckerTestCase): module = astroid.MANAGER.ast_from_module_name("beyond_top", REGR_DATA) import_from = module.body[0] - msg = Message(msg_id="relative-beyond-top-level", node=import_from) + msg = TestMessage(msg_id="relative-beyond-top-level", node=import_from) with self.assertAddsMessages(msg): self.checker.visit_importfrom(import_from) with self.assertNoMessages(): @@ -155,7 +155,7 @@ class TestImportsChecker(CheckerTestCase): module = astroid.MANAGER.ast_from_module_name("wildcard", REGR_DATA) import_from = module.body[0] - msg = Message( + msg = TestMessage( msg_id="wildcard-import", node=import_from, args="empty", |