def test_open(self): if hostname(False) in engine.supported_hosts(): engine.open() self.assertEqual(1,len(visit.GetEngineList())) engine.close() self.assertEqual(0,len(visit.GetEngineList())) else: self.assertRaises(VisItException,engine.open)
def launch(host, nprocs, ppn, method, part, bank, rtime, vdir, extra_args=None): """ Launch helper. """ msg = "[%s: opening engine" % host if not part is None: msg += " on %s" % part msg += "]" print msg nnodes = int(math.ceil(float(nprocs) / float(ppn))) ehost = host if host == hostname(False): ehost = "localhost" args = ["-l", method] if not method == "serial": args.extend(["-np", str(nprocs)]) if not vdir is None: args.extend(["-dir", vdir]) if not part is None: args.extend(["-nn", str(nnodes), "-p", part]) if not extra_args is None: args.extend(extra_args) if method == "msub/srun": rtime = str(rtime) if rtime.count(":") == 0: rtime += ":00" args.extend(["-b", bank, "-t", rtime]) if visit.OpenComputeEngine(ehost, args) != 1: raise VisItException("Failed to open compute engine on %s." % host) return visit.GetEngineList()[0]
def test_serial(self): ename = engine.open(nprocs=1,method="serial") self.assertEqual(1,len(visit.GetEngineList())) engine.close(ename) self.assertEqual(0,len(visit.GetEngineList()))