From df3bdccb06b5409fe3d9d49123dbdab43ee295fa Mon Sep 17 00:00:00 2001 From: Andrea Date: Fri, 31 Jan 2025 14:56:29 +0400 Subject: [PATCH] refactor: Make test_metabackend_list_available run on gpu as well --- src/qibojit/tests/test_backends.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/qibojit/tests/test_backends.py b/src/qibojit/tests/test_backends.py index 5a73a4e..9df653a 100644 --- a/src/qibojit/tests/test_backends.py +++ b/src/qibojit/tests/test_backends.py @@ -3,6 +3,8 @@ from qibojit.backends import MetaBackend +from .conftest import AVAILABLE_BACKENDS + def test_device_setter(backend): if backend.platform == "numba": @@ -120,5 +122,5 @@ def test_backend_eigh_sparse(backend, sparse_type, k): def test_metabackend_list_available(): - available_backends = dict(zip(("numba", "cupy", "cuquantum"), (True, False, False))) + available_backends = {backend: True for backend in AVAILABLE_BACKENDS} assert MetaBackend().list_available() == available_backends