def test_largest_sequence_search_traverses_grid():
q00 = cirq.GridQubit(0, 0)
q10 = cirq.GridQubit(1, 0)
q11 = cirq.GridQubit(1, 1)
q20 = cirq.GridQubit(2, 0)
q30 = cirq.GridQubit(3, 0)
q40 = cirq.GridQubit(4, 0)
q41 = cirq.GridQubit(4, 1)
q42 = cirq.GridQubit(4, 2)
q50 = cirq.GridQubit(5, 0)
qubits = [q00, q10, q11, q20, q30, q40, q50, q41, q42]
device = _create_device(qubits)
search = greedy._PickLargestArea(device, q20)
# + + +-+-+ +
# |
# + +
# |
# +
assert search._choose_next_qubit(q20, {q20}) == q30
assert search._choose_next_qubit(q30, {q20, q30}) == q40
assert search._choose_next_qubit(q40, {q20, q30, q40}) == q41
assert search._choose_next_qubit(q41, {q20, q30, q40, q41}) == q42
assert search._choose_next_qubit(q42, {q20, q30, q40, q41, q42}) is None
method = greedy.GreedySequenceSearchStrategy('largest_area')
assert method.place_line(device, 7) == GridQubitLineTuple(
[q00, q10, q20, q30, q40, q41, q42])
with pytest.raises(NotFoundError):
_ = method.place_line(device, 8)
def test_largest_collect_unused_collects():
q00 = cirq.GridQubit(0, 0)
q01 = cirq.GridQubit(0, 1)
q02 = cirq.GridQubit(0, 2)
q12 = cirq.GridQubit(1, 2)
qubits = [q00, q01, q02, q12]
start = q01
search = greedy._PickLargestArea(_create_device(qubits), start)
assert search._collect_unused(start, {q00, q01}) == {q01, q02, q12}
def test_largest_sequence_search_does_not_use_used():
q00 = cirq.GridQubit(0, 0)
q10 = cirq.GridQubit(1, 0)
q20 = cirq.GridQubit(2, 0)
q21 = cirq.GridQubit(2, 1)
qubits = [q00, q10, q20, q21]
search = greedy._PickLargestArea(_create_device(qubits), q10)
# +-* X
#
# +
assert search._choose_next_qubit(q10, {q10, q20}) == q00
def test_largest_collect_stops_on_used():
q00 = cirq.GridQubit(0, 0)
q01 = cirq.GridQubit(0, 1)
q02 = cirq.GridQubit(0, 2)
q03 = cirq.GridQubit(0, 3)
q04 = cirq.GridQubit(0, 4)
q05 = cirq.GridQubit(0, 5)
q11 = cirq.GridQubit(1, 1)
q14 = cirq.GridQubit(1, 4)
q24 = cirq.GridQubit(2, 4)
qubits = [q00, q01, q11, q02, q03, q04, q05, q14, q24]
start = q02
search = greedy._PickLargestArea(_create_device(qubits), start)
assert search._collect_unused(start, {start, q04}) == {q00, q01, q11, q02, q03}
