def test_run_executor_identical_circuits_batched(ncircuits, executor):
collector = Executor(executor=executor, max_batch_size=10)
circuits = [cirq.Circuit(cirq.H(cirq.LineQubit(0)))] * ncircuits
results = collector._run(circuits)
assert np.allclose(results, np.zeros(ncircuits))
assert collector.calls_to_executor == 1
def test_run_executor_preserves_order(s, b):
rng = np.random.RandomState(1)
collector = Executor(executor=executor_batched_unique, max_batch_size=b)
assert collector.can_batch
circuits = [
cirq.Circuit(cirq.H(cirq.LineQubit(0))),
cirq.Circuit([cirq.H(cirq.LineQubit(0))] * 2),
]
batch = [circuits[i] for i in rng.random_integers(low=0, high=1, size=s)]
assert np.allclose(collector._run(batch), executor_batched_unique(batch))
def test_run_executor_force_run_all_serial_executor_identical_circuits(
ncircuits, force_run_all):
collector = Executor(executor=executor_serial)
assert not collector.can_batch
circuits = [cirq.Circuit(cirq.H(cirq.LineQubit(0)))] * ncircuits
results = collector._run(circuits, force_run_all=force_run_all)
assert np.allclose(results, np.zeros(ncircuits))
if force_run_all:
assert collector.calls_to_executor == ncircuits
else:
assert collector.calls_to_executor == 1
def test_run_executor_all_unique(ncircuits, batch_size):
collector = Executor(executor=executor_batched, max_batch_size=batch_size)
assert collector.can_batch
random_state = np.random.RandomState(seed=1)
circuits = [
cirq.testing.random_circuit(qubits=4,
n_moments=10,
op_density=1,
random_state=random_state)
for _ in range(ncircuits)
]
results = collector._run(circuits)
assert np.allclose(results, np.zeros(ncircuits))
assert collector.calls_to_executor == np.ceil(ncircuits / batch_size)
def test_run_executor_nonidentical_pyquil_programs(batch_size):
collector = Executor(executor=executor_pyquil_batched,
max_batch_size=batch_size)
assert collector.can_batch
circuits = [
pyquil.Program(pyquil.gates.X(0)),
pyquil.Program(pyquil.gates.H(0)),
] * 10
results = collector._run(circuits)
assert np.allclose(results, np.zeros(len(circuits)))
if batch_size == 1:
assert collector.calls_to_executor == 2
else:
assert collector.calls_to_executor == 1