dmlc--dgl
Hongzhi (Steve), Chen
74c9d27d16
* [Misc] Auto-format tests. * more --------- Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
29 行
614 B
Python
29 行
614 B
Python
import unittest
|
|
|
|
import backend as F
|
|
|
|
import dgl
|
|
import numpy as np
|
|
|
|
|
|
@unittest.skipIf(
|
|
F._default_context_str == "gpu", reason="GPU random choice not implemented"
|
|
)
|
|
def test_rand_graph():
|
|
g = dgl.rand_graph(10000, 100000)
|
|
assert g.number_of_nodes() == 10000
|
|
assert g.number_of_edges() == 100000
|
|
# test random seed
|
|
dgl.random.seed(42)
|
|
g1 = dgl.rand_graph(100, 30)
|
|
dgl.random.seed(42)
|
|
g2 = dgl.rand_graph(100, 30)
|
|
u1, v1 = g1.edges()
|
|
u2, v2 = g2.edges()
|
|
assert F.array_equal(u1, u2)
|
|
assert F.array_equal(v1, v2)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_rand_graph()
|