def test_filter_function_inner_exception(self): dag = retworkx.PyDiGraph() dag.add_node('a') dag.add_child(0, 'b', None) def filter_function(node): raise IndexError("Things fail from time to time") with self.assertRaises(IndexError): retworkx.collect_runs(dag, filter_function)
def test_h_h_cx(self): dag = retworkx.PyDiGraph() q0, q1 = dag.add_nodes_from(['q0', 'q1']) h_1 = dag.add_child(q0, 'h', 'q0') h_2 = dag.add_child(q1, 'h', 'q1') cx_2 = dag.add_child(h_1, 'cx', 'q0') dag.add_edge(h_2, cx_2, 'q1') def filter_function(node): return node in ['cx', 'h'] res = retworkx.collect_runs(dag, filter_function) self.assertEqual([['h', 'cx'], ['h']], res)
def test_h_h_cx(self): dag = retworkx.PyDiGraph() q0, q1 = dag.add_nodes_from(["q0", "q1"]) h_1 = dag.add_child(q0, "h", "q0") h_2 = dag.add_child(q1, "h", "q1") cx_2 = dag.add_child(h_1, "cx", "q0") dag.add_edge(h_2, cx_2, "q1") def filter_function(node): return node in ["cx", "h"] res = retworkx.collect_runs(dag, filter_function) self.assertEqual([["h", "cx"], ["h"]], res)
def test_multiple_successor_edges(self): dag = retworkx.PyDiGraph() q0, q1 = dag.add_nodes_from(['q0', 'q1']) cx_1 = dag.add_child(q0, 'cx', 'q0') dag.add_edge(q1, cx_1, 'q1') cx_2 = dag.add_child(cx_1, 'cx', 'q0') dag.add_edge(q1, cx_2, 'q1') cx_3 = dag.add_child(cx_2, 'cx', 'q0') dag.add_edge(q1, cx_3, 'q1') def filter_function(node): return node == 'cx' res = retworkx.collect_runs(dag, filter_function) self.assertEqual([['cx', 'cx', 'cx']], res)
def test_multiple_successor_edges(self): dag = retworkx.PyDiGraph() q0, q1 = dag.add_nodes_from(["q0", "q1"]) cx_1 = dag.add_child(q0, "cx", "q0") dag.add_edge(q1, cx_1, "q1") cx_2 = dag.add_child(cx_1, "cx", "q0") dag.add_edge(q1, cx_2, "q1") cx_3 = dag.add_child(cx_2, "cx", "q0") dag.add_edge(q1, cx_3, "q1") def filter_function(node): return node == "cx" res = retworkx.collect_runs(dag, filter_function) self.assertEqual([["cx", "cx", "cx"]], res)
def test_dagcircuit_basic(self): dag = retworkx.PyDAG() qr_0_in = dag.add_node("qr[0]") qr_0_out = dag.add_node("qr[0]") qr_1_in = dag.add_node("qr[1]") qr_1_out = dag.add_node("qr[1]") cr_0_in = dag.add_node("cr[0]") cr_0_out = dag.add_node("cr[0]") cr_1_in = dag.add_node("cr[1]") cr_1_out = dag.add_node("cr[1]") h_gate = dag.add_child(qr_0_in, "h", "qr[0]") x_gate = dag.add_child(h_gate, "x", "qr[0]") cx_gate = dag.add_child(x_gate, "cx", "qr[0]") dag.add_edge(qr_1_in, cx_gate, "qr[1]") measure_qr_1 = dag.add_child(cx_gate, "measure", "qr[1]") dag.add_edge(cr_1_in, measure_qr_1, "cr[1]") x_gate = dag.add_child(measure_qr_1, "x", "qr[1]") dag.add_edge(measure_qr_1, x_gate, "cr[1]") dag.add_edge(cr_0_in, x_gate, "cr[0]") measure_qr_0 = dag.add_child(cx_gate, "measure", "qr[0]") dag.add_edge(measure_qr_0, qr_0_out, "qr[0]") dag.add_edge(measure_qr_0, cr_0_out, "cr[0]") dag.add_edge(x_gate, measure_qr_0, "cr[0]") measure_qr_1_out = dag.add_child(x_gate, "measure", "cr[1]") dag.add_edge(x_gate, measure_qr_1_out, "qr[1]") dag.add_edge(measure_qr_1_out, qr_1_out, "qr[1]") dag.add_edge(measure_qr_1_out, cr_1_out, "cr[1]") def filter_function(node): return node in ['h', 'x'] res = retworkx.collect_runs(dag, filter_function) expected = [['h', 'x'], ['x']] self.assertEqual(expected, res)
def test_empty(self): dag = retworkx.PyDAG() self.assertEqual([], retworkx.collect_runs(dag, lambda _: True))
def test_cycle(self): dag = retworkx.PyDiGraph() dag.extend_from_edge_list([(0, 1), (1, 2), (2, 0)]) with self.assertRaises(retworkx.DAGHasCycle): retworkx.collect_runs(dag, lambda _: True)