Removing QuantumJob

Qiskit · Jun 21, 2018 · 290e28d · 290e28d
1 parent 3eea61c
commit 290e28d
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 19 deletions.
diff --git a/test/python/_mockutils.py b/test/python/_mockutils.py
@@ -136,25 +136,23 @@ def error(self):
         return self._future.exception(timeout=0)
 
 
-class FakeQJob():
-    """Fakes qiskit._quantumjob.QuantumJob instances."""
-    def __init__(self):
-        self.backend = FakeBackend()
-        self.qobj = {
-            'id': 'test-id',
+def new_fake_qobj():
+    """Creates a fake qobj dictionary."""
+    return {
+        'id': 'test-id',
+        'config': {
+            'backend_name': 'test-backend',
+            'shots': 1024,
+            'max_credits': 100
+        },
+        'circuits': [{
+            'compiled_circuit_qasm': 'fake-code',
             'config': {
-                'backend_name': self.backend.name,
-                'shots': 1024,
-                'max_credits': 100
+                'seed': 123456
             },
-            'circuits': [{
-                'compiled_circuit_qasm': 'fake-code',
-                'config': {
-                    'seed': 123456
-                },
-                'compiled_circuit': {}
-            }]
-        }
+            'compiled_circuit': {}
+        }]
+    }
 
 
 class FakeBackend():

diff --git a/test/python/test_ibmqjob_states.py b/test/python/test_ibmqjob_states.py
@@ -16,7 +16,7 @@
 from qiskit.backends.jobstatus import JobStatus
 from qiskit.backends.ibmq.ibmqjob import IBMQJob, IBMQJobError
 from .common import QiskitTestCase
-from ._mockutils import FakeQJob
+from ._mockutils import new_fake_qobj
 
 
 class TestIBMQJobStates(QiskitTestCase):
@@ -221,7 +221,7 @@ def run_with_api(self, api):
         """Creates a new `IBMQJob` instance running with the provided API
         object."""
         self._current_api = api
-        self._current_qjob = IBMQJob(FakeQJob(), api, False)
+        self._current_qjob = IBMQJob(new_fake_qobj(), api, False)
         return self._current_qjob