add mock for Quantum_Neuron

amazon-braket · Aug 12, 2023 · b782a59 · b782a59
1 parent c0c2364
commit b782a59
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/test/integ_tests/default_mocks/default_mocks.py b/test/integ_tests/default_mocks/default_mocks.py
@@ -1,6 +1,9 @@
 def pre_run_inject(mock_utils):
+    print("This is inside default_mocks.py")
+
     mocker = mock_utils.Mocker()
     mock_utils.mock_default_device_calls(mocker)
+    mocker.set_create_job_result({"jobArn": "arn:aws:braket:us-east-1:000000:job/TestARN"})
 
 
 def post_run(tb):

diff --git a/test/integ_tests/textbook/Quantum_Neuron_mocks.py b/test/integ_tests/textbook/Quantum_Neuron_mocks.py
@@ -1,4 +1,6 @@
 def pre_run_inject(mock_utils):
+    print("This is inside Quantum_Neuron_mocks.py")
+
     mocker = mock_utils.Mocker()
     mock_utils.mock_default_device_calls(mocker)
     mocker.set_create_job_result({"jobArn": "arn:aws:braket:us-east-1:000000:job/TestARN"})