test pos_only_index

Author: InvincibleRMC

mypy/stubdoc.py

@@ -77,6 +77,8 @@ class FunctionSig(NamedTuple):
     name: str
     args: list[ArgSig]
     ret_type: str | None
+    pos_only_index: int | None = None
+    kwarg_only_index: int | None = None
     type_args: str = ""  # TODO implement in stubgenc and remove the default
     docstring: str | None = None
 
@@ -139,6 +141,11 @@ def format_sig(
 
             args.append(arg_def)
 
+        if self.pos_only_index:
+            args.insert(self.pos_only_index, '/')
+        if self.kwarg_only_index:
+            args.insert(self.kwarg_only_index, '*')
+
         retfield = ""
         ret_type = self.ret_type if self.ret_type else any_val
         if ret_type is not None:
@@ -342,7 +349,7 @@ def add_token(self, token: tokenize.TokenInfo) -> None:
 
             if self.found:
                 self.signatures.append(
-                    FunctionSig(name=self.function_name, args=self.args, ret_type=self.ret_type)
+                    FunctionSig(name=self.function_name, args=self.args, pos_only_index=self.pos_only, kwarg_only_index=self.keyword_only, ret_type=self.ret_type)
                 )
                 self.found = False
             self.args = []

mypy/test/teststubgen.py

@@ -477,17 +477,17 @@ def test_infer_sig_from_docstring_positional_only_arguments(self) -> None:
     def test_infer_sig_from_docstring_keyword_only_arguments(self) -> None:
         assert_equal(
             infer_sig_from_docstring("func(*, x) -> str", "func"),
-            [FunctionSig(name="func", args=[ArgSig(name="x")], ret_type="str")],
+            [FunctionSig(name="func", args=[ArgSig(name="x")], ret_type="str", pos_only_index=0)],
         )
 
         assert_equal(
             infer_sig_from_docstring("func(x, *, y) -> str", "func"),
-            [FunctionSig(name="func", args=[ArgSig(name="x"), ArgSig(name="y")], ret_type="str")],
+            [FunctionSig(name="func", args=[ArgSig(name="x"), ArgSig(name="y")], ret_type="str", pos_only_index=1)],
         )
 
         assert_equal(
             infer_sig_from_docstring("func(*, x, y) -> str", "func"),
-            [FunctionSig(name="func", args=[ArgSig(name="x"), ArgSig(name="y")], ret_type="str")],
+            [FunctionSig(name="func", args=[ArgSig(name="x"), ArgSig(name="y")], ret_type="str", pos_only_index=0)],
         )
 
         assert_equal(
@@ -497,6 +497,7 @@ def test_infer_sig_from_docstring_keyword_only_arguments(self) -> None:
                     name="func",
                     args=[ArgSig(name="x"), ArgSig(name="kwonly"), ArgSig("**kwargs")],
                     ret_type="str",
+                    pos_only_index=1,
                 )
             ],
         )
@@ -514,6 +515,8 @@ def test_infer_sig_from_docstring_pos_only_and_keyword_only_arguments(self) -> N
                     name="func",
                     args=[ArgSig(name="x"), ArgSig(name="y"), ArgSig(name="z")],
                     ret_type="str",
+                    pos_only_index=1,
+                    kwarg_only_index=3
                 )
             ],
         )
@@ -530,6 +533,8 @@ def test_infer_sig_from_docstring_pos_only_and_keyword_only_arguments(self) -> N
                         ArgSig("**kwargs"),
                     ],
                     ret_type="str",
+                    pos_only_index=1,
+                    kwarg_only_index=3
                 )
             ],
         )