test_dataset.py

import pytest
from dgl import graphbolt as gb


def test_Dataset():
    dataset = gb.Dataset()
    with pytest.raises(NotImplementedError):
        _ = dataset.train_set()
    with pytest.raises(NotImplementedError):
        _ = dataset.validation_set()
    with pytest.raises(NotImplementedError):
        _ = dataset.test_set()
    with pytest.raises(NotImplementedError):
        _ = dataset.graph()
    with pytest.raises(NotImplementedError):
        _ = dataset.feature()