def cleanup(dir='test'): for path, dirs, files in os.walk(dir): for fn in files: os.remove(os.path.join(path, fn)) for dir in dirs: os.removedirs(os.path.join(path, dir)) # basic test