better bfs edge test (dmlc#142)

liushuchun · Nov 12, 2018 · 7cbb83d · 7cbb83d
1 parent d6c80e0
commit 7cbb83d
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/pytorch/test_traversal.py b/tests/pytorch/test_traversal.py
@@ -17,10 +17,9 @@ def toset(x):
     return set(x.tolist())
 
 def test_bfs_nodes(n=1000):
+    g_nx = nx.random_tree(n)
     g = dgl.DGLGraph()
-    a = sp.random(n, n, 10 / n, data_rvs=lambda n: np.ones(n))
-    g.from_scipy_sparse_matrix(a)
-    g_nx = g.to_networkx()
+    g.from_networkx(g_nx)
 
     src = random.choice(range(n))