Fixed a bug with comments after function docstrings

Author: Nurdok

src/pep257.py

@@ -119,7 +119,6 @@ class Definition(Value):
     module = property(lambda self: self.parent.module)
     all = property(lambda self: self.module.all)
     _slice = property(lambda self: slice(self.start - 1, self.end))
-    source = property(lambda self: ''.join(self._source[self._slice]))
     is_class = False
 
     def __iter__(self):
@@ -129,6 +128,17 @@ def __iter__(self):
     def _publicity(self):
         return {True: 'public', False: 'private'}[self.is_public]
 
+    @property
+    def source(self):
+        """Return the source code for the definition."""
+        full_src = self._source[self._slice]
+
+        def predicate(line):
+            return line.strip() == '' or line.strip().startswith('#')
+
+        filtered_src = dropwhile(predicate, reversed(full_src))
+        return ''.join(reversed(list(filtered_src)))
+
     def __str__(self):
         return 'in %s %s `%s`' % (self._publicity, self._human, self.name)
 
@@ -428,7 +438,7 @@ def parse_module(self):
         return module
 
     def parse_definition(self, class_):
-        """Parse a defintion and return its value in a `class_` object."""
+        """Parse a definition and return its value in a `class_` object."""
         start = self.line
         self.consume(tk.NAME)
         name = self.current.value

src/tests/test_cases/comment_after_def_bug.py

@@ -0,0 +1,14 @@
+"""Check for a bug in parsing comments after definitions."""
+
+from .expected import Expectation
+
+expectation = Expectation()
+expect = expectation.expect
+
+
+def should_be_ok():
+    """Just a function without violations."""
+
+
+# This is a comment that triggers a bug that causes the previous function
+# to generate a D202 error.

src/tests/test_definitions.py

@@ -138,14 +138,19 @@ def test_token_stream():
 
 def test_pep257():
     """Run domain-specific tests from test.py file."""
-    test_cases = ('test', 'unicode_literals', 'nested_class', 'capitalization')
+    test_cases = (
+        'test',
+        'unicode_literals',
+        'nested_class',
+        'capitalization',
+        'comment_after_def_bug',
+    )
     for test_case in test_cases:
         case_module = __import__('test_cases.{0}'.format(test_case),
                                  globals=globals(),
                                  locals=locals(),
                                  fromlist=['expectation'],
                                  level=1)
-        # from .test_cases import test
         results = list(check([os.path.join(os.path.dirname(__file__),
                                            'test_cases', test_case + '.py')],
                              select=set(ErrorRegistry.get_error_codes())))