def local_tests(): print "local tests.." checkl("testfun1", modutil.find_modules([testfun1]),\ ["random", "os", "time"]) checkl("testfun2", modutil.find_modules([testfun2]),\ [('extramodule1', abspath("extramodule1.py"))]) checkl("testfun1+2", modutil.find_modules([testfun1, testfun2],\ send_modules = False),\ ["random", "os", "time", "extramodule1"]) try: modutil.find_modules([testfun3]) raise Exception("Test testfun3 failed: No exception raised") except ImportError: pass checkl("testfun4", modutil.find_modules([testfun4]),\ [("mod1", abspath("extra/mod1.py")),\ ("mod2", abspath("extra/mod2.py"))]) checkl("testfun4-nosend", modutil.find_modules([testfun4],\ send_modules = False), ["mod1"]) checkl("testfun4-norecurse", modutil.find_modules([testfun4],\ recurse = False), [("mod1", abspath("extra/mod1.py"))]) print "local tests ok"
def required_modules(self): return modutil.find_modules([self.map])
def test_missing(self): self.assertRaises(ModUtilImportError, lambda: modutil.find_modules([missing_module]))
def assertFindsModules(self, functions, modules, send_modules=True, recurse=True): self.assertEquals(sorted(modutil.find_modules(functions, send_modules=send_modules, recurse=recurse)), sorted(modules))
print "disco tests.." # default job = disco.new_job( name = "test_modutil1", input = tserver.makeurl(inputs), map = fun_map) checkl("test_modutil1", result_iterator(job.wait()), [("4.0", "")]) job.purge() print "test_modutil1 ok" job = disco.new_job( name = "test_modutil2", input = tserver.makeurl(inputs), required_modules = modutil.find_modules([fun_map]), map = fun_map) checkl("test_modutil2", result_iterator(job.wait()), [("4.0", "")]) job.purge() print "test_modutil2 ok" job = disco.new_job( name = "test_modutil3", input = tserver.makeurl(inputs), required_modules = ["math", ("mod1", "extra/mod1.py")], required_files = [modutil.locate_modules(["mod2"])[0][1]], map = fun_map) checkl("test_modutil3", result_iterator(job.wait()), [("4.0", "")]) job.purge() print "test_modutil3 ok"