def test_run_collector_identical_circuits_batched(ncircuits, executor):
collector = Collector(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_collector_preserves_order(s, b):
rng = np.random.RandomState(1)
collector = Collector(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_collector_force_run_all_serial_executor_identical_circuits(
ncircuits, force_run_all
):
collector = Collector(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_collector_all_unique(ncircuits, batch_size):
collector = Collector(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_collector_nonidentical_pyquil_programs(batch_size):
collector = Collector(
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
def test_collector_is_batched_executor():
assert Collector.is_batched_executor(executor=executor_batched)
assert not Collector.is_batched_executor(executor=executor_serial_typed)
assert not Collector.is_batched_executor(executor=executor_serial)
def test_collector_simple():
collector = Collector(executor=executor_batched, max_batch_size=10)
assert collector.can_batch
assert collector._max_batch_size == 10
assert collector.calls_to_executor == 0