Update mem2reg test

src/ir/mod.rs

@@ -321,7 +321,7 @@ impl fmt::Display for JumpArg {
             self.bid,
             self.args
                 .iter()
-                .map(|a| a.to_string())
+                .map(|a| format!("{}:{}", a, a.dtype()))
                 .collect::<Vec<_>>()
                 .join(", ")
         )

src/ir/write_ir.rs

@@ -259,12 +259,12 @@ impl WriteString for BlockExit {
                 default,
                 cases,
             } => format!(
-                "switch {} default: {} [\n{}\n  ]",
+                "switch {} default {} [\n{}\n  ]",
                 value.write_string(),
                 default,
                 cases
                     .iter()
-                    .map(|(v, b)| format!("    {}:{}, {}", v, v.dtype(), b))
+                    .map(|(v, b)| format!("    {}:{} {}", v, v.dtype(), b))
                     .collect::<Vec<_>>()
                     .join("\n")
             ),