Add typing with `PyAnnotate` to `./tests` (#4950)

* Add mypy_extensions to requirement for ``NoReturn``

Co-authored-by: Pierre Sassoulas <pierre.sassoulas@gmail.com>

1 files changed, 8 insertions, 6 deletions

diff --git a/tests/test_import_graph.py b/tests/test_import_graph.py
index b2061c006..802970718 100644
--- a/tests/test_import_graph.py
+++ b/tests/test_import_graph.py
@@ -21,8 +21,10 @@
 import os
 import shutil
 from os.path import exists
+from typing import Iterator, Union
 
 import pytest
+from _pytest.fixtures import SubRequest
 
 from pylint import testutils
 from pylint.checkers import imports, initialize
@@ -30,7 +32,7 @@ from pylint.lint import PyLinter
 
 
 @pytest.fixture
-def dest(request):
+def dest(request: SubRequest) -> Iterator[Union[Iterator, Iterator[str]]]:
     dest = request.param
     yield dest
     try:
@@ -44,7 +46,7 @@ POSSIBLE_DOT_FILENAMES = ["foo.dot", "foo.gv", "tests/regrtest_data/foo.dot"]
 
 
 @pytest.mark.parametrize("dest", POSSIBLE_DOT_FILENAMES, indirect=True)
-def test_dependencies_graph(dest):
+def test_dependencies_graph(dest: str) -> None:
     """DOC files are correctly generated, and the graphname is the basename"""
     imports._dependencies_graph(dest, {"labas": ["hoho", "yep"], "hoho": ["yep"]})
     with open(dest, encoding="utf-8") as stream:
@@ -70,21 +72,21 @@ URL="." node[shape="box"]
 @pytest.mark.skipif(
     any(shutil.which(x) for x in ("dot", "gv")), reason="dot or gv is installed"
 )
-def test_missing_graphviz(filename):
+def test_missing_graphviz(filename: str) -> None:
     """Raises if graphviz is not installed, and defaults to png if no extension given"""
     with pytest.raises(RuntimeError, match=r"Cannot generate `graph\.png`.*"):
         imports._dependencies_graph(filename, {"a": ["b", "c"], "b": ["c"]})
 
 
 @pytest.fixture
-def linter():
+def linter() -> PyLinter:
     pylinter = PyLinter(reporter=testutils.GenericTestReporter())
     initialize(pylinter)
     return pylinter
 
 
 @pytest.fixture
-def remove_files():
+def remove_files() -> Iterator:
     yield
     for fname in ("import.dot", "ext_import.dot", "int_import.dot"):
         try:
@@ -94,7 +96,7 @@ def remove_files():
 
 
 @pytest.mark.usefixtures("remove_files")
-def test_checker_dep_graphs(linter):
+def test_checker_dep_graphs(linter: PyLinter) -> None:
     linter.global_set_option("persistent", False)
     linter.global_set_option("reports", True)
     linter.global_set_option("enable", "imports")