Merge pull request #65 from ziadsawalha/more-tests

Add tests and fixed bug where new 209 was failing

Author: Vladimir Keleshev

pep257.py

@@ -609,9 +609,7 @@ def check_newline_after_last_paragraph(self, definition, docstring):
         if docstring:
             lines = [l for l in eval(docstring).split('\n') if not is_blank(l)]
             if len(lines) > 1:
-                lines = eval(docstring).split('\n')
-                blanks = len(list(takewhile(is_blank, reversed(lines))))
-                if blanks < 2:
+                if docstring.split("\n")[-1].strip() not in ['"""', "'''"]:
                     return Error('D209: Put multi-line docstring closing '
                                  'quotes on separate line')
 

test.py

@@ -1,5 +1,5 @@
-"""Docstring."""
 # encoding: utf-8
+# No docstring, so we can test D100
 import sys
 
 
@@ -201,9 +201,26 @@ def foobar():
     """Signature: foobar()."""
 
 
+def new_209():
+    """First line.
+
+    More lines.
+    """
+    pass
+
+
+def old_209():
+    """One liner.
+
+    Multi-line comments. OK to have extra blank line
+
+    """
+
+
 def run():
     """Run the functions above and check errors agains expected errors."""
     import pep257
+    expect(__file__, 'D100: Docstring missing')
     results = list(pep257.check([__file__]))
     assert set(map(type, results)) == set([pep257.Error]), results
     results = set([(e.definition.name, e.message) for e in results])