def genlist(args): dataset.create_datalist(args.dataset, args.output)
def test_create_datalist3(self):
""" create_datalist case3: search dataset not recursively """
dirpath = 'resources/test_train/dataset3/in'
filepath = 'resources/test_train/dataset3/out/image-list.txt'
dataset.create_datalist(dirpath, filepath)
assert os.path.exists(filepath)
def test_create_datalist4(self):
""" create_datalist case4: empty dataset (only text file) """
dirpath = 'resources/test_train/dataset4/in'
filepath = 'resources/test_train/dataset4/out/image-list.txt'
dataset.create_datalist(dirpath, filepath)
assert os.path.exists(filepath)
def test_create_datalist1(self):
""" create_datalist case1: empty dataset """
dirpath = 'resources/test_train/dataset1/in'
filepath = 'resources/test_train/dataset1/out/image-list.txt'
dataset.create_datalist(dirpath, filepath)
assert os.path.exists(filepath)
