Function test_oagbert_dataset

tests/datasets/test_oagbert_data.py:5–9

()

from cogdl.utils import build_args_from_dict

def test_oagbert_dataset():
    args = build_args_from_dict({'dataset': 'aff30'})
    data = build_dataset(args)
    assert isinstance(data.get_data(), dict)
    assert isinstance(data.get_candidates(), list)

if __name__ == '__main__':
    test_oagbert_dataset()