def mkoutdir(self, clobber=True): """ Create outdir as outpudir/module.class.method (destructively if clobber is True). """ funcname = ".".join(self.id().split(".")[-3:]) outdir = path.join(self.outputdir, funcname) mkdir(outdir, clobber) return outdir
loglevel = {0: logging.WARNING, 1: logging.INFO, 2: logging.DEBUG}.get(verbosity, logging.DEBUG) if verbosity > 1: logformat = "%(levelname)s %(module)s %(lineno)s %(message)s" else: logformat = "%(message)s" logging.basicConfig(file=sys.stdout, format=logformat, level=loglevel) log = logging.getLogger(__name__) # module data datadir = "testfiles" outputdir = "test_output" mkdir(outputdir) def get_testfile(fn): pth = path.join(datadir, fn) if not path.exists(pth): raise ValueError('no such file "{}"'.format(pth)) return pth class TestBase(unittest.TestCase): """ Base class for unit tests with methods for defining output directories based on method name. """