dataset_test Namespace Reference

Classes
class	TestDataset

Functions
def	create_dataset

Variables
string	BASE_DIR = 'datasets'

Function Documentation

def dataset_test.create_dataset	(		name,
			number_of_files,
			basedir = BASE_DIR
	)

Definition at line 9 of file dataset_test.py.

Referenced by dataset_test.TestDataset.test_local().

def create_dataset(name, number_of_files, basedir=BASE_DIR):
    if not os.path.isdir(basedir):
        os.mkdir(basedir)
    old = os.getcwd()
    os.chdir(basedir)
    if os.path.isdir(name):
        shutil.rmtree(name)
    os.mkdir(name)
    os.chdir(name)
    for i in range(number_of_files):
        os.system('touch file_{i:d}.root'.format(i=i))
    os.chdir(old)