def test_get_acquaintance_size():
qubits = cirq.LineQubit.range(8)
op = OtherOperation(qubits)
assert op.with_qubits(qubits) == op
assert cca.get_acquaintance_size(op) == 0
for s, _ in enumerate(qubits):
op = cca.acquaint(*qubits[:s + 1])
assert cca.get_acquaintance_size(op) == s + 1
part_lens = (2, 2, 2, 2)
acquaintance_size = 3
gate = cca.SwapNetworkGate(part_lens, acquaintance_size)
op = gate(*qubits[:sum(part_lens)])
assert cca.get_acquaintance_size(op) == 3
part_lens = (2, 2, 2, 2)
acquaintance_size = 4
gate = cca.SwapNetworkGate(part_lens, acquaintance_size)
op = gate(*qubits[:sum(part_lens)])
assert cca.get_acquaintance_size(op) == 0
part_lens = (2, 2, 2, 2)
acquaintance_size = 1
gate = cca.SwapNetworkGate(part_lens, acquaintance_size)
op = gate(*qubits[:sum(part_lens)])
assert cca.get_acquaintance_size(op) == 0
part_lens = (2, 2, 2, 2)
acquaintance_size = 1
gate = cca.SwapNetworkGate(part_lens, acquaintance_size)
op = gate(*qubits[:sum(part_lens)])
assert cca.get_acquaintance_size(op) == 0
def test_get_acquaintance_size():
with pytest.raises(TypeError):
cca.get_acquaintance_size(cirq.Circuit())
with pytest.raises(TypeError):
cca.get_acquaintance_size(3)
circuit = cirq.Circuit(device=cca.UnconstrainedAcquaintanceDevice)
cca.get_acquaintance_size(circuit)
def test_bipartite_swap_network_acquaintance_size():
qubits = cirq.LineQubit.range(4)
gate = cca.BipartiteSwapNetworkGate(cca.BipartiteGraphType.COMPLETE, 2)
assert cca.get_acquaintance_size(gate(*qubits)) == 2
def test_complete_acquaintance_strategy():
qubits = [cirq.NamedQubit(s) for s in alphabet]
with pytest.raises(ValueError):
_ = cca.complete_acquaintance_strategy(qubits, -1)
empty_strategy = cca.complete_acquaintance_strategy(qubits)
assert empty_strategy._moments == []
trivial_strategy = cca.complete_acquaintance_strategy(qubits[:4], 1)
actual_text_diagram = trivial_strategy.to_text_diagram().strip()
expected_text_diagram = """
a: ───█───
b: ───█───
c: ───█───
d: ───█───
""".strip()
assert actual_text_diagram == expected_text_diagram
assert cca.get_acquaintance_size(trivial_strategy) == 1
is_shift_or_lin_perm = lambda op: isinstance(op.gate, (
cca.CircularShiftGate, cca.LinearPermutationGate))
expand = cirq.ExpandComposite(no_decomp=is_shift_or_lin_perm)
quadratic_strategy = cca.complete_acquaintance_strategy(qubits[:8], 2)
actual_text_diagram = quadratic_strategy.to_text_diagram().strip()
expected_text_diagram = """
a: ───×(0,0)───
│
b: ───×(1,0)───
│
c: ───×(2,0)───
│
d: ───×(3,0)───
│
e: ───×(4,0)───
│
f: ───×(5,0)───
│
g: ───×(6,0)───
│
h: ───×(7,0)───
""".strip()
assert actual_text_diagram == expected_text_diagram
assert cca.get_acquaintance_size(quadratic_strategy) == 2
expand(quadratic_strategy)
actual_text_diagram = quadratic_strategy.to_text_diagram(
transpose=True).strip()
expected_text_diagram = '\n'.join(
("a b c d e f g h ".strip(),
"│ │ │ │ │ │ │ │ ".strip(),
"█───█ █───█ █───█ █───█ ".strip(),
"│ │ │ │ │ │ │ │ ".strip(),
"╲0╱─╱1╲ ╲0╱─╱1╲ ╲0╱─╱1╲ ╲0╱─╱1╲ ".strip(),
"│ │ │ │ │ │ │ │ ".strip(),
"│ █───█ █───█ █───█ │ ".strip(),
"│ │ │ │ │ │ │ │ ".strip(),
"│ ╲0╱─╱1╲ ╲0╱─╱1╲ ╲0╱─╱1╲ │ ".strip(),
"│ │ │ │ │ │ │ │ ".strip(),
"█───█ █───█ █───█ █───█ ".strip(),
"│ │ │ │ │ │ │ │ ".strip(),
"╲0╱─╱1╲ ╲0╱─╱1╲ ╲0╱─╱1╲ ╲0╱─╱1╲ ".strip(),
"│ │ │ │ │ │ │ │ ".strip(),
"│ █───█ █───█ █───█ │ ".strip(),
"│ │ │ │ │ │ │ │ ".strip(),
"│ ╲0╱─╱1╲ ╲0╱─╱1╲ ╲0╱─╱1╲ │ ".strip(),
"│ │ │ │ │ │ │ │ ".strip(),
"█───█ █───█ █───█ █───█ ".strip(),
"│ │ │ │ │ │ │ │ ".strip(),
"╲0╱─╱1╲ ╲0╱─╱1╲ ╲0╱─╱1╲ ╲0╱─╱1╲ ".strip(),
"│ │ │ │ │ │ │ │ ".strip(),
"│ █───█ █───█ █───█ │ ".strip(),
"│ │ │ │ │ │ │ │ ".strip(),
"│ ╲0╱─╱1╲ ╲0╱─╱1╲ ╲0╱─╱1╲ │ ".strip(),
"│ │ │ │ │ │ │ │ ".strip(),
"█───█ █───█ █───█ █───█ ".strip(),
"│ │ │ │ │ │ │ │ ".strip(),
"╲0╱─╱1╲ ╲0╱─╱1╲ ╲0╱─╱1╲ ╲0╱─╱1╲ ".strip(),
"│ │ │ │ │ │ │ │ ".strip(),
"│ █───█ █───█ █───█ │ ".strip(),
"│ │ │ │ │ │ │ │ ".strip(),
"│ ╲0╱─╱1╲ ╲0╱─╱1╲ ╲0╱─╱1╲ │ ".strip(),
"│ │ │ │ │ │ │ │ ".strip()))
assert actual_text_diagram == expected_text_diagram
assert cca.get_acquaintance_size(quadratic_strategy) == 2
cubic_strategy = cca.complete_acquaintance_strategy(qubits[:4], 3)
actual_text_diagram = cubic_strategy.to_text_diagram(
transpose=True).strip()
expected_text_diagram = """
a b c d
│ │ │ │
×(0,0)─×(0,1)─×(1,0)─×(1,1)
│ │ │ │
╱1╲────╲0╱ ╱1╲────╲0╱
│ │ │ │
×(0,0)─×(1,0)─×(1,1)─×(2,0)
│ │ │ │
│ ╲0╱────╱1╲ │
│ │ │ │
×(0,0)─×(0,1)─×(1,0)─×(1,1)
│ │ │ │
╱1╲────╲0╱ ╱1╲────╲0╱
│ │ │ │
×(0,0)─×(1,0)─×(1,1)─×(2,0)
│ │ │ │
│ ╲0╱────╱1╲ │
│ │ │ │
""".strip()
assert actual_text_diagram == expected_text_diagram
assert cca.get_acquaintance_size(cubic_strategy) == 3
