diff --git a/mypy/test/teststubgen.py b/mypy/test/teststubgen.py
index 29ccccce10bd..a91d4397f46f 100644
--- a/mypy/test/teststubgen.py
+++ b/mypy/test/teststubgen.py
@@ -391,7 +391,13 @@ def test_infer_sig_from_docstring(self) -> None:
 
         assert_equal(
             infer_sig_from_docstring("\nfunc(*args, **kwargs)", "func"),
-            [FunctionSig(name="func", args=[ArgSig(name="*args"), ArgSig(name="**kwargs")], ret_type="Any")],
+            [
+                FunctionSig(
+                    name="func",
+                    args=[ArgSig(name="*args"), ArgSig(name="**kwargs")],
+                    ret_type="Any",
+                )
+            ],
         )
 
     def test_infer_sig_from_docstring_duplicate_args(self) -> None:
@@ -447,11 +453,28 @@ def test_infer_sig_from_docstring_multiple_overloads(self) -> None:
 
 2. func(self: class_type, a: float, b: int) -> None
 """
-        assert_equal(infer_sig_from_docstring(input, "func"), [
-            FunctionSig(name="func", args=[ArgSig(name="self", type="class_type")], ret_type="None"),
-            FunctionSig(name="func", args=[ArgSig(name="self", type="class_type"), ArgSig(name="a", type="float"), ArgSig(name="b", type="int")], ret_type="None"),
-            FunctionSig(name="func", args=[ArgSig(name="*args"), ArgSig(name="**kwargs")], ret_type="Any")
-        ])
+        assert_equal(
+            infer_sig_from_docstring(input, "func"),
+            [
+                FunctionSig(
+                    name="func", args=[ArgSig(name="self", type="class_type")], ret_type="None"
+                ),
+                FunctionSig(
+                    name="func",
+                    args=[
+                        ArgSig(name="self", type="class_type"),
+                        ArgSig(name="a", type="float"),
+                        ArgSig(name="b", type="int"),
+                    ],
+                    ret_type="None",
+                ),
+                FunctionSig(
+                    name="func",
+                    args=[ArgSig(name="*args"), ArgSig(name="**kwargs")],
+                    ret_type="Any",
+                ),
+            ],
+        )
 
     def test_infer_sig_from_docstring_deeply_nested_types(self) -> None:
         assert_equal(
@@ -1520,6 +1543,7 @@ def module_to_path(out_dir: str, module: str) -> str:
             return alt_fnam
     return fnam
 
+
 def main():
     suite = StubgenUtilSuite()
     suite.test_infer_sig_from_docstring_invalid_signature()
@@ -1528,6 +1552,7 @@ def main():
     suite.test_infer_sig_from_docstring_duplicate_args()
     suite.test_infer_sig_from_docstring_deeply_nested_types()
     suite.test_infer_sig_from_docstring_multiple_overloads()
-    
+
+
 if __name__ == "__main__":
-    main()
\ No newline at end of file
+    main()