コード例 #1
0
ファイル: test.py プロジェクト: yihsuanliu/gpaw
    tests = tests[index:]

if opt.range:
    indices = opt.range.split(',')
    start_index = tests.index(indices[0])
    stop_index = tests.index(indices[1])
    tests = tests[start_index:stop_index]

for test in exclude:
    if test in tests:
        tests.remove(test)

from gpaw.test import TestRunner

old_hooks = hooks.copy()
hooks.clear()
if mpi.rank == 0:
    if opt.directory is None:
        tmpdir = tempfile.mkdtemp(prefix='gpaw-test-')
    else:
        tmpdir = opt.directory
        if os.path.isdir(tmpdir):
            opt.keep_tmpdir = True
        else:
            os.mkdir(tmpdir)
else:
    tmpdir = None
tmpdir = mpi.broadcast_string(tmpdir)
cwd = os.getcwd()
os.chdir(tmpdir)
if mpi.rank == 0:
コード例 #2
0
ファイル: test.py プロジェクト: eojons/gpaw-scme
def run():
    parser = OptionParser(usage='%prog [options] [tests]',
                          version='%prog 0.1')
    parser.add_option('-x', '--exclude',
                      type='string', default=None,
                      help='Exclude tests (comma separated list of tests).',
                      metavar='test1.py,test2.py,...')
    parser.add_option('-f', '--run-failed-tests-only',
                      action='store_true',
                      help='Run failed tests only.')
    parser.add_option('--from', metavar='TESTFILE', dest='from_test',
                      help='Run remaining tests, starting from TESTFILE')
    parser.add_option('--after', metavar='TESTFILE', dest='after_test',
                      help='Run remaining tests, starting after TESTFILE')
    parser.add_option('--range',
                      type='string', default=None,
                      help='Run tests in range test_i.py to test_j.py ' +
                      '(inclusive)',
                      metavar='test_i.py,test_j.py')
    parser.add_option('-j', '--jobs', type='int', default=1,
                      help='Run JOBS threads.')
    parser.add_option('--reverse', action='store_true',
                      help=('Run tests in reverse order (less overhead with '
                            'multiple jobs)'))
    parser.add_option('-k', '--keep-temp-dir', action='store_true',
                      dest='keep_tmpdir', help='Do not delete temporary files.')
    parser.add_option('-d', '--directory', help='Run test in this directory')
    parser.add_option('-s', '--show-output', action='store_true',
                      help='Show standard output from tests.')

    opt, tests = parser.parse_args()


    if len(tests) == 0:
        from gpaw.test import tests

    if opt.reverse:
        tests.reverse()

    if opt.run_failed_tests_only:
        tests = [line.strip() for line in open('failed-tests.txt')]

    exclude = []
    if opt.exclude is not None:
        exclude += opt.exclude.split(',')

    if opt.from_test:
        fromindex = tests.index(opt.from_test)
        tests = tests[fromindex:]

    if opt.after_test:
        index = tests.index(opt.after_test) + 1
        tests = tests[index:]

    if opt.range:
        # default start(stop) index is first(last) test
        indices = opt.range.split(',')
        try:
            start_index = tests.index(indices[0])
        except ValueError:
            start_index = 0
        try:
            stop_index = tests.index(indices[1]) + 1
        except ValueError:
            stop_index = len(tests)
        tests = tests[start_index:stop_index]

    if opt.jobs > 1:
        exclude.append('maxrss.py')

    for test in exclude:
        if test in tests:
            tests.remove(test)

    from gpaw.test import TestRunner

    if mpi.world.size > 8:
        if mpi.rank == 0:
            message = '!!!!!!!\n' \
                'GPAW regression test suite was not designed to run on more\n' \
                'than 8 MPI tasks. Re-run test suite using 1, 2, 4 or 8 MPI\n' \
                'tasks instead.'
            warnings.warn(message, RuntimeWarning)

    old_hooks = hooks.copy()
    hooks.clear()
    if mpi.rank == 0:
        if opt.directory is None:
            tmpdir = tempfile.mkdtemp(prefix='gpaw-test-')
        else:
            tmpdir = opt.directory
            if os.path.isdir(tmpdir):
                opt.keep_tmpdir = True
            else:
                os.mkdir(tmpdir)
    else:
        tmpdir = None
    tmpdir = mpi.broadcast_string(tmpdir)
    cwd = os.getcwd()
    os.chdir(tmpdir)
    operating_system = platform.system() + ' ' + platform.machine()
    operating_system += ' ' + ' '.join(platform.dist())
    python = platform.python_version() + ' ' + platform.python_compiler()
    python += ' ' + ' '.join(platform.architecture())
    if mpi.rank == 0:
        print('python %s on %s' % (python, operating_system))
        print('Running tests in %s' % tmpdir)
        print('Jobs: %d, Cores: %d, debug-mode: %r' % (opt.jobs, mpi.size,
                                                       debug))
    failed = TestRunner(tests, jobs=opt.jobs, show_output=opt.show_output).run()
    os.chdir(cwd)
    if mpi.rank == 0:
        if len(failed) > 0:
            open('failed-tests.txt', 'w').write('\n'.join(failed) + '\n')
        elif not opt.keep_tmpdir:
            os.system('rm -rf ' + tmpdir)
    hooks.update(old_hooks.items())
    return len(failed)
コード例 #3
0
ファイル: test.py プロジェクト: robwarm/gpaw-symm
def run():
    description = (
        "Run the GPAW test suite.  The test suite can be run in "
        "parallel with MPI through gpaw-python.  The test suite "
        "supports 1, 2, 4 or 8 CPUs although some tests are "
        "skipped for some parallelizations.  If no TESTs are "
        "given, run all tests supporting the parallelization."
    )

    parser = OptionParser(usage="%prog [OPTION...] [TEST...]", description=description, version="%%prog %s" % version)
    parser.add_option(
        "-x",
        "--exclude",
        type="string",
        default=None,
        help="Exclude tests (comma separated list of tests).",
        metavar="test1.py,test2.py,...",
    )
    parser.add_option("-f", "--run-failed-tests-only", action="store_true", help="Run failed tests only.")
    parser.add_option(
        "--from", metavar="TESTFILE", dest="from_test", help="Run remaining tests, starting from TESTFILE"
    )
    parser.add_option(
        "--after", metavar="TESTFILE", dest="after_test", help="Run remaining tests, starting after TESTFILE"
    )
    parser.add_option(
        "--range",
        type="string",
        default=None,
        help="Run tests in range test_i.py to test_j.py " "(inclusive)",
        metavar="test_i.py,test_j.py",
    )
    parser.add_option(
        "-j",
        "--jobs",
        type="int",
        default=1,
        help="Run JOBS threads.  Each test will be executed "
        "in serial by one thread.  This option cannot be used "
        "for parallelization together with MPI.",
    )
    parser.add_option(
        "--reverse", action="store_true", help=("Run tests in reverse order (less overhead with " "multiple jobs)")
    )
    parser.add_option(
        "-k", "--keep-temp-dir", action="store_true", dest="keep_tmpdir", help="Do not delete temporary files."
    )
    parser.add_option("-d", "--directory", help="Run test in this directory")
    parser.add_option("-s", "--show-output", action="store_true", help="Show standard output from tests.")

    opt, tests = parser.parse_args()

    if len(tests) == 0:
        from gpaw.test import tests

    if opt.reverse:
        tests.reverse()

    if opt.run_failed_tests_only:
        tests = [line.strip() for line in open("failed-tests.txt")]

    exclude = []
    if opt.exclude is not None:
        exclude += opt.exclude.split(",")

    if opt.from_test:
        fromindex = tests.index(opt.from_test)
        tests = tests[fromindex:]

    if opt.after_test:
        index = tests.index(opt.after_test) + 1
        tests = tests[index:]

    if opt.range:
        # default start(stop) index is first(last) test
        indices = opt.range.split(",")
        try:
            start_index = tests.index(indices[0])
        except ValueError:
            start_index = 0
        try:
            stop_index = tests.index(indices[1]) + 1
        except ValueError:
            stop_index = len(tests)
        tests = tests[start_index:stop_index]

    if opt.jobs > 1:
        exclude.append("maxrss.py")

    for test in exclude:
        if test in tests:
            tests.remove(test)

    from gpaw.test import TestRunner

    if mpi.world.size > 8:
        if mpi.rank == 0:
            message = (
                "!!!!!!!\n"
                "GPAW regression test suite was not designed to run on more\n"
                "than 8 MPI tasks. Re-run test suite using 1, 2, 4 or 8 MPI\n"
                "tasks instead."
            )
            warnings.warn(message, RuntimeWarning)

    old_hooks = hooks.copy()
    hooks.clear()
    if mpi.rank == 0:
        if opt.directory is None:
            tmpdir = tempfile.mkdtemp(prefix="gpaw-test-")
        else:
            tmpdir = opt.directory
            if os.path.isdir(tmpdir):
                opt.keep_tmpdir = True
            else:
                os.mkdir(tmpdir)
    else:
        tmpdir = None
    tmpdir = mpi.broadcast_string(tmpdir)
    cwd = os.getcwd()
    os.chdir(tmpdir)
    operating_system = platform.system() + " " + platform.machine()
    operating_system += " " + " ".join(platform.dist())
    python = platform.python_version() + " " + platform.python_compiler()
    python += " " + " ".join(platform.architecture())
    if mpi.rank == 0:
        print("python %s on %s" % (python, operating_system))
        print("Running tests in %s" % tmpdir)
        print("Jobs: %d, Cores: %d, debug-mode: %r" % (opt.jobs, mpi.size, debug))
    failed = TestRunner(tests, jobs=opt.jobs, show_output=opt.show_output).run()
    os.chdir(cwd)
    if mpi.rank == 0:
        if len(failed) > 0:
            open("failed-tests.txt", "w").write("\n".join(failed) + "\n")
        elif not opt.keep_tmpdir:
            os.system("rm -rf " + tmpdir)
    hooks.update(old_hooks.items())
    return len(failed)