Skip to content

Commit ba2f04a

Browse files
weinbe58weinbe58
committed
fixing all tests
1 parent 1bf77c2 commit ba2f04a

7 files changed

+66
-66
lines changed

‎tests/test_batch.py

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -205,8 +205,8 @@ def test_metadata_filter_vector():
205205
filtered_batch_all = batch.filter_metadata(**filters)
206206

207207

208-
@patch("bloqade.ir.routine.braket.BraketBackend")
209-
@patch("bloqade.task.batch.np.random.permutation")
208+
@patch("bloqade_analog.ir.routine.braket.BraketBackend")
209+
@patch("bloqade_analog.task.batch.np.random.permutation")
210210
def test_remote_batch_task_metric(permutation, BraketBackend):
211211
backend = BraketBackend(
212212
device_arn="arn:aws:braket:us-east-1::device/qpu/quera/Aquila"
@@ -245,7 +245,7 @@ def test_remote_batch_task_metric(permutation, BraketBackend):
245245
assert batch.total_nshots == 60
246246

247247

248-
@patch("bloqade.ir.routine.braket.BraketBackend")
248+
@patch("bloqade_analog.ir.routine.braket.BraketBackend")
249249
def test_pull(BraketBackend):
250250
backend = BraketBackend(
251251
device_arn="arn:aws:braket:us-east-1::device/qpu/quera/Aquila"
@@ -276,7 +276,7 @@ def test_pull(BraketBackend):
276276
assert len(new_batch.tasks) == 6
277277

278278

279-
@patch("bloqade.ir.routine.braket.BraketBackend")
279+
@patch("bloqade_analog.ir.routine.braket.BraketBackend")
280280
def test_retrieve(BraketBackend):
281281
backend = BraketBackend(
282282
device_arn="arn:aws:braket:us-east-1::device/qpu/quera/Aquila"
@@ -320,7 +320,7 @@ def test_retrieve(BraketBackend):
320320
assert len(new_batch.tasks) == 6
321321

322322

323-
@patch("bloqade.ir.routine.braket.BraketBackend")
323+
@patch("bloqade_analog.ir.routine.braket.BraketBackend")
324324
def test_remove_invalid_tasks(BraketBackend):
325325
backend = BraketBackend(
326326
device_arn="arn:aws:braket:us-east-1::device/qpu/quera/Aquila"
@@ -350,7 +350,7 @@ def test_remove_invalid_tasks(BraketBackend):
350350
assert len(new_batch.tasks) == 0
351351

352352

353-
@patch("bloqade.ir.routine.braket.BraketBackend")
353+
@patch("bloqade_analog.ir.routine.braket.BraketBackend")
354354
def test_test_filters(BraketBackend):
355355
backend = BraketBackend(
356356
device_arn="arn:aws:braket:us-east-1::device/qpu/quera/Aquila"
@@ -400,7 +400,7 @@ def test_test_filters(BraketBackend):
400400
assert len(new_batch.tasks) == 6
401401

402402

403-
@patch("bloqade.ir.routine.braket.BraketBackend")
403+
@patch("bloqade_analog.ir.routine.braket.BraketBackend")
404404
def test_cancel_tasks(BraketBackend):
405405
backend = BraketBackend(
406406
device_arn="arn:aws:braket:us-east-1::device/qpu/quera/Aquila"
@@ -430,7 +430,7 @@ def test_cancel_tasks(BraketBackend):
430430
assert backend.cancel_task.call_count == 6
431431

432432

433-
@patch("bloqade.ir.routine.braket.BraketBackend")
433+
@patch("bloqade_analog.ir.routine.braket.BraketBackend")
434434
def test_resubmit_tasks(BraketBackend):
435435
backend = BraketBackend(
436436
device_arn="arn:aws:braket:us-east-1::device/qpu/quera/Aquila"
@@ -467,7 +467,7 @@ def test_resubmit_tasks(BraketBackend):
467467
)
468468

469469

470-
@patch("bloqade.ir.routine.braket.BraketBackend")
470+
@patch("bloqade_analog.ir.routine.braket.BraketBackend")
471471
def test_report(BraketBackend):
472472
backend = BraketBackend(
473473
device_arn="arn:aws:braket:us-east-1::device/qpu/quera/Aquila"
@@ -512,7 +512,7 @@ def test_report(BraketBackend):
512512
assert all([len(ele) == 0 for ele in report.counts()])
513513

514514

515-
@patch("bloqade.task.quera.QuEraBackend")
515+
@patch("bloqade_analog.task.quera.QuEraBackend")
516516
def test_braket_task(QuEraBackend):
517517
backend = QuEraBackend()
518518

@@ -569,7 +569,7 @@ def test_braket_task(QuEraBackend):
569569
assert backend.cancel_task.call_count == 1
570570

571571

572-
@patch("bloqade.task.braket.BraketBackend")
572+
@patch("bloqade_analog.task.braket.BraketBackend")
573573
def test_quera_task(BraketBackend):
574574
backend = BraketBackend()
575575

‎tests/test_braket_submission_backend.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ def get_task_ir():
3737
)
3838

3939

40-
@patch("bloqade.submission.braket.AwsDevice")
40+
@patch("bloqade_analog.submission.braket.AwsDevice")
4141
def test_braket_submit(*args, **kwargs):
4242
task_ir = get_task_ir()
4343

@@ -49,7 +49,7 @@ def test_braket_submit(*args, **kwargs):
4949
mock_aws_device.run.assert_called_once()
5050

5151

52-
@patch("bloqade.submission.braket.AwsDevice")
52+
@patch("bloqade_analog.submission.braket.AwsDevice")
5353
def test_add_braket_user_agent_invoked(*args, **kwargs):
5454
backend = bloqade_analog.submission.braket.BraketBackend()
5555
expected_user_agent = f"Bloqade/{bloqade_analog.__version__}"
@@ -62,8 +62,8 @@ def test_add_braket_user_agent_invoked(*args, **kwargs):
6262
@pytest.mark.skip(
6363
reason="removed implementation for validation because of issue with empty queue."
6464
)
65-
@patch("bloqade.submission.braket.AwsDevice")
66-
@patch("bloqade.submission.braket.AwsQuantumTask")
65+
@patch("bloqade_analog.submission.braket.AwsDevice")
66+
@patch("bloqade_analog.submission.braket.AwsQuantumTask")
6767
def test_braket_validate_task(*args, **kwargs):
6868
task_ir = get_task_ir()
6969

@@ -101,7 +101,7 @@ def test_braket_validate_task(*args, **kwargs):
101101
mock_aws_device.run.assert_called_once()
102102

103103

104-
@patch("bloqade.submission.braket.AwsQuantumTask")
104+
@patch("bloqade_analog.submission.braket.AwsQuantumTask")
105105
def test_braket_fetch(*args, **kwargs):
106106
backend = bloqade_analog.submission.braket.BraketBackend()
107107
mock_aws_quantum_task = bloqade_analog.submission.braket.AwsQuantumTask("task_id")
@@ -111,7 +111,7 @@ def test_braket_fetch(*args, **kwargs):
111111
mock_aws_quantum_task.result.assert_called_once()
112112

113113

114-
@patch("bloqade.submission.braket.AwsQuantumTask")
114+
@patch("bloqade_analog.submission.braket.AwsQuantumTask")
115115
def test_braket_cancel(*args, **kwargs):
116116
backend = bloqade_analog.submission.braket.BraketBackend()
117117
mock_aws_quantum_task = bloqade_analog.submission.braket.AwsQuantumTask("task_id")
@@ -121,7 +121,7 @@ def test_braket_cancel(*args, **kwargs):
121121
mock_aws_quantum_task.cancel.assert_called_once()
122122

123123

124-
@patch("bloqade.submission.braket.AwsQuantumTask")
124+
@patch("bloqade_analog.submission.braket.AwsQuantumTask")
125125
def test_braket_status(*args, **kwargs):
126126
backend = bloqade_analog.submission.braket.BraketBackend()
127127
mock_aws_quantum_task = bloqade_analog.submission.braket.AwsQuantumTask("task_id")

‎tests/test_costum_submission.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ def create_response(
1818
return response
1919

2020

21-
@patch("bloqade.ir.routine.quera.request")
21+
@patch("bloqade_analog.ir.routine.quera.request")
2222
def test_custom_submission(request_mock):
2323
body_template = '{{"token": "token", "body":{task_ir}}}'
2424

@@ -46,7 +46,7 @@ def test_custom_submission(request_mock):
4646
assert response_json["task_id"] == task_id
4747

4848

49-
@patch("bloqade.ir.routine.quera.request")
49+
@patch("bloqade_analog.ir.routine.quera.request")
5050
def test_custom_submission_error_missing_task_ir_key(request_mock):
5151
body_template = '{{"token": "token", "body":}}'
5252

@@ -68,7 +68,7 @@ def test_custom_submission_error_missing_task_ir_key(request_mock):
6868
program.quera.custom().submit(100, "https://my_service.test", body_template)
6969

7070

71-
@patch("bloqade.ir.routine.quera.request")
71+
@patch("bloqade_analog.ir.routine.quera.request")
7272
def test_custom_submission_error_malformed_template(request_mock):
7373
body_template = '{{"token": token", "body":}}'
7474

0 commit comments

Comments
 (0)