diff options
Diffstat (limited to 'tests/message/conftest.py')
-rw-r--r-- | tests/message/conftest.py | 16 |
1 files changed, 9 insertions, 7 deletions
diff --git a/tests/message/conftest.py b/tests/message/conftest.py index 942119076..e30dbb535 100644 --- a/tests/message/conftest.py +++ b/tests/message/conftest.py @@ -3,10 +3,12 @@ # pylint: disable=redefined-outer-name +from typing import Dict, ValuesView + import pytest from pylint.checkers import BaseChecker -from pylint.message import MessageDefinitionStore, MessageIdStore +from pylint.message import MessageDefinition, MessageDefinitionStore, MessageIdStore @pytest.fixture @@ -20,12 +22,12 @@ def symbol(): @pytest.fixture -def empty_store(): +def empty_store() -> MessageDefinitionStore: return MessageDefinitionStore() @pytest.fixture -def store(): +def store() -> MessageDefinitionStore: store_ = MessageDefinitionStore() class Checker(BaseChecker): @@ -50,12 +52,12 @@ def store(): @pytest.fixture -def message_definitions(store): +def message_definitions(store: MessageDefinitionStore) -> ValuesView[MessageDefinition]: return store.messages @pytest.fixture -def msgids(): +def msgids() -> Dict[str, str]: return { "W1234": "warning-symbol", "W1235": "warning-symbol-two", @@ -65,12 +67,12 @@ def msgids(): @pytest.fixture -def empty_msgid_store(): +def empty_msgid_store() -> MessageIdStore: return MessageIdStore() @pytest.fixture -def msgid_store(msgids): +def msgid_store(msgids: Dict[str, str]) -> MessageIdStore: msgid_store = MessageIdStore() for msgid, symbol in msgids.items(): msgid_store.add_msgid_and_symbol(msgid, symbol) |