get tests to work with new annotation dialect

Author: johnzl-777

src/bloqade/stim/rewrite/get_record_util.py

@@ -2,7 +2,7 @@
 from kirin.dialects import py
 
 from bloqade.stim.dialects import auxiliary
-from bloqade.analysis.measure_id.lattice import MeasureIdBool, MeasureIdTuple
+from bloqade.analysis.measure_id.lattice import RawMeasureId, MeasureIdTuple
 
 
 def insert_get_records(
@@ -12,9 +12,9 @@ def insert_get_records(
     Insert GetRecord statements before the given node
     """
     get_record_ssas = []
-    for measure_id_bool in measure_id_tuple.data:
-        assert isinstance(measure_id_bool, MeasureIdBool)
-        target_rec_idx = (measure_id_bool.idx - 1) - meas_count_at_stmt
+    for measure_id in measure_id_tuple.data:
+        assert isinstance(measure_id, RawMeasureId)
+        target_rec_idx = (measure_id.idx - 1) - meas_count_at_stmt
         idx_stmt = py.constant.Constant(target_rec_idx)
         idx_stmt.insert_before(node)
         get_record_stmt = auxiliary.GetRecord(idx_stmt.result)

test/stim/passes/test_annotation_to_stim.py

@@ -68,12 +68,12 @@ def main():
 
         ms = squin.broadcast.measure(q)
 
-        if ms[0]:
+        if squin.is_one(ms[0]):
             squin.z(q[0])
             squin.broadcast.x([q[1], q[2], q[3]])
             squin.broadcast.z(q)
 
-        if ms[1]:
+        if squin.is_one(ms[1]):
             squin.x(q[0])
             squin.y(q[1])
 
@@ -99,7 +99,7 @@ def main():
 
         ms = squin.broadcast.measure(q)
 
-        if ms[0]:
+        if squin.is_one(ms[0]):
             squin.z(q[0])
         else:
             squin.x(q[0])
@@ -118,9 +118,9 @@ def main():
 
         ms = squin.broadcast.measure(q)
 
-        if ms[0]:
+        if squin.is_one(ms[0]):
             squin.z(q[0])
-            if ms[0]:
+            if squin.is_one(ms[0]):
                 squin.x(q[1])
 
         return