feat: add integration and rule-based tests for various QASM features (#9)

orangekame3 · web-flow · commit 77c0eda4dd75 · 2026-01-13T23:50:32.000+09:00
diff --git a/src/format.rs b/src/format.rs
@@ -334,7 +334,24 @@ fn format_barrier(b: ast::Barrier) -> Doc {
 }
 
 fn format_assignment(a: ast::AssignmentStmt) -> Doc {
-    Doc::text(a.syntax().text().to_string().trim())
+    let mut parts = vec![];
+
+    if let Some(idx) = a.indexed_identifier() {
+        parts.push(format_indexed_identifier(&idx));
+    } else if let Some(id) = a.identifier() {
+        parts.push(Doc::text(
+            id.ident_token().map(|t| t.to_string()).unwrap_or_default(),
+        ));
+    }
+
+    parts.push(Doc::text(" = "));
+
+    if let Some(rhs) = a.rhs() {
+        parts.push(format_expr(rhs));
+    }
+
+    parts.push(Doc::text(";"));
+    Doc::concat(parts)
 }
 
 fn format_classical_decl(c: ast::ClassicalDeclarationStatement) -> Doc {
diff --git a/tests/fixtures/assignment/input.qasm b/tests/fixtures/assignment/input.qasm
@@ -0,0 +1,4 @@
+OPENQASM 3.0;
+qubit q;
+bit c;
+c=measure q;
diff --git a/tests/fixtures/assignment/output.qasm b/tests/fixtures/assignment/output.qasm
@@ -0,0 +1,4 @@
+OPENQASM 3.0;
+qubit q;
+bit c;
+c = measure q;
diff --git a/tests/fixtures/barrier/input.qasm b/tests/fixtures/barrier/input.qasm
@@ -0,0 +1,3 @@
+OPENQASM 3.0;
+qubit[2] q;
+barrier q[0],q[1];
diff --git a/tests/fixtures/barrier/output.qasm b/tests/fixtures/barrier/output.qasm
@@ -0,0 +1,3 @@
+OPENQASM 3.0;
+qubit[2] q;
+barrier q[0], q[1];
diff --git a/tests/fixtures/declarations/input.qasm b/tests/fixtures/declarations/input.qasm
@@ -0,0 +1,3 @@
+OPENQASM 3.0;
+qubit[2]q;
+bit[4]c;
diff --git a/tests/fixtures/declarations/output.qasm b/tests/fixtures/declarations/output.qasm
@@ -0,0 +1,3 @@
+OPENQASM 3.0;
+qubit[2] q;
+bit[4] c;
diff --git a/tests/fixtures/gate_calls/input.qasm b/tests/fixtures/gate_calls/input.qasm
@@ -0,0 +1,4 @@
+OPENQASM 3.0;
+qubit[2] q;
+h q[0];
+cx q[0],q[1];
diff --git a/tests/fixtures/gate_calls/output.qasm b/tests/fixtures/gate_calls/output.qasm
@@ -0,0 +1,4 @@
+OPENQASM 3.0;
+qubit[2] q;
+h q[0];
+cx q[0], q[1];
diff --git a/tests/fixtures/gate_definition/input.qasm b/tests/fixtures/gate_definition/input.qasm
diff --git a/tests/fixtures/gate_definition/output.qasm b/tests/fixtures/gate_definition/output.qasm
diff --git a/tests/fixtures/include_statement/input.qasm b/tests/fixtures/include_statement/input.qasm
@@ -0,0 +1,2 @@
+OPENQASM 3.0;
+include"stdgates.inc";
diff --git a/tests/fixtures/include_statement/output.qasm b/tests/fixtures/include_statement/output.qasm
@@ -0,0 +1,2 @@
+OPENQASM 3.0;
+include "stdgates.inc";
diff --git a/tests/fixtures/minified/input.qasm b/tests/fixtures/minified/input.qasm
@@ -0,0 +1 @@
+OPENQASM 3.0;include"stdgates.inc";qubit[2]q;bit[2]c;h q[0];cx q[0],q[1];c=measure q;
diff --git a/tests/fixtures/minified/output.qasm b/tests/fixtures/minified/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;
diff --git a/tests/fixtures/version_declaration/input.qasm b/tests/fixtures/version_declaration/input.qasm
@@ -0,0 +1 @@
+OPENQASM  3.0;
diff --git a/tests/fixtures/version_declaration/output.qasm b/tests/fixtures/version_declaration/output.qasm
@@ -0,0 +1 @@
+OPENQASM 3.0;
diff --git a/tests/golden.rs b/tests/golden.rs
@@ -23,12 +23,56 @@ fn run_test(name: &str) {
     }
 }
 
+// =============================================================================
+// Integration Tests
+// =============================================================================
+
 #[test]
 fn test_basic() {
     run_test("basic");
 }
 
 #[test]
-fn test_gate_def() {
-    run_test("gate_def");
+fn test_minified() {
+    run_test("minified");
+}
+
+// =============================================================================
+// Rule-based Tests
+// See: docs/src/rules/overview.md
+// =============================================================================
+
+#[test]
+fn test_version_declaration() {
+    run_test("version_declaration");
+}
+
+#[test]
+fn test_include_statement() {
+    run_test("include_statement");
+}
+
+#[test]
+fn test_declarations() {
+    run_test("declarations");
+}
+
+#[test]
+fn test_gate_calls() {
+    run_test("gate_calls");
+}
+
+#[test]
+fn test_gate_definition() {
+    run_test("gate_definition");
+}
+
+#[test]
+fn test_assignment() {
+    run_test("assignment");
+}
+
+#[test]
+fn test_barrier() {
+    run_test("barrier");
 }

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+OPENQASM 3.0;`
	`2`	`+qubit[2] q;`
	`3`	`+barrier q[0],q[1];`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+OPENQASM 3.0;`
	`2`	`+qubit[2]q;`
	`3`	`+bit[4]c;`