dmlc--dgl
Xiangkun Hu
317be7f188
* PPIDataset * Revert "PPIDataset" This reverts commit 264bd0c960cfa698a7bb946dad132bf52c2d0c8a. * check whether the original labels are valid * add unit test for gin dataset * bug of asscalar in mxnet<=1.6 * mxnet<=1.6 asscalar requires ndarray to be shaped (1,) * skip gpu while testing datasets * Update test_data.py * test of gin dataset takes too much time
45 行
1.3 KiB
Python
45 行
1.3 KiB
Python
import dgl.data as data
|
|
import unittest
|
|
import backend as F
|
|
|
|
|
|
@unittest.skipIf(F._default_context_str == 'gpu', reason="Datasets don't need to be tested on GPU.")
|
|
def test_minigc():
|
|
ds = data.MiniGCDataset(16, 10, 20)
|
|
g, l = list(zip(*ds))
|
|
print(g, l)
|
|
|
|
|
|
@unittest.skipIf(F._default_context_str == 'gpu', reason="Datasets don't need to be tested on GPU.")
|
|
def test_gin():
|
|
ds_n_graphs = {
|
|
'MUTAG': 188,
|
|
'IMDBBINARY': 1000,
|
|
'IMDBMULTI': 1500,
|
|
'PROTEINS': 1113,
|
|
'PTC': 344,
|
|
}
|
|
for name, n_graphs in ds_n_graphs.items():
|
|
ds = data.GINDataset(name, self_loop=False, degree_as_nlabel=False)
|
|
assert len(ds) == n_graphs, (len(ds), name)
|
|
|
|
|
|
@unittest.skipIf(F._default_context_str == 'gpu', reason="Datasets don't need to be tested on GPU.")
|
|
def test_data_hash():
|
|
class HashTestDataset(data.DGLDataset):
|
|
def __init__(self, hash_key=()):
|
|
super(HashTestDataset, self).__init__('hashtest', hash_key=hash_key)
|
|
def _load(self):
|
|
pass
|
|
|
|
a = HashTestDataset((True, 0, '1', (1,2,3)))
|
|
b = HashTestDataset((True, 0, '1', (1,2,3)))
|
|
c = HashTestDataset((True, 0, '1', (1,2,4)))
|
|
assert a.hash == b.hash
|
|
assert a.hash != c.hash
|
|
|
|
if __name__ == '__main__':
|
|
test_minigc()
|
|
test_gin()
|
|
test_data_hash()
|