fix test

Author: csmoe

src/expr.rs

@@ -710,11 +710,12 @@ impl Rewrite for ast::Stmt {
                 };
 
                 let shape = shape.sub_width(suffix.len())?;
-                let expr_type = if stmt_is_expr(&self) {
-                    ExprType::SubExpression
-                } else {
-                    ExprType::Statement
-                };
+                let expr_type =
+                    if stmt_is_expr(&self) && context.snippet(self.span).starts_with("if ") {
+                        ExprType::SubExpression
+                    } else {
+                        ExprType::Statement
+                    };
                 format_expr(ex, expr_type, context, shape).map(|s| s + suffix)
             }
             ast::StmtKind::Mac(..) | ast::StmtKind::Item(..) => None,

tests/source/one_line_if.rs

@@ -0,0 +1,24 @@
+fn plain_if(x: bool) -> u8 {
+    if x {
+        0
+    } else {
+        1
+    }
+}
+
+fn paren_if(x: bool) -> u8 {
+    (if x { 0 } else { 1 })
+}
+
+fn let_if(x: bool) -> u8 {
+    let x = if x { 0 } else { 1 };
+    x
+}
+
+fn return_if(x: bool) -> u8 {
+    return if x {
+        0
+    } else {
+        1
+    };
+}

tests/target/one_line_if.rs

@@ -0,0 +1,16 @@
+fn plain_if(x: bool) -> u8 {
+    if x { 0 } else { 1 }
+}
+
+fn paren_if(x: bool) -> u8 {
+    (if x { 0 } else { 1 })
+}
+
+fn let_if(x: bool) -> u8 {
+    let x = if x { 0 } else { 1 };
+    x
+}
+
+fn return_if(x: bool) -> u8 {
+    return if x { 0 } else { 1 };
+}