add tests

Author: kaihsin

src/bloqade/stim/emit/impls.py

@@ -12,6 +12,6 @@ class EmitStimDebugMethods(MethodTable):
     def info(self, emit: EmitStimMain, frame: EmitStrFrame, stmt: Info):
 
         msg: str = frame.get(stmt.msg)
-        emit.writeln(frame, f"#{msg}")
+        emit.writeln(frame, f"# {msg}")
 
         return ()

test/stim/dialects/stim/emit/test_stim_debug.py

@@ -13,4 +13,4 @@ def test_debug_main():
 
     test_debug_main.print()
     out = codegen(test_debug_main)
-    assert out.strip() == "#debug message"
+    assert out.strip() == "# debug message"

test/stim/passes/test_squin_debug_to_stim.py

@@ -0,0 +1,44 @@
+import os
+
+from kirin import ir
+from kirin.dialects import py, debug
+
+from bloqade.squin import kernel
+from bloqade.stim.emit import EmitStimMain
+from bloqade.stim.passes import SquinToStimPass
+
+
+# Taken gratuitously from Kai's unit test
+def codegen(mt: ir.Method):
+    # method should not have any arguments!
+    emit = EmitStimMain()
+    emit.initialize()
+    emit.run(mt=mt, args=())
+    return emit.get_output()
+
+
+def as_int(value: int):
+    return py.constant.Constant(value=value)
+
+
+def as_float(value: float):
+    return py.constant.Constant(value=value)
+
+
+def load_reference_program(filename):
+    path = os.path.join(
+        os.path.dirname(__file__), "stim_reference_programs", "debug", filename
+    )
+    with open(path, "r") as f:
+        return f.read()
+
+
+def test_info():
+    @kernel
+    def test():
+        debug.info("debug message")
+        return
+
+    SquinToStimPass(test.dialects)(test)
+    base_stim_prog = load_reference_program("debug.stim")
+    assert codegen(test) == base_stim_prog.rstrip()