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

da0c92a2 · Quan (Andy) Gan · Minjie Wang · 21ff161d · da0c92a2
Commit da0c92a2 authored Jun 07, 2019 by Quan (Andy) Gan Committed by Minjie Wang Jun 06, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 2 deletions

tests/compute/test_traversal.py tests/compute/test_traversal.py +4 -2

No files found.
--- a/tests/compute/test_traversal.py
+++ b/tests/compute/test_traversal.py
@@ -31,8 +31,10 @@ def test_bfs(n=1000):
                edges_nx.append(edge_frontier)
                frontier = set([v])
                edge_frontier = set([g.edge_id(u, v)])
-        layers_nx.append(frontier)
+        # avoids case of no successors
-        edges_nx.append(edge_frontier)
+        if len(frontier) > 0 and len(edge_frontier) > 0:
+            layers_nx.append(frontier)
+            edges_nx.append(edge_frontier)
        return layers_nx, edges_nx
    g = dgl.DGLGraph()