From 4fe183cff156ece27e889a697d3b619134888a97 Mon Sep 17 00:00:00 2001 From: Cody Wang Date: Fri, 28 Jun 2024 13:10:20 -0700 Subject: [PATCH] Update test_device_creation.py --- test/integ_tests/test_device_creation.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/test/integ_tests/test_device_creation.py b/test/integ_tests/test_device_creation.py index 540c09f61..70a73aae5 100644 --- a/test/integ_tests/test_device_creation.py +++ b/test/integ_tests/test_device_creation.py @@ -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] @@ -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 @@ -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