jaunatisblue/final-qibotn
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Removed timing infrastructure from tests

Browse Source
This commit is contained in:
Vinitha Balachandran
2023-12-14 04:04:37 +08:00
parent b779227846
commit c593965a59
1 changed files with 8 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
tests/test_qasm_quimb_backend.py
View File

@@ -1,7 +1,5 @@
import copy import copy
import os import os
from timeit import default_timer as timer
import config import config
import numpy as np import numpy as np
import pytest import pytest
@@ -18,18 +16,10 @@ def create_init_state(nqubits):
def qibo_qft(nqubits, init_state, swaps): def qibo_qft(nqubits, init_state, swaps):
circ_qibo = QFT(nqubits, swaps) circ_qibo = QFT(nqubits, swaps)
state_vec = (circ_qibo(init_state)).state(numpy=True) state_vec = circ_qibo(init_state).state(numpy=True)
return circ_qibo, state_vec return circ_qibo, state_vec
def time(func):
start = timer()
res = func()
end = timer()
time = end - start
return time, res
@pytest.mark.parametrize("nqubits, tolerance, is_mps", @pytest.mark.parametrize("nqubits, tolerance, is_mps",
[(1, 1e-6, True), (2, 1e-6, False), (5, 1e-3, True), (10, 1e-3, False)]) [(1, 1e-6, True), (2, 1e-6, False), (5, 1e-3, True), (10, 1e-3, False)])
def test_eval(nqubits: int, tolerance: float, is_mps: bool): def test_eval(nqubits: int, tolerance: float, is_mps: bool):
@@ -45,19 +35,20 @@ def test_eval(nqubits: int, tolerance: float, is_mps: bool):
# Test qibo # Test qibo
qibo.set_backend(backend=config.qibo.backend, qibo.set_backend(backend=config.qibo.backend,
platform=config.qibo.platform) platform=config.qibo.platform)
qibo_time, (qibo_circ, result_sv) = time( #qibo_time, (qibo_circ, result_sv) = time(
lambda: qibo_qft(nqubits, init_state, swaps=True) #lambda: qibo_qft(nqubits, init_state, swaps=True)
) #)
qibo_circ, result_sv= qibo_qft(nqubits, init_state, swaps=True)
# Convert to qasm for other backends # Convert to qasm for other backends
qasm_circ = qibo_circ.to_qasm() qasm_circ = qibo_circ.to_qasm()
# Test quimb # Test quimb
quimb_time, result_tn = time( result_tn = qibotn.quimb.eval(
lambda: qibotn.quimb.eval(
qasm_circ, init_state_tn, is_mps, backend=config.quimb.backend qasm_circ, init_state_tn, is_mps, backend=config.quimb.backend
) )
)
assert np.allclose(result_sv, result_tn, assert np.allclose(result_sv, result_tn,
atol=tolerance), "Resulting dense vectors do not match" atol=tolerance), "Resulting dense vectors do not match"