fix: handle case where class contains only docstring

Author: finswimmer

src/docformatter/classify.py

@@ -463,7 +463,7 @@ def is_string_variable(
     try:
         _token_types = (tokenize.AWAIT, tokenize.OP)
     except AttributeError:
-        _token_types = (tokenize.OP,)  # type: ignore
+        _token_types = (tokenize.OP,)
 
     if prev_token.type in _token_types and (
         '= """' in token.line or token.line in prev_token.line

src/docformatter/format.py

@@ -260,11 +260,22 @@ def _get_class_docstring_newlines(
         The number of newlines to insert after the docstring.
     """
     j = index + 1
+    indention_level = tokens[index].start[1]
 
     # The docstring is followed by a comment.
     if tokens[j].string.startswith("#"):
         return 0
 
+    while j < len(tokens):
+        if tokens[j].type in (tokenize.NL, tokenize.NEWLINE):
+            j += 1
+            continue
+
+        if tokens[j].start[1] < indention_level:
+            return 2
+
+        break
+
     return 1
 
 

tests/_data/string_files/do_format_code.toml

@@ -1160,3 +1160,34 @@ expected='''def foo(bar):
     Description.
     """
 '''
+
+[two_lines_between_stub_classes]
+source='''class Foo:
+    """Foo class."""
+class Bar:
+    """Bar class."""
+'''
+expected='''class Foo:
+    """Foo class."""
+
+
+class Bar:
+    """Bar class."""
+'''
+
+[two_lines_between_stub_classes_with_preceding_comment]
+source='''class Foo:
+    """Foo class."""
+
+# A comment for class Bar
+class Bar:
+    """Bar class."""
+'''
+expected='''class Foo:
+    """Foo class."""
+
+
+# A comment for class Bar
+class Bar:
+    """Bar class."""
+'''

tests/formatter/test_do_format_code.py

@@ -135,6 +135,8 @@
         ("issue_187", NO_ARGS),
         ("issue_203", NO_ARGS),
         ("issue_243", NO_ARGS),
+        ("two_lines_between_stub_classes", NO_ARGS),
+        ("two_lines_between_stub_classes_with_preceding_comment", NO_ARGS),
     ],
 )
 def test_do_format_code(test_key, test_args, args):