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