factor out helper should_suppress_clause

Author: dylwil3

crates/ruff_python_formatter/src/statement/clause.rs

@@ -556,18 +556,7 @@ where
 
 impl<'ast> Format<PyFormatContext<'ast>> for FormatClause<'_, 'ast> {
     fn fmt(&self, f: &mut Formatter<PyFormatContext<'ast>>) -> FormatResult<()> {
-        let source = f.context().source();
-        if is_fix_fmt_skip_in_one_liners_enabled(f.context())
-            && let Some(last_child_in_clause) = self.format_header.header.last_child_in_clause()
-            && has_skip_comment(
-                f.context().comments().trailing(last_child_in_clause),
-                f.context().source(),
-            )
-            && !source.contains_line_break(TextRange::new(
-                self.format_header.header.range(source).unwrap().start(),
-                last_child_in_clause.end(),
-            ))
-        {
+        if should_suppress_clause(&self, f)? {
             write_suppressed_clause(self, f)
         } else {
             write!(f, [self.format_header, self.format_body])
@@ -644,6 +633,39 @@ fn colon_range(after_keyword_or_condition: TextSize, source: &str) -> FormatResu
     }
 }
 
+fn should_suppress_clause(
+    clause: &FormatClause<'_, '_>,
+    f: &mut Formatter<PyFormatContext<'_>>,
+) -> FormatResult<bool> {
+    // Check if preview behavior is enabled
+    if !is_fix_fmt_skip_in_one_liners_enabled(f.context()) {
+        return Ok(false);
+    };
+
+    let source = f.context().source();
+    let Some(last_child_in_clause) = clause.format_header.header.last_child_in_clause() else {
+        return Ok(false);
+    };
+
+    let clause_start = clause
+        .format_header
+        .header
+        .first_keyword_range(source)?
+        .start();
+
+    let clause_range = TextRange::new(clause_start, last_child_in_clause.end());
+
+    // Only applies to clauses on a single line
+    if source.contains_line_break(clause_range) {
+        return Ok(false);
+    };
+
+    Ok(has_skip_comment(
+        f.context().comments().trailing(last_child_in_clause),
+        f.context().source(),
+    ))
+}
+
 #[cold]
 fn write_suppressed_clause(
     clause: &FormatClause,
@@ -655,7 +677,9 @@ fn write_suppressed_clause(
 
     let comments = f.context().comments().clone();
 
-    let last_child = header.last_child_in_clause().unwrap();
+    let last_child = header
+        .last_child_in_clause()
+        .expect("last child to exist if `should_suppress_clause` is `Ok(true)`");
     let range_end = last_child.end();
 
     // Write the outer comments and format the node as verbatim