Allow Not to be in the same line if it is a single argument

FuelLabs · Dec 21, 2023 · 96514e8 · 96514e8
1 parent 4397e1c
commit 96514e8
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/swayfmt/src/utils/language/expr/mod.rs b/swayfmt/src/utils/language/expr/mod.rs
@@ -775,6 +775,10 @@ fn same_line_if_only_argument(expr: &Expr) -> bool {
         Expr::Struct { path: _, fields: _ }
             | Expr::Tuple(_)
             | Expr::Parens(_)
+            | Expr::Not {
+                bang_token: _,
+                expr: _
+            }
             | Expr::Path(_)
             | Expr::FuncApp { func: _, args: _ }
             | Expr::Match {

diff --git a/swayfmt/tests/mod.rs b/swayfmt/tests/mod.rs
@@ -2671,3 +2671,19 @@ fn test() {
 "#,
     );
 }
+
+#[test]
+fn single_argument_not() {
+    check(
+        r#"library;
+fn test() {
+    assert(!(U256::from((0, 0, 0, 1)) > U256::from((0, u64::max(), 0, 0))));
+}
+    "#,
+        r#"library;
+fn test() {
+    assert(!(U256::from((0, 0, 0, 1)) > U256::from((0, u64::max(), 0, 0))));
+}
+"#,
+    );
+}