refactor tests to check equivilance and fixing more potential bugs

Author: weinbe58

src/bloqade/native/stdlib/broadcast.py

@@ -225,9 +225,9 @@ def u3(theta: float, phi: float, lam: float, qubits: ilist.IList[qubit.Qubit, An
         lam (float): Z rotations in decomposition (radians).
         qubits (ilist.IList[qubit.Qubit, Any]): Target qubits.
     """
-    rz(lam, qubits)
-    ry(theta, qubits)
-    rz(phi, qubits)
+    _u3_turns(
+        _radian_to_turn(theta), _radian_to_turn(phi), _radian_to_turn(lam), qubits
+    )
 
 
 N = TypeVar("N")

src/bloqade/rewrite/passes/callgraph.py

@@ -98,8 +98,8 @@ def unsafe_run(self, mt: ir.Method) -> RewriteResult:
         mt_map = {}
 
         cg = CallGraph(mt)
-        all_methods = set([mt])
-        all_methods.update(cg.edges.keys())
+        all_methods = set(cg.edges.keys())
+        all_methods.add(mt)
         for original_mt in all_methods:
             if original_mt is mt:
                 new_mt = original_mt

test/native/upstream/test_squin2native.py

@@ -76,7 +76,8 @@ def ghz_native(angle: float):
 
     AggressiveUnroll(ghz_native.dialects).fixpoint(ghz_native)
     test_utils.assert_nodes(
-        ghz_native_rewrite.callable_region, ghz_native.callable_region
+        ghz_native_rewrite.callable_region.blocks[0],
+        ghz_native.callable_region.blocks[0],
     )
 
 
@@ -92,10 +93,13 @@ def ghz_native(theta: float, phi: float, lam: float):
         qubits = squin.qalloc(1)
         native.u3(theta, phi, lam, qubits[0])
 
+    # unroll first to check that gete rewrites happen in ghz body
+    AggressiveUnroll(ghz.dialects).fixpoint(ghz)
     ghz_native_rewrite = SquinToNative().emit(ghz)
     AggressiveUnroll(ghz.dialects).fixpoint(ghz_native_rewrite)
 
     AggressiveUnroll(ghz_native.dialects).fixpoint(ghz_native)
     test_utils.assert_nodes(
-        ghz_native_rewrite.callable_region, ghz_native.callable_region
+        ghz_native_rewrite.callable_region.blocks[0],
+        ghz_native.callable_region.blocks[0],
     )