Skip to content

Commit fffd492

Browse files
committed
Fix tests
1 parent a35fb61 commit fffd492

File tree

1 file changed

+27
-27
lines changed

1 file changed

+27
-27
lines changed

test/pyqrack/test_squin.py

Lines changed: 27 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -33,8 +33,8 @@ def m():
3333

3434
target = PyQrack(3)
3535
result = target.run(m)
36-
assert isinstance(result, list)
37-
assert result == [0, 0, 0]
36+
assert isinstance(result, ilist.IList)
37+
assert result.data == [0, 0, 0]
3838

3939
@squin.kernel
4040
def measure_and_reset():
@@ -44,8 +44,8 @@ def measure_and_reset():
4444

4545
target = PyQrack(3)
4646
result = target.run(measure_and_reset)
47-
assert isinstance(result, list)
48-
assert result == [0, 0, 0]
47+
assert isinstance(result, ilist.IList)
48+
assert result.data == [0, 0, 0]
4949

5050

5151
def test_x():
@@ -54,11 +54,11 @@ def main():
5454
q = squin.qubit.new(1)
5555
x = squin.op.x()
5656
squin.qubit.apply(x, q)
57-
return squin.qubit.measure(q)
57+
return squin.qubit.measure(q[0])
5858

5959
target = PyQrack(1)
6060
result = target.run(main)
61-
assert result == [1]
61+
assert result == 1
6262

6363

6464
@pytest.mark.parametrize(
@@ -139,14 +139,14 @@ def main():
139139
x = squin.op.x()
140140
id = squin.op.mult(x, x)
141141
squin.qubit.apply(id, q)
142-
return squin.qubit.measure(q)
142+
return squin.qubit.measure(q[0])
143143

144144
main.print()
145145

146146
target = PyQrack(1)
147147
result = target.run(main)
148148

149-
assert result == [0]
149+
assert result == 0
150150

151151

152152
def test_kron():
@@ -161,7 +161,7 @@ def main():
161161
target = PyQrack(2)
162162
result = target.run(main)
163163

164-
assert result == [1, 1]
164+
assert result == ilist.IList([1, 1])
165165

166166

167167
def test_scale():
@@ -174,11 +174,11 @@ def main():
174174
s = squin.op.scale(x, 2)
175175

176176
squin.qubit.apply(s, q)
177-
return squin.qubit.measure(q)
177+
return squin.qubit.measure(q[0])
178178

179179
target = PyQrack(1)
180180
result = target.run(main)
181-
assert result == [1]
181+
assert result == 1
182182

183183

184184
def test_phase():
@@ -192,11 +192,11 @@ def main():
192192
squin.qubit.apply(p, q)
193193

194194
squin.qubit.apply(h, q)
195-
return squin.qubit.measure(q)
195+
return squin.qubit.measure(q[0])
196196

197197
target = PyQrack(1)
198198
result = target.run(main)
199-
assert result == [1]
199+
assert result == 1
200200

201201

202202
def test_sp():
@@ -221,11 +221,11 @@ def main2():
221221
sp = squin.op.spin_p()
222222
squin.qubit.apply(sn, q)
223223
squin.qubit.apply(sp, q)
224-
return squin.qubit.measure(q)
224+
return squin.qubit.measure(q[0])
225225

226226
target = PyQrack(1)
227227
result = target.run(main2)
228-
assert result == [0]
228+
assert result == 0
229229

230230

231231
def test_adjoint():
@@ -235,11 +235,11 @@ def main():
235235
x = squin.op.x()
236236
xadj = squin.op.adjoint(x)
237237
squin.qubit.apply(xadj, q)
238-
return squin.qubit.measure(q)
238+
return squin.qubit.measure(q[0])
239239

240240
target = PyQrack(1)
241241
result = target.run(main)
242-
assert result == [1]
242+
assert result == 1
243243

244244
@squin.kernel
245245
def adj_that_does_something():
@@ -266,35 +266,35 @@ def main_x():
266266
x = squin.op.x()
267267
r = squin.op.rot(x, math.pi)
268268
squin.qubit.apply(r, q)
269-
return squin.qubit.measure(q)
269+
return squin.qubit.measure(q[0])
270270

271271
target = PyQrack(1)
272272
result = target.run(main_x)
273-
assert result == [1]
273+
assert result == 1
274274

275275
@squin.kernel
276276
def main_y():
277277
q = squin.qubit.new(1)
278278
y = squin.op.y()
279279
r = squin.op.rot(y, math.pi)
280280
squin.qubit.apply(r, q)
281-
return squin.qubit.measure(q)
281+
return squin.qubit.measure(q[0])
282282

283283
target = PyQrack(1)
284284
result = target.run(main_y)
285-
assert result == [1]
285+
assert result == 1
286286

287287
@squin.kernel
288288
def main_z():
289289
q = squin.qubit.new(1)
290290
z = squin.op.z()
291291
r = squin.op.rot(z, math.pi)
292292
squin.qubit.apply(r, q)
293-
return squin.qubit.measure(q)
293+
return squin.qubit.measure(q[0])
294294

295295
target = PyQrack(1)
296296
result = target.run(main_z)
297-
assert result == [0]
297+
assert result == 0
298298

299299

300300
def test_broadcast():
@@ -307,7 +307,7 @@ def main():
307307

308308
target = PyQrack(3)
309309
result = target.run(main)
310-
assert result == [1, 1, 1]
310+
assert result == ilist.IList([1, 1, 1])
311311

312312
@squin.kernel
313313
def non_bc_error():
@@ -365,7 +365,7 @@ def broadcast_adjoint():
365365

366366
target = PyQrack(3)
367367
result = target.run(broadcast_adjoint)
368-
assert result == [0, 0, 0]
368+
assert result == ilist.IList([0, 0, 0])
369369

370370

371371
def test_projectors():
@@ -396,7 +396,7 @@ def main_p1():
396396
assert result == 1
397397

398398

399-
def test_clifford_str():
399+
def test_pauli_str():
400400
@squin.kernel
401401
def main():
402402
q = squin.qubit.new(3)
@@ -406,7 +406,7 @@ def main():
406406

407407
target = PyQrack(3)
408408
result = target.run(main)
409-
assert result == [1, 1, 1]
409+
assert result == ilist.IList([1, 1, 1])
410410

411411

412412
@pytest.mark.xfail

0 commit comments

Comments
 (0)