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']