Refactor Q8SBackend initialization to allow optional client parameter…

… and update tests accordingly
vstirbu · Nov 15, 2024 · 50f707d · 50f707d
1 parent f03ea78
commit 50f707d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/app/qiskit/backend.py b/app/qiskit/backend.py
@@ -10,10 +10,13 @@
 class Q8SBackend(BackendV2, ABC):
     _client = None
 
-    def __init__(self, **kwargs):
+    def __init__(self, client: Client = None, **kwargs):
         super().__init__(**kwargs)
 
-        self._client = Client()
+        if client:
+            self._client = client
+        else:
+            self._client = Client()
 
     @classmethod
     def _default_options(self) -> Options:

diff --git a/app/test_main.py b/app/test_main.py
@@ -30,7 +30,7 @@ def test_run_circuit_invalid():
 
 
 def test_run_circuit_from_backend():
-    simulator = Q8SBackend()
+    simulator = Q8SBackend(client=client)
 
     job = simulator.run(circuit)