def run():
v = options.parse_arguments().verbosity
if not arb.mpi_is_initialized():
arb.mpi_init()
comm = arb.mpi_comm()
alloc = arb.proc_allocation()
ctx = arb.context(alloc, comm)
rank = ctx.rank
if rank == 0:
runner = unittest.TextTestRunner(verbosity=v)
else:
sys.stdout = open(os.devnull, 'w')
runner = unittest.TextTestRunner(stream=sys.stdout)
runner.run(suite())
if not arb.mpi_is_finalized():
arb.mpi_finalize()
def _finalize_mpi():
print("Context fixture finalizing mpi")
arbor.mpi_finalize()
def tearDownClass(self):
if self.local_mpi:
arb.mpi_finalize()
if __name__ == "__main__":
v = options.parse_arguments().verbosity
if not arb.mpi_is_initialized():
print(" Runner initializing mpi")
arb.mpi_init()
if mpi4py_enabled:
comm = arb.mpi_comm(mpi.COMM_WORLD)
elif mpi_enabled:
comm = arb.mpi_comm()
alloc = arb.proc_allocation()
ctx = arb.context(alloc, comm)
rank = ctx.rank
if rank == 0:
runner = unittest.TextTestRunner(verbosity=v)
else:
sys.stdout = open(os.devnull, 'w')
runner = unittest.TextTestRunner(stream=sys.stdout)
result = runner.run(suite())
if not arb.mpi_is_finalized():
arb.mpi_finalize()
sys.exit(not (result.wasSuccessful()))
