Add test of gate broadcasting

Qiskit · Jan 31, 2024 · 6712183 · 6712183
1 parent fcc79df
commit 6712183
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/test/python/qasm3/test_import.py b/test/python/qasm3/test_import.py
@@ -319,3 +319,35 @@ def test_set_include_path(self):
         expected = QuantumCircuit([Qubit()])
         expected.x(0)
         self.assertEqual(parsed, expected)
+
+    def test_gate_broadcast(self):
+        program = """
+            OPENQASM 3.0;
+            include "stdgates.inc";
+            qubit[2] q0;
+            qubit q1;
+            qubit[2] q2;
+            h q0;
+            cx q0, q1;
+            cx q0[0], q2;
+            cx q0, q2;
+            ccx q0[{1, 0}], q1, q2;
+        """
+        parsed = qasm3.loads_experimental(program)
+        q0, q1, q2 = QuantumRegister(2, "q0"), Qubit(), QuantumRegister(2, "q2")
+        expected = QuantumCircuit(q0, [q1], q2)
+        expected.h(q0[0])
+        expected.h(q0[1])
+        #
+        expected.cx(q0[0], q1)
+        expected.cx(q0[1], q1)
+        #
+        expected.cx(q0[0], q2[0])
+        expected.cx(q0[0], q2[1])
+        #
+        expected.cx(q0[0], q2[0])
+        expected.cx(q0[1], q2[1])
+        #
+        expected.ccx(q0[1], q1, q2[0])
+        expected.ccx(q0[0], q1, q2[1])
+        self.assertEqual(parsed, expected)