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()))