X-Git-Url: http://mj.ucw.cz/gitweb/?a=blobdiff_plain;f=t%2Fmoe%2Futil.py;h=d287ff864c1230709b3172d658af4e1f3f157297;hb=5ad8b4f69ab4fdf4d75f5a56fc72ef2988f1da1f;hp=e083f0fb6f84dbfe285d342c8482036211d05d2e;hpb=5701572b35e7ebc7c2d96cc603cd720fb2649ac4;p=eval.git

diff --git a/t/moe/util.py b/t/moe/util.py
index e083f0f..d287ff8 100644
--- a/t/moe/util.py
+++ b/t/moe/util.py
@@ -1,21 +1,22 @@
 #!/usr/bin/env python
 
 import os
+import sys
 import os.path
 import shutil
 
 def mkdir_tree(name):
     try:
 	os.mkdir(name)	
-    except OSError, e:
-	if e.errno == os.errno.ENOENT:
+    except OSError, err:
+	if err.errno == os.errno.ENOENT:
 	    head, tail = os.path.split(name)
 	    mkdir_tree(head)
 	    os.mkdir(name)
-	elif e.errno == os.errno.EEXIST:
+	elif err.errno == os.errno.EEXIST:
 	    pass
 	else:
-	    raise e
+	    raise err
 
 def link_or_copy(src, dest):
     try:
@@ -33,3 +34,10 @@ def remove_tree_contents(dir):
 		shutil.rmtree(os.path.join(dir, f))
 	    else:
 		raise err
+
+def load_module(modname):
+    """Return the module `modname` (full name) if loaded, or try to import it. 
+    Returns the module or raises `ImportError`."""
+    if not sys.modules.has_key(modname):
+	__import__(modname)
+    return sys.modules[modname]