[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

mypy/installtypes.py

@@ -21,7 +21,8 @@
 def normalize_distribution_name(name: str) -> str:
     return _DIST_NORMALIZE_RE.sub("-", name).lower()
 
-DIST_TO_MODULE_NAME: dict[str, str] = { #FIX
+
+DIST_TO_MODULE_NAME: dict[str, str] = {  # FIX
     "python-dateutil": "dateutil",
     "pyyaml": "yaml",
     "python-xlib": "Xlib",
@@ -76,7 +77,7 @@ def resolve_stub_packages_from_lock(locked: Mapping[str, str | None]) -> list[st
     #         dist_name,
     #         dist_name.replace("-", "_"),
     #     }
-    
+
     #     for module_name in candidates:
     #         stub = stub_distribution_name(module_name)
     #         if stub:
@@ -89,10 +90,7 @@ def resolve_stub_packages_from_lock(locked: Mapping[str, str | None]) -> list[st
         if dist_name.startswith("types-"):
             continue
 
-        candidates = {
-            dist_name,
-            dist_name.replace("-", "_"),
-        }
+        candidates = {dist_name, dist_name.replace("-", "_")}
 
         mapped_module = DIST_TO_MODULE_NAME.get(dist_name)
         if mapped_module is not None:
@@ -115,4 +113,4 @@ def make_runtime_constraints(locked: Mapping[str, str | None]) -> list[str]:
     for name, version in sorted(locked.items()):
         if version:
             constraints.append(f"{name}=={version}")
-    return constraints
+    return constraints

mypy/main.py

@@ -30,7 +30,11 @@
 from mypy.errors import CompileError
 from mypy.find_sources import InvalidSourceList, create_source_list
 from mypy.fscache import FileSystemCache
-from mypy.installtypes import make_runtime_constraints, read_locked_packages, resolve_stub_packages_from_lock
+from mypy.installtypes import (
+    make_runtime_constraints,
+    read_locked_packages,
+    resolve_stub_packages_from_lock,
+)
 from mypy.modulefinder import (
     BuildSource,
     FindModuleCache,
@@ -136,11 +140,7 @@ def main(
     if options.install_types_from_pylock is not None and not os.path.isfile(
         options.install_types_from_pylock
     ):
-        fail(
-            f"error: Can't find lock file '{options.install_types_from_pylock}'",
-            stderr,
-            options,
-        )
+        fail(f"error: Can't find lock file '{options.install_types_from_pylock}'", stderr, options)
 
     if options.install_types and not options.incremental:
         fail(

mypy/test/testinstalltypes.py

@@ -4,21 +4,20 @@
 import tempfile
 import textwrap
 import unittest
-from types import SimpleNamespace #ADDED
-from unittest.mock import patch #ADDED
+from types import SimpleNamespace  # ADDED
+from unittest.mock import patch  # ADDED
 
 from mypy.installtypes import (
     make_runtime_constraints,
     read_locked_packages,
     resolve_stub_packages_from_lock,
 )
-from mypy.main import install_types #ADDED
+from mypy.main import install_types  # ADDED
 
 
 class TestInstallTypesFromPylock(unittest.TestCase):
     def test_read_locked_packages(self) -> None:
-        content = textwrap.dedent(
-            """
+        content = textwrap.dedent("""
             [[package]]
             name = "requests"
             version = "2.32.3"
@@ -30,8 +29,7 @@ def test_read_locked_packages(self) -> None:
             [[package]]
             name = "types-requests"
             version = "2.32.0"
-            """
-        )
+            """)
         with tempfile.NamedTemporaryFile("w", suffix=".toml", delete=False, encoding="utf-8") as f:
             f.write(content)
             path = f.name
@@ -88,32 +86,24 @@ def test_read_locked_packages_empty_file(self) -> None:
         assert locked == {}
 
     def test_resolve_stub_packages_from_lock(self) -> None:
-        locked = {
-            "requests": "2.32.3",
-            "python-dateutil": "2.9.0",
-            "types-requests": "2.32.0",
-        }
+        locked = {"requests": "2.32.3", "python-dateutil": "2.9.0", "types-requests": "2.32.0"}
         stubs = resolve_stub_packages_from_lock(locked)
         # requests already worked before fix because distribution name == module name.
         assert "types-requests" in stubs
 
         # FIX: Before my fix, this failed because the lock file used the distribution name
         # "python-dateutil" but stubinfo.py knows the module/import name "dateutil".
         # After adding the explicit distribution->module mapping, it should resolve.
-        assert "types-python-dateutil" in stubs #FIX
+        assert "types-python-dateutil" in stubs  # FIX
 
-    #TEST: checks explicit distribution->module mapping
+    # TEST: checks explicit distribution->module mapping
     def test_resolve_stub_packages_from_lock_handles_distribution_module_mismatch(self) -> None:
-        locked = {
-            "python-dateutil": "2.9.0",
-        }
+        locked = {"python-dateutil": "2.9.0"}
         stubs = resolve_stub_packages_from_lock(locked)
         assert "types-python-dateutil" in stubs
-    
+
     def test_resolve_stub_packages_skips_types_packages(self) -> None:
-        locked = {
-            "types-requests": "2.32.0",
-        }
+        locked = {"types-requests": "2.32.0"}
         stubs = resolve_stub_packages_from_lock(locked)
         # Should not produce "types-types-requests"
         assert "types-types-requests" not in stubs
@@ -133,11 +123,7 @@ def test_resolve_stub_packages_pyyaml_mapping(self) -> None:
         assert "types-PyYAML" in stubs
 
     def test_make_runtime_constraints(self) -> None:
-        locked = {
-            "requests": "2.32.3",
-            "python-dateutil": "2.9.0",
-            "no-version": None,
-        }
+        locked = {"requests": "2.32.3", "python-dateutil": "2.9.0", "no-version": None}
         constraints = make_runtime_constraints(locked)
         assert constraints == ["python-dateutil==2.9.0", "requests==2.32.3"]
 
@@ -156,46 +142,41 @@ def test_make_runtime_constraints_is_sorted(self) -> None:
         constraints = make_runtime_constraints(locked)
         assert constraints == sorted(constraints)
 
-#FIX: stub/mock object that pretends to be a formatter so tests don’t crash.
+
+# FIX: stub/mock object that pretends to be a formatter so tests don’t crash.
 class DummyFormatter:
     def style(self, text: str, *args: object, **kwargs: object) -> str:
         return text
 
-#TEST: integrations tests
+
+# TEST: integrations tests
 class TestInstallTypesFromPylockIntegration(unittest.TestCase):
     def make_options(self) -> SimpleNamespace:
-        return SimpleNamespace(
-            python_executable="python",
-            cache_dir="unused",
-        )
+        return SimpleNamespace(python_executable="python", cache_dir="unused")
 
     @patch("mypy.main.subprocess.run")
     def test_install_types_builds_correct_pip_command(self, mock_run) -> None:
-        content = textwrap.dedent(
-            """
+        content = textwrap.dedent("""
             [[package]]
             name = "requests"
             version = "2.32.3"
 
             [[package]]
             name = "python-dateutil"
             version = "2.9.0"
-            """
-        )
+            """)
 
-        with tempfile.NamedTemporaryFile(
-            "w", suffix=".toml", delete=False, encoding="utf-8"
-        ) as f:
+        with tempfile.NamedTemporaryFile("w", suffix=".toml", delete=False, encoding="utf-8") as f:
             f.write(content)
             path = f.name
 
         try:
             options = self.make_options()
-            formatter = DummyFormatter() #FIX
+            formatter = DummyFormatter()  # FIX
 
             result = install_types(
                 # formatter=None,
-                formatter=formatter, # FIX
+                formatter=formatter,  # FIX
                 options=options,
                 non_interactive=True,
                 pylock_path=path,
@@ -229,27 +210,23 @@ def test_install_types_builds_correct_pip_command(self, mock_run) -> None:
 
     @patch("mypy.main.subprocess.run")
     def test_no_stubs_found_skips_install(self, mock_run) -> None:
-        content = textwrap.dedent(
-            """
+        content = textwrap.dedent("""
             [[package]]
             name = "unknown-lib"
             version = "1.0.0"
-            """
-        )
+            """)
 
-        with tempfile.NamedTemporaryFile(
-            "w", suffix=".toml", delete=False, encoding="utf-8"
-        ) as f:
+        with tempfile.NamedTemporaryFile("w", suffix=".toml", delete=False, encoding="utf-8") as f:
             f.write(content)
             path = f.name
 
         try:
             options = self.make_options()
-            formatter = DummyFormatter() #FIX
+            formatter = DummyFormatter()  # FIX
 
             result = install_types(
                 # formatter=None,
-                formatter=formatter, #FIX
+                formatter=formatter,  # FIX
                 options=options,
                 non_interactive=True,
                 pylock_path=path,
@@ -263,13 +240,11 @@ def test_no_stubs_found_skips_install(self, mock_run) -> None:
 
     @patch("mypy.main.subprocess.run")
     def test_constraint_file_cleaned_up_after_success(self, mock_run) -> None:
-        content = textwrap.dedent(
-            """
+        content = textwrap.dedent("""
             [[package]]
             name = "requests"
             version = "2.32.3"
-            """
-        )
+            """)
         with tempfile.NamedTemporaryFile("w", suffix=".toml", delete=False, encoding="utf-8") as f:
             f.write(content)
             path = f.name
@@ -300,13 +275,11 @@ def capture_cmd(cmd: list[str], **kwargs: object) -> None:
 
     @patch("mypy.main.subprocess.run")
     def test_constraint_file_cleaned_up_even_if_subprocess_fails(self, mock_run) -> None:
-        content = textwrap.dedent(
-            """
+        content = textwrap.dedent("""
             [[package]]
             name = "requests"
             version = "2.32.3"
-            """
-        )
+            """)
         with tempfile.NamedTemporaryFile("w", suffix=".toml", delete=False, encoding="utf-8") as f:
             f.write(content)
             path = f.name
@@ -335,4 +308,4 @@ def capture_and_raise(cmd: list[str], **kwargs: object) -> None:
         self.assertFalse(
             os.path.exists(captured[0]),
             "Temp constraint file should be deleted even when subprocess raises",
-        )
+        )