def test_display_mapping():
indices = [4, 2, 0, 1, 3]
qubits = cirq.LineQubit.range(len(indices))
circuit = cca.complete_acquaintance_strategy(qubits, 2)
cca.expose_acquaintance_gates(circuit)
initial_mapping = dict(zip(qubits, indices))
cca.display_mapping(circuit, initial_mapping)
expected_diagram = """
0: ───4───█───4───╲0╱───2───────2─────────2───█───2───╲0╱───1───────1─────────1───█───1───╲0╱───3───
│ │ │ │ │ │
1: ───2───█───2───╱1╲───4───█───4───╲0╱───1───█───1───╱1╲───2───█───2───╲0╱───3───█───3───╱1╲───1───
│ │ │ │
2: ───0───█───0───╲0╱───1───█───1───╱1╲───4───█───4───╲0╱───3───█───3───╱1╲───2───█───2───╲0╱───0───
│ │ │ │ │ │
3: ───1───█───1───╱1╲───0───█───0───╲0╱───3───█───3───╱1╲───4───█───4───╲0╱───0───█───0───╱1╲───2───
│ │ │ │
4: ───3───────3─────────3───█───3───╱1╲───0───────0─────────0───█───0───╱1╲───4───────4─────────4───
"""
cirq.testing.assert_has_diagram(circuit, expected_diagram)
def __str__(self) -> str:
circuit = self.circuit.copy()
cca.display_mapping(circuit, self.initial_mapping)
return str(circuit)