[9d3784]: / aiagents4pharma / talk2knowledgegraphs / tests / test_datasets_dataset.py

Download this file

30 lines (24 with data), 566 Bytes

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
"""
Test cases for datasets/dataset.py
"""
from ..datasets.dataset import Dataset
class MockDataset(Dataset):
"""
Mock dataset class for testing purposes.
"""
def setup(self):
pass
def load_data(self):
pass
def test_dataset_setup():
"""
Test the setup method of the Dataset class.
"""
dataset = MockDataset()
assert dataset.setup() is None
def test_dataset_load_data():
"""
Test the load_data method of the Dataset class.
"""
dataset = MockDataset()
assert dataset.load_data() is None