TST: test source_file_def_line with decorators

Author: sdiebolt

numpydoc/tests/test_validate.py

@@ -1,5 +1,6 @@
 import pytest
 import warnings
+from inspect import getsourcelines
 
 from numpydoc import validate
 import numpydoc.tests
@@ -1528,6 +1529,35 @@ def test_bad_docstrings(self, capsys, klass, func, msgs):
             assert msg in " ".join(err[1] for err in result["errors"])
 
 
+def decorator(x):
+    """Test decorator."""
+    return x
+
+
+@decorator
+@decorator
+class DecoratorClass:
+    """
+    Class and methods with decorators.
+
+    `DecoratorClass` has two decorators, `DecoratorClass.test_no_decorator` has no
+    decorator and `DecoratorClass.test_three_decorators` has three decorators.
+    `Validator.source_file_def_line` should return the `def` or `class` line number, not
+    the line of the first decorator.
+    """
+
+    def test_no_decorator(self):
+        """Test method without decorators."""
+        pass
+
+    @decorator
+    @decorator
+    @decorator
+    def test_three_decorators(self):
+        """Test method with three decorators."""
+        pass
+
+
 class TestValidatorClass:
     @pytest.mark.parametrize("invalid_name", ["unknown_mod", "unknown_mod.MyClass"])
     def test_raises_for_invalid_module_name(self, invalid_name):
@@ -1544,3 +1574,28 @@ def test_raises_for_invalid_attribute_name(self, invalid_name):
         msg = f"'{obj_name}' has no attribute '{invalid_attr_name}'"
         with pytest.raises(AttributeError, match=msg):
             numpydoc.validate.Validator._load_obj(invalid_name)
+
+    @pytest.mark.parametrize(
+        ["decorated_obj", "def_line"],
+        [
+            [
+                "numpydoc.tests.test_validate.DecoratorClass",
+                getsourcelines(DecoratorClass)[-1] + 2,
+            ],
+            [
+                "numpydoc.tests.test_validate.DecoratorClass.test_no_decorator",
+                getsourcelines(DecoratorClass.test_no_decorator)[-1],
+            ],
+            [
+                "numpydoc.tests.test_validate.DecoratorClass.test_three_decorators",
+                getsourcelines(DecoratorClass.test_three_decorators)[-1] + 3,
+            ],
+        ],
+    )
+    def test_source_file_def_line_with_decorators(self, decorated_obj, def_line):
+        doc = numpydoc.validate.Validator(
+            numpydoc.docscrape.get_doc_object(
+                numpydoc.validate.Validator._load_obj(decorated_obj)
+            )
+        )
+        assert doc.source_file_def_line == def_line

numpydoc/validate.py

@@ -298,14 +298,14 @@ def source_file_def_line(self):
             sourcelines = inspect.getsourcelines(self.code_obj)
             # getsourcelines will return the line of the first decorator found for the
             # current function. We have to find the def declaration after that.
-            def_lines = [
+            def_line = next(
                 i
                 for i, x in enumerate(
-                    [re.match("^ *(def|class) ", s) for s in sourcelines[0]]
+                    re.match("^ *(def|class) ", s) for s in sourcelines[0]
                 )
                 if x is not None
-            ]
-            return sourcelines[-1] + def_lines[0]
+            )
+            return sourcelines[-1] + def_line
         except (OSError, TypeError):
             # In some cases the object is something complex like a cython
             # object that can't be easily introspected. An it's better to