From b782a5990cd18bbd97b113e897034687619fde4f Mon Sep 17 00:00:00 2001 From: YutoTakaki Date: Sat, 12 Aug 2023 03:33:42 +0000 Subject: [PATCH] add mock for Quantum_Neuron --- test/integ_tests/default_mocks/default_mocks.py | 3 +++ test/integ_tests/textbook/Quantum_Neuron_mocks.py | 2 ++ 2 files changed, 5 insertions(+) diff --git a/test/integ_tests/default_mocks/default_mocks.py b/test/integ_tests/default_mocks/default_mocks.py index 523d961f..65856b67 100644 --- 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 index f525f4ba..6b3f652e 100644 --- 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"})