From 725c92e75dc73c3e0b6d315d79f3089c961c10ec Mon Sep 17 00:00:00 2001 From: Liwei Yang Date: Wed, 19 Apr 2023 15:12:08 +0800 Subject: [PATCH] Remove unused imports, format it using black --- tests/test_cuquantum_cutensor_backend.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/tests/test_cuquantum_cutensor_backend.py b/tests/test_cuquantum_cutensor_backend.py index 30823ef..74cf2e1 100644 --- a/tests/test_cuquantum_cutensor_backend.py +++ b/tests/test_cuquantum_cutensor_backend.py @@ -1,5 +1,3 @@ -import copy -import os from timeit import default_timer as timer import config @@ -22,18 +20,22 @@ def time(func): time = end - start return time, res + @pytest.mark.gpu @pytest.mark.parametrize("nqubits", [1, 2, 5, 10]) def test_eval(nqubits: int): import qibotn.cutn # Test qibo - qibo.set_backend(backend=config.qibo.backend, platform=config.qibo.platform) - qibo_time, (qibo_circ, result_sv) = time(lambda: qibo_qft(nqubits, swaps=True)) + qibo.set_backend(backend=config.qibo.backend, + platform=config.qibo.platform) + qibo_time, (qibo_circ, result_sv) = time( + lambda: qibo_qft(nqubits, swaps=True)) # Test Cuquantum data_type = "complex128" - cutn_time, result_tn = time(lambda: qibotn.cutn.eval(qibo_circ,data_type)) + cutn_time, result_tn = time(lambda: qibotn.cutn.eval(qibo_circ, data_type)) assert 1e-2 * qibo_time < cutn_time < 1e2 * qibo_time - assert np.allclose(result_sv, result_tn), "Resulting dense vectors do not match" + assert np.allclose( + result_sv, result_tn), "Resulting dense vectors do not match"