def test_equality(result_str_1, result_str_2):
result_1 = AnnealingQuantumTaskResult.from_string(result_str_1)
result_2 = AnnealingQuantumTaskResult.from_string(result_str_1)
other_result = AnnealingQuantumTaskResult.from_string(result_str_2)
non_result = "not a quantum task result"
assert result_1 == result_2
assert result_1 is not result_2
assert result_1 != other_result
assert result_1 != non_result
def test_result_annealing(annealing_task):
_mock_metadata(annealing_task._aws_session, "COMPLETED")
_mock_s3(annealing_task._aws_session, MockS3.MOCK_S3_RESULT_ANNEALING)
expected = AnnealingQuantumTaskResult.from_string(
MockS3.MOCK_S3_RESULT_ANNEALING)
assert annealing_task.result() == expected
s3_bucket = annealing_task.metadata()["outputS3Bucket"]
s3_object_key = annealing_task.metadata()["outputS3Directory"]
annealing_task._aws_session.retrieve_s3_object_body.assert_called_with(
s3_bucket, f"{s3_object_key}/results.json")
def test_from_string(
result_str_1,
solutions,
values,
solution_counts,
variable_count,
problem_type,
task_metadata,
additional_metadata,
):
result = AnnealingQuantumTaskResult.from_string(result_str_1)
solutions = np.asarray(solutions, dtype=int)
values = np.asarray(values, dtype=float)
solution_counts = np.asarray(solution_counts, dtype=int)
assert result.variable_count == variable_count
assert result.problem_type == problem_type
assert result.task_metadata == task_metadata
assert result.additional_metadata == additional_metadata
np.testing.assert_equal(
result.record_array,
AnnealingQuantumTaskResult._create_record_array(
solutions, solution_counts, values),
)
def test_from_object_equal_to_from_string(result_str_1):
assert AnnealingQuantumTaskResult.from_object(
AnnealingTaskResult.parse_raw(result_str_1)
) == AnnealingQuantumTaskResult.from_string(result_str_1)
def test_from_string_solution_counts_empty_list(result_str_3, solutions):
result = AnnealingQuantumTaskResult.from_string(result_str_3)
np.testing.assert_equal(result.record_array.solution_count,
np.ones(len(solutions), dtype=int))