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)