def test_add_reg_duplicate_name(self):
"""Adding quantum registers with the same name is not allowed."""
dag = DAGDependency()
qr1 = QuantumRegister(3, 'qr')
dag.add_qreg(qr1)
qr2 = QuantumRegister(2, 'qr')
self.assertRaises(DAGDependencyError, dag.add_qreg, qr2)
def setUp(self):
self.dag = DAGDependency()
self.qreg = QuantumRegister(2, 'qr')
self.creg = ClassicalRegister(2, 'cr')
self.dag.add_qreg(self.qreg)
self.dag.add_creg(self.creg)
def setUp(self):
super().setUp()
self.dag = DAGDependency()
self.qreg = QuantumRegister(2, "qr")
self.creg = ClassicalRegister(2, "cr")
self.dag.add_qreg(self.qreg)
self.dag.add_creg(self.creg)
def test_add_duplicate_registerless_bits(self):
"""Verify we raise when adding a bit already present in the circuit."""
qubits = [Qubit() for _ in range(5)]
clbits = [Clbit() for _ in range(3)]
dag = DAGDependency()
dag.add_qubits(qubits)
dag.add_clbits(clbits)
with self.assertRaisesRegex(DAGDependencyError, r"duplicate qubits"):
dag.add_qubits(qubits[:1])
with self.assertRaisesRegex(DAGDependencyError, r"duplicate clbits"):
dag.add_clbits(clbits[:1])
def test_dag_get_qubits(self):
"""get_qubits() method """
dag = DAGDependency()
dag.add_qreg(QuantumRegister(1, 'qr1'))
dag.add_qreg(QuantumRegister(1, 'qr10'))
dag.add_qreg(QuantumRegister(1, 'qr0'))
dag.add_qreg(QuantumRegister(1, 'qr3'))
dag.add_qreg(QuantumRegister(1, 'qr4'))
dag.add_qreg(QuantumRegister(1, 'qr6'))
self.assertListEqual(dag.qubits, [QuantumRegister(1, 'qr1')[0],
QuantumRegister(1, 'qr10')[0],
QuantumRegister(1, 'qr0')[0],
QuantumRegister(1, 'qr3')[0],
QuantumRegister(1, 'qr4')[0],
QuantumRegister(1, 'qr6')[0]])
def test_dag_get_qubits(self):
"""get_qubits() method"""
dag = DAGDependency()
dag.add_qreg(QuantumRegister(1, "qr1"))
dag.add_qreg(QuantumRegister(1, "qr10"))
dag.add_qreg(QuantumRegister(1, "qr0"))
dag.add_qreg(QuantumRegister(1, "qr3"))
dag.add_qreg(QuantumRegister(1, "qr4"))
dag.add_qreg(QuantumRegister(1, "qr6"))
self.assertListEqual(
dag.qubits,
[
QuantumRegister(1, "qr1")[0],
QuantumRegister(1, "qr10")[0],
QuantumRegister(1, "qr0")[0],
QuantumRegister(1, "qr3")[0],
QuantumRegister(1, "qr4")[0],
QuantumRegister(1, "qr6")[0],
],
)
def test_add_qreg_creg(self):
"""add_qreg() and add_creg() methods"""
dag = DAGDependency()
dag.add_qreg(QuantumRegister(2, 'qr'))
dag.add_creg(ClassicalRegister(1, 'cr'))
self.assertDictEqual(dag.qregs, {'qr': QuantumRegister(2, 'qr')})
self.assertDictEqual(dag.cregs, {'cr': ClassicalRegister(1, 'cr')})
def test_add_qreg_creg(self):
"""add_qreg() and add_creg() methods"""
dag = DAGDependency()
dag.add_qreg(QuantumRegister(2, "qr"))
dag.add_creg(ClassicalRegister(1, "cr"))
self.assertDictEqual(dag.qregs, {"qr": QuantumRegister(2, "qr")})
self.assertDictEqual(dag.cregs, {"cr": ClassicalRegister(1, "cr")})
def test_add_registerless_bits(self):
"""Verify we can add are retrieve bits without an associated register."""
qubits = [Qubit() for _ in range(5)]
clbits = [Clbit() for _ in range(3)]
dag = DAGDependency()
dag.add_qubits(qubits)
dag.add_clbits(clbits)
self.assertEqual(dag.qubits, qubits)
self.assertEqual(dag.clbits, clbits)
def test_add_reg_bad_type(self):
"""add_qreg with a classical register is not allowed."""
dag = DAGDependency()
cr = ClassicalRegister(2)
self.assertRaises(DAGDependencyError, dag.add_qreg, cr)
def test_add_reg_duplicate(self):
"""add_qreg with the same register twice is not allowed."""
dag = DAGDependency()
qr = QuantumRegister(2)
dag.add_qreg(qr)
self.assertRaises(DAGDependencyError, dag.add_qreg, qr)
class TestDagNodeSelection(QiskitTestCase):
"""Test methods that select successors and predecessors"""
def setUp(self):
self.dag = DAGDependency()
self.qreg = QuantumRegister(2, 'qr')
self.creg = ClassicalRegister(2, 'cr')
self.dag.add_qreg(self.qreg)
self.dag.add_creg(self.creg)
def test_successors_predecessors(self):
"""Test the method direct_successors."""
circuit = QuantumCircuit(self.qreg, self.creg)
circuit.h(self.qreg[0])
circuit.x(self.qreg[0])
circuit.h(self.qreg[0])
circuit.x(self.qreg[1])
circuit.h(self.qreg[0])
circuit.measure(self.qreg[0], self.creg[0])
self.dag = circuit_to_dagdependency(circuit)
dir_successors_second = self.dag.direct_successors(1)
self.assertEqual(dir_successors_second, [2, 4])
dir_successors_fourth = self.dag.direct_successors(3)
self.assertEqual(dir_successors_fourth, [])
successors_second = self.dag.successors(1)
self.assertEqual(successors_second, [2, 4, 5])
successors_fourth = self.dag.successors(3)
self.assertEqual(successors_fourth, [])
dir_predecessors_sixth = self.dag.direct_predecessors(5)
self.assertEqual(dir_predecessors_sixth, [2, 4])
dir_predecessors_fourth = self.dag.direct_predecessors(3)
self.assertEqual(dir_predecessors_fourth, [])
predecessors_sixth = self.dag.predecessors(5)
self.assertEqual(predecessors_sixth, [0, 1, 2, 4])
predecessors_fourth = self.dag.predecessors(3)
self.assertEqual(predecessors_fourth, [])
class TestDagNodeEdge(QiskitTestCase):
"""Test adding nodes and edges to a dag and related functions."""
def setUp(self):
self.dag = DAGDependency()
self.qreg = QuantumRegister(2, 'qr')
self.creg = ClassicalRegister(2, 'cr')
self.dag.add_qreg(self.qreg)
self.dag.add_creg(self.creg)
def test_node(self):
"""Test the methods add_op_node(), get_node() and get_nodes()"""
circuit = QuantumCircuit(self.qreg, self.creg)
circuit.h(self.qreg[0])
self.dag.add_op_node(circuit.data[0][0], circuit.data[0][1],
circuit.data[0][2])
self.assertIsInstance(self.dag.get_node(0).op, HGate)
circuit.measure(self.qreg[0], self.creg[0])
self.dag.add_op_node(circuit.data[1][0], circuit.data[1][1],
circuit.data[1][2])
self.assertIsInstance(self.dag.get_node(1).op, Measure)
nodes = list(self.dag.get_nodes())
self.assertEqual(len(list(nodes)), 2)
for node in nodes:
self.assertIsInstance(node.op, Instruction)
node_1 = nodes.pop()
node_2 = nodes.pop()
self.assertIsInstance(node_1.op, Measure)
self.assertIsInstance(node_2.op, HGate)
def test_add_edge(self):
"""Test that add_edge(), get_edges(), get_all_edges(),
get_in_edges() and get_out_edges(). """
circuit = QuantumCircuit(self.qreg, self.creg)
circuit.h(self.qreg[0])
circuit.x(self.qreg[1])
circuit.cx(self.qreg[1], self.qreg[0])
circuit.measure(self.qreg[0], self.creg[0])
self.dag = circuit_to_dagdependency(circuit)
second_edge = self.dag.get_edges(1, 2)
self.assertEqual(second_edge[0]['commute'], False)
all_edges = self.dag.get_all_edges()
self.assertEqual(len(all_edges), 3)
for edges in all_edges:
self.assertEqual(edges[2]['commute'], False)
in_edges = self.dag.get_in_edges(2)
self.assertEqual(len(list(in_edges)), 2)
out_edges = self.dag.get_out_edges(2)
self.assertEqual(len(list(out_edges)), 1)
