test: added test for list_available

qiboteam · Feb 27, 2024 · 814ee43 · 814ee43
1 parent f66bd12
commit 814ee43
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/src/qibojit/backends/__init__.py b/src/qibojit/backends/__init__.py
@@ -5,12 +5,12 @@
 
 QibojitBackend = NumbaBackend | CupyBackend | CuQuantumBackend
 
+PLATFORMS = ("numba", "cupy", "cuquantum")
+
 
 class MetaBackend:
  """Meta-backend class which takes care of loading the qibojit backends."""
 
- PLATFORMS = ("numba", "cupy", "cuquantum")
-
  @staticmethod
  def load(platform: str) -> QibojitBackend:
  """Loads the backend.

diff --git a/src/qibojit/tests/test_backends.py b/src/qibojit/tests/test_backends.py
@@ -1,6 +1,8 @@
 import numpy as np
 import pytest
 
+from qibojit.backends import MetaBackend
+
 
 def test_device_setter(backend):
  if backend.platform == "numba":
@@ -114,3 +116,8 @@ def test_backend_eigh_sparse(backend, sparse_type, k):
  eigvals1 = backend.to_numpy(eigvals1)
  eigvals2 = backend.to_numpy(eigvals2)
  backend.assert_allclose(sorted(eigvals1), sorted(eigvals2))
+
+
+def test_metabackend_list_available():
+ available_backends = dict(zip(("numba", "cupy", "cuquantum"), (True, False, False)))
+ assert MetaBackend().list_available() == available_backends