Fix additional mypy errors and expand CI checks to more files (#2541)

* Fix additional mypy errors and expand CI checks to more files

* Use an assertion instead of a type ignore

* Move an assert outside of a type-checking block

Author: correctmost

astroid/_ast.py

@@ -32,7 +32,7 @@ def parse(
 
 def parse_function_type_comment(type_comment: str) -> FunctionType | None:
     """Given a correct type comment, obtain a FunctionType object."""
-    func_type = ast.parse(type_comment, "<type_comment>", "func_type")  # type: ignore[attr-defined]
+    func_type = ast.parse(type_comment, "<type_comment>", "func_type")
     return FunctionType(argtypes=func_type.argtypes, returns=func_type.returns)
 
 

astroid/brain/brain_argparse.py

@@ -4,9 +4,10 @@
 
 from __future__ import annotations
 
-from astroid import arguments, inference_tip, nodes
+from astroid import arguments, nodes
 from astroid.context import InferenceContext
 from astroid.exceptions import UseInferenceDefault
+from astroid.inference_tip import inference_tip
 from astroid.manager import AstroidManager
 
 

astroid/brain/brain_boto3.py

@@ -4,14 +4,14 @@
 
 """Astroid hooks for understanding ``boto3.ServiceRequest()``."""
 
-from astroid import extract_node
+from astroid.builder import extract_node
 from astroid.manager import AstroidManager
 from astroid.nodes.scoped_nodes import ClassDef
 
 BOTO_SERVICE_FACTORY_QUALIFIED_NAME = "boto3.resources.base.ServiceResource"
 
 
-def service_request_transform(node):
+def service_request_transform(node: ClassDef) -> ClassDef:
     """Transform ServiceResource to look like dynamic classes."""
     code = """
     def __getattr__(self, attr):
@@ -22,7 +22,7 @@ def __getattr__(self, attr):
     return node
 
 
-def _looks_like_boto3_service_request(node) -> bool:
+def _looks_like_boto3_service_request(node: ClassDef) -> bool:
     return node.qname() == BOTO_SERVICE_FACTORY_QUALIFIED_NAME
 
 

astroid/brain/brain_builtin_inference.py

@@ -11,7 +11,7 @@
 from functools import partial
 from typing import TYPE_CHECKING, Any, NoReturn, Union, cast
 
-from astroid import arguments, helpers, inference_tip, nodes, objects, util
+from astroid import arguments, helpers, nodes, objects, util
 from astroid.builder import AstroidBuilder
 from astroid.context import InferenceContext
 from astroid.exceptions import (
@@ -21,6 +21,7 @@
     MroError,
     UseInferenceDefault,
 )
+from astroid.inference_tip import inference_tip
 from astroid.manager import AstroidManager
 from astroid.nodes import scoped_nodes
 from astroid.typing import (
@@ -844,7 +845,7 @@ def _class_or_tuple_to_container(
     return class_container
 
 
-def infer_len(node, context: InferenceContext | None = None):
+def infer_len(node, context: InferenceContext | None = None) -> nodes.Const:
     """Infer length calls.
 
     :param nodes.Call node: len call to infer
@@ -867,7 +868,7 @@ def infer_len(node, context: InferenceContext | None = None):
         raise UseInferenceDefault(str(exc)) from exc
 
 
-def infer_str(node, context: InferenceContext | None = None):
+def infer_str(node, context: InferenceContext | None = None) -> nodes.Const:
     """Infer str() calls.
 
     :param nodes.Call node: str() call to infer
@@ -926,7 +927,7 @@ def infer_dict_fromkeys(node, context: InferenceContext | None = None):
         will be inferred instead.
     """
 
-    def _build_dict_with_elements(elements):
+    def _build_dict_with_elements(elements: list) -> nodes.Dict:
         new_node = nodes.Dict(
             col_offset=node.col_offset,
             lineno=node.lineno,

astroid/brain/brain_crypt.py

@@ -2,12 +2,13 @@
 # For details: https://github.com/pylint-dev/astroid/blob/main/LICENSE
 # Copyright (c) https://github.com/pylint-dev/astroid/blob/main/CONTRIBUTORS.txt
 
+from astroid import nodes
 from astroid.brain.helpers import register_module_extender
 from astroid.builder import parse
 from astroid.manager import AstroidManager
 
 
-def _re_transform():
+def _re_transform() -> nodes.Module:
     return parse(
         """
     from collections import namedtuple

astroid/brain/brain_ctypes.py

@@ -12,12 +12,13 @@
 """
 import sys
 
+from astroid import nodes
 from astroid.brain.helpers import register_module_extender
 from astroid.builder import parse
 from astroid.manager import AstroidManager
 
 
-def enrich_ctypes_redefined_types():
+def enrich_ctypes_redefined_types() -> nodes.Module:
     """
     For each ctypes redefined types, overload 'value' and '_type_' members
     definition.

astroid/brain/brain_curses.py

@@ -2,12 +2,13 @@
 # For details: https://github.com/pylint-dev/astroid/blob/main/LICENSE
 # Copyright (c) https://github.com/pylint-dev/astroid/blob/main/CONTRIBUTORS.txt
 
+from astroid import nodes
 from astroid.brain.helpers import register_module_extender
 from astroid.builder import parse
 from astroid.manager import AstroidManager
 
 
-def _curses_transform():
+def _curses_transform() -> nodes.Module:
     return parse(
         """
     A_ALTCHARSET = 1

astroid/brain/brain_datetime.py

@@ -2,13 +2,14 @@
 # For details: https://github.com/pylint-dev/astroid/blob/main/LICENSE
 # Copyright (c) https://github.com/pylint-dev/astroid/blob/main/CONTRIBUTORS.txt
 
+from astroid import nodes
 from astroid.brain.helpers import register_module_extender
 from astroid.builder import AstroidBuilder
 from astroid.const import PY312_PLUS
 from astroid.manager import AstroidManager
 
 
-def datetime_transform():
+def datetime_transform() -> nodes.Module:
     """The datetime module was C-accelerated in Python 3.12, so use the
     Python source."""
     return AstroidBuilder(AstroidManager()).string_build("from _pydatetime import *")

astroid/brain/brain_dateutil.py

@@ -6,12 +6,13 @@
 
 import textwrap
 
+from astroid import nodes
 from astroid.brain.helpers import register_module_extender
 from astroid.builder import AstroidBuilder
 from astroid.manager import AstroidManager
 
 
-def dateutil_transform():
+def dateutil_transform() -> nodes.Module:
     return AstroidBuilder(AstroidManager()).string_build(
         textwrap.dedent(
             """

astroid/brain/brain_functools.py

@@ -10,7 +10,8 @@
 from functools import partial
 from itertools import chain
 
-from astroid import BoundMethod, arguments, extract_node, nodes, objects
+from astroid import BoundMethod, arguments, nodes, objects
+from astroid.builder import extract_node
 from astroid.context import InferenceContext
 from astroid.exceptions import InferenceError, UseInferenceDefault
 from astroid.inference_tip import inference_tip