Add unit test with multiple corr errors and move roundtrip test

Author: rafaelha

test/stim/dialects/stim/emit/test_stim_noise.py

@@ -1,6 +1,16 @@
 from bloqade import stim
+from bloqade.stim.emit import EmitStimMain
+from bloqade.stim.parse import loads
+from bloqade.stim.dialects import noise
 
-from .base import codegen
+emit = EmitStimMain()
+
+
+def codegen(mt):
+    # method should not have any arguments!
+    emit.initialize()
+    emit.run(mt=mt, args=())
+    return emit.get_output()
 
 
 def test_noise():
@@ -49,3 +59,39 @@ def test_correlated_qubit_loss():
 
     out = codegen(test_correlated_qubit_loss)
     assert out.strip() == "I_ERROR[correlated_loss:0](0.10000000) 0 1 2"
+
+
+def test_correlated_qubit_loss_multiple():
+
+    @stim.main
+    def test_correlated_qubit_loss_multiple():
+        stim.correlated_qubit_loss(probs=(0.1,), targets=(0, 1))
+        stim.correlated_qubit_loss(probs=(0.1,), targets=(2, 3))
+
+    for i in range(2):  # repeat the test to ensure the identifier is reset each time
+        out = codegen(test_correlated_qubit_loss_multiple).strip()
+        print(out)
+        assert (
+            out.strip()
+            == "I_ERROR[correlated_loss:0](0.10000000) 0 1\n"
+            + "I_ERROR[correlated_loss:1](0.10000000) 2 3"
+        )
+
+
+def test_correlated_qubit_codegen_roundtrip():
+    @stim.main
+    def test():
+        stim.correlated_qubit_loss(probs=(0.1,), targets=(0, 1, 2))
+        stim.qubit_loss(probs=(0.2,), targets=(2,))
+        stim.correlated_qubit_loss(probs=(0.3,), targets=(3, 4))
+
+    stim_str = codegen(test)
+
+    mt = loads(
+        stim_str,
+        nonstim_noise_ops={
+            "loss": noise.QubitLoss,
+            "correlated_loss": noise.CorrelatedQubitLoss,
+        },
+    )
+    assert codegen(mt) == stim_str

test/stim/passes/test_squin_noise_to_stim.py

@@ -10,11 +10,9 @@
 from bloqade.squin import noise, kernel
 from bloqade.types import Qubit, QubitType
 from bloqade.stim.emit import EmitStimMain
-from bloqade.stim.parse import loads
 from bloqade.stim.passes import SquinToStimPass, flatten
 from bloqade.stim.rewrite import SquinNoiseToStim
 from bloqade.squin.rewrite import WrapAddressAnalysis
-from bloqade.stim.dialects import noise as stim_noise
 from bloqade.analysis.address import AddressAnalysis
 
 
@@ -244,29 +242,6 @@ def test():
     assert codegen(test) == expected
 
 
-def test_correlated_qubit_loss_codegen_roundtrip():
-
-    @kernel
-    def test():
-        q = sq.qalloc(4)
-        sq.correlated_qubit_loss(0.1, qubits=q[:2])
-        sq.qubit_loss(0.2, qubit=q[2])
-        sq.broadcast.qubit_loss(0.3, qubits=q)
-        sq.broadcast.correlated_qubit_loss(0.1, qubits=[q[:2], q[2:]])
-
-    SquinToStimPass(test.dialects)(test)
-    stim_str = codegen(test)
-
-    mt = loads(
-        stim_str,
-        nonstim_noise_ops={
-            "loss": stim_noise.QubitLoss,
-            "correlated_loss": stim_noise.CorrelatedQubitLoss,
-        },
-    )
-    assert codegen(mt) == stim_str
-
-
 def test_correlated_qubit_loss_codegen_with_offset():
 
     @kernel