Treat ... in __doc__ as separator between varargs and kwonly args

sebastinas · sebastinas · commit 30018e427412 · 2021-03-16T23:39:43.000+01:00
diff --git a/bpython/inspection.py b/bpython/inspection.py
@@ -203,6 +203,9 @@ def getpydocspec(f, func):
             varkwargs = arg[2:]
         elif arg.startswith("*"):
             varargs = arg[1:]
+        elif arg == "...":
+            # At least print denotes "..." as separator between varargs and kwonly args.
+            varargs = ""
         else:
             arg, _, default = arg.partition("=")
             if varargs is not None:
diff --git a/bpython/test/test_inspection.py b/bpython/test/test_inspection.py
@@ -118,6 +118,17 @@ def test_get_source_file(self):
         )
         self.assertEqual(encoding, "utf-8")
 
+    def test_getfuncprops_print(self):
+        props = inspection.getfuncprops("print", print)
+
+        self.assertEqual(props.func, "print")
+        self.assertIn("end", props.argspec.kwonly)
+        self.assertIn("file", props.argspec.kwonly)
+        self.assertIn("flush", props.argspec.kwonly)
+        self.assertIn("sep", props.argspec.kwonly)
+        self.assertEqual(props.argspec.kwonly_defaults["file"], "sys.stdout")
+        self.assertEqual(props.argspec.kwonly_defaults["flush"], "False")
+
     @unittest.skipUnless(numpy is not None, "requires numpy")
     def test_getfuncprops_numpy_array(self):
         props = inspection.getfuncprops("array", numpy.array)
