def test_largest_collect_unused_collects_all_for_empty():
q00 = GridQubit(0, 0)
q01 = GridQubit(0, 1)
q02 = GridQubit(0, 2)
q12 = GridQubit(1, 2)
qubits = [q00, q01, q02, q12]
start = q01
search = LargestAreaGreedySequenceSearch(_create_device(qubits), start)
assert search._collect_unused(start, set()) == set(qubits)
assert search._collect_unused(start, {start}) == set(qubits)
def test_largest_collect_unused_collects_all_for_empty():
q00 = GridQubit(0, 0)
q01 = GridQubit(0, 1)
q02 = GridQubit(0, 2)
q12 = GridQubit(1, 2)
qubits = [q00, q01, q02, q12]
start = q01
search = LargestAreaGreedySequenceSearch(_create_device(qubits), start)
assert search._collect_unused(start, set()) == set(qubits)
assert search._collect_unused(start, {start}) == set(qubits)
def test_largest_collect_unused_collects():
q00 = XmonQubit(0, 0)
q01 = XmonQubit(0, 1)
q02 = XmonQubit(0, 2)
q12 = XmonQubit(1, 2)
qubits = [q00, q01, q02, q12]
start = q01
search = LargestAreaGreedySequenceSearch(_create_device(qubits), start)
assert search._collect_unused(start, {q00, q01}) == {q01, q02, q12}
def test_largest_collect_stops_on_used():
q00 = GridQubit(0, 0)
q01 = GridQubit(0, 1)
q02 = GridQubit(0, 2)
q03 = GridQubit(0, 3)
q04 = GridQubit(0, 4)
q05 = GridQubit(0, 5)
q11 = GridQubit(1, 1)
q14 = GridQubit(1, 4)
q24 = GridQubit(2, 4)
qubits = [q00, q01, q11, q02, q03, q04, q05, q14, q24]
start = q02
search = LargestAreaGreedySequenceSearch(_create_device(qubits), start)
assert search._collect_unused(start,
{start, q04}) == {q00, q01, q11, q02, q03}
def test_largest_collect_stops_on_used():
q00 = GridQubit(0, 0)
q01 = GridQubit(0, 1)
q02 = GridQubit(0, 2)
q03 = GridQubit(0, 3)
q04 = GridQubit(0, 4)
q05 = GridQubit(0, 5)
q11 = GridQubit(1, 1)
q14 = GridQubit(1, 4)
q24 = GridQubit(2, 4)
qubits = [q00, q01, q11, q02, q03, q04, q05, q14, q24]
start = q02
search = LargestAreaGreedySequenceSearch(_create_device(qubits), start)
assert search._collect_unused(start, {start, q04}) == {q00, q01, q11, q02,
q03}