def test_create_dataset1(self):
""" create_dataset case1: empty dataset """
indir = 'resources/test_train/dataset1/in'
outdir = 'resources/test_train/dataset1/out'
if os.path.exists(outdir):
shutil.rmtree(outdir)
os.mkdir(outdir)
dataset.create_dataset(indir, outdir, 0, 0)
assert os.listdir(os.path.join(outdir, '')) == []
def test_create_dataset4(self):
""" create_dataset case4: dataset not exists """
indir = 'resources/test_train/dataset4/in'
outdir = 'resources/test_train/dataset4/out'
if os.path.exists(outdir):
shutil.rmtree(outdir)
os.mkdir(outdir)
dataset.create_dataset(indir, outdir, 64, 64)
assert os.listdir(os.path.join(outdir, '')) == []
def test_create_dataset3(self):
""" create_dataset case3: out directory not exists """
indir = 'resources/test_train/dataset3/in'
outdir = 'resources/test_train/dataset3/out'
if os.path.exists(outdir):
shutil.rmtree(outdir)
os.mkdir(outdir)
dataset.create_dataset(indir, outdir, 64, 64)
files = os.listdir(outdir)
res = ['test1.png', 'test2.png']
assert sorted(files) == sorted(res)
def test_create_dataset2(self):
""" create_dataset case2: out directory exists """
indir = 'resources/test_train/dataset2/in'
outdir = 'resources/test_train/dataset2/out'
if os.path.exists(outdir):
shutil.rmtree(outdir)
os.mkdir(outdir)
else:
os.mkdir(outdir)
dataset.create_dataset(indir, outdir, 64, 64)
dir0 = os.path.join(outdir, '0')
dir1 = os.path.join(outdir, '1')
xs = os.walk(outdir)
files = []
for path, ds, fs in xs:
if len(fs) > 0:
for f in fs:
files.append(os.path.join(path, f))
res = [os.path.join(dir0, 'test1.png'), os.path.join(dir1, 'test2.png')]
assert sorted(files) == sorted(res)
def gendata(args): dataset.create_dataset(args.rawdata, args.output, args.size, args.size)
