add in unit tests

johnzl-777 · johnzl-777 · commit 98be61d1bb2f · 2025-11-06T14:33:58.000-05:00
diff --git a/src/bloqade/annotate/stmts.py b/src/bloqade/annotate/stmts.py
@@ -11,7 +11,7 @@
 @statement
 class ConsumesMeasurementResults(ir.Statement):
     traits = frozenset({lowering.FromPythonCall()})
-    inputs: ir.SSAValue = info.argument(
+    measurements: ir.SSAValue = info.argument(
         ilist.IListType[MeasurementResultType, kirin_types.Any]
     )
 
diff --git a/src/bloqade/stim/rewrite/set_detector_to_stim.py b/src/bloqade/stim/rewrite/set_detector_to_stim.py
@@ -52,7 +52,7 @@ def rewrite_SetDetector(self, node: SetDetector) -> RewriteResult:
             coord_ssas.append(coord_stmt.result)
             coord_stmt.insert_before(node)
 
-        measure_ids = self.measure_id_frame.entries[node.inputs]
+        measure_ids = self.measure_id_frame.entries[node.measurements]
         assert isinstance(measure_ids, MeasureIdTuple)
 
         get_record_list = insert_get_records(
diff --git a/src/bloqade/stim/rewrite/set_observable_to_stim.py b/src/bloqade/stim/rewrite/set_observable_to_stim.py
@@ -36,7 +36,7 @@ def rewrite_SetObservable(self, node: SetObservable) -> RewriteResult:
         idx_stmt = auxiliary.ConstInt(value=0)
         idx_stmt.insert_before(node)
 
-        measure_ids = self.measure_id_frame.entries[node.inputs]
+        measure_ids = self.measure_id_frame.entries[node.measurements]
         assert isinstance(measure_ids, MeasureIdTuple)
 
         get_record_list = insert_get_records(
diff --git a/test/stim/passes/stim_reference_programs/annotate/broadcast_with_alias.stim b/test/stim/passes/stim_reference_programs/annotate/broadcast_with_alias.stim
@@ -0,0 +1,2 @@
+
+X 0 1
diff --git a/test/stim/passes/stim_reference_programs/annotate/clean_surface_code.stim b/test/stim/passes/stim_reference_programs/annotate/clean_surface_code.stim
@@ -0,0 +1,38 @@
+
+RZ 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16
+H 13 14 15 16
+CX 13 14 16 1 3 7 0 2 4 10 11 12
+CX 13 14 16 2 4 8 3 5 7 10 11 12
+CX 13 15 16 0 4 6 1 3 5 9 10 11
+CX 13 15 16 1 5 7 4 6 8 9 10 11
+H 13 14 15 16
+MZ(0.00000000) 13 14 15 16
+MZ(0.00000000) 9 10 11 12
+RZ 13 14 15 16 9 10 11 12
+DETECTOR(0, 0) rec[-4]
+DETECTOR(0, 0) rec[-3]
+DETECTOR(0, 0) rec[-2]
+DETECTOR(0, 0) rec[-1]
+H 13 14 15 16
+CX 13 14 16 1 3 7 0 2 4 10 11 12
+CX 13 14 16 2 4 8 3 5 7 10 11 12
+CX 13 15 16 0 4 6 1 3 5 9 10 11
+CX 13 15 16 1 5 7 4 6 8 9 10 11
+H 13 14 15 16
+MZ(0.00000000) 13 14 15 16
+MZ(0.00000000) 9 10 11 12
+RZ 13 14 15 16 9 10 11 12
+DETECTOR(0, 0) rec[-4] rec[-12]
+DETECTOR(0, 0) rec[-3] rec[-11]
+DETECTOR(0, 0) rec[-2] rec[-10]
+DETECTOR(0, 0) rec[-1] rec[-9]
+DETECTOR(0, 0) rec[-8] rec[-16]
+DETECTOR(0, 0) rec[-7] rec[-15]
+DETECTOR(0, 0) rec[-6] rec[-14]
+DETECTOR(0, 0) rec[-5] rec[-13]
+MZ(0.00000000) 0 1 2 3 4 5 6 7 8
+DETECTOR(0, 0) rec[-9] rec[-8] rec[-13]
+DETECTOR(0, 0) rec[-8] rec[-7] rec[-5] rec[-4] rec[-12]
+DETECTOR(0, 0) rec[-6] rec[-5] rec[-3] rec[-2] rec[-11]
+DETECTOR(0, 0) rec[-2] rec[-1] rec[-10]
+OBSERVABLE_INCLUDE(0) rec[-9] rec[-8] rec[-7]
diff --git a/test/stim/passes/stim_reference_programs/annotate/detector_coords_as_args.stim b/test/stim/passes/stim_reference_programs/annotate/detector_coords_as_args.stim
@@ -0,0 +1,9 @@
+
+MZ(0.00000000) 0 1
+DETECTOR(0, 1) rec[-2] rec[-1]
+DETECTOR(3, 4) rec[-2] rec[-1]
+DETECTOR(0, 5.00000000) rec[-2] rec[-1]
+DETECTOR(5.00000000, 3) rec[-2] rec[-1]
+DETECTOR(1, 2, 5.00000000) rec[-2] rec[-1]
+DETECTOR(1, 2) rec[-2] rec[-1]
+DETECTOR(1, 2, 5.00000000) rec[-2] rec[-1]
diff --git a/test/stim/passes/stim_reference_programs/annotate/kernel_base_op.stim b/test/stim/passes/stim_reference_programs/annotate/kernel_base_op.stim
@@ -0,0 +1,6 @@
+
+H 1 3 2
+MZ(0.00000000) 1 3 2
+DETECTOR(0, 0) rec[-3]
+DETECTOR(0, 0) rec[-2]
+DETECTOR(0, 0) rec[-1]
diff --git a/test/stim/passes/stim_reference_programs/annotate/linear_program_rewrite.stim b/test/stim/passes/stim_reference_programs/annotate/linear_program_rewrite.stim
@@ -0,0 +1,11 @@
+
+X 0
+Y 1
+Z 2
+CX 0 1
+CX 0 1 2 3
+Z 0 1 2 3
+MZ(0.00000000) 0 1 2 3
+DETECTOR(0.00000000, 0.00000000) rec[-4] rec[-3]
+DETECTOR(1.00000000, 0.00000000) rec[-3] rec[-2]
+OBSERVABLE_INCLUDE(0) rec[-2]
diff --git a/test/stim/passes/stim_reference_programs/annotate/measure_desugar.stim b/test/stim/passes/stim_reference_programs/annotate/measure_desugar.stim
@@ -0,0 +1,6 @@
+
+MZ(0.00000000) 0
+MZ(0.00000000) 0
+MZ(0.00000000) 0
+MZ(0.00000000) 0
+MZ(0.00000000) 0
diff --git a/test/stim/passes/stim_reference_programs/annotate/nested_for.stim b/test/stim/passes/stim_reference_programs/annotate/nested_for.stim
@@ -0,0 +1,7 @@
+
+MZ(0.00000000) 0 1
+DETECTOR(0, 0) rec[-2]
+DETECTOR(0, 0) rec[-1]
+MZ(0.00000000) 0 1
+DETECTOR(0, 0) rec[-2]
+DETECTOR(0, 0) rec[-1]
diff --git a/test/stim/passes/stim_reference_programs/annotate/no_kernel_base_op.stim b/test/stim/passes/stim_reference_programs/annotate/no_kernel_base_op.stim
@@ -0,0 +1,6 @@
+
+X 0 2 1
+MZ(0.00000000) 0 2 1
+DETECTOR(0, 0) rec[-3]
+DETECTOR(0, 0) rec[-2]
+DETECTOR(0, 0) rec[-1]
diff --git a/test/stim/passes/stim_reference_programs/annotate/rep_code.stim b/test/stim/passes/stim_reference_programs/annotate/rep_code.stim
@@ -0,0 +1,23 @@
+
+RZ 0 1 2 3 4
+CX 0 1 2 3
+CX 2 1 4 3
+MZ(0.00000000) 1 3
+DETECTOR(0, 0) rec[-2]
+DETECTOR(0, 0) rec[-1]
+CX 0 1 2 3
+CX 2 1 4 3
+MZ(0.00000000) 1 3
+DETECTOR(0, 0) rec[-4] rec[-2]
+DETECTOR(0, 0) rec[-3] rec[-1]
+CX 0 1 2 3
+CX 2 1 4 3
+DEPOLARIZE2(0.01000000) 0 1 2 3
+I_ERROR[loss](0.00100000) 0 1 2 3 4
+MZ(0.00000000) 1 3
+DETECTOR(0, 0) rec[-4] rec[-2]
+DETECTOR(0, 0) rec[-3] rec[-1]
+MZ(0.00000000) 0 2 4
+DETECTOR(0, 0) rec[-3] rec[-2] rec[-5]
+DETECTOR(0, 0) rec[-2] rec[-1] rec[-4]
+OBSERVABLE_INCLUDE(0) rec[-1]
diff --git a/test/stim/passes/stim_reference_programs/annotate/set_detector_with_alias.stim b/test/stim/passes/stim_reference_programs/annotate/set_detector_with_alias.stim
@@ -0,0 +1,3 @@
+
+MZ(0.00000000) 0 1
+DETECTOR(0, 0) rec[-2] rec[-1]
diff --git a/test/stim/passes/stim_reference_programs/annotate/simple_if_rewrite.stim b/test/stim/passes/stim_reference_programs/annotate/simple_if_rewrite.stim
@@ -0,0 +1,10 @@
+
+MZ(0.00000000) 0 1 2 3
+CZ rec[-4] 0
+CX rec[-4] 1 rec[-4] 2 rec[-4] 3
+CZ rec[-4] 0 rec[-4] 1 rec[-4] 2 rec[-4] 3
+CX rec[-3] 0
+CY rec[-3] 1
+MZ(0.00000000) 0 1 2 3
+DETECTOR(0.00000000, 0.00000000) rec[-4] rec[-3]
+OBSERVABLE_INCLUDE(0) rec[-2]
diff --git a/test/stim/passes/test_annotation_to_stim.py b/test/stim/passes/test_annotation_to_stim.py
diff --git a/test/stim/passes/test_code_basic_operations.py b/test/stim/passes/test_code_basic_operations.py

Original file line number	Diff line number	Diff line change
`@@ -11,7 +11,7 @@`
`11`	`11`	`@statement`
`12`	`12`	`class ConsumesMeasurementResults(ir.Statement):`
`13`	`13`	`traits = frozenset({lowering.FromPythonCall()})`
`14`		`- inputs: ir.SSAValue = info.argument(`
	`14`	`+ measurements: ir.SSAValue = info.argument(`
`15`	`15`	`ilist.IListType[MeasurementResultType, kirin_types.Any]`
`16`	`16`	`)`
`17`	`17`