fix tests

Author: Roger-luo

src/bloqade/qasm2/dialects/glob.py

@@ -33,7 +33,8 @@ class GlobEmit(interp.MethodTable):
     @interp.impl(UGate)
     def ugate(self, emit: EmitQASM2Gate, frame: EmitQASM2Frame, stmt: UGate):
         registers = [
-            emit.assert_node(ast.Name, reg) for reg in frame.get(stmt.registers)
+            emit.assert_node(ast.Name, reg)
+            for reg in frame.get_casted(stmt.registers, ilist.IList)
         ]
         theta = emit.assert_node(ast.Expr, frame.get(stmt.theta))
         phi = emit.assert_node(ast.Expr, frame.get(stmt.phi))

src/bloqade/qasm2/passes/py2qasm.py

@@ -32,11 +32,12 @@ class _Py2QASM(RewriteRule):
 
     def rewrite_Statement(self, node: ir.Statement) -> RewriteResult:
         if isinstance(node, py.Constant):
-            if isinstance(node.value, int):
-                node.replace_by(expr.ConstInt(value=node.value))
+            value = node.value.unwrap()
+            if isinstance(value, int):
+                node.replace_by(expr.ConstInt(value=value))
                 return RewriteResult(has_done_something=True)
-            elif isinstance(node.value, float):
-                node.replace_by(expr.ConstFloat(value=node.value))
+            elif isinstance(value, float):
+                node.replace_by(expr.ConstFloat(value=value))
                 return RewriteResult(has_done_something=True)
         elif isinstance(node, py.BinOp):
             if (pystmt := self.BINARY_OPS.get(type(node))) is not None:

test/qasm2/emit/test_extended.py

@@ -44,7 +44,7 @@ def main():
     my_gate(qreg[0], qreg[1])
 
 
-# main.print()
+main.print()
 
 target = qasm2.emit.QASM2()
 ast = target.emit(main)