Exemple #1
0
    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
Exemple #2
0
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.
    """