Skip to content

Commit

Permalink
Update test_device_creation.py
Browse files Browse the repository at this point in the history
  • Loading branch information
speller26 authored Jun 28, 2024
1 parent 59d9151 commit 4fe183c
Showing 1 changed file with 3 additions and 5 deletions.
8 changes: 3 additions & 5 deletions test/integ_tests/test_device_creation.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,12 +20,11 @@
RIGETTI_ARN = "arn:aws:braket:us-west-1::device/qpu/rigetti/Aspen-M-3"
IONQ_ARN = "arn:aws:braket:us-east-1::device/qpu/ionq/Harmony"
SIMULATOR_ARN = "arn:aws:braket:::device/quantum-simulator/amazon/sv1"
OQC_ARN = "arn:aws:braket:eu-west-2::device/qpu/oqc/Lucy"
PULSE_ARN = "arn:aws:braket:us-west-1::device/qpu/rigetti/Aspen-M-3"


@pytest.mark.parametrize(
"arn", [(RIGETTI_ARN), (IONQ_ARN), (OQC_ARN), (SIMULATOR_ARN), (PULSE_ARN)]
"arn", [(RIGETTI_ARN), (IONQ_ARN), (SIMULATOR_ARN), (PULSE_ARN)]
)
def test_device_creation(arn, created_braket_devices):
device = created_braket_devices[arn]
Expand All @@ -48,10 +47,9 @@ def test_device_across_regions(aws_session, created_braket_devices):
# assert QPUs across different regions can be created using the same aws_session
created_braket_devices[RIGETTI_ARN]
created_braket_devices[IONQ_ARN]
created_braket_devices[OQC_ARN]


@pytest.mark.parametrize("arn", [(RIGETTI_ARN), (IONQ_ARN), (OQC_ARN), (SIMULATOR_ARN)])
@pytest.mark.parametrize("arn", [(RIGETTI_ARN), (IONQ_ARN), (SIMULATOR_ARN)])
def test_get_devices_arn(arn):
results = AwsDevice.get_devices(arns=[arn])
assert results[0].arn == arn
Expand All @@ -77,7 +75,7 @@ def test_get_devices_others():

def test_get_devices_all(braket_devices):
result_arns = [result.arn for result in braket_devices]
for arn in [RIGETTI_ARN, IONQ_ARN, SIMULATOR_ARN, OQC_ARN]:
for arn in [RIGETTI_ARN, IONQ_ARN, SIMULATOR_ARN]:
assert arn in result_arns


Expand Down

0 comments on commit 4fe183c

Please sign in to comment.