"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "7684518377180879a6e45611820e92c9dec7d4b6"
test_actor.py 582 Bytes
Newer Older
Mufei Li's avatar
Mufei Li committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
import unittest

import backend as F

import dgl


@unittest.skipIf(
    F._default_context_str == "gpu",
    reason="Datasets don't need to be tested on GPU.",
)
@unittest.skipIf(
    dgl.backend.backend_name != "pytorch", reason="only supports pytorch"
)
def test_actor():
    transform = dgl.AddSelfLoop(allow_duplicate=True)

    g = dgl.data.ActorDataset(force_reload=True)[0]
    assert g.num_nodes() == 7600
    assert g.num_edges() == 33391
    g2 = dgl.data.ActorDataset(force_reload=True, transform=transform)[0]
    assert g2.num_edges() - g.num_edges() == g.num_nodes()