Added unfix(), corrected fix()

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

diff --git a/t/moe/status.py b/t/moe/status.py
index 5a5adf22c5b4851b572ff955ca08a0814eacb903..76681c7b4739fb2f499e92259002234ae9147786 100644 (file)
--- a/t/moe/status.py
+++ b/t/moe/status.py
@@ -9,7 +9,7 @@ key_pattern = re.compile("^[A-Za-z0-9_-]+$")
 class MoeStatusInvalid(Exception):
     pass
 
-class MoeStatus:
+class Status:
     """Moe status file."""
 
     def __init__(self):
@@ -47,7 +47,7 @@ class MoeStatus:
            if type(v) == types.ListType: vals = v
            else: vals = [v]
            for w in vals:
-               if isinstance(w, MoeStatus):
+               if isinstance(w, Status):
                    l.append(prefix + k + "(\n")
                    l.extend(self.str_lines(prefix+"  "))
                    l.append(prefix + ")\n")
@@ -102,7 +102,7 @@ class MoeStatus:
                    this.read_val(k, v)
                elif x.endswith("("):
                    k = x[:-1]
-                   new = MoeStatus()
+                   new = Status()
                    this.read_val(k, new)
                    stk.append(this)
                    this = new