Match spacing after module docstring between black and --black

src/docformatter/format.py

@@ -345,25 +345,14 @@ def _get_function_docstring_newlines(  # noqa: PLR0911
     return 0
 
 
-def _get_module_docstring_newlines(black: bool = False) -> int:
+def _get_module_docstring_newlines() -> int:
     """Return number of newlines after a module docstring.
 
-    docformatter_8.2: One blank line after a module docstring.
-    docformatter_8.2.1: Two blank lines after a module docstring when in black mode.
-
-    Parameters
-    ----------
-    black : bool
-        Indicates whether we're using black formatting rules.
-
     Returns
     -------
     newlines : int
         The number of newlines to insert after the docstring.
     """
-    if black:
-        return 2
-
     return 1
 
 
@@ -395,7 +384,7 @@ def _get_newlines_by_type(
         return 0
     elif _classify.is_module_docstring(tokens, index):
         # print("Module")
-        return _get_module_docstring_newlines(black)
+        return _get_module_docstring_newlines()
     elif _classify.is_class_docstring(tokens, index):
         # print("Class")
         return _get_class_docstring_newlines(tokens, index)

tests/_data/string_files/do_format_code.toml

@@ -1233,3 +1233,14 @@ expected="""foo = f'''
   bar
 '''
 """
+
+[issue_331_black_module_docstring]
+source='''"""A."""
+
+
+pass
+'''
+expected='''"""A."""
+
+pass
+'''

tests/_data/string_files/format_functions.toml

@@ -15,7 +15,7 @@ expected = 1
 expected = 1
 
 [module_docstring_in_black]
-expected = 2
+expected = 1
 
 [class_docstring_followed_by_statement]
 source = '''

tests/formatter/test_do_format_code.py

@@ -140,6 +140,7 @@
         ("ellipses_is_code_line", NO_ARGS),
         ("do_not_break_f_string_double_quotes", NO_ARGS),
         ("do_not_break_f_string_single_quotes", NO_ARGS),
+        ("issue_331_black_module_docstring", ["--black", ""]),
     ],
 )
 def test_do_format_code(test_key, test_args, args):

tests/formatter/test_format_functions.py

@@ -64,19 +64,19 @@ def _get_docstring_token_and_index(tokens):
 
 @pytest.mark.unit
 @pytest.mark.parametrize(
-    "test_key, black",
+    "test_key",
     [
-        ("module_docstring_followed_by_string", False),
-        ("module_docstring_followed_by_code", False),
-        ("module_docstring_followed_by_comment_then_code", False),
-        ("module_docstring_followed_by_comment_then_string", False),
-        ("module_docstring_in_black", True),
+        "module_docstring_followed_by_string",
+        "module_docstring_followed_by_code",
+        "module_docstring_followed_by_comment_then_code",
+        "module_docstring_followed_by_comment_then_string",
+        "module_docstring_in_black",
     ],
 )
-def test_module_docstring_newlines(test_key, black):
+def test_module_docstring_newlines(test_key):
     expected = TEST_STRINGS[test_key]["expected"]
 
-    result = _format._get_module_docstring_newlines(black)
+    result = _format._get_module_docstring_newlines()
     assert (
         result == expected
     ), f"\nFailed {test_key}:\nExpected {expected}\nGot {result}"