Skip to content

Commit da07d26

Browse files
committed
fix unit test, remove leftover print statements
1 parent 311fb4d commit da07d26

File tree

1 file changed

+3
-9
lines changed

1 file changed

+3
-9
lines changed

test/analysis/address/test_analysis.py

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,8 @@
11
from kirin import ir, types
2-
from bloqade import qasm2, squin
32
from kirin.passes import Fold
43
from kirin.dialects import py, func, ilist
4+
5+
from bloqade import qasm2, squin
56
from bloqade.analysis import address
67

78

@@ -74,10 +75,7 @@ def test_multiple_unwrap():
7475
stmts: list[ir.Statement] = [
7576
# Create qubit register
7677
(n_qubits := as_int(2)),
77-
# (qreg := qasm2.core.QRegNew(n_qubits=n_qubits.result)),
78-
(qreg := squin.qubit.New(n_qubits=n_qubits.result)),
79-
(idx0 := as_int(0)),
80-
# (q0 := py.GetItem(obj=qreg.result, index=idx0.result)),
78+
(qreg := qasm2.core.QRegNew(n_qubits=n_qubits.result)),
8179
# Get qubits out
8280
(idx0 := as_int(0)),
8381
(q0 := qasm2.core.QRegGet(reg=qreg.result, idx=idx0.result)),
@@ -117,8 +115,6 @@ def test_multiple_unwrap():
117115
fold_pass = Fold(squin_with_qasm_core)
118116
fold_pass(constructed_method)
119117

120-
constructed_method.print()
121-
122118
frame, _ = address.AddressAnalysis(constructed_method.dialects).run_analysis(
123119
constructed_method, no_raise=False
124120
)
@@ -190,8 +186,6 @@ def test_multiple_wire_apply():
190186
constructed_method, no_raise=False
191187
)
192188

193-
constructed_method.print(analysis=frame.entries)
194-
195189
address_wire_parent_qubit_0 = []
196190
address_wire_parent_qubit_1 = []
197191
address_types = frame.entries.values() # dict[SSAValue, Address]

0 commit comments

Comments
 (0)