test_dataset.py 389 Bytes
Newer Older
1
import pytest
2

3
4
5
6
7
8
from dgl import graphbolt as gb


def test_Dataset():
    dataset = gb.Dataset()
    with pytest.raises(NotImplementedError):
9
        _ = dataset.tasks
10
    with pytest.raises(NotImplementedError):
11
        _ = dataset.graph
12
    with pytest.raises(NotImplementedError):
13
        _ = dataset.feature
14
15
    with pytest.raises(NotImplementedError):
        _ = dataset.dataset_name