Skip to content

GitLab

Commit da0c92a2 authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by Minjie Wang
Browse files

[BUGFix][Test] fix random bfs failure (#612)

parent 21ff161d
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
tests/compute/test_traversal.py
View file @ da0c92a2
...@@ -31,6 +31,8 @@ def test_bfs(n=1000): ...@@ -31,6 +31,8 @@ def test_bfs(n=1000):
edges_nx.append(edge_frontier) edges_nx.append(edge_frontier)
frontier = set([v]) frontier = set([v])
edge_frontier = set([g.edge_id(u, v)]) edge_frontier = set([g.edge_id(u, v)])
# avoids case of no successors
if len(frontier) > 0 and len(edge_frontier) > 0:
layers_nx.append(frontier) layers_nx.append(frontier)
edges_nx.append(edge_frontier) edges_nx.append(edge_frontier)
return layers_nx, edges_nx return layers_nx, edges_nx
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment