def test_sampler():
service = ionq.Service(remote_host='http://example.com', api_key='key')
mock_client = mock.MagicMock()
service._client = mock_client
job_dict = {
'id': '1',
'status': 'completed',
'qubits': '1',
'target': 'qpu',
'metadata': {
'shots': 4,
'measurement0': f'a{chr(31)}0'
},
'data': {
'histogram': {
'0': '0.25',
'1': '0.75'
}
},
}
mock_client.create_job.return_value = job_dict
mock_client.get_job.return_value = job_dict
sampler = service.sampler(target='qpu', seed=10)
q0 = cirq.LineQubit(0)
circuit = cirq.Circuit(cirq.X(q0), cirq.measure(q0, key='a'))
results = sampler.sample(program=circuit, repetitions=4)
pd.testing.assert_frame_equal(
results,
pd.DataFrame(columns=['a'],
index=[0, 1, 2, 3],
data=[[0], [1], [1], [1]]))
mock_client.create_job.assert_called_once()
def test_service_get_current_calibration():
service = ionq.Service(remote_host='http://example.com', api_key='key')
mock_client = mock.MagicMock()
calibration_dict = {'qubits': 11}
mock_client.get_current_calibration.return_value = calibration_dict
service._client = mock_client
cal = service.get_current_calibration()
assert cal.num_qubits() == 11
mock_client.get_current_calibration.assert_called_once()
def test_service_get_job():
service = ionq.Service(remote_host='http://example.com', api_key='key')
mock_client = mock.MagicMock()
job_dict = {'id': 'job_id', 'status': 'ready'}
mock_client.get_job.return_value = job_dict
service._client = mock_client
job = service.get_job('job_id')
assert job.job_id() == 'job_id'
mock_client.get_job.assert_called_with(job_id='job_id')
def test_service_list_jobs():
service = ionq.Service(remote_host='http://example.com', api_key='key')
mock_client = mock.MagicMock()
jobs = [{'id': '1'}, {'id': '2'}]
mock_client.list_jobs.return_value = jobs
service._client = mock_client
listed_jobs = service.list_jobs(status='completed', limit=10, batch_size=2)
assert listed_jobs[0].job_id() == '1'
assert listed_jobs[1].job_id() == '2'
mock_client.list_jobs.assert_called_with(status='completed',
limit=10,
batch_size=2)
def test_service_run(target, expected_results):
service = ionq.Service(remote_host='http://example.com', api_key='key')
mock_client = mock.MagicMock()
mock_client.create_job.return_value = {
'id': 'job_id',
'status': 'ready',
}
mock_client.get_job.return_value = {
'id': 'job_id',
'status': 'completed',
'target': target,
'metadata': {
'shots': '4',
'measurement0': f'a{chr(31)}0'
},
'qubits': '1',
'data': {
'histogram': {
'0': '0.25',
'1': '0.75'
}
},
'status': 'completed',
}
service._client = mock_client
a = sympy.Symbol('a')
q = cirq.LineQubit(0)
circuit = cirq.Circuit((cirq.X**a)(q), cirq.measure(q, key='a'))
params = cirq.ParamResolver({'a': 0.5})
result = service.run(
circuit=circuit,
repetitions=4,
target=target,
name='bacon',
param_resolver=params,
seed=2,
)
assert result == cirq.Result(params=params,
measurements={'a': expected_results})
create_job_kwargs = mock_client.create_job.call_args[1]
# Serialization induces a float, so we don't validate full circuit.
assert create_job_kwargs['serialized_program'].body['qubits'] == 1
assert create_job_kwargs['serialized_program'].metadata == {
'measurement0': f'a{chr(31)}0'
}
assert create_job_kwargs['repetitions'] == 4
assert create_job_kwargs['target'] == target
assert create_job_kwargs['name'] == 'bacon'
def test_service_list_calibrations():
service = ionq.Service(remote_host='http://example.com', api_key='key')
mock_client = mock.MagicMock()
calibrations = [{'id': '1', 'qubits': '1'}, {'id': '2', 'qubits': 2}]
mock_client.list_calibrations.return_value = calibrations
service._client = mock_client
start = datetime.datetime.utcfromtimestamp(1284286794)
end = datetime.datetime.utcfromtimestamp(1284286795)
listed_calibrations = service.list_calibrations(start=start,
end=end,
limit=10,
batch_size=2)
assert listed_calibrations[0].num_qubits() == 1
assert listed_calibrations[1].num_qubits() == 2
mock_client.list_calibrations.assert_called_with(start=start,
end=end,
limit=10,
batch_size=2)
def test_service_create_job():
service = ionq.Service(remote_host='http://example.com', api_key='key')
mock_client = mock.MagicMock()
mock_client.create_job.return_value = {'id': 'job_id', 'status': 'ready'}
mock_client.get_job.return_value = {'id': 'job_id', 'status': 'completed'}
service._client = mock_client
circuit = cirq.Circuit(cirq.X(cirq.LineQubit(0)))
job = service.create_job(circuit=circuit,
repetitions=100,
target='qpu',
name='bacon')
assert job.status() == 'completed'
create_job_kwargs = mock_client.create_job.call_args[1]
# Serialization induces a float, so we don't validate full circuit.
assert create_job_kwargs['serialized_program'].body['qubits'] == 1
assert create_job_kwargs['repetitions'] == 100
assert create_job_kwargs['target'] == 'qpu'
assert create_job_kwargs['name'] == 'bacon'
def test_service_no_url_default():
service = ionq.Service(api_key='tomyheart')
assert service.remote_host == 'https://api.ionq.co/v0.1'
def test_service_no_param_or_env_variable():
with pytest.raises(EnvironmentError):
_ = ionq.Service(remote_host='http://example.com')
def test_service_remote_host_via_env():
os.environ['IONQ_REMOTE_HOST'] = 'http://example.com'
service = ionq.Service(api_key='tomyheart')
assert service.remote_host == 'http://example.com'
del os.environ['IONQ_REMOTE_HOST']
def test_service_api_key_via_env():
os.environ['IONQ_API_KEY'] = 'tomyheart'
service = ionq.Service(remote_host='http://example.com')
assert service.api_key == 'tomyheart'
del os.environ['IONQ_API_KEY']
