def test_generate_file_lists(): dataset_path = './tests/data' dataset = Dataset(dataset_path) dataset.generate_file_lists() assert dataset.file_lists == {'fold1': [], 'fold2': [], 'fold3': []}
def test_generate_file_lists(): dataset_path = "./tests/data" dataset = Dataset(dataset_path) dataset.generate_file_lists() assert dataset.file_lists == {"fold1": [], "fold2": [], "fold3": []}