feat: add integration tests for basic and gate definition formatting (#5)

Author: orangekame3

src/format.rs

@@ -40,14 +40,11 @@ pub fn format_source(source: &str, config: &FormatConfig) -> Result<String, Form
 }
 
 fn format_file(ctx: &FormatContext, file: &ast::SourceFile) -> Doc {
-    let mut docs = Vec::new();
-
-    for stmt in file.statements() {
-        docs.push(format_stmt(ctx, stmt));
-        docs.push(Doc::hardline());
-    }
-
-    Doc::concat(docs)
+    let stmts: Vec<Doc> = file
+        .statements()
+        .map(|stmt| format_stmt(ctx, stmt))
+        .collect();
+    join(stmts, Doc::hardline())
 }
 
 fn format_stmt(ctx: &FormatContext, stmt: ast::Stmt) -> Doc {

tests/fixtures/basic/input.qasm

@@ -0,0 +1,10 @@
+OPENQASM   3.0;
+include "stdgates.inc";
+
+qubit[2]   q;
+bit[2] c;
+
+h    q[0];
+cx q[0],q[1];
+
+c = measure q;

tests/fixtures/basic/output.qasm

@@ -0,0 +1,7 @@
+OPENQASM 3.0;
+include "stdgates.inc";
+qubit[2] q;
+bit[2] c;
+h q[0];
+cx q[0], q[1];
+c = measure q;

tests/fixtures/gate_def/input.qasm

@@ -0,0 +1,8 @@
+gate   mygate(theta)   q   {
+rz(theta)   q;
+x q;
+}
+
+gate   cx   a,b{
+ctrl @ x a,b;
+}

tests/fixtures/gate_def/output.qasm

@@ -0,0 +1,7 @@
+gate mygate(theta) q {
+    rz(theta) q;
+    x q;
+}
+gate cx a,b {
+    ctrl @ x a,b;
+}

tests/golden.rs

@@ -0,0 +1,34 @@
+use std::fs;
+use std::path::Path;
+
+use qasmfmt::format;
+
+fn run_test(name: &str) {
+    let testdata = Path::new(env!("CARGO_MANIFEST_DIR"))
+        .join("tests/fixtures")
+        .join(name);
+
+    let input = fs::read_to_string(testdata.join("input.qasm"))
+        .unwrap_or_else(|e| panic!("{}/input.qasm: {}", name, e));
+    let expected = fs::read_to_string(testdata.join("output.qasm"))
+        .unwrap_or_else(|e| panic!("{}/output.qasm: {}", name, e));
+
+    let actual = format(&input).unwrap_or_else(|e| panic!("{}: {}", name, e));
+
+    if actual != expected {
+        panic!(
+            "\n\n=== {} ===\n\nExpected:\n{}\n\nActual:\n{}\n",
+            name, expected, actual
+        );
+    }
+}
+
+#[test]
+fn test_basic() {
+    run_test("basic");
+}
+
+#[test]
+fn test_gate_def() {
+    run_test("gate_def");
+}