Batch task testing works (sort of)

[moe.git] / t / moe / __init__.py

diff --git a/t/moe/__init__.py b/t/moe/__init__.py
index 19957e28a55f8fec2c2f4c73a1ddcbcd06ae77ad..26e4f24ebf7951e5fc46bce03361b014eb3fd75d 100644 (file)
--- a/t/moe/__init__.py
+++ b/t/moe/__init__.py
@@ -4,4 +4,13 @@ class MoeErr(Exception):
     pass
 
 class SolutionErr(Exception):
-    pass
+
+    def __init__(self, message, stat_code=None):
+       self.stat_code = stat_code
+       self.message = message
+
+    def __str__(self):
+       if self.stat_code is None:
+           return self.message
+       else:
+           return "%s: %s" % (self.stat_code, self.message)