Skip to content

GitLab

Unverified Commit 7cbb83db authored by Gan Quan's avatar Gan Quan Committed by GitHub
Browse files

better bfs edge test (#142)

parent d6c80e0c
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
tests/pytorch/test_traversal.py
View file @ 7cbb83db
...@@ -17,10 +17,9 @@ def toset(x): ...@@ -17,10 +17,9 @@ def toset(x):
return set(x.tolist()) return set(x.tolist())
def test_bfs_nodes(n=1000): def test_bfs_nodes(n=1000):
g_nx = nx.random_tree(n)
g = dgl.DGLGraph() g = dgl.DGLGraph()
a = sp.random(n, n, 10 / n, data_rvs=lambda n: np.ones(n)) g.from_networkx(g_nx)
g.from_scipy_sparse_matrix(a)
g_nx = g.to_networkx()
src = random.choice(range(n)) src = random.choice(range(n))
......
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