def test_get_drmaa(self):
"""Check SGERunner._get_drmaa()"""
class DummyDRMAA(object):
class Session(object):
def initialize(self): pass
def exit(self): pass
sys.modules['drmaa'] = DummyDRMAA()
r = SGERunner('test.sh')
d, s = r._get_drmaa()
self.assert_(isinstance(d, DummyDRMAA))
self.assert_(isinstance(s, DummyDRMAA.Session))
SGERunner._drmaa = None
del sys.modules['drmaa']
def test_get_drmaa(self):
"""Check SGERunner._get_drmaa()"""
class DummyDRMAA(object):
class Session(object):
def initialize(self):
pass
def exit(self):
pass
sys.modules['drmaa'] = DummyDRMAA()
r = SGERunner('test.sh')
d, s = r._get_drmaa()
self.assert_(isinstance(d, DummyDRMAA))
self.assert_(isinstance(s, DummyDRMAA.Session))
SGERunner._drmaa = None
del sys.modules['drmaa']
def test_sge_name(self):
"""Check SGERunner.set_sge_name()"""
r = SGERunner('echo foo', interpreter='/bin/csh')
r.set_sge_name('test\t job ')
self.assertEqual(r._name, 'testjob')
r.set_sge_name('TestJob')
self.assertEqual(r._name, 'TestJob')
r.set_sge_name('1234')
self.assertEqual(r._name, 'J1234')
def test_sge_name(self):
"""Check SGERunner.set_sge_name()"""
r = SGERunner('echo foo', interpreter='/bin/csh')
r.set_sge_name('test\t job ')
self.assertEqual(r._name, 'testjob')
r.set_sge_name('TestJob')
self.assertEqual(r._name, 'TestJob')
r.set_sge_name('1234')
self.assertEqual(r._name, 'J1234')
r.set_sge_name('None')
self.assertEqual(r._name, 'JNone')
r.set_sge_name('ALL')
self.assertEqual(r._name, 'JALL')
r.set_sge_name('template')
self.assertEqual(r._name, 'Jtemplate')
