diff options
author | Daniƫl van Noord <13665637+DanielNoord@users.noreply.github.com> | 2022-09-04 13:04:45 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-09-04 13:04:45 +0200 |
commit | c2989ad5c71b3e1be0f0a7e5297f9b7e47fa2766 (patch) | |
tree | 89f867543fdf13470f0dddfc509686fe137c60a8 /tests/checkers | |
parent | cd7761d4fcdf1d6d3ad19b34a426a7033b41cc1a (diff) | |
download | pylint-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.py | 15 |
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 |