def test_no_extension(self):
"""There are no virtual qubits to extend."""
layout = Layout({self.qr3[0]: 0, self.qr3[1]: 1, self.qr3[2]: 2})
pass_ = EnlargeWithAncilla(layout)
after = pass_.run(self.dag)
qregs = list(after.qregs.values())
self.assertEqual(1, len(qregs))
self.assertEqual(self.qr3, qregs[0])
def test_no_extension(self):
"""There are no idle physical bits to extend."""
layout = Layout([(self.qr3, 0), (self.qr3, 1), (self.qr3, 2)])
pass_ = EnlargeWithAncilla(layout)
after = pass_.run(self.dag)
qregs = list(after.qregs.values())
self.assertEqual(1, len(qregs))
self.assertEqual(self.qr3, qregs[0])
def test_with_extension(self):
"""There are 2 idle physical bits to extend."""
layout = Layout([(self.qr3, 0), None, (self.qr3, 1), None,
(self.qr3, 2)])
pass_ = EnlargeWithAncilla(layout)
after = pass_.run(self.dag)
qregs = list(after.qregs.values())
self.assertEqual(2, len(qregs))
self.assertEqual(self.qr3, qregs[0])
self.assertEqual(QuantumRegister(2, name='ancilla'), qregs[1])
def test_with_extension(self):
"""There are 2 virtual qubit to extend."""
ancilla = QuantumRegister(2, 'ancilla')
layout = Layout({0: self.qr3[0], 1: ancilla[0],
2: self.qr3[1], 3: ancilla[1],
4: self.qr3[2]})
pass_ = EnlargeWithAncilla(layout)
after = pass_.run(self.dag)
qregs = list(after.qregs.values())
self.assertEqual(2, len(qregs))
self.assertEqual(self.qr3, qregs[0])
self.assertEqual(ancilla, qregs[1])
def test_name_collision(self):
"""Name collision during ancilla extension."""
qr_ancilla = QuantumRegister(3, 'ancilla')
circuit = QuantumCircuit(qr_ancilla)
circuit.h(qr_ancilla)
dag = circuit_to_dag(circuit)
layout = Layout([(qr_ancilla, 0), None, (qr_ancilla, 1), None,
(qr_ancilla, 2)])
pass_ = EnlargeWithAncilla(layout)
after = pass_.run(dag)
qregs = list(after.qregs.values())
self.assertEqual(2, len(qregs))
self.assertEqual(qr_ancilla, qregs[0])
self.assertEqual(2, qregs[1].size)
self.assertRegex(qregs[1].name, r'^ancilla\d+$')
def test_with_extension(self):
"""There are 2 idle physical bits to extend."""
layout = Layout([(self.qr3, 0),
None,
(self.qr3, 1),
None,
(self.qr3, 2)])
pass_ = EnlargeWithAncilla(layout)
after = pass_.run(self.dag)
final_layout = {0: (QuantumRegister(3, 'qr'), 0), 1: (QuantumRegister(2, 'ancilla'), 0),
2: (QuantumRegister(3, 'qr'), 1), 3: (QuantumRegister(2, 'ancilla'), 1),
4: (QuantumRegister(3, 'qr'), 2)}
qregs = list(after.qregs.values())
self.assertEqual(2, len(qregs))
self.assertEqual(self.qr3, qregs[0])
self.assertEqual(QuantumRegister(2, name='ancilla'), qregs[1])
self.assertEqual(final_layout, pass_.property_set['layout'].get_physical_bits())