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)
