add tests

Author: kaihsin

src/bloqade/qasm2/emit/gate.py

@@ -1,4 +1,3 @@
-from typing import Any
 from dataclasses import field, dataclass
 
 from kirin import ir, types, interp
@@ -87,4 +86,3 @@ def emit_err(self, emit: EmitQASM2Gate, frame: EmitQASM2Frame, stmt):
     @interp.impl(func.ConstantNone)
     def ignore(self, emit: EmitQASM2Gate, frame: EmitQASM2Frame, stmt):
         return ()
-

src/bloqade/qasm2/emit/impls/__init__.py

@@ -1 +1 @@
-from . import noise_native as noise_native
+from . import noise_native as noise_native

src/bloqade/qasm2/emit/impls/noise_native.py

@@ -1,11 +1,13 @@
-
 from typing import Any
+
 from kirin import interp
 from kirin.dialects import ilist
+
 from bloqade.noise import native
 from bloqade.qasm2.parse import ast
 from bloqade.qasm2.emit.gate import EmitQASM2Gate, EmitQASM2Frame
 
+
 @native.dialect.register(key="emit.qasm2.gate")
 class NativeNoise(interp.MethodTable):
 

src/bloqade/qasm2/emit/target.py

@@ -11,9 +11,9 @@
 from bloqade.qasm2.passes.py2qasm import Py2QASM
 from bloqade.qasm2.passes.parallel import ParallelToUOp
 
+from . import impls as impls  # register the tables
 from .gate import EmitQASM2Gate
 from .main import EmitQASM2Main
-from . import impls as impls # register the tables
 
 
 class QASM2:

test/qasm2/emit/test_extended_noise.py

@@ -14,12 +14,11 @@ def main():
 
         qasm2.u(qreg[2], theta=0.1, phi=0.2, lam=0.3)
 
-
     main.print()
 
     target = qasm2.emit.QASM2(allow_noise=True)
     out = target.emit_str(main)
-    
+
     expected = """OPENQASM 2.0;
 include "qelib1.inc";
 qreg qreg[4];
@@ -40,13 +39,12 @@ def main():
         qreg = qasm2.qreg(4)
 
         qasm2.cx(qreg[0], qreg[1])
-        qasm2.parallel.u([qreg[2],qreg[3]], theta=0.1, phi=0.2, lam=0.3)
+        qasm2.parallel.u([qreg[2], qreg[3]], theta=0.1, phi=0.2, lam=0.3)
 
         noise.native.pauli_channel(qargs=[qreg[0], qreg[1]], px=0.1, py=0.2, pz=0.3)
 
         qasm2.u(qreg[2], theta=0.1, phi=0.2, lam=0.3)
 
-
     main.print()
 
     target = qasm2.emit.QASM2(allow_noise=True, allow_parallel=True)
@@ -65,5 +63,3 @@ def main():
 """
 
     assert out == expected
-
-