unit tests for correctly documented parameters and return values (all 3 styles)

1 files changed, 53 insertions, 2 deletions

diff --git a/pylint/extensions/test/test_check_docs.py b/pylint/extensions/test/test_check_docs.py
index 642ceb1..47b264f 100644
--- a/pylint/extensions/test/test_check_docs.py
+++ b/pylint/extensions/test/test_check_docs.py
@@ -209,9 +209,9 @@ class SpinxDocCheckerTest(CheckerTestCase):
         ):
             self.checker.visit_class(node)
 
-    def test_existing_func_params_in_docstring(self):
+    def test_existing_func_params_in_sphinx_docstring(self):
         """Example of a function with correctly documented parameters and
-        return values
+        return values (Sphinx style)
         """
         node = test_utils.extract_node("""
         def function_foo(xarg, yarg, zarg):
@@ -233,6 +233,57 @@ class SpinxDocCheckerTest(CheckerTestCase):
         with self.assertNoMessages():
             self.checker.visit_function(node)
 
+    def test_existing_func_params_in_google_docstring(self):
+        """Example of a function with correctly documented parameters and
+        return values (Google style)
+        """
+        node = test_utils.extract_node("""
+        def function_foo(xarg, yarg, zarg):
+            '''function foo ...
+
+            Args:
+                xarg (int): bla xarg
+                yarg (float): bla
+                    bla yarg
+        
+                zarg (int): bla zarg
+
+            Returns:
+                sum (float)
+            '''
+            return xarg + yarg
+        """)
+        with self.assertNoMessages():
+            self.checker.visit_function(node)
+
+    def test_existing_func_params_in_numpy_docstring(self):
+        """Example of a function with correctly documented parameters and
+        return values (Numpy style)
+        """
+        node = test_utils.extract_node("""
+        def function_foo(xarg, yarg, zarg):
+            '''function foo ...
+
+            Parameters
+            ----------
+            xarg: int
+                bla xarg
+            yarg: float
+                bla yarg
+
+            zarg: int
+                bla zarg
+
+            Returns
+            -------
+            float
+                sum
+            '''
+            return xarg + yarg
+        """)
+        with self.assertNoMessages():
+            self.checker.visit_function(node)
+
     def test_wrong_name_of_func_params_in_docstring(self):
         """Example of functions with inconsistent parameter names in the
         signature and in the documentation