def pick_next_node(G, node): probs, edges = [], [] for e in G.out_edges_iter(node): prob = G.get_edge_data(*e)["prob"] probs.append(prob) edges.append(e) return edges[utilities.pick_from_distr(probs)][1]
def test_pick_from_distr(self): d = mc.pick_from_distr(mc.get_circular_distribution(5)) self.assertGreaterEqual(d, 0) self.assertLess(d, 5)