fixing bug in standard library

Author: weinbe58

src/bloqade/native/stdlib/broadcast.py

@@ -33,7 +33,7 @@ def _ry_turns(angle: float, qubits: ilist.IList[qubit.Qubit, Any]):
 
 @kernel
 def _rz_turns(angle: float, qubits: ilist.IList[qubit.Qubit, Any]):
-    native.r(0.5, angle, qubits)
+    native.rz(angle, qubits)
 
 
 @kernel
@@ -126,7 +126,7 @@ def rz(angle: float, qubits: ilist.IList[qubit.Qubit, Any]):
         angle (float): Rotation angle in radians.
         qubits (ilist.IList[qubit.Qubit, Any]): Target qubits.
     """
-    native.rz(_radian_to_turn(angle), qubits)
+    _rz_turns(_radian_to_turn(angle), qubits)
 
 
 @kernel

test/native/upstream/test_squin2native.py

@@ -2,10 +2,9 @@
 import pytest
 from kirin.analysis import callgraph
 
-from bloqade import squin, native
+from bloqade import squin, native, test_utils
 from bloqade.squin import gate
 from bloqade.pyqrack import StackMemorySimulator
-from bloqade.test_utils import assert_methods
 from bloqade.rewrite.passes import AggressiveUnroll
 from bloqade.native.dialects import gate as native_gate
 from bloqade.native.upstream import GateRule, SquinToNative
@@ -70,4 +69,6 @@ def ghz_native(angle: float):
 
     AggressiveUnroll(ghz_native.dialects).fixpoint(ghz_native)
     ghz_native.print()
-    assert_methods(ghz_native_rewrite, ghz_native)
+    test_utils.assert_nodes(
+        ghz_native_rewrite.callable_region, ghz_native.callable_region
+    )