def test_quantum_job_constructor_explicit_session(mock_session,
quantum_job_arn, job_region):
aws_session_mock = Mock(braket_client=Mock(meta=Mock(
region_name=job_region)))
job = AwsQuantumJob(quantum_job_arn, aws_session_mock)
assert job._aws_session == aws_session_mock
assert job.arn == quantum_job_arn
mock_session.assert_not_called()
def test_equality(quantum_job_arn, aws_session, job_region):
new_aws_session = Mock(braket_client=Mock(meta=Mock(
region_name=job_region)))
quantum_job_1 = AwsQuantumJob(quantum_job_arn, aws_session)
quantum_job_2 = AwsQuantumJob(quantum_job_arn, aws_session)
quantum_job_3 = AwsQuantumJob(quantum_job_arn, new_aws_session)
other_quantum_job = AwsQuantumJob(
"arn:aws:braket:us-west-2:875981177017:job/other-job", aws_session)
non_quantum_job = quantum_job_1.arn
assert quantum_job_1 == quantum_job_2
assert quantum_job_1 == quantum_job_3
assert quantum_job_1 is not quantum_job_2
assert quantum_job_1 is not quantum_job_3
assert quantum_job_1 is quantum_job_1
assert quantum_job_1 != other_quantum_job
assert quantum_job_1 != non_quantum_job
def test_cancel_job(quantum_job_arn, aws_session,
generate_cancel_job_response):
cancellation_status = "CANCELLING"
aws_session.cancel_job.return_value = generate_cancel_job_response(
cancellationStatus=cancellation_status)
quantum_job = AwsQuantumJob(quantum_job_arn, aws_session)
status = quantum_job.cancel()
aws_session.cancel_job.assert_called_with(quantum_job_arn)
assert status == cancellation_status
def test_quantum_job_constructor_default_session(aws_session_mock,
mock_session, arn,
expected_region):
mock_boto_session = Mock()
aws_session_mock.return_value = Mock()
mock_session.return_value = mock_boto_session
job = AwsQuantumJob(arn)
mock_session.assert_called_with(region_name=expected_region)
aws_session_mock.assert_called_with(boto_session=mock_boto_session)
assert job.arn == arn
assert job._aws_session == aws_session_mock.return_value
def test_name(quantum_job_arn, quantum_job_name, aws_session):
quantum_job = AwsQuantumJob(quantum_job_arn, aws_session)
assert quantum_job.name == quantum_job_name
def test_arn(quantum_job_arn, aws_session):
quantum_job = AwsQuantumJob(quantum_job_arn, aws_session)
assert quantum_job.arn == quantum_job_arn
def test_quantum_job_constructor_invalid_region(aws_session):
arn = "arn:aws:braket:unknown-region:875981177017:job/quantum_job_name"
AwsQuantumJob(arn, aws_session)
def quantum_job(quantum_job_arn, aws_session):
return AwsQuantumJob(quantum_job_arn, aws_session)
def test_quantum_job_constructor_invalid_region(aws_session):
region_mismatch = "The aws session region does not match the region for the supplied arn."
arn = "arn:aws:braket:unknown-region:875981177017:job/quantum_job_name"
with pytest.raises(ValueError, match=region_mismatch):
AwsQuantumJob(arn, aws_session)