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)