test_dataset.py 388 Bytes
Newer Older
1
2
3
4
5
6
7
import pytest
from dgl import graphbolt as gb


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