simple site verification test

Author: johnzl-777

test/squin/stim/stim.py

@@ -1,3 +1,4 @@
+import pytest
 from kirin import ir, types
 from kirin.passes import Fold
 from kirin.dialects import py, func, ilist
@@ -384,10 +385,51 @@ def test_wire_measure_and_reset():
     constructed_method.print()
 
 
+def test_wire_apply_site_verification():
+
+    stmts: list[ir.Statement] = [
+        # Create qubit register
+        (n_qubits := as_int(3)),
+        (qreg := qasm2.core.QRegNew(n_qubits=n_qubits.result)),
+        # Get qubis out
+        (idx0 := as_int(0)),
+        (q0 := qasm2.core.QRegGet(reg=qreg.result, idx=idx0.result)),
+        (idx1 := as_int(1)),
+        (q1 := qasm2.core.QRegGet(reg=qreg.result, idx=idx1.result)),
+        (idx2 := as_int(2)),
+        (q2 := qasm2.core.QRegGet(reg=qreg.result, idx=idx2.result)),
+        # Unwrap to get wires
+        (w0 := squin.wire.Unwrap(qubit=q0.result)),
+        (w1 := squin.wire.Unwrap(qubit=q1.result)),
+        (w2 := squin.wire.Unwrap(qubit=q2.result)),
+        # set up control gate
+        (op1 := squin.op.stmts.X()),
+        (cx := squin.op.stmts.Control(op1.result, n_controls=1)),
+        # improper application, dangling qubit that verification should catch!
+        (app := squin.wire.Apply(cx.result, w0.result, w1.result, w2.result)),
+        # wrap things back
+        (squin.wire.Wrap(wire=app.results[0], qubit=q0.result)),
+        (squin.wire.Wrap(wire=app.results[1], qubit=q1.result)),
+        (squin.wire.Wrap(wire=app.results[2], qubit=q2.result)),
+        (ret_none := func.ConstantNone()),
+        (func.Return(ret_none)),
+    ]
+
+    constructed_method = gen_func_from_stmts(stmts)
+    constructed_method.print()
+
+    squin_to_stim = squin_passes.SquinToStim(constructed_method.dialects)
+
+    with pytest.raises(ValueError):
+        squin_to_stim(constructed_method)
+
+
 # test_wire_measure_and_reset()
 # test_qubit_measure_and_reset()
 # test_wire_reset()
 
 # test_parallel_qubit_1q_application()
 # test_parallel_wire_1q_application()
 # test_parallel_control_gate_wire_application()
+
+# test_wire_apply_site_verification()