better bfs edge test (#142)

7cbb83db · Gan Quan · GitHub · d6c80e0c · 7cbb83db
Unverified Commit 7cbb83db authored Nov 11, 2018 by Gan Quan Committed by GitHub Nov 11, 2018
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 3 deletions

tests/pytorch/test_traversal.py tests/pytorch/test_traversal.py +2 -3

No files found.
--- 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))