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

lili0710432 · Jun 6, 2019 · da0c92a · da0c92a
1 parent 21ff161
commit da0c92a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/compute/test_traversal.py b/tests/compute/test_traversal.py
@@ -31,8 +31,10 @@ def _bfs_nx(g_nx, src):
                 edges_nx.append(edge_frontier)
                 frontier = set([v])
                 edge_frontier = set([g.edge_id(u, v)])
-        layers_nx.append(frontier)
-        edges_nx.append(edge_frontier)
+        # avoids case of no successors
+        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()