summaryrefslogtreecommitdiff
path: root/tests/checkers
diff options
context:
space:
mode:
authorDaniƫl van Noord <13665637+DanielNoord@users.noreply.github.com>2022-09-04 13:04:45 +0200
committerGitHub <noreply@github.com>2022-09-04 13:04:45 +0200
commitc2989ad5c71b3e1be0f0a7e5297f9b7e47fa2766 (patch)
tree89f867543fdf13470f0dddfc509686fe137c60a8 /tests/checkers
parentcd7761d4fcdf1d6d3ad19b34a426a7033b41cc1a (diff)
downloadpylint-git-c2989ad5c71b3e1be0f0a7e5297f9b7e47fa2766.tar.gz
Complete typing of all generic types (#7406)
And update ``mypy`` configuration
Diffstat (limited to 'tests/checkers')
-rw-r--r--tests/checkers/unittest_stdlib.py15
1 files changed, 9 insertions, 6 deletions
diff --git a/tests/checkers/unittest_stdlib.py b/tests/checkers/unittest_stdlib.py
index 2f47a4075..66747deb9 100644
--- a/tests/checkers/unittest_stdlib.py
+++ b/tests/checkers/unittest_stdlib.py
@@ -6,23 +6,26 @@ from __future__ import annotations
import contextlib
from collections.abc import Callable, Iterator
-from typing import Any
+from typing import Any, Type
import astroid
from astroid import nodes
+from astroid.context import InferenceContext
from astroid.manager import AstroidManager
from pylint.checkers import stdlib
from pylint.testutils import CheckerTestCase
+_NodeNGT = Type[nodes.NodeNG]
+
@contextlib.contextmanager
def _add_transform(
manager: AstroidManager,
- node: type,
- transform: Callable,
+ node: _NodeNGT,
+ transform: Callable[[_NodeNGT], _NodeNGT],
predicate: Any | None = None,
-) -> Iterator:
+) -> Iterator[None]:
manager.register_transform(node, transform, predicate)
try:
yield
@@ -43,8 +46,8 @@ class TestStdlibChecker(CheckerTestCase):
def infer_func(
inner_node: nodes.Name,
- context: Any | None = None, # pylint: disable=unused-argument
- ) -> Iterator[Iterator | Iterator[nodes.AssignAttr]]:
+ context: InferenceContext | None = None, # pylint: disable=unused-argument
+ ) -> Iterator[nodes.AssignAttr]:
new_node = nodes.AssignAttr(attrname="alpha", parent=inner_node)
yield new_node