Implement no_blank_line_before_class_docstring preview style

crates/ruff_python_formatter/src/preview.rs

@@ -24,3 +24,12 @@ pub(crate) const fn is_prefer_splitting_right_hand_side_of_assignments_enabled(
 ) -> bool {
     context.is_preview()
 }
+
+/// Returns `true` if the [`no_blank_line_before_class_docstring`] preview style is enabled.
+///
+/// [`no_blank_line_before_class_docstring`]: https://github.com/astral-sh/ruff/issues/8888
+pub(crate) const fn is_no_blank_line_before_class_docstring_enabled(
+    context: &PyFormatContext,
+) -> bool {
+    context.is_preview()
+}

crates/ruff_python_formatter/src/statement/suite.rs

@@ -11,6 +11,7 @@ use crate::comments::{
 use crate::context::{NodeLevel, TopLevelStatementPosition, WithIndentLevel, WithNodeLevel};
 use crate::expression::expr_string_literal::ExprStringLiteralKind;
 use crate::prelude::*;
+use crate::preview::is_no_blank_line_before_class_docstring_enabled;
 use crate::statement::stmt_expr::FormatStmtExpr;
 use crate::verbatim::{
     suppressed_node, write_suppressed_statements_starting_with_leading_comment,
@@ -108,14 +109,24 @@ impl FormatRule<Suite, PyFormatContext<'_>> for FormatSuite {
                     if !comments.has_leading(first)
                         && lines_before(first.start(), source) > 1
                         && !source_type.is_stub()
+                        && !is_no_blank_line_before_class_docstring_enabled(f.context())
                     {
                         // Allow up to one empty line before a class docstring, e.g., this is
                         // stable formatting:
+                        //
                         // ```python
                         // class Test:
                         //
                         //     """Docstring"""
                         // ```
+                        //
+                        // But, in preview mode, we don't want to allow any empty lines before a
+                        // class docstring, e.g., this is preview formatting:
+                        //
+                        // ```python
+                        // class Test:
+                        //   """Docstring"""
+                        // ```
                         empty_line().fmt(f)?;
                     }
 

crates/ruff_python_formatter/tests/snapshots/black_compatibility@cases__raw_docstring.py.snap

@@ -27,30 +27,21 @@ class UpperCaseR:
 ```diff
 --- Black
 +++ Ruff
-@@ -1,4 +1,5 @@
- class C:
-+
-     r"""Raw"""
- 
- 
-@@ -7,8 +8,9 @@
+@@ -7,7 +7,7 @@
  
  
  class SingleQuotes:
 -    r'''Raw'''
- 
 +    r"""Raw"""
-+
+ 
  
  class UpperCaseR:
-     R"""Raw"""
 ```
 
 ## Ruff Output
 
 ```python
 class C:
-
     r"""Raw"""
 
 
@@ -59,7 +50,6 @@ def f():
 
 
 class SingleQuotes:
-
     r"""Raw"""
 
 

crates/ruff_python_formatter/tests/snapshots/format@preview.py.snap

@@ -198,7 +198,6 @@ def reference_docstring_newlines():
 
 
 class RemoveNewlineBeforeClassDocstring:
-
     """Black's `Preview.no_blank_line_before_class_docstring`"""
 
 

crates/ruff_python_formatter/tests/snapshots/format@statement__class_definition.py.snap

@@ -513,7 +513,30 @@ class QuerySet(AltersData):
  
  
  class Test(
-@@ -159,20 +158,17 @@
+@@ -94,7 +93,6 @@
+ 
+ 
+ class Test:
+-
+     """Docstring"""
+ 
+ 
+@@ -111,14 +109,12 @@
+ 
+ 
+ class Test:
+-
+     """Docstring"""
+ 
+     x = 1
+ 
+ 
+ class Test:
+-
+     """Docstring"""
+ 
+     # comment
+@@ -159,20 +155,17 @@
  
  @dataclass
  # Copied from transformers.models.clip.modeling_clip.CLIPOutput with CLIP->AltCLIP