Refactor: Remove unused import and add reset method to Emit classes

Author: zhenrongliew

src/bloqade/native/upstream/__init__.py

@@ -1,5 +1,4 @@
 from .squin2native import (
     GateRule as GateRule,
     SquinToNative as SquinToNative,
-    SquinToNativePass as SquinToNativePass,
 )

src/bloqade/qasm2/emit/base.py

@@ -82,3 +82,6 @@ def assert_node(
         if not isinstance(node, typ):
             raise TypeError(f"expected {typ}, got {type(node)}")
         return node
+
+    def reset(self):
+        pass

src/bloqade/stim/emit/stim_str.py

@@ -48,6 +48,10 @@ def get_attribute(self, frame: EmitStimFrame, node: ir.Attribute) -> str:
             raise ValueError(f"Method not found for node: {node}")
         return method(self, frame, node)
 
+    def reset(self):
+        self.io.truncate(0)
+        self.io.seek(0)
+
 
 @func.dialect.register(key="emit.stim")
 class FuncEmit(interp.MethodTable):

test/stim/dialects/stim/test_stim_circuits.py

@@ -58,6 +58,9 @@ def test_controlled_two_qubit_gates():
     # print(interp.get_output())
 
 
+test_gates()
+
+
 def test_noise():
     @stim.main
     def test_depolarize():

test/stim/passes/test_squin_debug_to_stim.py

@@ -1,8 +1,10 @@
+import io
 import os
 
 from kirin import ir
 from kirin.dialects import py, debug
 
+from bloqade import stim
 from bloqade.squin import kernel
 from bloqade.stim.emit import EmitStimMain
 from bloqade.stim.passes import SquinToStimPass
@@ -11,10 +13,11 @@
 # Taken gratuitously from Kai's unit test
 def codegen(mt: ir.Method):
     # method should not have any arguments!
-    emit = EmitStimMain()
+    buf = io.StringIO()
+    emit = EmitStimMain(dialects=stim.main, io=buf)
     emit.initialize()
-    emit.run(mt=mt, args=())
-    return emit.get_output()
+    emit.run(mt)
+    return buf.getvalue().strip()
 
 
 def as_int(value: int):