summaryrefslogtreecommitdiff
path: root/tests/pyreverse/test_inspector.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/pyreverse/test_inspector.py')
-rw-r--r--tests/pyreverse/test_inspector.py20
1 files changed, 11 insertions, 9 deletions
diff --git a/tests/pyreverse/test_inspector.py b/tests/pyreverse/test_inspector.py
index 0c8873f97..014838564 100644
--- a/tests/pyreverse/test_inspector.py
+++ b/tests/pyreverse/test_inspector.py
@@ -17,23 +17,25 @@
# pylint: disable=redefined-outer-name
import os
+from typing import Callable
import astroid
import pytest
from astroid import nodes
from pylint.pyreverse import inspector
+from pylint.pyreverse.inspector import Project
@pytest.fixture
-def project(get_project):
+def project(get_project: Callable) -> Project:
project = get_project("data", "data")
linker = inspector.Linker(project)
linker.visit(project)
return project
-def test_class_implements(project):
+def test_class_implements(project: Project) -> None:
klass = project.get_module("data.clientmodule_test")["Ancestor"]
assert hasattr(klass, "implements")
assert len(klass.implements) == 1
@@ -41,13 +43,13 @@ def test_class_implements(project):
assert klass.implements[0].name == "Interface"
-def test_class_implements_specialization(project):
+def test_class_implements_specialization(project: Project) -> None:
klass = project.get_module("data.clientmodule_test")["Specialization"]
assert hasattr(klass, "implements")
assert len(klass.implements) == 0
-def test_locals_assignment_resolution(project):
+def test_locals_assignment_resolution(project: Project) -> None:
klass = project.get_module("data.clientmodule_test")["Specialization"]
assert hasattr(klass, "locals_type")
type_dict = klass.locals_type
@@ -60,7 +62,7 @@ def test_locals_assignment_resolution(project):
assert type_dict["top"][0].value == "class"
-def test_instance_attrs_resolution(project):
+def test_instance_attrs_resolution(project: Project) -> None:
klass = project.get_module("data.clientmodule_test")["Specialization"]
assert hasattr(klass, "instance_attrs_type")
type_dict = klass.instance_attrs_type
@@ -74,7 +76,7 @@ def test_instance_attrs_resolution(project):
assert type_dict["_id"][0] is astroid.Uninferable
-def test_concat_interfaces():
+def test_concat_interfaces() -> None:
cls = astroid.extract_node(
'''
class IMachin: pass
@@ -96,7 +98,7 @@ def test_concat_interfaces():
assert [i.name for i in interfaces] == ["IMachin"]
-def test_interfaces():
+def test_interfaces() -> None:
module = astroid.parse(
"""
class Interface(object): pass
@@ -122,12 +124,12 @@ def test_interfaces():
assert [i.name for i in inspector.interfaces(klass)] == interfaces
-def test_from_directory(project):
+def test_from_directory(project: Project) -> None:
expected = os.path.join("tests", "data", "__init__.py")
assert project.name == "data"
assert project.path.endswith(expected)
-def test_project_node(project):
+def test_project_node(project: Project) -> None:
expected = ["data", "data.clientmodule_test", "data.suppliermodule_test"]
assert sorted(project.keys()) == expected